From cde6b5ac604ec005754facb96803980f349f49a4 Mon Sep 17 00:00:00 2001 From: Francesco Balassone <77972200+duel0@users.noreply.github.com> Date: Wed, 28 May 2025 16:34:17 +0200 Subject: [PATCH 1/3] Fixed Thought Agent --- src/cai/agents/thought.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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], ) From d0c38388741491cd66cae42b42b9a323135af8be Mon Sep 17 00:00:00 2001 From: Francesco Balassone <77972200+duel0@users.noreply.github.com> Date: Wed, 28 May 2025 16:40:13 +0200 Subject: [PATCH 2/3] Added .env.example --- .env.example | 5 +++++ 1 file changed, 5 insertions(+) create mode 100644 .env.example diff --git a/.env.example b/.env.example new file mode 100644 index 00000000..b228aee9 --- /dev/null +++ b/.env.example @@ -0,0 +1,5 @@ +OPENAI_API_KEY="sk-1234" +ANTHROPIC_API_KEY="" +OLLAMA="" +PROMPT_TOOLKIT_NO_CPR=1 +CAI_STREAM=false From f0aaa38d0038508ac3457d2c8d33c2447f8c6afa Mon Sep 17 00:00:00 2001 From: Francesco Balassone <77972200+duel0@users.noreply.github.com> Date: Wed, 28 May 2025 16:41:37 +0200 Subject: [PATCH 3/3] Added CAI_STREAM to env examples --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 0190e731..933cf7a7 100644 --- a/README.md +++ b/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