dacp 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

dacp/tools.py

@@ -1,55 +1,105 @@
-from typing import Dict, Any, Callable
+"""
+DACP Tools - Built-in tool implementations.
+
+This module provides the core tool functionality including tool registry,
+execution, and built-in tools like file_writer.
+"""
+
+import logging
 from pathlib import Path
+from typing import Dict, Any, Callable
+
+logger = logging.getLogger("dacp.tools")
+
+# Global tool registry
+TOOL_REGISTRY: Dict[str, Callable[[Dict[str, Any]], Dict[str, Any]]] = {}
+
+
+def register_tool(name: str, func: Callable[[Dict[str, Any]], Dict[str, Any]]) -> None:
+    """
+    Register a tool function.
+
+    Args:
+        name: Name of the tool
+        func: Function that takes args dict and returns result dict
+    """
+    TOOL_REGISTRY[name] = func
+    logger.info(f"🔧 Tool '{name}' registered")
 
-TOOL_REGISTRY: Dict[str, Callable[..., Dict[str, Any]]] = {}
 
+def execute_tool(name: str, args: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Execute a tool by name with given arguments.
+
+    Args:
+        name: Name of the tool to execute
+        args: Arguments to pass to the tool
 
-def register_tool(tool_id: str, func: Callable[..., Dict[str, Any]]) -> None:
-    """Register a tool function."""
-    TOOL_REGISTRY[tool_id] = func
+    Returns:
+        Tool execution result
 
+    Raises:
+        ValueError: If tool is not found
+    """
+    if name not in TOOL_REGISTRY:
+        available_tools = list(TOOL_REGISTRY.keys())
+        logger.error(
+            f"❌ Tool '{name}' not found. " f"Available tools: {available_tools}"
+        )
+        raise ValueError(f"Tool '{name}' not found. Available tools: {available_tools}")
 
-def run_tool(tool_id: str, args: Dict[str, Any]) -> Dict[str, Any]:
-    """Run a registered tool with the given arguments."""
-    if tool_id not in TOOL_REGISTRY:
-        raise ValueError(f"Unknown tool: {tool_id}")
+    logger.debug(f"🛠️  Executing tool '{name}' with args: {args}")
 
-    tool_func = TOOL_REGISTRY[tool_id]
-    return tool_func(**args)
+    try:
+        result = TOOL_REGISTRY[name](args)
+        logger.debug(f"✅ Tool '{name}' completed successfully")
+        return result
+    except Exception as e:
+        logger.error(f"❌ Tool '{name}' failed: {type(e).__name__}: {e}")
+        raise
 
 
-def file_writer(path: str, content: str) -> Dict[str, Any]:
+def file_writer(args: Dict[str, Any]) -> Dict[str, Any]:
     """
-    Write content to a file, creating parent directories if they don't exist.
+    Write content to a file, creating directories as needed.
 
     Args:
-        path: File path to write to
-        content: Content to write to the file
+        args: Dictionary containing 'path' and 'content'
 
     Returns:
-        Dict with success status and file path
+        Success status and file information
     """
+    path = args.get("path")
+    content = args.get("content", "")
+
+    if not path:
+        raise ValueError("file_writer requires 'path' argument")
+
     try:
         # Create parent directories if they don't exist
-        Path(path).parent.mkdir(parents=True, exist_ok=True)
+        file_path = Path(path)
+        file_path.parent.mkdir(parents=True, exist_ok=True)
 
-        # Write the content to the file
-        with open(path, "w", encoding="utf-8") as f:
+        # Write content to file
+        with open(file_path, "w", encoding="utf-8") as f:
             f.write(content)
 
+        logger.info(f"✅ File written successfully: {path}")
+
         return {
             "success": True,
-            "path": path,
+            "path": str(file_path),
             "message": f"Successfully wrote {len(content)} characters to {path}",
         }
+
     except Exception as e:
+        logger.error(f"❌ Failed to write file {path}: {e}")
         return {
             "success": False,
             "path": path,
             "error": str(e),
-            "message": f"Failed to write to {path}: {e}",
         }
 
 
-# Register the built-in file_writer tool
+# Register built-in tools
 register_tool("file_writer", file_writer)