Update agent.py

agent.py

@@ -16,9 +16,6 @@ from langchain_core.tools import tool
 #from langchain.tools.retriever import create_retriever_tool
 #from supabase.client import Client, create_client
 
-#from huggingface_hub import login
-#login(token=HF_TOKEN)
-
 load_dotenv()
 
 @tool
@@ -87,15 +84,15 @@ def wiki_search(query: str) -> str:
         ])
     return {"wiki_results": formatted_search_docs}
 
-"""
+
 @tool
 def web_search(query: str) -> str:
-"""
-"""Search Tavily for a query and return maximum 3 results.
+
+    """Search Tavily for a query and return maximum 3 results.
     
     Args:
         query: The search query."""
-"""
+
     search_docs = TavilySearchResults(max_results=3).invoke(query=query)
     formatted_search_docs = "\n\n---\n\n".join(
         [
@@ -104,7 +101,6 @@ def web_search(query: str) -> str:
         ])
     return {"web_results": formatted_search_docs}
 
-"""
 
 @tool
 def arvix_search(query: str) -> str:
@@ -165,7 +161,7 @@ tools = [
     divide,
     modulus,
     wiki_search,
-    #web_search,
+    web_search,
     arvix_search,
 ]