Enhancement Suite: Anti-loop protection, Kokoro TTS integration, ComfyUI status monitoring, and auto-open image functionality

build/lib/ollama_cli/tools/web.py

@@ -0,0 +1,67 @@
+import requests
+import urllib.parse
+from .base import tool
+
+@tool(
+    name="web_search",
+    description="Search the web for information",
+    parameters={
+        "query": {"type": "string", "description": "The search query"}
+    }
+)
+def web_search(query: str) -> str:
+    """Enhanced web search with DuckDuckGo"""
+    try:
+        encoded = urllib.parse.quote(query)
+        # Use DuckDuckGo Instant Answer API
+        api_url = f"https://api.duckduckgo.com/?q={encoded}&format=json"
+        response = requests.get(api_url, timeout=10)
+        data = response.json()
+        
+        results = []
+        if data.get("AbstractText"):
+            results.append(f"Summary: {data['AbstractText']}")
+        
+        for topic in data.get("RelatedTopics", [])[:5]:
+            if isinstance(topic, dict) and "Text" in topic:
+                results.append(f"- {topic['Text']}")
+        
+        return "\n".join(results) if results else "No direct results found. Try a different query."
+    except Exception as e:
+        return f"Search error: {str(e)}"
+
+@tool(
+    name="read_url",
+    description="Read and extract text content from a URL",
+    parameters={
+        "url": {"type": "string", "description": "The URL to read"}
+    }
+)
+def read_url(url: str) -> str:
+    try:
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()
+        
+        try:
+            from bs4 import BeautifulSoup
+            soup = BeautifulSoup(response.text, 'html.parser')
+            
+            # Remove scripts and styles
+            for script in soup(["script", "style"]):
+                script.decompose()
+                
+            text = soup.get_text()
+            
+            # Clean up whitespace
+            lines = (line.strip() for line in text.splitlines())
+            chunks = (phrase.strip() for line in lines for phrase in line.split("  "))
+            text = '\n'.join(chunk for chunk in chunks if chunk)
+            
+            return text[:10000] + ("\n... (truncated)" if len(text) > 10000 else "")
+            
+        except ImportError:
+            # Fallback if bs4 is missing (though we checked)
+            return response.text[:10000]
+            
+    except Exception as e:
+        return f"Error reading URL: {str(e)}"