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

{veris_ai-1.6.0 → veris_ai-1.8.0}/.github/workflows/test.yml

@@ -37,7 +37,7 @@ jobs:
 
     - name: Run type checks with mypy
       run: |
-        uv run mypy src/veris_ai tests
+        uv run mypy src/veris_ai
 
     - name: Run tests with pytest
       run: |

{veris_ai-1.6.0 → veris_ai-1.8.0}/.pre-commit-config.yaml

@@ -3,21 +3,21 @@ repos:
     hooks:
       - id: ruff-check
         name: Ruff lint
-        entry: uv run ruff check .
+        entry: uv run ruff check --fix .
         language: system
         pass_filenames: false
         types: [python]
 
       - id: ruff-format-check
         name: Ruff format (check)
-        entry: uv run ruff format --check .
+        entry: uv run ruff format 
         language: system
         pass_filenames: false
         types: [python]
 
       - id: mypy
         name: Mypy type check
-        entry: uv run mypy src/veris_ai tests
+        entry: uv run mypy src/veris_ai
         language: system
         pass_filenames: false
         types: [python]

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veris-ai
-Version: 1.6.0
+Version: 1.8.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
@@ -9,6 +9,7 @@ License-Expression: MIT
 License-File: LICENSE
 Requires-Python: >=3.11
 Requires-Dist: httpx>=0.24.0
+Requires-Dist: logfire>=4.3.3
 Requires-Dist: opentelemetry-api>=1.34.1
 Requires-Dist: opentelemetry-exporter-otlp>=1.34.1
 Requires-Dist: opentelemetry-instrumentation-fastapi>=0.55b1
@@ -19,11 +20,12 @@ Requires-Dist: opentelemetry-instrumentation>=0.55b1
 Requires-Dist: opentelemetry-sdk>=1.34.1
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: requests>=2.31.0
-Requires-Dist: traceloop-sdk>=0.45.4
+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.0.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'
@@ -45,7 +47,7 @@ A Python package for Veris AI tools with simulation capabilities and FastAPI MCP
 ## Quick Reference
 
 **Purpose**: Tool mocking, tracing, and FastAPI MCP integration for AI agent development  
