oridio 0.1.0__py3-none-any.whl

oridio/__init__.py

@@ -0,0 +1,99 @@
+"""
+Oridio Python SDK — AI Control Plane
+
+Drop-in OpenAI-compatible proxy with policy enforcement,
+semantic caching, cost controls, and observability.
+"""
+
+__version__ = "0.1.0"
+
+from .client import OridioClient, AsyncOridioClient
+from .decorators import oridio_protected
+from .context import (
+    set_agent_id, get_agent_id,
+    set_session_id, get_session_id,
+    set_run_id, get_run_id
+)
+from .bootstrap import init
+from .tools.bridge import wrap_tool, register_autogen_tool
+from .tools.schema import extract_mcp_schema
+from .interceptors import (
+    create_openai_client,
+    create_async_openai_client,
+    create_anthropic_client,
+    create_async_anthropic_client,
+    spawn_child,
+)
+from .exceptions import (
+    OridioError,
+    OridioAPIError,
+    OridioAuthError,
+    OridioRateLimitError,
+    OridioPolicyBlockedError,
+    OridioTimeoutError,
+    OridioHITLPendingError,
+    OridioHITLTimeoutError,
+    OridioHITLDeniedError,
+)
+from .models import (
+    ChatMessage,
+    ChatCompletionRequest,
+    ChatCompletionResponse,
+    McpGrant,
+    McpServer,
+    McpTool,
+    Policy,
+    PolicyRule,
+    ApiKey,
+    UsageRecord,
+    AgentStatus,
+    AgentIdentity,
+    MCPMeta,
+    MCPParams,
+    MCPPayload,
+    FirewallVerdict,
+    OridioResponse,
+)
+
+__all__ = [
+    "__version__",
+    # Client
+    "OridioClient",
+    "AsyncOridioClient",
+    "oridio_protected",
+    "create_openai_client",
+    "create_async_openai_client",
+    "create_anthropic_client",
+    "create_async_anthropic_client",
+    "spawn_child",
+    # Exceptions
+    "OridioError",
+    "OridioAPIError",
+    "OridioAuthError",
+    "OridioRateLimitError",
+    "OridioPolicyBlockedError",
+    "OridioTimeoutError",
+    "OridioHITLPendingError",
+    "OridioHITLTimeoutError",
+    "OridioHITLDeniedError",
+    "extract_mcp_schema",
+    # Models
+    "ChatMessage",
+    "ChatCompletionRequest",
+    "ChatCompletionResponse",
+    "McpGrant",
+    "McpServer",
+    "McpTool",
+    "Policy",
+    "PolicyRule",
+    "ApiKey",
+    "UsageRecord",
+    "AgentStatus",
+    "AgentIdentity",
+    "MCPMeta",
+    "MCPParams",
+    "MCPPayload",
+    "FirewallVerdict",
+    "OridioResponse",
+]
+

oridio/bootstrap.py

@@ -0,0 +1,45 @@
+import os
+import logging
+from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
+from opentelemetry.instrumentation.requests import RequestsInstrumentor
+from .context import get_agent_id
+
+logger = logging.getLogger(__name__)
+
+def _request_hook(span, request):
+    """Inject Oridio headers into outbound HTTP requests."""
+    agent_id = get_agent_id()
+    if agent_id:
+        request.headers["X-Oridio-Agent-Id"] = agent_id
+
+    # The JWT is typically already in the environment or set by the user,
+    # but if we wanted to auto-inject a known key here we could.
+    oridio_api_key = os.environ.get("ORIDIO_API_KEY")
+    if oridio_api_key and "Authorization" not in request.headers:
+        request.headers["Authorization"] = f"Bearer {oridio_api_key}"
+
+def init():
+    """
+    Initialize global Oridio interception for Python.
+    
+    1. Sets OPENAI_BASE_URL and ANTHROPIC_BASE_URL to point to the proxy.
+    2. Hooks httpx and requests via OpenTelemetry to inject identity headers.
+    """
+    proxy_url = os.environ.get("ORIDIO_PROXY_URL", "http://localhost:8080")
+    
+    # Force OpenAI and Anthropic SDKs to use the Oridio proxy
+    os.environ.setdefault("OPENAI_BASE_URL", f"{proxy_url}/v1")
+    os.environ.setdefault("ANTHROPIC_BASE_URL", f"{proxy_url}/v1")
+    
+    try:
+        HTTPXClientInstrumentor().instrument(request_hook=_request_hook)
+        logger.info("Successfully instrumented httpx.")
+    except Exception as e:
+        logger.warning(f"Failed to instrument httpx: {e}")
+
+    try:
+        RequestsInstrumentor().instrument(request_hook=_request_hook)
+        logger.info("Successfully instrumented requests.")
+    except Exception as e:
+        logger.warning(f"Failed to instrument requests: {e}")
+