Update agent.py

agent.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, LiteLLMModel, Tool
+from smolagents import CodeAgent, LiteLLMModel, Tool, DuckDuckGoSearchTool
 from token_bucket import Limiter, MemoryStorage
 from tenacity import retry, stop_after_attempt, wait_exponential
 from sentence_transformers import SentenceTransformer
@@ -188,14 +188,14 @@ class MagAgent:
             ToolRouter()
         ]
 
-        with open("prompts.yaml") as f:
-            self.prompt_templates = yaml.safe_load(f)
+#        with open("prompts.yaml") as f:
+#            self.prompt_templates = yaml.safe_load(f)
 
         self.agent = CodeAgent(
             model=self.model,
             tools=self.tools,
             verbosity_level=2,
-            prompt_templates=self.prompt_templates,
+#            prompt_templates=self.prompt_templates,
             max_steps=20
         )