tactus 0.31.2__py3-none-any.whl

tactus/primitives/tool.py

@@ -0,0 +1,375 @@
+"""
+Tool Primitive - Tool call tracking, result access, and direct invocation.
+
+Provides:
+- Tool.called(name) - Check if tool was called
+- Tool.last_result(name) - Get last result from named tool
+- Tool.last_call(name) - Get full call info
+- Tool.get(name) - Get a callable handle to an external tool (MCP, plugin)
+"""
+
+import logging
+from typing import Any, Optional, Dict, List, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from tactus.primitives.tool_handle import ToolHandle
+
+logger = logging.getLogger(__name__)
+
+
+class ToolCall:
+    """Represents a single tool call with arguments and result."""
+
+    def __init__(self, name: str, args: Dict[str, Any], result: Any):
+        self.name = name
+        self.args = args
+        self.result = result
+        self.timestamp = None  # Could add timestamp tracking
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dictionary for Lua access."""
+        return {"name": self.name, "args": self.args, "result": self.result}
+
+    def __repr__(self) -> str:
+        return f"ToolCall({self.name}, args={self.args})"
+
+
+class ToolPrimitive:
+    """
+    Tracks tool calls and provides access to results.
+
+    Maintains a history of tool calls and their results, allowing
+    Lua code to check what tools were used and access their outputs.
+
+    Also supports tool lookup via __call__:
+        Tool("done")({args})  -- Look up and call tool
+        Tool.called("done")   -- Check if tool was called (existing)
+    """
+
+    def __init__(
+        self, log_handler=None, agent_name: Optional[str] = None, procedure_id: Optional[str] = None
+    ):
+        """Initialize tool tracking."""
+        self._tool_calls: List[ToolCall] = []
+        self._last_calls: Dict[str, ToolCall] = {}  # name -> last call
+        self.log_handler = log_handler
+        self.agent_name = agent_name
+        self.procedure_id = procedure_id
+        self._runtime = None  # Will be set by runtime for Tool.get() support
+        self._tool_registry: Dict[str, "ToolHandle"] = {}  # For Tool("name") lookup
+        logger.debug("ToolPrimitive initialized")
+
+    def set_tool_registry(self, registry: Dict[str, "ToolHandle"]) -> None:
+        """
+        Set the tool registry for Tool("name") lookup.
+
+        Called by dsl_stubs.create_dsl_stubs() after tools are registered.
+
+        Args:
+            registry: Dict mapping tool names to ToolHandle instances
+        """
+        self._tool_registry = registry
+        logger.debug(f"ToolPrimitive tool registry set with {len(registry)} tools")
+
+    def __call__(self, tool_name: str) -> "ToolHandle":
+        """
+        Look up a tool by name for direct invocation.
+
+        This makes Tool dual-purpose:
+        - Tool("done")({args}) -- lookup and call
+        - Tool.called("done")  -- tracking (existing method)
+
+        Args:
+            tool_name: Name of the tool to look up
+
+        Returns:
+            ToolHandle that can be called directly
+
+        Raises:
+            ValueError: If tool not found in registry
+
+        Example (Lua):
+            Tool("done")({reason = "finished"})
+        """
+        if tool_name not in self._tool_registry:
+            available = list(self._tool_registry.keys())
+            raise ValueError(f"Tool '{tool_name}' not defined. Available tools: {available}")
+        return self._tool_registry[tool_name]
+
+    def set_runtime(self, runtime) -> None:
+        """
+        Set runtime reference for toolset lookup.
+
+        Called by the runtime after initialization to enable Tool.get().
+
+        Args:
+            runtime: TactusRuntime instance with toolset_registry
+        """
+        self._runtime = runtime
+        logger.debug("ToolPrimitive connected to runtime for toolset lookup")
+
+    def get(self, tool_name: str) -> "ToolHandle":
+        """
+        Get a callable handle to an external tool (MCP, plugin).
+
+        For Lua-defined tools, use the return value of tool() instead.
+
+        Args:
+            tool_name: Name of the external tool to retrieve
+
+        Returns:
+            ToolHandle that can be called directly
+
+        Raises:
+            ValueError: If tool is not found in registry
+
+        Example (Lua):
+            local search = Tool.get("web_search")  -- MCP tool
+            local result = search({query = "weather"})
+
+            local analyze = Tool.get("analyze_sentiment")  -- Plugin tool
+            local sentiment = analyze({text = "I love this!"})
+        """
+        from tactus.primitives.tool_handle import ToolHandle
+
+        logger.debug(f"Tool.get('{tool_name}') called")
+
+        # Look up toolset from runtime registry
+        toolset = self._get_toolset(tool_name)
+        if toolset is None:
+            raise ValueError(
+                f"Tool '{tool_name}' not found. "
+                f"Make sure it's registered as an MCP tool, plugin, or in a toolset."
+            )
+
+        # Extract the callable function from the toolset
+        tool_fn = self._extract_tool_function(toolset, tool_name)
+
+        logger.debug(f"Tool.get('{tool_name}') returning ToolHandle")
+        return ToolHandle(tool_name, tool_fn, self)
+
+    def _get_toolset(self, name: str) -> Optional[Any]:
+        """
+        Look up toolset from runtime registry.
+
+        Args:
+            name: Toolset name
+
+        Returns:
+            Toolset instance or None if not found
+        """
+        if self._runtime is None:
+            logger.warning("Tool.get() called but runtime not connected")
+            return None
+
+        if not hasattr(self._runtime, "toolset_registry"):
+            logger.warning("Runtime does not have toolset_registry")
+            return None
+
+        return self._runtime.toolset_registry.get(name)
+
+    def _extract_tool_function(self, toolset: Any, tool_name: str) -> Any:
+        """
+        Extract callable function from a pydantic-ai toolset.
+
+        Handles different toolset types (FunctionToolset, MCP, etc.)
+
+        Args:
+            toolset: Pydantic-ai toolset instance
+            tool_name: Name of the tool
+
+        Returns:
+            Callable function that executes the tool
+        """
+        # Try to get the tool function from common toolset patterns
+
+        # Pattern 1: FunctionToolset with tools list
+        if hasattr(toolset, "tools"):
+            tools = toolset.tools
+            if isinstance(tools, list):
+                for tool in tools:
+                    if hasattr(tool, "name") and tool.name == tool_name:
+                        # Return the tool's function
+                        if hasattr(tool, "function"):
+                            return tool.function
+                        elif callable(tool):
+                            return tool
+            # If tools is a dict
+            elif isinstance(tools, dict) and tool_name in tools:
+                tool = tools[tool_name]
+                if hasattr(tool, "function"):
+                    return tool.function
+                elif callable(tool):
+                    return tool
+
+        # Pattern 2: Toolset with a single tool (like individual Lua tools)
+        if hasattr(toolset, "function"):
+            return toolset.function
+
+        # Pattern 3: Toolset that is itself callable
+        if callable(toolset):
+            return toolset
+
+        # Pattern 4: MCP toolset - these wrap MCP tools
+        if hasattr(toolset, "_tools") or hasattr(toolset, "get_tool"):
+            # Try to get the specific tool
+            if hasattr(toolset, "get_tool"):
+                tool = toolset.get_tool(tool_name)
+                if tool:
+                    return tool
+
+            # Return a wrapper that calls through the toolset
+            def mcp_wrapper(args):
+                return toolset.call_tool(tool_name, args)
+
+            return mcp_wrapper
+
+        # Fallback: assume the toolset itself contains tool functions
+        logger.warning(
+            f"Could not extract tool function for '{tool_name}' from toolset type {type(toolset)}"
+        )
+
+        # Return a wrapper that attempts to call through the toolset
+        def fallback_wrapper(args):
+            if hasattr(toolset, "call"):
+                return toolset.call(tool_name, args)
+            raise RuntimeError(f"Cannot call tool '{tool_name}' - toolset type not supported")
+
+        return fallback_wrapper
+
+    def called(self, tool_name: str) -> bool:
+        """
+        Check if a tool was called at least once.
+
+        Args:
+            tool_name: Name of the tool to check
+
+        Returns:
+            True if the tool was called
+
+        Example (Lua):
+            if Tool.called("done") then
+                Log.info("Done tool was called")
+            end
+        """
+        called = tool_name in self._last_calls
+        logger.debug(f"Tool.called('{tool_name}') = {called}")
+        return called
+
+    def last_result(self, tool_name: str) -> Any:
+        """
+        Get the last result from a named tool.
+
+        Args:
+            tool_name: Name of the tool
+
+        Returns:
+            Last result from the tool, or None if never called
+
+        Example (Lua):
+            local result = Tool.last_result("search")
+            if result then
+                Log.info("Search found: " .. result)
+            end
+        """
+        if tool_name not in self._last_calls:
+            logger.debug(f"Tool.last_result('{tool_name}') = None (never called)")
+            return None
+
+        result = self._last_calls[tool_name].result
+        logger.debug(f"Tool.last_result('{tool_name}') = {result}")
+        return result
+
+    def last_call(self, tool_name: str) -> Optional[Dict[str, Any]]:
+        """
+        Get full information about the last call to a tool.
+
+        Args:
+            tool_name: Name of the tool
+
+        Returns:
+            Dictionary with 'name', 'args', 'result' or None if never called
+
+        Example (Lua):
+            local call = Tool.last_call("search")
+            if call then
+                Log.info("Search was called with: " .. call.args.query)
+                Log.info("Result: " .. call.result)
+            end
+        """
+        if tool_name not in self._last_calls:
+            logger.debug(f"Tool.last_call('{tool_name}') = None (never called)")
+            return None
+
+        call_dict = self._last_calls[tool_name].to_dict()
+        logger.debug(f"Tool.last_call('{tool_name}') = {call_dict}")
+        return call_dict
+
+    def record_call(
+        self, tool_name: str, args: Dict[str, Any], result: Any, agent_name: Optional[str] = None
+    ) -> None:
+        """
+        Record a tool call (called by runtime after tool execution).
+
+        Args:
+            tool_name: Name of the tool
+            args: Arguments passed to the tool
+            result: Result returned by the tool
+            agent_name: Optional name of agent that called the tool
+
+        Note: This is called internally by the runtime, not from Lua
+        """
+        call = ToolCall(tool_name, args, result)
+        self._tool_calls.append(call)
+        self._last_calls[tool_name] = call
+
+        logger.debug(f"Tool call recorded: {tool_name} -> {len(self._tool_calls)} total calls")
+
+        # Emit ToolCallEvent if we have a log handler
+        if self.log_handler:
+            try:
+                from tactus.protocols.models import ToolCallEvent
+
+                event = ToolCallEvent(
+                    agent_name=agent_name or self.agent_name or "unknown",
+                    tool_name=tool_name,
+                    tool_args=args,
+                    tool_result=result,
+                    procedure_id=self.procedure_id,
+                )
+                self.log_handler.log(event)
+            except Exception as e:
+                logger.warning(f"Failed to log tool call event: {e}")
+
+    def get_all_calls(self) -> List[ToolCall]:
+        """
+        Get all tool calls (for debugging/logging).
+
+        Returns:
+            List of all ToolCall objects
+        """
+        return self._tool_calls.copy()
+
+    def get_call_count(self, tool_name: Optional[str] = None) -> int:
+        """
+        Get the number of times a tool was called.
+
+        Args:
+            tool_name: Name of tool (or None for total count)
+
+        Returns:
+            Number of calls
+        """
+        if tool_name is None:
+            return len(self._tool_calls)
+
+        return sum(1 for call in self._tool_calls if call.name == tool_name)
+
+    def reset(self) -> None:
+        """Reset tool tracking (mainly for testing)."""
+        self._tool_calls.clear()
+        self._last_calls.clear()
+        logger.debug("Tool tracking reset")
+
+    def __repr__(self) -> str:
+        return f"ToolPrimitive({len(self._tool_calls)} calls)"

tactus/primitives/tool_handle.py

@@ -0,0 +1,279 @@
+"""
+Tool Handle - Callable wrapper for direct tool invocation.
+
+Provides OOP-style tool access where tool() returns a callable handle
+that can be invoked directly without going through an agent.
+"""
+
+import asyncio
+import logging
+from typing import Any, Callable, Dict, Optional, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from tactus.primitives.tool import ToolPrimitive
+
+logger = logging.getLogger(__name__)
+
+
+class ToolHandle:
+    """
+    Callable wrapper around a tool for direct invocation.
+
+    Returned by tool() for Lua-defined tools and Tool.get() for external tools.
+    Can be called directly from Lua: result = handle({args})
+
+    Example (Lua):
+        local calculate_tip = tool({...}, function(args) ... end)
+        local result = calculate_tip({bill_amount = 50, tip_percentage = 20})
+    """
+
+    def __init__(
+        self,
+        name: str,
+        impl_fn: Callable,
+        tool_primitive: Optional["ToolPrimitive"] = None,
+        is_async: bool = False,
+        record_calls: bool = True,
+    ):
+        """
+        Initialize a tool handle.
+
+        Args:
+            name: Tool name for tracking/logging
+            impl_fn: The actual function to execute
+            tool_primitive: Optional ToolPrimitive for call recording
+            is_async: Whether impl_fn is async (for MCP tools)
+        """
+        self.name = name
+        self.impl_fn = impl_fn
+        self.tool_primitive = tool_primitive
+        self.is_async = is_async
+        self.record_calls = record_calls
+
+        logger.debug(f"ToolHandle created for '{name}' (async={is_async})")
+
+    def call(self, args: Dict[str, Any]) -> Any:
+        """
+        Execute the tool with given arguments.
+
+        Args:
+            args: Dictionary of arguments to pass to the tool
+
+        Returns:
+            Tool result
+
+        Example (Lua):
+            local result = my_tool:call({arg1 = "value"})
+        """
+        logger.debug(f"ToolHandle.call('{self.name}') with args: {args}")
+
+        try:
+            # Convert Lua table to Python dict if needed
+            if hasattr(args, "items"):
+                args = self._lua_table_to_dict(args)
+
+            # Execute the implementation
+            if self.is_async or asyncio.iscoroutinefunction(self.impl_fn):
+                result = self._run_async(args)
+            else:
+                result = self.impl_fn(args)
+
+            # Record the call for tracking
+            if self.tool_primitive and self.record_calls:
+                self.tool_primitive.record_call(self.name, args, result)
+
+            logger.debug(f"ToolHandle.call('{self.name}') returned: {result}")
+            return result
+
+        except Exception as e:
+            logger.error(f"ToolHandle.call('{self.name}') failed: {e}", exc_info=True)
+            raise
+
+    def __call__(self, args: Dict[str, Any]) -> Any:
+        """
+        Make handle callable for shorthand syntax.
+
+        Example (Lua):
+            local result = my_tool({arg1 = "value"})
+        """
+        return self.call(args)
+
+    def called(self) -> bool:
+        """
+        Check if this tool has been called at least once.
+
+        Returns:
+            True if tool was called, False otherwise
+
+        Example (Lua):
+            if done.called() then
+                Log.info("Task completed!")
+            end
+        """
+        if not self.tool_primitive:
+            logger.warning(f"ToolHandle.called('{self.name}'): No tool_primitive attached")
+            return False
+
+        result = self.tool_primitive.called(self.name)
+        logger.debug(f"ToolHandle.called('{self.name}') = {result}")
+        return result
+
+    def last_call(self) -> Optional[Dict[str, Any]]:
+        """
+        Get the last call record for this tool.
+
+        Returns:
+            Dictionary with 'name', 'args', 'result' or None if never called
+
+        Example (Lua):
+            local call = multiply.last_call()
+            if call then
+                Log.info("Last multiply: " .. call.args.a .. " * " .. call.args.b)
+            end
+        """
+        if not self.tool_primitive:
+            logger.warning(f"ToolHandle.last_call('{self.name}'): No tool_primitive attached")
+            return None
+
+        result = self.tool_primitive.last_call(self.name)
+        logger.debug(f"ToolHandle.last_call('{self.name}') = {result}")
+        return result
+
+    def last_result(self) -> Any:
+        """
+        Get the result from the last call to this tool.
+
+        Returns:
+            Result value from last call, or None if never called
+
+        Example (Lua):
+            local answer = done.last_result()
+            return { result = answer }
+        """
+        if not self.tool_primitive:
+            logger.warning(f"ToolHandle.last_result('{self.name}'): No tool_primitive attached")
+            return None
+
+        result = self.tool_primitive.last_result(self.name)
+        logger.debug(f"ToolHandle.last_result('{self.name}') = {result}")
+        return result
+
+    def call_count(self) -> int:
+        """
+        Get the number of times this tool has been called.
+
+        Returns:
+            Number of calls (0 if never called)
+
+        Example (Lua):
+            local count = multiply.call_count()
+            Log.info("Multiply was called " .. count .. " times")
+        """
+        if not self.tool_primitive:
+            logger.warning(f"ToolHandle.call_count('{self.name}'): No tool_primitive attached")
+            return 0
+
+        # Count all calls with this tool name
+        count = sum(1 for call in self.tool_primitive._tool_calls if call.name == self.name)
+        logger.debug(f"ToolHandle.call_count('{self.name}') = {count}")
+        return count
+
+    def reset(self) -> None:
+        """
+        Clear all recorded calls for this tool.
+
+        This is useful when reusing the same tool handle in multiple sequential
+        operations within a single procedure, allowing called() checks to work
+        independently for each operation.
+
+        Example (Lua):
+            -- First agent uses done
+            agent1()
+            if done.called() then
+                Log.info("Agent 1 completed")
+            end
+
+            -- Reset for second agent
+            done.reset()
+
+            -- Second agent uses done independently
+            agent2()
+            if done.called() then
+                Log.info("Agent 2 completed")
+            end
+        """
+        if not self.tool_primitive:
+            logger.warning(f"ToolHandle.reset('{self.name}'): No tool_primitive attached")
+            return
+
+        # Remove all calls for this tool
+        self.tool_primitive._tool_calls = [
+            call for call in self.tool_primitive._tool_calls if call.name != self.name
+        ]
+        logger.debug(f"ToolHandle.reset('{self.name}'): Cleared all call records")
+
+    def _run_async(self, args: Dict[str, Any]) -> Any:
+        """
+        Run async function from sync context.
+
+        Handles the complexity of running async code from Lua's sync context.
+        """
+        try:
+            # Try to get a running event loop
+            loop = asyncio.get_running_loop()
+
+            # We're in an async context - use nest_asyncio if available
+            try:
+                import nest_asyncio
+
+                nest_asyncio.apply(loop)
+                return asyncio.run(self.impl_fn(args))
+            except ImportError:
+                # nest_asyncio not available, fall back to threading
+                import threading
+
+                result_container = {"value": None, "exception": None}
+
+                def run_in_thread():
+                    try:
+                        new_loop = asyncio.new_event_loop()
+                        asyncio.set_event_loop(new_loop)
+                        try:
+                            result_container["value"] = new_loop.run_until_complete(
+                                self.impl_fn(args)
+                            )
+                        finally:
+                            new_loop.close()
+                    except Exception as e:
+                        result_container["exception"] = e
+
+                thread = threading.Thread(target=run_in_thread)
+                thread.start()
+                thread.join()
+
+                if result_container["exception"]:
+                    raise result_container["exception"]
+                return result_container["value"]
+
+        except RuntimeError:
+            # No event loop running - safe to use asyncio.run()
+            return asyncio.run(self.impl_fn(args))
+
+    def _lua_table_to_dict(self, lua_table) -> Dict[str, Any]:
+        """Convert a Lua table to Python dict recursively."""
+        if lua_table is None:
+            return {}
+
+        if not hasattr(lua_table, "items"):
+            return lua_table
+
+        result = {}
+        for key, value in lua_table.items():
+            if hasattr(value, "items"):
+                result[key] = self._lua_table_to_dict(value)
+            else:
+                result[key] = value
+        return result
+
+    def __repr__(self) -> str:
+        return f"ToolHandle('{self.name}')"