veris-ai 1.13.0__py3-none-any.whl → 1.14.1__py3-none-any.whl

veris_ai/api_client.py

@@ -20,9 +20,15 @@ class SimulatorAPIClient:
         changes reflected without recreating the singleton.
     """
 
-    def __init__(self) -> None:
+    def __init__(self, timeout: float | None = None, base_url: str | None = None) -> None:
         """Initialize the API client with static timeout configuration."""
-        self.timeout = float(os.getenv("VERIS_MOCK_TIMEOUT", "90.0"))
+        self._timeout = timeout or float(os.getenv("VERIS_MOCK_TIMEOUT", "300.0"))
+        self._base_url = base_url
+
+    @property
+    def base_url(self) -> str:
+        """Get the resolved base URL."""
+        return self._get_base_url()
 
     def _get_base_url(self) -> str:
         """Resolve the base URL from environment.
@@ -33,7 +39,7 @@ class SimulatorAPIClient:
               (do not fall back). This supports tests expecting connection
               failures when an invalid endpoint is provided.
         """
-        return os.getenv("VERIS_API_URL") or "https://simulator.api.veris.ai"
+        return self._base_url or os.getenv("VERIS_API_URL") or "https://simulator.api.veris.ai"
 
     def _build_headers(self) -> dict[str, str] | None:
         """Build headers including OpenTelemetry tracing and API key."""
@@ -55,7 +61,7 @@ class SimulatorAPIClient:
         if not endpoint.startswith(("http://", "https://")):
             raise httpx.ConnectError("Invalid endpoint URL (not absolute): {endpoint}")
 
-        with httpx.Client(timeout=self.timeout) as client:
+        with httpx.Client(timeout=self._timeout) as client:
             response = client.post(endpoint, json=payload, headers=headers)
             response.raise_for_status()
             return response.json() if response.content else None
@@ -73,7 +79,7 @@ class SimulatorAPIClient:
             error_msg = f"Invalid endpoint URL (not absolute): {endpoint}"
             raise httpx.ConnectError(error_msg)
 
-        async with httpx.AsyncClient(timeout=self.timeout) as client:
+        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
@@ -81,15 +87,15 @@ class SimulatorAPIClient:
     @property
     def tool_mock_endpoint(self) -> str:
         """Get the tool mock endpoint URL."""
-        return urljoin(self._get_base_url(), "v3/tool_mock")
+        return urljoin(self.base_url, "v3/tool_mock")
 
     def get_log_tool_call_endpoint(self, session_id: str) -> str:
         """Get the log tool call endpoint URL."""
-        return urljoin(self._get_base_url(), f"v3/log_tool_call?session_id={session_id}")
+        return urljoin(self.base_url, f"v3/log_tool_call?session_id={session_id}")
 
     def get_log_tool_response_endpoint(self, session_id: str) -> str:
         """Get the log tool response endpoint URL."""
-        return urljoin(self._get_base_url(), f"v3/log_tool_response?session_id={session_id}")
+        return urljoin(self.base_url, f"v3/log_tool_response?session_id={session_id}")
 
 
 # Global singleton instance
@@ -99,3 +105,9 @@ _api_client = SimulatorAPIClient()
 def get_api_client() -> SimulatorAPIClient:
     """Get the global API client instance."""
     return _api_client
+
+
+def set_api_client_params(base_url: str | None = None, timeout: float | None = None) -> None:
+    """Set the global API client instance for testing purposes."""
+    global _api_client  # noqa: PLW0603
+    _api_client = SimulatorAPIClient(base_url=base_url, timeout=timeout)

veris_ai/tool_mock.py

@@ -16,14 +16,16 @@ from typing import (
 
 
 from veris_ai.models import ResponseExpectation, ToolCallOptions
-from veris_ai.api_client import get_api_client
+from veris_ai.api_client import get_api_client, set_api_client_params
 from veris_ai.utils import (
     convert_to_type,
     execute_callback,
+    execute_combined_callback,
     extract_json_schema,
     get_function_parameters,
     get_input_parameters,
     launch_callback_task,
+    launch_combined_callback_task,
 )
 
 logger = logging.getLogger(__name__)
@@ -90,6 +92,8 @@ class VerisSDK:
 
             self._set_session_id(token_data["session_id"])
             self._set_thread_id(token_data["thread_id"])
+            if token_data.get("api_url"):
+                set_api_client_params(base_url=token_data["api_url"])
             logger.info(
                 f"Session ID set to {token_data['session_id']}, "
                 f"Thread ID set to {token_data['thread_id']} - mocking enabled"
@@ -245,13 +249,14 @@ class VerisSDK:
 
         return decorator
 
-    def mock(  # noqa: C901, PLR0915
+    def mock(  # noqa: C901, PLR0915, PLR0913
         self,
         mode: Literal["tool", "function"] = "tool",
         expects_response: bool | None = None,
         cache_response: bool | None = None,
         input_callback: Callable[..., Any] | None = None,
         output_callback: Callable[[Any], Any] | None = None,
+        combined_callback: Callable[..., Any] | None = None,
     ) -> Callable:
         """Decorator for mocking tool calls.
 
