updated web search function

pipeline.py

@@ -39,7 +39,7 @@ gemini_llm = ChatGoogleGenerativeAI(
     # Additional parameters or safety_settings can be added here if needed
 )
 
-web_gemini_llm = LiteLLMModel(model_id="gemini/gemini-pro", api_key=os.environ.get("GEMINI_API_KEY"))
+# web_gemini_llm = LiteLLMModel(model_id="gemini/gemini-pro", api_key=os.environ.get("GEMINI_API_KEY"))
 
 ################################################################################
 # Pydantic Models
@@ -315,9 +315,26 @@ def do_cached_web_search(query: str) -> str:
     # 2) If no suitable cached answer, do a new search
     try:
         print("DEBUG: Performing a new web search...")
+        model =  LiteLLMModel(model_id="gemini/gemini-pro", api_key=os.environ.get("GEMINI_API_KEY"))
         search_tool = DuckDuckGoSearchTool()
-        search_agent = ManagedAgent(llm=web_gemini_llm, tools=[search_tool])
-        new_search_result = search_agent.run(f"Search for information about: {query}")
+        web_agent = CodeAgent(
+        tools=[search_tool],
+        model=model
+        )
+        
+        managed_web_agent = ManagedAgent(
+        agent=web_agent,
+        name="web_search",
+        description="Runs a web search for you. Provide your query as an argument."
+        )
+        
+        manager_agent = CodeAgent(
+        tools=[],  # If you have additional tools for the manager, add them here
+        model=model,
+        managed_agents=[managed_web_agent]
+        )
+        
+        new_search_result = manager_agent.run(f"Search for information about: {query}")
         
         # 3) Store in cache for future reuse
         store_websearch_result(query, new_search_result)