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

veris_ai/api_client.py

@@ -2,6 +2,7 @@
 
 import logging
 import os
+from contextvars import ContextVar
 from typing import Any
 
 import httpx
@@ -9,6 +10,9 @@ from urllib.parse import urljoin
 
 logger = logging.getLogger(__name__)
 
+# Context variable to store base_url for per-request multi-tenant scenarios
+_base_url_context: ContextVar[str | None] = ContextVar("veris_base_url", default=None)
+
 
 class SimulatorAPIClient:
     """Centralized client for making requests to VERIS simulation endpoints.
@@ -20,10 +24,9 @@ class SimulatorAPIClient:
         changes reflected without recreating the singleton.
     """
 
-    def __init__(self, timeout: float | None = None, base_url: str | None = None) -> None:
+    def __init__(self, timeout: float | None = None) -> None:
         """Initialize the API client with static timeout configuration."""
         self._timeout = timeout or float(os.getenv("VERIS_MOCK_TIMEOUT", "300.0"))
-        self._base_url = base_url
 
     @property
     def base_url(self) -> str:
@@ -31,7 +34,12 @@ class SimulatorAPIClient:
         return self._get_base_url()
 
     def _get_base_url(self) -> str:
-        """Resolve the base URL from environment.
+        """Resolve the base URL from context, environment, or default.
+
+        Priority order:
+            1. Context variable (for per-request multi-tenant scenarios)
+            2. VERIS_API_URL environment variable
+            3. Default production URL
 
         Behavior:
             - If VERIS_API_URL is unset, default to the dev simulator URL.
@@ -39,7 +47,9 @@ class SimulatorAPIClient:
               (do not fall back). This supports tests expecting connection
               failures when an invalid endpoint is provided.
         """
-        return self._base_url or os.getenv("VERIS_API_URL") or "https://simulator.api.veris.ai"
+        # Priority: context > env var > default
+        context_url = _base_url_context.get()
+        return context_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."""
@@ -107,7 +117,7 @@ def get_api_client() -> SimulatorAPIClient:
     return _api_client
 
 
-def set_api_client_params(base_url: str | None = None, timeout: float | None = None) -> None:
+def set_api_client_params(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)
+    _api_client = SimulatorAPIClient(timeout=timeout)

veris_ai/tool_mock.py

@@ -16,7 +16,7 @@ from typing import (
 
 
 from veris_ai.models import ResponseExpectation, ToolCallOptions
-from veris_ai.api_client import get_api_client, set_api_client_params
+from veris_ai.api_client import _base_url_context, get_api_client
 from veris_ai.utils import (
     convert_to_type,
     execute_callback,
@@ -93,7 +93,7 @@ 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"])
+                _base_url_context.set(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"
@@ -144,10 +144,11 @@ class VerisSDK:
         _thread_id_context.set(None)
 
     def clear_context(self) -> None:
-        """Clear the session_id and thread_id from context variables."""
+        """Clear the session_id, thread_id, and base_url from context variables."""
         self._clear_session_id()
         self._clear_thread_id()
-        logger.info("Session ID and Thread ID cleared - mocking disabled")
+        _base_url_context.set(None)
+        logger.info("Session ID, Thread ID, and Base URL cleared - mocking disabled")
 
     def clear_session_id(self) -> None:
         """DEPRECATED: Use clear_context() instead."""

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veris-ai
-Version: 1.14.1
+Version: 1.14.2
 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.14.1.dist-info → veris_ai-1.14.2.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=ZtxskNdAG_ASx_y2VhzV0vdG2kW5Ph5gPDBn-FG44-4,4654
+veris_ai/api_client.py,sha256=LevQnX9oK4wU1ehXUJiBxoJAkgnrZi1dQf88PkQfLRM,5040
 veris_ai/models.py,sha256=xKeheSJQle2tBeJG1DsGJzMDwv24p5jECjX6RAa39n4,495
 veris_ai/observability.py,sha256=eSIXmk6fpOAoWM-sDbsvzyUASh1ZwU6tRIPduy09RxY,4206
-veris_ai/tool_mock.py,sha256=A1k_llfjbMsMXP9J1M9pLhYVyNGxaLACaL6vsTc0zoI,25108
+veris_ai/tool_mock.py,sha256=D3a-w5j8_3oFALZYCSXkBGqxCeF0JUSULypjNbPQXjo,25153
 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.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,,
+veris_ai-1.14.2.dist-info/METADATA,sha256=UbnAoKMupmF6CyK_D32FWZJLAyGtwYdDWCR4bIwoq8M,16684
+veris_ai-1.14.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+veris_ai-1.14.2.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
+veris_ai-1.14.2.dist-info/RECORD,,