@@ -261,6 +266,7 @@ class VerisSDK:
             cache_response: Whether to cache the response
             input_callback: Callable that receives input parameters as individual arguments
             output_callback: Callable that receives the output value
+            combined_callback: Callable that receives both input parameters and mock_output
         """
         response_expectation = (
             ResponseExpectation.NONE
@@ -306,6 +312,7 @@ class VerisSDK:
                 input_params = get_input_parameters(func, args, kwargs)
                 launch_callback_task(input_callback, input_params, unpack=True)
                 launch_callback_task(output_callback, result, unpack=False)
+                launch_combined_callback_task(combined_callback, input_params, result)
 
                 return result
 
@@ -337,6 +344,7 @@ class VerisSDK:
                 input_params = get_input_parameters(func, args, kwargs)
                 execute_callback(input_callback, input_params, unpack=True)
                 execute_callback(output_callback, result, unpack=False)
+                execute_combined_callback(combined_callback, input_params, result)
 
                 return result
 
@@ -350,6 +358,7 @@ class VerisSDK:
         return_value: Any,  # noqa: ANN401
         input_callback: Callable[..., Any] | None = None,
         output_callback: Callable[[Any], Any] | None = None,
+        combined_callback: Callable[..., Any] | None = None,
     ) -> Callable:
         """Decorator for stubbing tool calls.
 
@@ -357,6 +366,7 @@ class VerisSDK:
             return_value: The value to return when the function is stubbed
             input_callback: Callable that receives input parameters as individual arguments
             output_callback: Callable that receives the output value
