t

tools.py

@@ -62,12 +62,22 @@ def web_search(query: str) -> str:
     
     Args:
         query: The search query. The query fulfils the argument that is input."""
-    search_docs = TavilySearch(max_results=5, topic="general", include_answer=True, include_image_descriptions=True).invoke(input=query)
-    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
-        ])
+    tavily = TavilySearch(max_results=5, topic="general", include_answer=True, include_image_descriptions=True)
+    raw_results = tavily.invoke(query)
+
+    docs = []
+    for idx, text in enumerate(raw_results):
+        docs.append(Document(
+            page_content=text,
+            metadata={"source": f"Tavily result #{idx+1}"}
+        ))
+
+    formatted = "\n\n---\n\n".join(
+        f'<Document source="{doc.metadata["source"]}">\n'
+        f'{doc.page_content}\n'
+        f'</Document>'
+        for doc in docs
+    )
     return {"web_results": formatted_search_docs}
     
 web_search_tool = Tool(