veris-ai 1.8.0__tar.gz → 1.8.1__tar.gz

{veris_ai-1.8.0 → veris_ai-1.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veris-ai
-Version: 1.8.0
+Version: 1.8.1
 Summary: A Python package for Veris AI tools
 Project-URL: Homepage, https://github.com/veris-ai/veris-python-sdk
 Project-URL: Bug Tracker, https://github.com/veris-ai/veris-python-sdk/issues
@@ -168,7 +168,7 @@ async def your_function(param1: str, param2: int) -> dict:
     return {"result": "actual implementation"}
 
 # Spy mode: Executes function but logs calls/responses
-@veris.mock(mode="spy")
+@veris.spy()
 async def monitored_function(data: str) -> dict:
     return process_data(data)
 

{veris_ai-1.8.0 → veris_ai-1.8.1}/README.md

@@ -126,7 +126,7 @@ async def your_function(param1: str, param2: int) -> dict:
     return {"result": "actual implementation"}
 
 # Spy mode: Executes function but logs calls/responses
-@veris.mock(mode="spy")
+@veris.spy()
 async def monitored_function(data: str) -> dict:
     return process_data(data)
 

{veris_ai-1.8.0 → veris_ai-1.8.1}/examples/README.md

@@ -64,7 +64,7 @@ if os.getenv("USE_FASTAPI") == "true":
 
 ### Decorator Usage
 - **Function Mocking**: `@veris.mock()` for simulation mode
-- **Spy Mode**: `@veris.mock(mode="spy")` for logging
+- **Spy Mode**: `@veris.spy()` for logging
 - **Stub Mode**: `@veris.stub(return_value={})` for fixed responses
 
 ### Error Handling

{veris_ai-1.8.0 → veris_ai-1.8.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "veris-ai"
-version = "1.8.0"
+version = "1.8.1"
 description = "A Python package for Veris AI tools"
 readme = "README.md"
 requires-python = ">=3.11"

veris_ai-1.8.1/src/veris_ai/agents_wrapper.py

@@ -0,0 +1,283 @@
+"""OpenAI Agents wrapper for automatic tool mocking via Veris SDK."""
+
+import logging
+from collections.abc import Callable
+from typing import Any
+
+from agents import RunContextWrapper, RunResult, Runner
+from pydantic import BaseModel
+
+from veris_ai import veris
+from veris_ai.tool_mock import mock_tool_call
+
+logger = logging.getLogger(__name__)
+
+
+def _wrap(
+    include_tools: list[str] | None = None,
+    exclude_tools: list[str] | None = None,
+) -> Callable:
+    """Private wrapper for OpenAI agents Runner to intercept tool calls through Veris SDK.
+
+    This function transparently intercepts tool calls from OpenAI agents and
+    routes them through the Veris SDK's mocking infrastructure.
+
+    Args:
+        include_tools: Optional list of tool names to intercept (only these if provided)
+        exclude_tools: Optional list of tool names to NOT intercept (these run normally)
+
+    Returns:
+        A wrapped Runner.run function
+
+    Raises:
+        ValueError: If both include_tools and exclude_tools are specified
+        ImportError: If agents package is not installed
+    """
+    if include_tools and exclude_tools:
+        msg = "Cannot specify both include_tools and exclude_tools"
+        raise ValueError(msg)
+
+    def wrapped_run_func(run_func: Callable) -> Callable:
+        """Inner wrapper that takes the actual Runner.run function."""
+        try:
+            from agents import FunctionTool  # type: ignore[import-untyped] # noqa: PLC0415
+        except ImportError as e:
+            msg = "openai-agents package not installed. Install with: pip install veris-ai[agents]"
+            raise ImportError(msg) from e
+
+        async def wrapped_run(starting_agent: Any, input_text: str, **kwargs: Any) -> Any:  # noqa: ANN401
+            """Wrapped version of Runner.run that intercepts tool calls."""
+            # Store a mapping of tools to their original functions
+            tool_functions = {}
+
+            if hasattr(starting_agent, "tools") and starting_agent.tools:
+                for tool in starting_agent.tools:
+                    if isinstance(tool, FunctionTool):
+                        tool_name = getattr(tool, "name", None)
+
+                        # Check if we should patch this tool
+                        if tool_name and _should_intercept_tool(
+                            tool_name, include_tools, exclude_tools
+                        ):
+                            # Extract the original function before patching
+                            original_func = _extract_the_func(tool.on_invoke_tool)
+                            if original_func:
+                                tool_functions[id(tool)] = original_func
+
+                            # Store original on_invoke_tool
+                            original_on_invoke = tool.on_invoke_tool
+
+                            def make_wrapped_on_invoke_tool(
+                                tool_id: int, orig_invoke: Callable
+                            ) -> Callable:
+                                """Create a wrapped on_invoke_tool with proper closure."""
+
+                                async def wrapped_on_invoke_tool(
+                                    ctx: RunContextWrapper[Any], parameters: str
+                                ) -> Any:  # noqa: ANN401
+                                    """Wrapped on_invoke_tool that intercepts the tool function."""
+                                    session_id = veris.session_id
+                                    the_func = tool_functions.get(tool_id)
+                                    if the_func and session_id:
+                                        # mock_tool_call is synchronous, don't await it
+                                        return mock_tool_call(
+                                            the_func, session_id, parameters, None
+                                        )
+                                    # Fall back to original if we couldn't extract the function
+                                    return await orig_invoke(ctx, parameters)
+
+                                return wrapped_on_invoke_tool
+
+                            tool.on_invoke_tool = make_wrapped_on_invoke_tool(
+                                id(tool), original_on_invoke
+                            )
+            return await run_func(starting_agent, input_text, **kwargs)
+
+        return wrapped_run
+
+    return wrapped_run_func
+
+
+def _should_intercept_tool(
+    tool_name: str,
+    include_tools: list[str] | None,
+    exclude_tools: list[str] | None,
+) -> bool:
+    """Determine if a tool should be intercepted based on include/exclude lists.
+
+    Args:
+        tool_name: Name of the tool
+        include_tools: If provided, only these tools are intercepted
+        exclude_tools: If provided, these tools are NOT intercepted
+
+    Returns:
+        True if the tool should be intercepted, False otherwise
+    """
+    if include_tools:
+        return tool_name in include_tools
+    if exclude_tools:
+        return tool_name not in exclude_tools
+    return True
+
+
+def _extract_the_func(on_invoke_tool: Callable) -> Callable | None:
+    """Extract the original user function from the on_invoke_tool closure.
+
+    This function attempts multiple strategies to extract the original function:
+    1. Direct attribute access (if the tool stores it)
+    2. Closure inspection for known patterns
+    3. Deep closure traversal as a fallback
+
+    Args:
+        on_invoke_tool: The on_invoke_tool function from FunctionTool
+
+    Returns:
+        The original user function if found, None otherwise
+    """
+
+    # Strategy 1: Check if the tool has stored the original function as an attribute
+    # (This would be the cleanest approach if the agents library supported it)
+    if hasattr(on_invoke_tool, "__wrapped__"):
+        return on_invoke_tool.__wrapped__
+
+    # Strategy 2: Look for the function in the closure using known structure
+    # Based on the agents library implementation, we know:
+    # - on_invoke_tool has _on_invoke_tool_impl in its closure
+    # - _on_invoke_tool_impl has the_func in its closure
+
+    if not hasattr(on_invoke_tool, "__closure__") or not on_invoke_tool.__closure__:
+        return None
+
+    # Find _on_invoke_tool_impl by looking for a function with that name pattern
+    for cell in on_invoke_tool.__closure__:
+        try:
+            obj = cell.cell_contents
+            if not callable(obj):
+                continue
+
+            # Check if this looks like _on_invoke_tool_impl
+            if (
+                hasattr(obj, "__name__")
+                and "_on_invoke_tool_impl" in obj.__name__
+                and hasattr(obj, "__closure__")
+                and obj.__closure__
+            ):
+                # Now look for the_func in its closure
+                return _find_user_function_in_closure(obj.__closure__)
+        except (ValueError, AttributeError):
+            continue
+
+    # Strategy 3: Fallback - do a broader search in the closure
+    return _find_user_function_in_closure(on_invoke_tool.__closure__)
+
+
+def _find_user_function_in_closure(closure: tuple) -> Callable | None:
+    """Find the user function in a closure by filtering out known library functions.
+
+    Args:
+        closure: The closure tuple to search
+
+    Returns:
+        The user function if found, None otherwise
+    """
+    import inspect
+
+    # List of module prefixes that indicate library/framework code
+    library_modules = ("json", "inspect", "agents", "pydantic", "openai", "typing")
+
+    for cell in closure:
+        try:
+            obj = cell.cell_contents
+
+            # Must be callable but not a class
+            if not callable(obj) or isinstance(obj, type):
+                continue
+
+            # Skip private/internal functions
+            if hasattr(obj, "__name__") and obj.__name__.startswith("_"):
+                continue
+
+            # Check the module to filter out library code
+            module = inspect.getmodule(obj)
+            if module:
+                # Skip if it's from a known library
+                if module.__name__.startswith(library_modules):
+                    continue
+
+                # Skip if it's from site-packages (library code)
+                if (
+                    hasattr(module, "__file__")
+                    and module.__file__
+                    and "site-packages" in module.__file__
+                    # Unless it's user code installed as a package
+                    # (this is a heuristic - may need adjustment)
+                    and not any(pkg in module.__name__ for pkg in ["my_", "custom_", "app_"])
+                ):
+                    continue
+
+            # If we made it here, this is likely the user function
+            return obj
+
+        except (ValueError, AttributeError):
+            continue
+
+    return None
+
+
+class VerisConfig(BaseModel):
+    """Configuration for the Veris SDK."""
+
+    include_tools: list[str] | None = None
+    exclude_tools: list[str] | None = None
+
+
+def veris_runner(
+    starting_agent: Any,  # noqa: ANN401
+    input_text: str,
+    veris_config: VerisConfig | None = None,
+    **kwargs: Any,  # noqa: ANN401
+) -> RunResult:  # noqa: ANN401
+    """Veris-wrapped version of OpenAI agents Runner.run.
+
+    This function wraps the OpenAI agents Runner.run to intercept tool calls
+    and route them through the Veris SDK's mocking infrastructure. It can be
+    used as a drop-in replacement for Runner.run with an additional veris_config parameter.
+
+    Args:
+        starting_agent: The OpenAI agent to run
+        input_text: The input text to process
+        veris_config: Optional configuration for Veris SDK tool interception
+        **kwargs: Additional arguments to pass to Runner.run
+
+    Returns:
+        The result from Runner.run
+
+    Example:
+        ```python
+        from veris_ai import veris_runner, VerisConfig
+        from agents import Agent, FunctionTool
+
+        # Define your agent with tools
+        agent = Agent(...)
+
+        # Use veris_runner instead of Runner.run
+        result = await veris_runner(agent, "Process this input")
+
+        # Or with specific tool configuration
+        config = VerisConfig(include_tools=["calculator", "search"])
+        result = await veris_runner(agent, "Calculate 2+2", veris_config=config)
+        ```
+    """
+
+    # Extract config values
+    include_tools = None
+    exclude_tools = None
+    if veris_config:
+        include_tools = veris_config.include_tools
+        exclude_tools = veris_config.exclude_tools
+
+    # Create the wrapped version of Runner.run with the config
+    wrapped_run = _wrap(include_tools=include_tools, exclude_tools=exclude_tools)(Runner.run)
+
+    # Execute the wrapped run function
+    return wrapped_run(starting_agent, input_text, **kwargs)

{veris_ai-1.8.0 → veris_ai-1.8.1}/src/veris_ai/api_client.py

@@ -29,15 +29,7 @@ class SimulatorAPIClient:
 
         return headers
 
-    async def post_async(self, endpoint: str, payload: dict[str, Any]) -> Any:  # noqa: ANN401
-        """Make an async POST request to the specified endpoint."""
-        headers = self._build_headers()
-        async with httpx.AsyncClient(timeout=self.timeout) as client:
-            response = await client.post(endpoint, json=payload, headers=headers)
-            response.raise_for_status()
-            return response.json() if response.content else None
-
-    def post_sync(self, endpoint: str, payload: dict[str, Any]) -> Any:  # noqa: ANN401
+    def post(self, endpoint: str, payload: dict[str, Any]) -> Any:  # noqa: ANN401
         """Make a synchronous POST request to the specified endpoint."""
         headers = self._build_headers()
         with httpx.Client(timeout=self.timeout) as client:

veris_ai-1.8.1/src/veris_ai/logging.py

@@ -0,0 +1,46 @@
+"""Logging utilities for VERIS tool calls and responses."""
+
+import json
+import logging
+from typing import Any
+
+from veris_ai.api_client import get_api_client
+
+logger = logging.getLogger(__name__)
+
+
+def log_tool_call(
+    session_id: str,
+    function_name: str,
+    parameters: str,
+    docstring: str,
+) -> None:
+    """Log tool call synchronously to the VERIS logging endpoint."""
+    api_client = get_api_client()
+    endpoint = api_client.get_log_tool_call_endpoint(session_id)
+    payload = {
+        "function_name": function_name,
+        "parameters": json.loads(parameters),
+        "docstring": docstring,
+    }
+    try:
+        api_client.post(endpoint, payload)
+        logger.debug(f"Tool call logged for {function_name}")
+    except Exception as e:
+        logger.warning(f"Failed to log tool call for {function_name}: {e}")
+
+
+def log_tool_response(session_id: str, response: Any) -> None:  # noqa: ANN401
+    """Log tool response synchronously to the VERIS logging endpoint."""
+    api_client = get_api_client()
+    endpoint = api_client.get_log_tool_response_endpoint(session_id)
+
+    payload = {
+        "response": json.dumps(response, default=str),
+    }
+
+    try:
+        api_client.post(endpoint, payload)
+        logger.debug("Tool response logged")
+    except Exception as e:
+        logger.warning(f"Failed to log tool response: {e}")

veris_ai-1.8.1/src/veris_ai/models.py

@@ -0,0 +1,22 @@
+"""Models for the VERIS SDK."""
+
+from enum import Enum
+from typing import Literal
+
+from pydantic import BaseModel
+
+
+class ResponseExpectation(str, Enum):
+    """Expected response behavior for tool mocking."""
+
+    AUTO = "auto"
+    REQUIRED = "required"
+    NONE = "none"
+
+
+class ToolCallOptions(BaseModel):
+    """Options for tool call."""
+
+    response_expectation: ResponseExpectation = ResponseExpectation.AUTO
+    cache_response: bool = False
+    mode: Literal["tool", "function", "spy"] = "tool"