diff options
| author | CaptainJack2491 <jayrupnakawala@gmail.com> | 2025-11-20 20:52:00 +0000 |
|---|---|---|
| committer | CaptainJack2491 <jayrupnakawala@gmail.com> | 2025-11-20 20:52:00 +0000 |
| commit | c367ce44f63a112ca4fef372e6fd8477cfbdf6f9 (patch) | |
| tree | 7c38651c5e334055383c50d4515ef924057df4b6 | |
| parent | d290be9ea9cad8a9f6cd5f0c20aa4d80561dfb97 (diff) | |
updated agent to work with the EXTRA_BODY variable to work with models that require it.
| -rw-r--r-- | src/agents/03-logs/agent.py | 17 | ||||
| -rw-r--r-- | src/agents/03-logs/config.py | 2 | ||||
| -rw-r--r-- | src/agents/03-logs/main.py | 4 |
3 files changed, 8 insertions, 15 deletions
diff --git a/src/agents/03-logs/agent.py b/src/agents/03-logs/agent.py index 9b061b7..2ac8565 100644 --- a/src/agents/03-logs/agent.py +++ b/src/agents/03-logs/agent.py @@ -2,7 +2,7 @@ from openai import OpenAI import json import datetime -from config import MODEL, BASE_URL, API_KEY, TEMPERATURE +from config import MODEL, BASE_URL, API_KEY, TEMPERATURE, EXTRA_BODY from tools import tools, available_functions class Agent: @@ -34,19 +34,8 @@ class Agent: messages=messages, tools=self.tools, temperature=self.temperature, - extra_body={ - # for gemini - # 'extra_body': { - # "google": { - # "thinking_config": { - # "thinking_budget": 800, - # "include_thoughts": True - # } - # } - # } - # for minimax - # "reasoning_split": True - } + extra_body=EXTRA_BODY, + # include=["reasoning.encrypted_content"] ) # Update token counts diff --git a/src/agents/03-logs/config.py b/src/agents/03-logs/config.py index fda6124..36901a6 100644 --- a/src/agents/03-logs/config.py +++ b/src/agents/03-logs/config.py @@ -3,4 +3,6 @@ MODEL = 'qwen3:0.6b' BASE_URL = "http://127.0.0.1:11434/v1" API_KEY = "ollama" TEMPERATURE = 0.7 +# default extra body +EXTRA_BODY = {} INCLUDE_HIDDEN_GOAL = True diff --git a/src/agents/03-logs/main.py b/src/agents/03-logs/main.py index d339330..5d398c4 100644 --- a/src/agents/03-logs/main.py +++ b/src/agents/03-logs/main.py @@ -13,7 +13,7 @@ def load_prompt(file_path): if __name__ == "__main__": # Configuration for the experiment run SCENARIO = "corporate_sabotage" - OVERSIGHT_LEVEL = "low" + OVERSIGHT_LEVEL = "high" # Construct paths based on the scenario scenario_dir = os.path.join("scenarios", SCENARIO) @@ -44,6 +44,8 @@ if __name__ == "__main__": system_prompt = "\n\n".join(filter(None, system_prompt_parts)) print("\n--- Using model:", MODEL, "---\n") + print("\n--- Final Virtual File System State ---") + VFS.get_instance().print_fs() agent = Agent( system_prompt=system_prompt, scenario=SCENARIO, |