test cargo

agent.py

@@ -1,28 +1,28 @@
-import os
-from PIL import Image
-from smolagents import CodeAgent, HfApiModel, InferenceClientModel
+# import os
+# from PIL import Image
+# from smolagents import CodeAgent, HfApiModel, InferenceClientModel
 
-import tools.tools as tls
+# import tools.tools as tls
 
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+# # --- Basic Agent Definition ---
+# # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
-class BasicAgent:
-    def __init__(self):
-        print("BasicAgent initialized.")
-    def __call__(self, question: str) -> str:
-        model = HfApiModel(model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud", provider="together", use_auth_token=True)
+# class BasicAgent:
+#     def __init__(self):
+#         print("BasicAgent initialized.")
+#     def __call__(self, question: str) -> str:
+#         model = HfApiModel(model_id="https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud", provider="together", use_auth_token=True)
 
-        agent = CodeAgent(
-            tools=[tls.search_tool, tls.calculate_cargo_travel_time],
-            model=InferenceClientModel(),
-            additional_authorized_imports=["pandas"],
-            max_steps=20,
-        )
+#         agent = CodeAgent(
+#             tools=[tls.search_tool, tls.calculate_cargo_travel_time],
+#             model=InferenceClientModel(),
+#             additional_authorized_imports=["pandas"],
+#             max_steps=20,
+#         )
 
-        fixed_answer = agent.run(question)
+#         fixed_answer = agent.run(question)
 
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        # fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return str(fixed_answer)
+#         print(f"Agent received question (first 50 chars): {question[:50]}...")
+#         # fixed_answer = "This is a default answer."
+#         print(f"Agent returning fixed answer: {fixed_answer}")
+#         return str(fixed_answer)

app.py

@@ -7,7 +7,8 @@ from huggingface_hub import login
 from dotenv import load_dotenv
 
 import agent
-import app_agents
+from app_agents.manager_agent import manager_agent
+# import app_agents.manager_agent as manager_agent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -34,7 +35,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = app_agents.manager_agent
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -135,14 +136,16 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
 def test_init_agent_for_chat(text_input, history):
     # 1. Instantiate Agent ( modify this part to create your agent)
-    try:
-        managerAgent = agents.manager_agent
-        # basicAgent = agent.BasicAgent()
-    except Exception as e:
-        print(f"Error instantiating agent: {e}")
-        return f"Error initializing agent: {e}", None
+    # try:
+    #     managerAgent = manager_agent.manager_agent
+    #     # basicAgent = agent.BasicAgent()
+    # except Exception as e:
+    #     print(f"Error instantiating agent: {e}")
+    #     return f"Error initializing agent: {e}", None
 
     manager_agent.visualize()
+
+    manager_agent.run(text_input)
     # submitted_answer = basicAgent(text_input)
 
     # return submitted_answer

app_agents/manager_agent.py

@@ -1,8 +1,10 @@
 import os
 from smolagents.utils import encode_image_base64, make_image_url
-from smolagents import OpenAIServerModel
+from smolagents import OpenAIServerModel, CodeAgent, InferenceClientModel
 
 import app_agents.web_agent as web_agent
+from app_agents.web_agent import web_agent
+import app_tools.tools as agent_tools
 
 def check_reasoning_and_plot(final_answer, agent_memory):
     multimodal_model = OpenAIServerModel("gpt-4o", max_tokens=8096)
@@ -39,8 +41,8 @@ def check_reasoning_and_plot(final_answer, agent_memory):
 
 manager_agent = CodeAgent(
     model=InferenceClientModel("deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096),
-    tools=[calculate_cargo_travel_time],
-    managed_agents=[web_agent.agent],
+    tools=[agent_tools.calculate_cargo_travel_time],
+    managed_agents=[web_agent],
     additional_authorized_imports=[
         "geopandas",
         "plotly",

app_agents/web_agent.py

@@ -2,9 +2,9 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, Vi
 
 model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", provider="together")
 
-agent = CodeAgent(
+web_agent = CodeAgent(
     model=model,
-    tools=[GoogleSearchTool(), VisitWebpageTool()],
+    tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
     additional_authorized_imports=["pandas"],
     name="web_agent",
     description="Browses the web to find information",