veris-ai 1.10.0__tar.gz → 1.11.0__tar.gz

{veris_ai-1.10.0 → veris_ai-1.11.0}/.github/workflows/test.yml

@@ -28,7 +28,7 @@ jobs:
 
     - name: Install dependencies
       run: |
-        uv sync --all-extras
+        uv sync --all-groups
 
     - name: Run code quality checks with Ruff
       run: |

{veris_ai-1.10.0 → veris_ai-1.11.0}/PKG-INFO

@@ -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 → veris_ai-1.11.0}/README.md

@@ -1,5 +1,7 @@
 # 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
@@ -108,7 +110,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
@@ -169,6 +171,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 → veris_ai-1.11.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "veris-ai"
-version = "1.10.0"
+version = "1.11.0"
 description = "A Python package for Veris AI tools"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -28,30 +28,6 @@ dependencies = [
     "tenacity>=9.1.2",
 ]
 
-[project.optional-dependencies]
-dev = [
-    "ruff>=0.11.4",
-    "pytest>=7.4.0",
-    "pytest-asyncio>=0.21.1",
-    "pytest-cov>=4.1.0",
-    "black>=23.7.0",
-    "mypy>=1.5.1",
-    "pre-commit>=3.3.3",
-    "openai-agents>=0.2.5",
-]
-fastapi = [
-    "fastapi",
-    "fastapi-mcp>=0.4.0",
-]
-agents = [
-    "openai-agents>=0.0.1",
-]
-instrument = [
-    "wrapt",
-    "opentelemetry-api",
-    "opentelemetry-sdk",
-]
-
 [project.urls]
 "Homepage" = "https://github.com/veris-ai/veris-python-sdk"
 "Bug Tracker" = "https://github.com/veris-ai/veris-python-sdk/issues"
@@ -63,6 +39,17 @@ packages = ["src/veris_ai"]
 testpaths = ["tests"]
 python_files = ["test_*.py"]
 
+[tool.mypy]
+ignore_missing_imports = false
+
+[[tool.mypy.overrides]]
+module = [
+    "fastapi.*",
+    "fastapi_mcp.*",
+    "agents.*",
+]
+ignore_missing_imports = true
+
 [tool.ruff]
 line-length = 100
 target-version = "py311"
@@ -171,13 +158,22 @@ veris-ai = { workspace = true }
 
 [dependency-groups]
 dev = [
-    "fastapi>=0.116.1",
-    "fastapi-mcp>=0.4.0",
-    "mypy>=1.17.0",
-    "openai-agents>=0.0.1",
+    "ruff>=0.12.4",
     "pytest>=8.4.1",
     "pytest-asyncio>=1.1.0",
     "pytest-cov>=6.2.1",
-    "ruff>=0.12.4",
+    "mypy>=1.17.0",
     "types-requests>=2.32.4.20250611",
+    "pre-commit>=3.3.3",
+    "uvicorn>=0.30.0",
+]
+fastapi = [
+    "fastapi>=0.116.1",
+    "fastapi-mcp>=0.4.0",
+]
+agents = [
+    "openai-agents>=0.2.5",
+]
+instrument = [
+    "wrapt>=1.14.0",
 ]

{veris_ai-1.10.0 → veris_ai-1.11.0}/src/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-1.10.0 → veris_ai-1.11.0}/src/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 → veris_ai-1.11.0}/tests/test_agents_wrapper_simple.py

@@ -244,7 +244,7 @@ async def test_runner_with_session_calls_endpoint(
 
     # Verify the call had correct structure
     first_call = mock_veris_endpoint["calls"][0]
-    assert first_call["endpoint"] == "http://localhost:8000/api/v2/tool_mock"
+    assert first_call["endpoint"] == "http://localhost:8000/v2/tool_mock"
     assert first_call["payload"]["session_id"] == "test-session-123"
     assert "add_numbers" in str(first_call["payload"]) or "veris_tool_function" in str(
         first_call["payload"]

{veris_ai-1.10.0 → veris_ai-1.11.0}/tests/test_tool_mock.py

@@ -91,8 +91,9 @@ async def test_mock_http_error(simulation_env):
 
 @pytest.mark.asyncio
 async def test_mock_missing_endpoint():
-    """Test that missing endpoint raises ValueError when function is called in simulation mode."""
-    with patch.dict(os.environ, {"VERIS_API_URL": ""}):
+    """Test that missing endpoint raises ConnectError when invalid URL is used."""
+    # Use invalid URL that will trigger ConnectError
+    with patch.dict(os.environ, {"VERIS_API_URL": "http://invalid-nonexistent-domain-12345.local"}):
 
         @veris.mock()
         async def test_func():
@@ -101,15 +102,18 @@ async def test_mock_missing_endpoint():
         # Set session_id to enable simulation mode
         veris.set_session_id("test-session")
 
-        # Since VERIS_API_URL has a default value, the function will try to connect
-        # We'll test that the connection fails with httpx.ConnectError
+        # Invalid domain will cause ConnectError
         with pytest.raises(httpx.ConnectError):
             await test_func()
 
 
 @pytest.mark.asyncio
 async def test_mock_invalid_endpoint(simulation_env):
-    with patch.dict(os.environ, {"VERIS_API_URL": ""}), pytest.raises(httpx.ConnectError):
+    # Use invalid URL that will trigger ConnectError
+    with (
+        patch.dict(os.environ, {"VERIS_API_URL": "http://invalid-nonexistent-domain-67890.local"}),
+        pytest.raises(httpx.ConnectError),
+    ):
 
         @veris.mock()
         async def test_func():