Update agent.py

agent.py

@@ -83,30 +83,11 @@ def wiki_search(query: str) -> str:
 
 @tool
 def web_search(query: str) -> str:
-    """Search Tavily and format up to 3 documents."""
-    retriever = TavilyRetriever(k=3)
-    docs = retriever.get_relevant_documents(query)
-
-    formatted_docs = "\n\n---\n\n".join(
-        [
-            f'<Document source="{doc.metadata.get("source", "unknown")}" page="{doc.metadata.get("page", "")}">\n{doc.page_content}\n</Document>'
-            for doc in docs
-        ]
-    )
-    return {"web_results": formatted_docs}
-
-
-# def web_search(query: str) -> str:
-#     """Search Tavily for a query and return maximum 3 results.
-#     Args:
-#         query: The search query."""
-#     search_docs = TavilySearchResults(max_results=3).invoke(query)  # Fixed: pass query as positional argument
-#     formatted_search_docs = "\n\n---\n\n".join(
-#         [
-#             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
-#             for doc in search_docs
-#         ])
-#     return formatted_search_docs  # Also fixed the return type issue
+    """Search Tavily for a query and return maximum 3 results.
+    Args:
+        query: The search query."""
+    search_docs = TavilySearchResults(max_results=3).invoke(query)  # Fixed: pass query as positional argument
+    return {"web_results": search_docs}  # Also fixed the return type issue
 
 @tool
 def arvix_search(query: str) -> str: