Update app.py

app.py

@@ -11,6 +11,7 @@ from langchain_community.tools import DuckDuckGoSearchResults
 from langchain_google_community import GoogleSearchAPIWrapper
 from langchain_community.document_loaders import YoutubeLoader,PyPDFLoader,Docx2txtLoader,TextLoader,ArxivLoader
 from langchain_community.document_loaders import WikipediaLoader
+from langchain_tavily import TavilySearch
 import wikipedia
 import speech_recognition as sr
 import tempfile
@@ -234,7 +235,29 @@ def wiki_search(query: str) -> str:
         ]
     )
     return {"wiki_results": formatted_search_docs}
-        
+
+@tool
+def web_search(query: str) -> str:
+    """Search Tavily on the web for a query and return maximum 3 results.
+    
+    Args:
+        query: The search query."""
+    search_docs = TavilySearch(max_results=3).invoke({'query': query})
+
+    print(f"Search query: {query}")
+    print(f"Search results count: {len(search_docs)}")
+    
+    if isinstance(search_docs, dict):
+        search_docs = search_docs.get('results', [])
+    
+    
+    formatted_search_docs = "\n\n---\n\n".join(
+        [
+            f"""<Document source="{doc.get('url', '')}" page="{doc.get('title', '')}"/>\n{doc.get('content', '')}\n</Document>"""
+            for doc in search_docs
+        ])
+    return {"web_results": formatted_search_docs}
+    
 #### add duckduckGoSearch into tool
 tools = [
     divide,
@@ -250,6 +273,7 @@ tools = [
     # youtube_transcript_tool,
     # transcribe_audio,
     # analyze_python_code,
+    web_search,
     arvix_search,
     wiki_search,
     GoogleSearchAPIWrapper(k=10).run,
@@ -309,6 +333,7 @@ def assistant(state: AgentState, llm_with_tools):
     Tools:
     1. GoogleSearchAPIWrapper(k=10).run
     2. PDF access
+    3. web_search
 
     - FINAL ANSWER: 0.1777
 
@@ -336,7 +361,8 @@ class BasicAgent:
             raise ValueError("GOOGLE_CSE_ID environment variable not set or loaded.")
             
         # self.visionLLM = ChatOpenAI(model="gpt-4o",api_key=api_key) # multi-modal LLM
-        self.LLM = ChatOpenAI(model="gpt-4o",api_key=api_key) # manager LLM
+        # self.LLM = ChatOpenAI(model="gpt-4o",api_key=api_key) # manager LLM
+        self.LLM = ChatOpenAI(model="gpt-4.1",api_key=api_key) # manager LLM
         self.LLM_with_tools = self.LLM.bind_tools(tools, parallel_tool_calls=False)
 
         assistant_with_llm = lambda state:assistant(state, self.LLM_with_tools)