Update TavilySearch

agent.py

@@ -23,6 +23,9 @@ from supabase import create_client, Client
 
 load_dotenv()
 
+os.environ["TAVILY_API_KEY"] = os.environ.get("TAVILY_API_KEY")
+os.environ["GOOGLE_API_KEY"] = os.environ.get("GOOGLE_API_KEY")
+
 __embeddings = HuggingFaceEmbeddings(
     model_name="sentence-transformers/all-mpnet-base-v2",
     model_kwargs= { 'device': 'cpu' })
@@ -129,9 +132,9 @@ def web_search(query: str) -> str:
     Args:
         query: The search query.
     """
-    search_docs = TavilySearchResults(max_results=3).invoke(query=query)
+    search_docs = TavilySearchResults(max_results=3).invoke(input=query)
     formatted_search_docs = "\n\n---\n\n".join([
-        f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n\t{doc.page_content}\n<Document>'
+        f'<Document source="{doc["url"]}"/>\n\t{doc["content"]}\n<Document>'
         for doc in search_docs
     ])
     return { "web_results": formatted_search_docs }
@@ -205,7 +208,7 @@ def build_graph():
 # Test
 if __name__ == "__main__":
     # question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
-    question = "Data feed 100ms happened once. If 50second "
+    question = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
     graph = build_graph()
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({ "messages": messages })