Update agent.py

agent.py

@@ -1,5 +1,5 @@
 from smolagents import HfApiModel, CodeAgent, DuckDuckGoSearchTool, WikipediaSearchTool, Tool, LiteLLMModel
-from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_tavily import TavilySearch
 from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
 import os, time, math, pandas, numpy
 
@@ -173,7 +173,7 @@ class TavilySearchTool(Tool):
     output_type = "string"
 
     def forward(self, query: str) -> str:
-        documents = TavilySearchResults(max_results=3).invoke(query=query)
+        documents = TavilySearch(max_results=3).invoke(query=query)
         condensed_docs = "\n\n---\n\n".join(
         [
             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
@@ -204,7 +204,7 @@ class ArvixSearchTool(Tool):
         documents = ArxivLoader(query=query, load_max_docs=3).load()
         condensed_docs = "\n\n---\n\n".join(
         [
-            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
+            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.content[:1000]}\n</Document>'
             for doc in documents
         ])
         return {"arvix_search_results": condensed_docs}