add back thinking tool and reformat benchmarks.py

TransformerOptimus · Sep 13, 2023 · 6e8991a · 6e8991a
1 parent 294ca5d
commit 6e8991a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/agbenchmark/benchmarks.py b/agbenchmark/benchmarks.py
@@ -15,7 +15,6 @@ def create_endpoint(
     response = requests.request(
         method, f"{baseUrl}{endpoint}", data=data, headers=headers
     )
-    logger.info(response.json())
     if response.status_code not in [200, 201] and exit_on_error:
         logger.info(f"Error connecting to API endpoint: {baseUrl}{endpoint}")
         sys.exit(1)
@@ -104,8 +103,6 @@ def setup(self):
             ),
             headers=self.headers,
         )
-        with open("config.yaml", "w") as file:
-            yaml.dump(self.config_data, file)
 
         return self.api_key
 

diff --git a/superagi/agent/agent_iteration_step_handler.py b/superagi/agent/agent_iteration_step_handler.py
@@ -146,7 +146,7 @@ def _build_agent_prompt(self, iteration_workflow: IterationWorkflow, agent_confi
         return prompt
 
     def _build_tools(self, agent_config: dict, agent_execution_config: dict):
-        agent_tools = []
+        agent_tools = [ThinkingTool()]
 
         config_data = AgentConfiguration.get_model_api_key(self.session, self.agent_id, agent_config["model"])
         model_api_key = config_data['api_key']