Update pipeline.py

pipeline.py

@@ -41,7 +41,22 @@ from mistralai import Mistral
 from langchain_google_genai import ChatGoogleGenerativeAI
 
 # Web search
-from smolagents import DuckDuckGoSearchTool, ManagedAgent, HfApiModel, CodeAgent
+# from smolagents import DuckDuckGoSearchTool, ManagedAgent, HfApiModel, CodeAgent
+from openinference.instrumentation.smolagents import SmolagentsInstrumentor
+from phoenix.otel import register
+
+
+register()
+SmolagentsInstrumentor().instrument(skip_dep_check=True)
+
+
+from smolagents import (
+    CodeAgent,
+    DuckDuckGoSearchTool,
+    HfApiModel,
+    ToolCallingAgent,
+    VisitWebpageTool,
+)
 
 # Import new prompts
 from prompts import (
@@ -217,14 +232,20 @@ def do_web_search(query: str) -> str:
             return cached
 
         logger.info("Performing a new web search for: '%s'", query)
-        model = HfApiModel()
-        search_tool = DuckDuckGoSearchTool()
-        web_agent = CodeAgent(tools=[search_tool], model=model)
+        # model = HfApiModel()
+        # search_tool = DuckDuckGoSearchTool()
+        # web_agent = CodeAgent(tools=[search_tool], model=model)
         
-        managed_web_agent = ManagedAgent(
-            agent=web_agent,
-            name="web_search",
-            description="Runs a web search. Provide your query."
+        # managed_web_agent = ManagedAgent(
+        #     agent=web_agent,
+        #     name="web_search",
+        #     description="Runs a web search. Provide your query."
+        # )
+        search_agent = ToolCallingAgent(
+        tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
+        model=HfApiModel(),
+        name="search_agent",
+        description="This is an agent that can do web search.",
         )
         
         manager_agent = CodeAgent(