Update src/tools/tools.py

src/tools/tools.py

@@ -365,7 +365,11 @@ def arxiv_search(
     
 
 @tool
-def web_extract(urls : List[str]) -> str:
+def web_extract(
+    urls: List[str] | str,
+    include_images: bool = False,
+    extract_depth: str = "basic",
+) -> str:
     """
     Extract text content from web pages using TavilyExtract.
 
@@ -377,14 +381,19 @@ def web_extract(urls : List[str]) -> str:
         include_images (bool) – add image URLs if True
         extract_depth (str) – "basic" (default) or "advanced"
     """
+    # нормализуем вход
+    if isinstance(urls, str):
+        urls = [urls]
 
     tool = TavilyExtract(
-    extract_depth="basic",
-    include_images=False,
-)
-    results = tool.invoke(urls)
+        extract_depth=extract_depth,
+        include_images=include_images,
+    )
+    # ВАЖНО: .invoke ждёт словарь по схеме TavilyExtractInput
+    results = tool.invoke({"urls": urls})
     return json.dumps(results)
 
+
 @tool
 def extract_youtube_transcript(url: str, chars: int = 10_00) -> str:
     """