RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed Thought Agent

This commit is contained in:
Francesco Balassone 2025-05-28 16:34:17 +02:00 committed by GitHub
parent 02bbcc0da7
commit cde6b5ac60
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/cai/agents/thought.py
View File

@ -5,7 +5,7 @@ using reasoner as a tool call
support meta agent may better @cai.sdk.agents.meta.reasoner_support support meta agent may better @cai.sdk.agents.meta.reasoner_support
""" """
from cai.tools.misc.reasoning import thought from cai.tools.misc.reasoning import think
from cai.sdk.agents import Agent, OpenAIChatCompletionsModel # pylint: disable=import-error from cai.sdk.agents import Agent, OpenAIChatCompletionsModel # pylint: disable=import-error
from openai import AsyncOpenAI from openai import AsyncOpenAI
from cai.util import load_prompt_template from cai.util import load_prompt_template
@ -15,7 +15,7 @@ thought_agent_system_prompt = load_prompt_template("prompts/system_thought_route
# Thought Process Agent for analysis and planning # Thought Process Agent for analysis and planning
thought_agent = Agent( thought_agent = Agent(
name="ThoughAgent", name="ThoughtAgent",
model=OpenAIChatCompletionsModel( model=OpenAIChatCompletionsModel(
model=os.getenv('CAI_MODEL', "qwen2.5:14b"), model=os.getenv('CAI_MODEL', "qwen2.5:14b"),
openai_client=AsyncOpenAI(), openai_client=AsyncOpenAI(),
@ -23,5 +23,5 @@ thought_agent = Agent(
description="""Agent focused on analyzing and planning the next steps description="""Agent focused on analyzing and planning the next steps
in a security assessment or CTF challenge.""", in a security assessment or CTF challenge.""",
instructions=thought_agent_system_prompt, instructions=thought_agent_system_prompt,
tools=[thought], tools=[think],
) )