Update pipeline.py

pipeline.py

@@ -12,7 +12,7 @@ from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.vectorstores import FAISS
 from langchain.chains import RetrievalQA
 
-from smolagents import DuckDuckGoSearchTool, ManagedAgent, LiteLLMModel ,CodeAgent
+from smolagents import DuckDuckGoSearchTool, ManagedAgent, LiteLLMModel ,CodeAgent, 
 from pydantic import BaseModel, Field, ValidationError, validator
 from mistralai import Mistral
 
@@ -315,7 +315,8 @@ 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"))
+        # model =  LiteLLMModel(model_id="gemini/gemini-pro", api_key=os.environ.get("GEMINI_API_KEY"))
+        model=HfApiModel()
         search_tool = DuckDuckGoSearchTool()
         web_agent = CodeAgent(
         tools=[search_tool],