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

build/lib/ollama_cli/tools/base.py

@@ -0,0 +1,55 @@
+from typing import Dict, Any, Callable, List, Optional
+import json
+import inspect
+
+class BaseTool:
+    def __init__(self, name: str, description: str, parameters: Dict[str, Any]):
+        self.name = name
+        self.description = description
+        self.parameters = parameters
+
+    def execute(self, **kwargs) -> Any:
+        raise NotImplementedError
+
+class ToolRegistry:
+    def __init__(self):
+        self.tools: Dict[str, BaseTool] = {}
+
+    def register(self, tool: BaseTool):
+        self.tools[tool.name] = tool
+
+    def get_tool(self, name: str) -> Optional[BaseTool]:
+        return self.tools.get(name)
+
+    def list_tools(self) -> List[Dict[str, Any]]:
+        return [
+            {
+                "name": t.name,
+                "description": t.description,
+                "parameters": t.parameters
+            }
+            for t in self.tools.values()
+        ]
+
+    def generate_system_prompt_snippet(self) -> str:
+        snippet = "Available tools:\n"
+        for t in self.tools.values():
+            snippet += f"- {t.name}: {t.description}\n"
+            snippet += f"  Parameters: {json.dumps(t.parameters)}\n"
+        return snippet
+
+# Singleton registry
+registry = ToolRegistry()
+
+def tool(name: str, description: str, parameters: Dict[str, Any]):
+    """Decorator to register a function as a tool"""
+    def decorator(func: Callable):
+        class DecoratedTool(BaseTool):
+            def __init__(self):
+                super().__init__(name, description, parameters)
+            def execute(self, **kwargs):
+                return func(**kwargs)
+        
+        registry.register(DecoratedTool())
+        return func
+    return decorator