+            combined_callback: Callable that receives both input parameters and mock_output
         """
 
         def decorator(func: Callable) -> Callable:
@@ -380,6 +390,7 @@ class VerisSDK:
                 input_params = get_input_parameters(func, args, kwargs)
                 launch_callback_task(input_callback, input_params, unpack=True)
                 launch_callback_task(output_callback, return_value, unpack=False)
+                launch_combined_callback_task(combined_callback, input_params, return_value)
 
                 return return_value
 
@@ -397,6 +408,7 @@ class VerisSDK:
                 input_params = get_input_parameters(func, args, kwargs)
                 execute_callback(input_callback, input_params, unpack=True)
                 execute_callback(output_callback, return_value, unpack=False)
+                execute_combined_callback(combined_callback, input_params, return_value)
 
                 return return_value
 

veris_ai/utils.py

@@ -489,3 +489,75 @@ def launch_callback_task(
     except RuntimeError:
         # If no event loop is running, log a warning
         logger.warning("Cannot launch callback task: no event loop running")
+
+
+def execute_combined_callback(
+    callback: Callable | None,
+    input_params: dict[str, Any],
+    mock_output: Any,  # noqa: ANN401
+) -> None:
+    """Execute a combined callback synchronously with input parameters and mock output.
+
+    Args:
+        callback: The callback callable to execute
+        input_params: Dictionary of input parameters
+        mock_output: The output from the mock/stub call
+
+    Note:
+        Exceptions in callbacks are caught and logged to prevent breaking the main flow.
+    """
+    if callback is None:
+        return
+
+    try:
+        # Combine input params with mock_output
+        combined_data = {**input_params, "mock_output": mock_output}
+        # Filter parameters to match callback signature
+        filtered_data = filter_callback_parameters(callback, combined_data)
+        callback(**filtered_data)
+    except Exception as e:
+        logger.warning(f"Combined callback execution failed: {e}", exc_info=True)
+
+
+def launch_combined_callback_task(
+    callback: Callable | None,
+    input_params: dict[str, Any],
+    mock_output: Any,  # noqa: ANN401
+) -> None:
+    """Launch a combined callback as a background task (fire-and-forget).
+
+    Args:
+        callback: The callback callable to execute (can be sync or async)
+        input_params: Dictionary of input parameters
+        mock_output: The output from the mock/stub call
+
+    Note:
+        This launches the callback without blocking. Errors are logged but won't
+        affect the main execution flow.
+    """
+    if callback is None:
+        return
+
+    async def _run_callback() -> None:
+        """Wrapper to run combined callback with error handling."""
+        try:
+            # Combine input params with mock_output
+            combined_data = {**input_params, "mock_output": mock_output}
+            # Filter parameters to match callback signature
+            filtered_data = filter_callback_parameters(callback, combined_data)
+
+            if inspect.iscoroutinefunction(callback):
+                await callback(**filtered_data)
+            else:
+                result = callback(**filtered_data)
+                if inspect.iscoroutine(result):
+                    await result
+        except Exception as e:
+            logger.warning(f"Combined callback execution failed: {e}", exc_info=True)
+
+    # Create task without awaiting (fire-and-forget)
+    try:
+        asyncio.create_task(_run_callback())
+    except RuntimeError:
+        # If no event loop is running, log a warning
+        logger.warning("Cannot launch combined callback task: no event loop running")

{veris_ai-1.13.0.dist-info → veris_ai-1.14.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veris-ai
-Version: 1.13.0
+Version: 1.14.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

{veris_ai-1.13.0.dist-info → veris_ai-1.14.1.dist-info}/RECORD

@@ -1,16 +1,16 @@
 veris_ai/README.md,sha256=Mxg9fyNt6hFkQaFBYZq85Kw9akw4cN2uu6j_mXZtNCs,3871
 veris_ai/__init__.py,sha256=enl_gEa6BQAjWvzCtsn_hFVJVVlJZ_dMsLL--E5W3nU,1907
 veris_ai/agents_wrapper.py,sha256=gLUd_0TyCVsqqilQLvsSJIpsU5uu2CdjjWOQ4QJjoJk,12786
-veris_ai/api_client.py,sha256=I1XyQ7J0ZU_JK9sZjF3XqFv5gGsrdKF38euOZmW8BG0,4150
+veris_ai/api_client.py,sha256=ZtxskNdAG_ASx_y2VhzV0vdG2kW5Ph5gPDBn-FG44-4,4654
 veris_ai/models.py,sha256=xKeheSJQle2tBeJG1DsGJzMDwv24p5jECjX6RAa39n4,495
 veris_ai/observability.py,sha256=eSIXmk6fpOAoWM-sDbsvzyUASh1ZwU6tRIPduy09RxY,4206
-veris_ai/tool_mock.py,sha256=olb_ywR88meK_FyoDVHpVnIi0h0B9oWM5IHV7k3hcko,24240
-veris_ai/utils.py,sha256=2fzXcsKQGLm1Q1ntjtuC_Z5l6Atj56zQE723m3zXdGg,16357
+veris_ai/tool_mock.py,sha256=A1k_llfjbMsMXP9J1M9pLhYVyNGxaLACaL6vsTc0zoI,25108
+veris_ai/utils.py,sha256=qwPPxS0CrsS36OoN_924hricz9jGRXx9FSDgLok0wgY,18940
 veris_ai/jaeger_interface/README.md,sha256=kd9rKcE5xf3EyNaiHu0tjn-0oES9sfaK6Ih-OhhTyCM,2821
 veris_ai/jaeger_interface/__init__.py,sha256=KD7NSiMYRG_2uF6dOLKkGG5lNQe4K9ptEwucwMT4_aw,1128
 veris_ai/jaeger_interface/client.py,sha256=yJrh86wRR0Dk3Gq12DId99WogcMIVbL0QQFqVSevvlE,8772
 veris_ai/jaeger_interface/models.py,sha256=e64VV6IvOEFuzRUgvDAMQFyOZMRb56I-PUPZLBZ3rX0,1864
-veris_ai-1.13.0.dist-info/METADATA,sha256=5yQZuNyfZdG4QiJd30nTHkwm_e2B8WRp4DbCXequnmk,16684
-veris_ai-1.13.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-veris_ai-1.13.0.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
-veris_ai-1.13.0.dist-info/RECORD,,
+veris_ai-1.14.1.dist-info/METADATA,sha256=C2JsiZwQNZcdW5uIX45fN2-sU3feqUyZJYMLR--VWKk,16684
+veris_ai-1.14.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+veris_ai-1.14.1.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
+veris_ai-1.14.1.dist-info/RECORD,,