tavily ret

agent_utils.py

@@ -1,5 +1,7 @@
 from langchain_community.document_loaders import WikipediaLoader
 import arxiv
+import os
+from tavily import TavilyClient
 
 def wiki_ret(question: str) -> str:
     """ Retrieve docs from wikipedia """
@@ -37,3 +39,23 @@ def arxiv_ret(query: str, max_results: int = 3) -> str:
         return "No relevant papers found."
     
     return "\n\n".join(results)
+
+
+def tavily_ret(query: str, max_results: int = 3) -> str:
+    """Use Tavily to retrieve web-based information about a topic."""
+    api_key = os.getenv("TAVILY_API_KEY")
+    if not api_key:
+        return "Tavily API key not found."
+
+    client = TavilyClient(api_key=api_key)
+
+    results = client.search(query=query, search_depth="basic", max_results=max_results)
+
+    if not results["results"]:
+        return "No relevant information found."
+
+    summaries = []
+    for item in results["results"]:
+        summaries.append(f"Title: {item['title']}\nURL: {item['url']}\nSnippet: {item['content']}\n")
+
+    return "\n\n".join(summaries)

app.py

@@ -17,7 +17,7 @@ from langchain_core.messages import HumanMessage
 from langgraph.graph import START, END, StateGraph
 from langgraph.prebuilt import ToolNode, tools_condition
 
-from agent_utils import wiki_ret, arxiv_ret
+from agent_utils import wiki_ret, arxiv_ret, tavily_ret
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -82,7 +82,7 @@ class BasicAgent:
     def create_graph(self):
         builder = StateGraph(GraphState)
         builder.add_node("agent", self.agent)
-        builder.add_node("tools", ToolNode(tools = [wiki_ret, arxiv_ret]))
+        builder.add_node("tools", ToolNode(tools = [wiki_ret, arxiv_ret, tavily_ret]))
 
         builder.add_edge(START, "agent")
         builder.add_conditional_edges("agent", tools_condition)

requirements.txt

@@ -5,4 +5,5 @@ langchain-community
 langchain-core
 wikipedia
 langgraph
-arxiv
+arxiv
+tavily-python