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