Update agent.py

agent.py

@@ -1,34 +1,51 @@
-from smolagent import CodeAgent,  DuckDuckGoSearchTool, InferenceClientModel, tool, VisitWebpageTool, Tool, FinalAnswerTool, WebSearchTool, WikipediaSearchTool
-  
-import numpy as np
-import time
-import datetime
+from smolagents import CodeAgent, HfApiModel
+from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, FinalAnswerTool
+from calculator_tool import CalculatorTool
 
-#Creating multyagent structure
-web_agent = CodeAgent(
+
+model = HfApiModel(
+    max_token = 2096,
+    temperature = 0.5, 
+    model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
+    custom_role_conversations=None
+)
+#or 
+#model=InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", provider="together")
+
+#with open("prompts.yaml", 'r') as stream:
+ #   prompt_templates = yaml.safe_load(stream)
+
+#Creating the agent
+agent = CodeAgent(
+    model=model,
     tools=[
-        DuckDuckGoSearchTool(),
         VisitWebpageTool(),
-        FinalAnswerTool()
+        WebSearchTool(),
+        WikipediaSearchTool(),
+        CalculatorTool(),
+        FinalAnswerTool(),
     ], 
-    model=InferenceClientModel(),
-    name="web_agent",
-    description="Brows the web to find information",
     max_steps=10,
-    verbosity_level=0,
+    verbosity_level = 1,        #or other parameter
+    grammar = None,             
+    planning_interval=None,      
+    #name="web_agent",            
+    #description="Brows the web to find information",      
+    #additional_autorized_imports = ["pandas"],
 )
 
-manager_agent = CodeAgent(
-    model=InferenceClientModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096),
-    managed_agent=[web_agent],
-    additional_authorized_imports=[
-        "datetime",
-        "pandas",
-        "numpy",
-        "json"
-    ],
-    planning_intervals=5,
-    verbosity_level=2,
-    max_steps=15,
-)
+#manager_agent = CodeAgent(
+ #   model=InferenceClientModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096),
+  #  managed_agent=[web_agent],
+   # additional_authorized_imports=[
+    #    "datetime",
+     #   "pandas",
+      #  "numpy",
+       # "json"
+   # ],
+    #planning_intervals=5,
+    #verbosity_level=2,
+    #max_steps=15,
+#)
+