Update tools.py

tools.py

@@ -1,3 +1,4 @@
+import io
 import os
 import re
 import sys
@@ -55,28 +56,7 @@ def web_search(query: str, k: int = 6) -> str:
         hits = ddg_search(query, k)
         if hits:
             return json.dumps(hits, ensure_ascii=False)
-    except Exception:  # fall through to Tavily
-        pass
 
-    try:
-        tavily_results = TavilySearchResults(
-            max_results=5,
-            # include_answer=True,
-            # search_depth="advanced",
-        )
-        search_result = tavily_results.invoke({"query": query})
-        print(
-            f"[TOOL] TAVILY search is triggered with following response: {search_result}"
-        )
-        formatted = [
-            {
-                "title": d.get("title", "")[:500],
-                "snippet": d.get("content", "")[:750],
-                "link": d.get("url", "")[:300],
-            }
-            for d in search_result
-        ]
-        return json.dumps(formatted, ensure_ascii=False)
     except Exception as exc:
         return f"search_error:{exc}"
 
@@ -105,22 +85,18 @@ def get_youtube_transcript(video_url: str) -> str:
         return "TRANSCRIPT_UNAVAILABLE"
 
 @tool
-def describe_image(img_bytes, question: str) -> str:
+def describe_image(img_bytes: bytes, question: str) -> str:
     """Use a vision model to interpret or answer questions about an image file.
 
     Args:
-        image_path: The local path to the image file (.png, .jpg, .jpeg, .gif, .webp).
+        img_bytes: Raw image bytes.
         question: Specific question to ask about the image content.
 
     Returns:
         A text description or answer about the image content.
     """
-
-    mime_map = {"jpg": "image/jpeg", "jpeg": "image/jpeg", "png": "image/png",
-                "gif": "image/gif", "webp": "image/webp"}
     mime_type = "image/png"
-
-    image_data = base64.standard_b64encode(img_bytes.read()).decode("utf-8")
+    image_data = base64.standard_b64encode(img_bytes).decode("utf-8")
 
     payload = {
         "model": "nvidia/nemotron-nano-12b-v2-vl:free",