Update agent.py

agent.py

@@ -7,12 +7,12 @@ agent.py  –  LangGraph-Agent mit
 import os, base64, mimetypes, subprocess, json, tempfile
 from typing import Any
 
-from langgraph.graph           import START, StateGraph, MessagesState
-from langgraph.prebuilt        import tools_condition, ToolNode
-from langchain_core.tools      import tool
-from langchain_core.messages   import SystemMessage, HumanMessage
-from langchain_google_genai    import ChatGoogleGenerativeAI
-from langchain_community.tools.duckduckgo_search import DuckDuckGoSearchResults as DDGS
+from langgraph.graph import START, StateGraph, MessagesState
+from langgraph.prebuilt import tools_condition, ToolNode
+from langchain_core.tools import tool
+from langchain_core.messages import SystemMessage, HumanMessage
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_community.tools.tavily_search import TavilySearchResults
 
 # ----------------------------------------------------------------------
 # 1 ── ENV / LLM
@@ -134,13 +134,14 @@ def ocr_image(file_path: str, lang: str = "eng") -> str:
 # ----------------------------------------------------------------------
 @tool
 @error_guard
-def web_search(query: str) -> str:
-    """DuckDuckGo text search – top 5 results."""
-    with DDGS() as ddgs:
-        results = ddgs.text(query, max_results=5)
-    if not results:
+def web_search(query: str, max_results: int = 5) -> str:
+    """Tavily web search – returns markdown list of results."""
+    search = TavilySearchResults(max_results=max_results)
+    hits = search.invoke(query)
+    if not hits:
         return "ERROR: no results."
-    return "\n\n".join(f"{r['title']} – {r['href']}" for r in results)
+    return "\n\n".join(f"{hit['title']} – {hit['url']}" for hit in hits)
+
 
 # ----------------------------------------------------------------------
 # 7 ── SYSTEM-PROMPT