Merge pull request #170 from duel0/main

Fixed #163 + Fix Typo in Thought Agent
This commit is contained in:
PRT 2025-05-28 17:24:50 +02:00 committed by GitHub
commit cb843a0696
3 changed files with 11 additions and 6 deletions

5
.env.example Normal file
View File

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

View File

@ -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

View File

@ -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],
)