veris-ai 1.10.0__py3-none-any.whl → 1.11.0__py3-none-any.whl

veris_ai/api_client.py

@@ -11,28 +11,50 @@ logger = logging.getLogger(__name__)
 
 
 class SimulatorAPIClient:
-    """Centralized client for making requests to VERIS simulation endpoints."""
+    """Centralized client for making requests to VERIS simulation endpoints.
+
+    Note:
+        This client intentionally reads configuration (base URL and API key)
+        from environment variables at call-time instead of at construction
+        time. This allows tests to patch environment variables and have those
+        changes reflected without recreating the singleton.
+    """
 
     def __init__(self) -> None:
-        """Initialize the API client with configuration from environment variables."""
-        self.base_url = os.getenv("VERIS_API_URL", "https://simulation.api.veris.ai/")
-        self.api_key = os.getenv("VERIS_API_KEY")
+        """Initialize the API client with static timeout configuration."""
         self.timeout = float(os.getenv("VERIS_MOCK_TIMEOUT", "90.0"))
 
+    def _get_base_url(self) -> str:
+        """Resolve the base URL from environment.
+
+        Behavior:
+            - If VERIS_API_URL is unset, default to the dev simulator URL.
+            - If VERIS_API_URL is set to an empty string, treat it as empty
+              (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"
+
     def _build_headers(self) -> dict[str, str] | None:
         """Build headers including OpenTelemetry tracing and API key."""
         headers: dict[str, str] | None = None
         # Add API key header if available
-        if self.api_key:
+        api_key = os.getenv("VERIS_API_KEY")
+        if api_key:
             if headers is None:
                 headers = {}
-            headers["x-api-key"] = self.api_key
+            headers["x-api-key"] = api_key
 
         return headers
 
     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()
+        # Validate endpoint URL; raise ConnectError for non-absolute URLs to
+        # mirror connection failures in tests when base URL is intentionally invalid.
+        if not endpoint.startswith(("http://", "https://")):
+            raise httpx.ConnectError("Invalid endpoint URL (not absolute): {endpoint}")
+
         with httpx.Client(timeout=self.timeout) as client:
             response = client.post(endpoint, json=payload, headers=headers)
             response.raise_for_status()
@@ -41,15 +63,15 @@ class SimulatorAPIClient:
     @property
     def tool_mock_endpoint(self) -> str:
         """Get the tool mock endpoint URL."""
-        return urljoin(self.base_url, "v2/tool_mock")
+        return urljoin(self._get_base_url(), "v2/tool_mock")
 
     def get_log_tool_call_endpoint(self, session_id: str) -> str:
         """Get the log tool call endpoint URL."""
-        return urljoin(self.base_url, f"v2/simulations/{session_id}/log_tool_call")
+        return urljoin(self._get_base_url(), f"v2/simulations/{session_id}/log_tool_call")
 
     def get_log_tool_response_endpoint(self, session_id: str) -> str:
         """Get the log tool response endpoint URL."""
-        return urljoin(self.base_url, f"v2/simulations/{session_id}/log_tool_response")
+        return urljoin(self._get_base_url(), f"v2/simulations/{session_id}/log_tool_response")
 
 
 # Global singleton instance

veris_ai/tool_mock.py

@@ -62,13 +62,14 @@ class VerisSDK:
             FastApiMCP,
         )
 
-        oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
+        oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token", auto_error=False)
 
         async def authenticate_request(
-            _: Request,
-            token: str = Depends(oauth2_scheme),  # noqa: ARG001
+            request: Request,  # noqa: ARG001
+            token: str | None = Depends(oauth2_scheme),
         ) -> None:
-            self.set_session_id(token)
+            if token:
+                self.set_session_id(token)
 
         # Create auth config with dependencies
         auth_config = AuthConfig(
@@ -267,6 +268,17 @@ def mock_tool_call(
     return_type_obj = type_hints.pop("return", Any)
     # Get function docstring
     docstring = inspect.getdoc(func) or ""
+
+    # Clean up parameters for V3 - just send values, not the nested dict
+    clean_params: dict[str, Any] = {}
+    for key, value in parameters.items():
+        if isinstance(value, dict) and "value" in value:
+            # Extract just the value from the nested structure
+            clean_params[key] = value["value"]
+        else:
+            # Already clean or unexpected format
+            clean_params[key] = value
+
     # Determine response expectation
     payload = {
         "session_id": session_id,
@@ -274,7 +286,7 @@ def mock_tool_call(
         "cache_response": bool(options.cache_response),
         "tool_call": {
             "function_name": func.__name__,
-            "parameters": parameters,
+            "parameters": clean_params,
             "return_type": json.dumps(extract_json_schema(return_type_obj)),
             "docstring": docstring,
         },
@@ -299,9 +311,20 @@ def log_tool_call(
     """Log tool call synchronously to the VERIS logging endpoint."""
     api_client = get_api_client()
     endpoint = api_client.get_log_tool_call_endpoint(session_id)
+
+    # Clean up parameters for V3 - just send values, not the nested dict
+    clean_params: dict[str, Any] = {}
+    for key, value in parameters.items():
+        if isinstance(value, dict) and "value" in value:
+            # Extract just the value from the nested structure
+            clean_params[key] = value["value"]
+        else:
+            # Already clean or unexpected format
+            clean_params[key] = value
+
     payload = {
         "function_name": function_name,
-        "parameters": parameters,
+        "parameters": clean_params,
         "docstring": docstring,
     }
     try:

{veris_ai-1.10.0.dist-info → veris_ai-1.11.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veris-ai
-Version: 1.10.0
+Version: 1.11.0
 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
@@ -21,28 +21,12 @@ Requires-Dist: opentelemetry-sdk>=1.34.1
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: requests>=2.31.0
 Requires-Dist: tenacity>=9.1.2
-Provides-Extra: agents
-Requires-Dist: openai-agents>=0.0.1; extra == 'agents'
-Provides-Extra: dev
-Requires-Dist: black>=23.7.0; extra == 'dev'
-Requires-Dist: mypy>=1.5.1; extra == 'dev'
-Requires-Dist: openai-agents>=0.2.5; extra == 'dev'
-Requires-Dist: pre-commit>=3.3.3; extra == 'dev'
-Requires-Dist: pytest-asyncio>=0.21.1; extra == 'dev'
-Requires-Dist: pytest-cov>=4.1.0; extra == 'dev'
-Requires-Dist: pytest>=7.4.0; extra == 'dev'
-Requires-Dist: ruff>=0.11.4; extra == 'dev'
-Provides-Extra: fastapi
-Requires-Dist: fastapi; extra == 'fastapi'
-Requires-Dist: fastapi-mcp>=0.4.0; extra == 'fastapi'
-Provides-Extra: instrument
-Requires-Dist: opentelemetry-api; extra == 'instrument'
-Requires-Dist: opentelemetry-sdk; extra == 'instrument'
-Requires-Dist: wrapt; extra == 'instrument'
 Description-Content-Type: text/markdown
 
 # Veris AI Python SDK
 
+For more information visit us at https://veris.ai
+
 A Python package for Veris AI tools with simulation capabilities and FastAPI MCP (Model Context Protocol) integration.
 
 ## Quick Reference
@@ -151,7 +135,7 @@ What this enables:
 
 End-to-end propagation with the simulator:
 - The simulator injects W3C headers when connecting to your FastAPI MCP endpoints
-- The SDK injects W3C headers on `/api/v2/tool_mock` and logging requests back to the simulator
+- The SDK injects W3C headers on `/v2/tool_mock` and logging requests back to the simulator
 - Result: customer agent spans and tool mocks appear under the same distributed trace
 
 ## Function Mocking
@@ -212,6 +196,58 @@ async def get_data() -> dict:
 
 **Implementation**: See [`src/veris_ai/tool_mock.py`](src/veris_ai/tool_mock.py) for decorator logic and API integration.
 
+### Core Instrument
+
+Instrument OpenAI agents without changing tool code by using the SDK's Runner (extends OpenAI's Runner) and an optional VerisConfig for fine control.
+
+Requirements:
+- Install extras: `uv add "veris-ai[agents]"`
+- Set `ENV=simulation` to enable mock behavior (otherwise passes through)
+
+Minimal usage:
+
+```python
+from veris_ai import Runner
+
+result = await Runner.run(agent, "What's 10 + 5?")
+```
+
+Select tools to intercept (include/exclude):
+
+```python
+from veris_ai import Runner, VerisConfig
+
+config = VerisConfig(include_tools=["calculator", "search_web"])  # or exclude_tools=[...]
+result = await Runner.run(agent, "Process this", veris_config=config)
+```
+
+Per‑tool behavior via ToolCallOptions:
+
+```python
+from veris_ai import Runner, VerisConfig, ToolCallOptions, ResponseExpectation
+
+config = VerisConfig(
+    tool_options={
+        "calculator": ToolCallOptions(
+            response_expectation=ResponseExpectation.REQUIRED,
+            cache_response=True,
+            mode="tool",
+        ),
+        "search_web": ToolCallOptions(
+            response_expectation=ResponseExpectation.NONE,
+            cache_response=False,
+            mode="spy",
+        ),
+    }
+)
+
+result = await Runner.run(agent, "Calculate and search", veris_config=config)
+```
+
+Notes:
+- Runner is a drop‑in enhancement over OpenAI's Runner (full details in [OpenAI Agents Integration](#openai-agents-integration))
+- See complete examples in [`examples/openai_agents_example.py`](examples/openai_agents_example.py)
+
 ## OpenAI Agents Integration
 
 **Semantic Tag**: `openai-agents`

{veris_ai-1.10.0.dist-info → veris_ai-1.11.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
 veris_ai/README.md,sha256=Ht1abGu-2_O0ZHJoJC7eAfG5WhO2CstAuV3hXPq1yAs,3875
 veris_ai/__init__.py,sha256=enl_gEa6BQAjWvzCtsn_hFVJVVlJZ_dMsLL--E5W3nU,1907
 veris_ai/agents_wrapper.py,sha256=N-YZ0qvIEJ8rlFZAzVKg6eWKa7I_qIt-khlg_zL-qY8,11931
-veris_ai/api_client.py,sha256=Gu_ovqaW-hZ6yPee2GVconMeg_xOxlt7IKu0HJQNiD0,2192
+veris_ai/api_client.py,sha256=385u11QuKq4Qk_WkI4Z5yOxAWJcV57VCGLYU5NIfnmo,3201
 veris_ai/models.py,sha256=xKeheSJQle2tBeJG1DsGJzMDwv24p5jECjX6RAa39n4,495
 veris_ai/observability.py,sha256=eSIXmk6fpOAoWM-sDbsvzyUASh1ZwU6tRIPduy09RxY,4206
-veris_ai/tool_mock.py,sha256=rU_ZwYWYIfgiOojEjmi5BdBsNzP3RphLPmMG1pyYUVw,11954
+veris_ai/tool_mock.py,sha256=--t0ZhjplJXzBt1F--XPE57ZyMBjB9SOjO2OzuoWkbk,12861
 veris_ai/utils.py,sha256=hJetCiN8Bubhy0nqSoS1C2awN9cdkKuHM1v7YhtwtTs,10066
 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.10.0.dist-info/METADATA,sha256=SRkcA5PGLQ5eNcz-yCNuXMECuRypD6olI9fjbFff-qQ,14471
-veris_ai-1.10.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-veris_ai-1.10.0.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
-veris_ai-1.10.0.dist-info/RECORD,,
+veris_ai-1.11.0.dist-info/METADATA,sha256=4ktkFw8WMexIqWwCtuRbFriJSVgAJp3_v0uny4R8lic,15259
+veris_ai-1.11.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+veris_ai-1.11.0.dist-info/licenses/LICENSE,sha256=2g4i20atAgtD5einaKzhQrIB-JrPhyQgD3bC0wkHcCI,1065
+veris_ai-1.11.0.dist-info/RECORD,,