Merge pull request #170 from duel0/main

Fixed #163 + Fix Typo in Thought Agent

Author: Lidiasaes

.env.example

@@ -0,0 +1,5 @@
+OPENAI_API_KEY="sk-1234"
+ANTHROPIC_API_KEY=""
+OLLAMA=""
+PROMPT_TOOLKIT_NO_CPR=1
+CAI_STREAM=false

README.md

@@ -173,7 +173,7 @@ python3.12 -m venv cai_env
 source cai_env/bin/activate && pip install cai-framework
 
 # Generate a .env file and set up with defaults
-echo -e 'OPENAI_API_KEY="sk-1234"\nANTHROPIC_API_KEY=""\nOLLAMA=""\nPROMPT_TOOLKIT_NO_CPR=1' > .env
+echo -e 'OPENAI_API_KEY="sk-1234"\nANTHROPIC_API_KEY=""\nOLLAMA=""\nPROMPT_TOOLKIT_NO_CPR=1\nCAI_STREAM=false' > .env
 
 # Launch CAI
 cai  # first launch it can take up to 30 seconds
@@ -195,7 +195,7 @@ python3.12 -m venv cai_env
 source cai_env/bin/activate && pip install cai-framework
 
 # Generate a .env file and set up with defaults
-echo -e 'OPENAI_API_KEY="sk-1234"\nANTHROPIC_API_KEY=""\nOLLAMA=""\nPROMPT_TOOLKIT_NO_CPR=1' > .env
+echo -e 'OPENAI_API_KEY="sk-1234"\nANTHROPIC_API_KEY=""\nOLLAMA=""\nPROMPT_TOOLKIT_NO_CPR=1\nCAI_STREAM=false' > .env
 
 # Launch CAI
 cai  # first launch it can take up to 30 seconds
@@ -218,7 +218,7 @@ python3 -m venv cai_env
 source cai_env/bin/activate && pip install cai-framework
 
 # Generate a .env file and set up with defaults
-echo -e 'OPENAI_API_KEY="sk-1234"\nANTHROPIC_API_KEY=""\nOLLAMA=""\nPROMPT_TOOLKIT_NO_CPR=1' > .env
+echo -e 'OPENAI_API_KEY="sk-1234"\nANTHROPIC_API_KEY=""\nOLLAMA=""\nPROMPT_TOOLKIT_NO_CPR=1\nCAI_STREAM=false' > .env
 
 # Launch CAI
 cai  # first launch it can take up to 30 seconds

src/cai/agents/thought.py

@@ -5,7 +5,7 @@
 
 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,13 +15,13 @@
 
 # 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(),
     ),
     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],
 )