-**Core Components**: [`tool_mock`](#function-mocking) • [`observability`](#sdk-observability-helpers) • [`fastapi_mcp`](#fastapi-mcp-integration) • [`jaeger_interface`](#jaeger-trace-interface)  
+**Core Components**: [`tool_mock`](#function-mocking) • [`api_client`](src/veris_ai/api_client.py) • [`observability`](#sdk-observability-helpers) • [`fastapi_mcp`](#fastapi-mcp-integration) • [`jaeger_interface`](#jaeger-trace-interface)  
 **Deep Dive**: [`Module Architecture`](src/veris_ai/README.md) • [`Testing Guide`](tests/README.md) • [`Usage Examples`](examples/README.md)  
 **Source of Truth**: Implementation details in [`src/veris_ai/`](src/veris_ai/) source code
 
@@ -84,13 +86,14 @@ from veris_ai import init_observability, instrument_fastapi_app  # Provided by S
 
 | Variable | Purpose | Default |
 |----------|---------|---------|
-| `VERIS_ENDPOINT_URL` | Mock server endpoint | *Required* |
+| `VERIS_API_KEY` | API authentication key | None |
 | `VERIS_MOCK_TIMEOUT` | Request timeout (seconds) | `90.0` |
 | `ENV` | Set to `"simulation"` for mock mode | Production |
-| `VERIS_SERVICE_NAME` | Tracing service name | Auto-detected |
-| `VERIS_OTLP_ENDPOINT` | OpenTelemetry collector | *Required for tracing* |
 
-**Configuration Details**: See [`src/veris_ai/tool_mock.py`](src/veris_ai/tool_mock.py) for environment handling logic.
+**Advanced Configuration** (rarely needed):
+- `VERIS_API_URL`: Override default API endpoint (defaults to production)
+
+**Configuration Details**: See [`src/veris_ai/api_client.py`](src/veris_ai/api_client.py) for API configuration and [`src/veris_ai/tool_mock.py`](src/veris_ai/tool_mock.py) for environment handling logic.
 
 
 ### SDK Observability Helpers
@@ -102,7 +105,7 @@ from fastapi import FastAPI
 from veris_ai import init_observability, instrument_fastapi_app
 
 # Initialize tracing/export early (no-op if dependencies are absent)
-init_observability(service_name="my-customer-service")
+init_observability()
 
 app = FastAPI()
 
@@ -110,6 +113,35 @@ app = FastAPI()
 instrument_fastapi_app(app)
 ```
 
+#### Observability Environment
+
+Set these environment variables to enable exporting traces via OTLP (Logfire) and ensure consistent service naming:
+
+| Variable | Example | Notes |
+|----------|---------|-------|
+| `OTEL_SERVICE_NAME` | `simulation-server` | Should match `VERIS_SERVICE_NAME` used elsewhere to keep traces aligned |
+| `OTEL_EXPORTER_OTLP_ENDPOINT` | `https://logfire-api.pydantic.dev` | OTLP HTTP endpoint |
+| `LOGFIRE_TOKEN` | `FILL_IN` | Logfire API token used by the exporter |
+| `OTEL_EXPORTER_OTLP_HEADERS` | `'Authorization=FILL_IN'` | Include quotes to preserve the `=`; often `Authorization=Bearer <LOGFIRE_TOKEN>` |
+
+Quick setup example:
+
+```bash
+export OTEL_SERVICE_NAME="simulation-server"
+export OTEL_EXPORTER_OTLP_ENDPOINT="https://logfire-api.pydantic.dev"
+export LOGFIRE_TOKEN="<your-token>"
+export OTEL_EXPORTER_OTLP_HEADERS="Authorization=${LOGFIRE_TOKEN}"
+```
+
+Then initialize in code early in your process:
+
+```python
+from veris_ai import init_observability, instrument_fastapi_app
+init_observability()
+app = FastAPI()
+instrument_fastapi_app(app)
+```
+
 What this enables:
 - Sets global W3C propagator (TraceContext + Baggage)
 - Optionally instruments FastAPI, requests, httpx, MCP client if installed
@@ -222,7 +254,7 @@ pytest --cov=veris_ai # Test with coverage
 
 **Semantic Tag**: `module-architecture`
 
-**Core Modules**: `tool_mock` (mocking), `jaeger_interface` (trace queries), `utils` (schema conversion)
+**Core Modules**: `tool_mock` (mocking), `api_client` (centralized API), `jaeger_interface` (trace queries), `utils` (schema conversion)
 
 **Complete Architecture**: See [`src/veris_ai/README.md`](src/veris_ai/README.md) for module overview, implementation flows, and configuration details. 
 

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

@@ -5,7 +5,7 @@ A Python package for Veris AI tools with simulation capabilities and FastAPI MCP
 ## Quick Reference
 
 **Purpose**: Tool mocking, tracing, and FastAPI MCP integration for AI agent development  
-**Core Components**: [`tool_mock`](#function-mocking) • [`observability`](#sdk-observability-helpers) • [`fastapi_mcp`](#fastapi-mcp-integration) • [`jaeger_interface`](#jaeger-trace-interface)  
+**Core Components**: [`tool_mock`](#function-mocking) • [`api_client`](src/veris_ai/api_client.py) • [`observability`](#sdk-observability-helpers) • [`fastapi_mcp`](#fastapi-mcp-integration) • [`jaeger_interface`](#jaeger-trace-interface)  
 **Deep Dive**: [`Module Architecture`](src/veris_ai/README.md) • [`Testing Guide`](tests/README.md) • [`Usage Examples`](examples/README.md)  
 **Source of Truth**: Implementation details in [`src/veris_ai/`](src/veris_ai/) source code
 
@@ -44,13 +44,14 @@ from veris_ai import init_observability, instrument_fastapi_app  # Provided by S
 
 | Variable | Purpose | Default |
 |----------|---------|---------|
-| `VERIS_ENDPOINT_URL` | Mock server endpoint | *Required* |
+| `VERIS_API_KEY` | API authentication key | None |
 | `VERIS_MOCK_TIMEOUT` | Request timeout (seconds) | `90.0` |
 | `ENV` | Set to `"simulation"` for mock mode | Production |
-| `VERIS_SERVICE_NAME` | Tracing service name | Auto-detected |
-| `VERIS_OTLP_ENDPOINT` | OpenTelemetry collector | *Required for tracing* |
 
-**Configuration Details**: See [`src/veris_ai/tool_mock.py`](src/veris_ai/tool_mock.py) for environment handling logic.
+**Advanced Configuration** (rarely needed):
+- `VERIS_API_URL`: Override default API endpoint (defaults to production)
+
+**Configuration Details**: See [`src/veris_ai/api_client.py`](src/veris_ai/api_client.py) for API configuration and [`src/veris_ai/tool_mock.py`](src/veris_ai/tool_mock.py) for environment handling logic.
 
 
 ### SDK Observability Helpers
@@ -62,7 +63,7 @@ from fastapi import FastAPI
 from veris_ai import init_observability, instrument_fastapi_app
 
 # Initialize tracing/export early (no-op if dependencies are absent)
-init_observability(service_name="my-customer-service")
+init_observability()
 
 app = FastAPI()
 
@@ -70,6 +71,35 @@ app = FastAPI()
 instrument_fastapi_app(app)
 ```
 
+#### Observability Environment
+
+Set these environment variables to enable exporting traces via OTLP (Logfire) and ensure consistent service naming:
+
+| Variable | Example | Notes |
+|----------|---------|-------|
+| `OTEL_SERVICE_NAME` | `simulation-server` | Should match `VERIS_SERVICE_NAME` used elsewhere to keep traces aligned |
+| `OTEL_EXPORTER_OTLP_ENDPOINT` | `https://logfire-api.pydantic.dev` | OTLP HTTP endpoint |
+| `LOGFIRE_TOKEN` | `FILL_IN` | Logfire API token used by the exporter |
+| `OTEL_EXPORTER_OTLP_HEADERS` | `'Authorization=FILL_IN'` | Include quotes to preserve the `=`; often `Authorization=Bearer <LOGFIRE_TOKEN>` |
+
+Quick setup example:
+
+```bash
+export OTEL_SERVICE_NAME="simulation-server"
+export OTEL_EXPORTER_OTLP_ENDPOINT="https://logfire-api.pydantic.dev"
+export LOGFIRE_TOKEN="<your-token>"
+export OTEL_EXPORTER_OTLP_HEADERS="Authorization=${LOGFIRE_TOKEN}"
+```
+
+Then initialize in code early in your process:
+
+```python
+from veris_ai import init_observability, instrument_fastapi_app
+init_observability()
+app = FastAPI()
+instrument_fastapi_app(app)
+```
+
 What this enables:
 - Sets global W3C propagator (TraceContext + Baggage)
 - Optionally instruments FastAPI, requests, httpx, MCP client if installed
@@ -182,7 +212,7 @@ pytest --cov=veris_ai # Test with coverage
 
 **Semantic Tag**: `module-architecture`
 
-**Core Modules**: `tool_mock` (mocking), `jaeger_interface` (trace queries), `utils` (schema conversion)
+**Core Modules**: `tool_mock` (mocking), `api_client` (centralized API), `jaeger_interface` (trace queries), `utils` (schema conversion)
 
 **Complete Architecture**: See [`src/veris_ai/README.md`](src/veris_ai/README.md) for module overview, implementation flows, and configuration details. 
 

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

@@ -93,6 +93,17 @@ ENV=simulation VERIS_ENDPOINT_URL=http://localhost:8000 python import_options.py
 ENABLE_TRACING=true USE_FASTAPI=true python import_options.py
 ```
 
+w### Observability Environment (optional)
+
+If you want traces exported while running examples, set the following before execution:
+
+```bash
+export OTEL_SERVICE_NAME="simulation-server"
+export OTEL_EXPORTER_OTLP_ENDPOINT="https://logfire-api.pydantic.dev"
+export LOGFIRE_TOKEN="<your-token>"
+export OTEL_EXPORTER_OTLP_HEADERS="Authorization=${LOGFIRE_TOKEN}"
+```
+
 ## Development Notes
 
 **Semantic Tag**: `example-development`

veris_ai-1.8.0/examples/openai_agents_example.py

@@ -0,0 +1,176 @@
+"""Example of using the Veris SDK with OpenAI agents to intercept tool calls."""
+
+import asyncio
+import os
+import sys
+
+from veris_ai import veris, wrap
+
+# Ensure the openai-agents package is installed
+try:
+    from agents import Agent, Runner, function_tool
+except ImportError:
+    print("Please install the openai-agents package: pip install veris-ai[agents]")
+    sys.exit(1)
+
+
+# Define some example tools
+@function_tool
+def calculator(x: int, y: int, operation: str = "add") -> int:
+    """Performs arithmetic operations on two numbers."""
+    if operation == "add":
+        return x + y
+    if operation == "multiply":
+        return x * y
+    if operation == "subtract":
+        return x - y
+    if operation == "divide":
+        return x // y if y != 0 else 0
+    return 0
+
+
+@function_tool
+def search_web(query: str) -> str:
+    """Searches the web for information."""
+    return f"Search results for: {query}"
+
+
+@function_tool
+def get_weather(location: str) -> str:
+    """Gets current weather for a location."""
+    return f"Weather in {location}: Sunny, 72°F"
+
+
+async def example_basic_wrap() -> None:
+    """Basic example: Wrap all tools."""
+    print("\n" + "=" * 60)
+    print("Example 1: Basic Wrap - All Tools Intercepted")
+    print("=" * 60)
+
+    # Create an agent with tools
+    agent = Agent(
+        name="Assistant",
+        model="gpt-4",
+        tools=[calculator, search_web, get_weather],
+        instructions="You are a helpful assistant with various tools.",
+    )
+
+    # Wrap the Runner.run method - all tools will be intercepted
+    wrapped_run = wrap(Runner.run)
+
+    # Use it like normal Runner.run
+    result = await wrapped_run(agent, "What's 10 + 5? Also search for Python tutorials.")
+    print(f"Result: {result.output}")
+
+
+async def example_selective_wrap() -> None:
+    """Example with selective tool interception."""
+    print("\n" + "=" * 60)
+    print("Example 2: Selective Wrap - Only Specific Tools")
+    print("=" * 60)
+
+    agent = Agent(
+        name="Assistant",
+        model="gpt-4",
+        tools=[calculator, search_web, get_weather],
+        instructions="You are a helpful assistant.",
+    )
+
+    # Only intercept calculator and search_web
+    wrapped_run = wrap(Runner.run, include_tools=["calculator", "search_web"])
+
+    print("📝 Only 'calculator' and 'search_web' will be mocked via Veris")
+    print("   'get_weather' will run normally")
+
+    result = await wrapped_run(agent, "Calculate 5+3 and check weather in NYC")
+    print(f"Result: {result.output}")
+
+
+async def example_exclude_tools() -> None:
+    """Example excluding specific tools from interception."""
+    print("\n" + "=" * 60)
+    print("Example 3: Exclude Tools - Keep Some Tools Normal")
+    print("=" * 60)
+
+    agent = Agent(
+        name="Assistant",
+        model="gpt-4",
+        tools=[calculator, search_web, get_weather],
+        instructions="You are a helpful assistant.",
+    )
+
+    # Intercept everything EXCEPT get_weather
+    wrapped_run = wrap(Runner.run, exclude_tools=["get_weather"])
+
+    print("📝 All tools EXCEPT 'get_weather' will be mocked via Veris")
+
+    result = await wrapped_run(agent, "Calculate 10*5 and check weather in London")
+    print(f"Result: {result.output}")
+
+
+async def example_coroutine_wrap() -> None:
+    """Example wrapping a coroutine directly."""
+    print("\n" + "=" * 60)
+    print("Example 4: Direct Coroutine Wrap")
+    print("=" * 60)
+
+    agent = Agent(
+        name="Assistant",
+        model="gpt-4",
+        tools=[calculator],
+        instructions="You are a helpful assistant.",
+    )
+
+    # Wrap the coroutine directly
+    result = await wrap(Runner.run(agent, "What's 7 * 8?"))
+    print(f"Result: {result.output}")
+
+
+async def main() -> None:
+    """Run all examples."""
+    print("\n🚀 VERIS SDK - OPENAI AGENTS WRAPPER EXAMPLES")
+    print("\nThese examples demonstrate different ways to intercept tool calls")
+    print("from OpenAI agents and route them through the Veris SDK.\n")
+
+    # Set up the environment
+    # In production, these would be set as environment variables
+    if not os.getenv("VERIS_ENDPOINT_URL"):
+        print("⚠️  VERIS_ENDPOINT_URL not set. Using demo endpoint.")
+        os.environ["VERIS_ENDPOINT_URL"] = "http://demo.veris.ai"
+
+    # Set a session ID to enable mocking
+    veris.set_session_id("example-session-123")
+    print(f"✅ Session ID set: {veris.session_id}")
+
+    # Check if OpenAI API key is set
+    if not os.getenv("OPENAI_API_KEY"):
+        print("\n⚠️  Note: OPENAI_API_KEY not set.")
+        print("   The examples will show the pattern but won't actually run.")
+        print("   Set your API key to see real agent execution.\n")
+
+    try:
+        # Run examples
+        await example_basic_wrap()
+        await example_selective_wrap()
+        await example_exclude_tools()
+        await example_coroutine_wrap()
+    except Exception as e:
+        print(f"\n❌ Error: {e}")
+        print("\nMake sure to:")
+        print("1. Set OPENAI_API_KEY environment variable")
+        print("2. Set VERIS_ENDPOINT_URL to your Veris endpoint")
+        print("3. Install dependencies: pip install veris-ai[agents]")
+    finally:
+        # Clear session when done
+        veris.clear_session_id()
+        print("\n✅ Session cleared")
+
+    print("\n✨ Examples complete!")
+    print("\nNext steps:")
+    print("1. Set up your Veris endpoint")
+    print("2. Configure your OpenAI API key")
+    print("3. Use wrap() to intercept tool calls in your agents")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "veris-ai"
-version = "1.6.0"
+version = "1.8.0"
 description = "A Python package for Veris AI tools"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -23,8 +23,8 @@ dependencies = [
     "opentelemetry-instrumentation-requests>=0.55b1",
     "opentelemetry-instrumentation-httpx>=0.55b1",
     "opentelemetry-instrumentation-mcp>=0.44.1",
-    "traceloop-sdk>=0.45.4",
     "opentelemetry-api>=1.34.1",
+    "logfire>=4.3.3",
 ]
 
 [project.optional-dependencies]
@@ -36,12 +36,15 @@ dev = [
     "black>=23.7.0",
     "mypy>=1.5.1",
     "pre-commit>=3.3.3",
-    "openai-agents>=0.0.1",
+    "openai-agents>=0.2.5",
 ]
 fastapi = [
     "fastapi",
     "fastapi-mcp>=0.4.0",
 ]
+agents = [
+    "openai-agents>=0.0.1",
+]
 instrument = [
     "wrapt",
     "opentelemetry-api",

{veris_ai-1.6.0 → veris_ai-1.8.0}/src/veris_ai/README.md

@@ -49,8 +49,33 @@ Environment variables are processed in [`tool_mock.py`](tool_mock.py):
 - `VERIS_ENDPOINT_URL`: Mock server endpoint
 - `VERIS_MOCK_TIMEOUT`: Request timeout (default: 90s)  
 - `ENV`: Set to `"simulation"` for mock mode
-- `VERIS_SERVICE_NAME`: Tracing service identifier
-- `VERIS_OTLP_ENDPOINT`: OpenTelemetry collector endpoint
+
+### Observability (OTLP / Logfire)
+
+When using the observability helpers (`init_observability`, `instrument_fastapi_app`), configure the following environment variables so traces export correctly and are attributed to the right service name:
+
+- `OTEL_SERVICE_NAME` — e.g. `simulation-server` (keep consistent with any `VERIS_SERVICE_NAME` you use)
+- `OTEL_EXPORTER_OTLP_ENDPOINT` — e.g. `https://logfire-api.pydantic.dev`
+- `LOGFIRE_TOKEN` — API token for Logfire
+- `OTEL_EXPORTER_OTLP_HEADERS` — e.g. `Authorization=Bearer <LOGFIRE_TOKEN>` (quote the value)
+
+Minimal shell setup:
+
+```bash
+export OTEL_SERVICE_NAME="simulation-server"
+export OTEL_EXPORTER_OTLP_ENDPOINT="https://logfire-api.pydantic.dev"
+export LOGFIRE_TOKEN="<your-token>"
+export OTEL_EXPORTER_OTLP_HEADERS="Authorization=${LOGFIRE_TOKEN}"
+```
+
+Then in code:
+
+```python
+from veris_ai import init_observability, instrument_fastapi_app
+init_observability()
+app = FastAPI()
+instrument_fastapi_app(app)
+```
 
 ## Development Notes
 

veris_ai-1.8.0/src/veris_ai/__init__.py

@@ -0,0 +1,70 @@
+"""Veris AI Python SDK."""
+
+from typing import Any
+
+__version__ = "0.1.0"
+
+# Import lightweight modules that only use base dependencies
+from .jaeger_interface import JaegerClient
+from .models import ResponseExpectation
+from .observability import init_observability, instrument_fastapi_app
+from .tool_mock import veris
+
+# Lazy import for modules with heavy dependencies
+_veris_runner = None
+_VerisConfig = None
+
+
+def veris_runner(*args: Any, **kwargs: Any) -> Any:  # noqa: ANN401
+    """Lazy loader for the veris_runner function from agents_wrapper.
+
+    This function wraps OpenAI agents Runner.run to intercept tool calls
+    through the Veris SDK's mocking infrastructure.
+
+    This function requires the 'agents' extra dependencies:
+    pip install veris-ai[agents]
+    """
+    global _veris_runner  # noqa: PLW0603
+    if _veris_runner is None:
+        try:
+            from .agents_wrapper import veris_runner as _veris_runner_impl  # noqa: PLC0415
+
+            _veris_runner = _veris_runner_impl
+        except ImportError as e:
+            error_msg = (
+                "The 'veris_runner' function requires additional dependencies. "
+                "Please install them with: pip install veris-ai[agents]"
+            )
+            raise ImportError(error_msg) from e
+    return _veris_runner(*args, **kwargs)
+
+
+def __getattr__(name: str) -> Any:  # noqa: ANN401
+    """Lazy load VerisConfig class."""
+    global _VerisConfig  # noqa: PLW0603
+    if name == "VerisConfig":
+        if _VerisConfig is None:
+            try:
+                from .agents_wrapper import VerisConfig as _VerisConfig_impl  # noqa: PLC0415
+
+                _VerisConfig = _VerisConfig_impl
+            except ImportError as e:
+                error_msg = (
+                    "The 'VerisConfig' class requires additional dependencies. "
+                    "Please install them with: pip install veris-ai[agents]"
+                )
+                raise ImportError(error_msg) from e
+        return _VerisConfig
+    msg = f"module {__name__!r} has no attribute {name!r}"
+    raise AttributeError(msg)
+
+
+__all__ = [
+    "veris",
+    "JaegerClient",
+    "ResponseExpectation",
+    "init_observability",
+    "instrument_fastapi_app",
+    "veris_runner",
+    "VerisConfig",
+]