Add langchain-tavily dependency and implement TavilySearch tool

pyproject.toml

@@ -12,6 +12,7 @@ dependencies = [
     "ipywidgets>=8.1.7",
     "langchain-community>=0.3.27",
     "langchain-google-genai>=2.1.9",
+    "langchain-tavily>=0.2.11",
     "langchain[google-genai,googlegenai,openai]>=0.3.26",
     "langgraph>=0.4.8",
     "matplotlib>=3.10.5",

tools/__init__.py

@@ -1,8 +1,6 @@
-from .wikipedia import fetch_wikipedia_content
+from .wikipedia import wiki_search
+from .search import web_search
 from langchain_core.tools import BaseTool
-from langchain_community.document_loaders import WikipediaLoader
-from langchain_core.tools import render_text_description_and_args
-from langchain_core.tools import tool
 
 def get_all_tools() -> list[BaseTool]:
     """
@@ -11,9 +9,10 @@ def get_all_tools() -> list[BaseTool]:
     Returns:
         List of BaseTool instances ready for use with LangChain agents
     """
-    tools = []
-
-    tools.append(fetch_wikipedia_content)
+    tools = [
+        wiki_search,
+        web_search
+    ]
 
     return tools
 

tools/search.py

@@ -0,0 +1,4 @@
+from langchain_core.tools import tool
+from langchain_tavily import TavilySearch
+
+web_search = TavilySearch(max_results=3, topic="general")

tools/wikipedia.py

@@ -1,6 +1,5 @@
 from langchain_core.tools import tool
 from langchain_community.document_loaders import WikipediaLoader
-from typing import Optional
 
 @tool
 def wiki_search(query: str) -> str:

uv.lock

@@ -1260,6 +1260,21 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/ac/f2/a6a73beec15e90605e6a24c4498a8592d79a72c8e81c18ed0f5e9b7308e9/langchain_openai-0.3.29-py3-none-any.whl", hash = "sha256:71ae6791b3e017ec892a8062f993edc882c6665fd8385aa66e9dc3bff8205996", size = 74316, upload-time = "2025-08-08T15:12:30.794Z" },
 ]
 
+[[package]]
+name = "langchain-tavily"
+version = "0.2.11"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "aiohttp" },
+    { name = "langchain" },
+    { name = "langchain-core" },
+    { name = "requests" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/17/bb/63ce4058684dddf525af3c8e5dcfab15c5f17515d20241ef6e726ac9e8b7/langchain_tavily-0.2.11.tar.gz", hash = "sha256:ab4f5d0f7fcb276a3905aef2e38c21a334b6cbfc86b405a3238fdc9c6eae1290", size = 22382, upload-time = "2025-07-25T17:26:33.41Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/a8/5a/9326f125b4d3055a96200a5035016efe1aac46149cdafc7182e56710fcfe/langchain_tavily-0.2.11-py3-none-any.whl", hash = "sha256:358317c18fbb26500bca665301450e38945f1f4f6a6f4e06406c7674a76c8d5c", size = 26187, upload-time = "2025-07-25T17:26:32.324Z" },
+]
+
 [[package]]
 name = "langchain-text-splitters"
 version = "0.3.9"
@@ -1781,6 +1796,7 @@ dependencies = [
     { name = "langchain", extra = ["google-genai", "openai"] },
     { name = "langchain-community" },
     { name = "langchain-google-genai" },
+    { name = "langchain-tavily" },
     { name = "langgraph" },
     { name = "matplotlib" },
     { name = "openinference-instrumentation-langchain" },
@@ -1803,6 +1819,7 @@ requires-dist = [
     { name = "langchain", extras = ["google-genai", "googlegenai", "openai"], specifier = ">=0.3.26" },
     { name = "langchain-community", specifier = ">=0.3.27" },
     { name = "langchain-google-genai", specifier = ">=2.1.9" },
+    { name = "langchain-tavily", specifier = ">=0.2.11" },
     { name = "langgraph", specifier = ">=0.4.8" },
     { name = "matplotlib", specifier = ">=3.10.5" },
     { name = "openinference-instrumentation-langchain", specifier = ">=0.1.43" },