diff --git a/src/cai/agents/thought.py b/src/cai/agents/thought.py index 032ac35a..14888296 100644 --- a/src/cai/agents/thought.py +++ b/src/cai/agents/thought.py @@ -5,7 +5,7 @@ using reasoner as a tool call 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 openai import AsyncOpenAI 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_agent = Agent( - name="ThoughAgent", + name="ThoughtAgent", model=OpenAIChatCompletionsModel( model=os.getenv('CAI_MODEL', "qwen2.5:14b"), openai_client=AsyncOpenAI(), @@ -23,5 +23,5 @@ thought_agent = Agent( description="""Agent focused on analyzing and planning the next steps in a security assessment or CTF challenge.""", instructions=thought_agent_system_prompt, - tools=[thought], + tools=[think], )