add agent.py

agent.py

@@ -0,0 +1,42 @@
+from smolagents import (
+    ToolCallingAgent,
+    CodeAgent,
+    DuckDuckGoSearchTool,
+    VisitWebpageTool,
+    InferenceClientModel,
+)
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+def get_agent():
+    llm_qwen = InferenceClientModel(
+        model_id="Qwen/Qwen2.5-Coder-32B-Instruct", provider="together"
+    )
+    llm_deepseek = InferenceClientModel(
+        "deepseek-ai/DeepSeek-R1", provider="together", max_tokens=8096
+    )
+
+    web_agent = ToolCallingAgent(
+        tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
+        model=llm_qwen,
+        max_steps=10,
+        name="Web_Agent",
+        description="A web agent that can search the web and visit webpages.",
+    )
+
+    manager_agent = CodeAgent(
+        tools=[],
+        managed_agents=[web_agent],
+        model=llm_deepseek,
+        max_steps=10,
+    )
+    return manager_agent
+
+
+if __name__ == "__main__":
+    agent = get_agent()
+    agent.run(
+        "What is the latest news about AI? Please search the web and summarize the results."
+    )

app.py

@@ -2,13 +2,6 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import (
-    ToolCallingAgent,
-    CodeAgent,
-    DuckDuckGoSearchTool,
-    HfApiModel,
-    VisitWebpageTool,
-)
 from dotenv import load_dotenv
 import base64
 from opentelemetry.sdk.trace import TracerProvider
@@ -16,6 +9,7 @@ from openinference.instrumentation.smolagents import SmolagentsInstrumentor
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 from opentelemetry import trace
+from .agent import get_agent
 
 load_dotenv()
 
@@ -56,23 +50,6 @@ SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-def get_agent():
-    llm = HfApiModel(
-        model_id="Qwen/Qwen2.5-Coder-32B-Instruct", token=os.getenv("HF_TOKEN")
-    )
-    web_agent = ToolCallingAgent(
-        tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
-        model=llm,
-        max_steps=10,
-        description="A web agent that can search the web and visit webpages.",
-    )
-
-    manager_agent = CodeAgent(
-        tools=[web_agent],
-        model=llm,
-        max_steps=10,
-    )
-    return manager_agent
 
 
 class BasicAgent: