Update pipeline.py

pipeline.py

@@ -209,10 +209,32 @@ def build_rag_chain(vectorstore: FAISS) -> RetrievalQA:
     
 def do_web_search(query: str) -> str:
     try:
+        print("DEBUG: Performing a new web search...")
+        # model =  LiteLLMModel(model_id="gemini/gemini-pro", api_key=os.environ.get("GEMINI_API_KEY"))
+        model=HfApiModel()
         search_tool = DuckDuckGoSearchTool()
-        search_agent = ManagedAgent(llm=gemini_llm, tools=[search_tool])
-        search_result = search_agent.run(f"Search for information about: {query}")
-        return str(search_result).strip()
+        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)
+        return str(new_search_result).strip()
     except Exception as e:
         print(f"Web search failed: {e}")
         return ""