copilotkit 0.1.88__tar.gz → 0.1.89__tar.gz

{copilotkit-0.1.88 → copilotkit-0.1.89}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: copilotkit
-Version: 0.1.88
+Version: 0.1.89
 Summary: CopilotKit python SDK
 License: MIT
 Keywords: copilot,copilotkit,langgraph,langchain,ai,langsmith,langserve

{copilotkit-0.1.88 → copilotkit-0.1.89}/copilotkit/__init__.py

@@ -7,6 +7,7 @@ from .agent import Agent
 from .langgraph_agui_agent import LangGraphAGUIAgent
 from .copilotkit_lg_middleware import CopilotKitMiddleware
 from ag_ui_langgraph.middlewares.state_streaming import StateStreamingMiddleware, StateItem
+from .header_propagation import set_forwarded_headers, get_forwarded_headers, install_httpx_hook
 
 
 
@@ -24,4 +25,7 @@ __all__ = [
     "CopilotKitMiddleware",
     "StateStreamingMiddleware",
     "StateItem",
+    "set_forwarded_headers",
+    "get_forwarded_headers",
+    "install_httpx_hook",
 ]

copilotkit-0.1.89/copilotkit/header_propagation.py

@@ -0,0 +1,58 @@
+"""Header propagation for forwarding x-* prefixed headers to outgoing LLM calls.
+
+Uses Python contextvars for per-request ambient state in async FastAPI handlers.
+An httpx event hook reads the ContextVar and injects headers on outgoing requests.
+Matches the CopilotKit runtime's extractForwardableHeaders() behavior.
+"""
+
+import contextvars
+import warnings
+from typing import Dict
+
+# Ambient per-request state for headers to forward to LLM calls
+_forwarded_headers: contextvars.ContextVar[Dict[str, str]] = contextvars.ContextVar(
+    'copilotkit_forwarded_headers'
+)
+
+
+def set_forwarded_headers(headers: Dict[str, str]) -> None:
+    """Store headers to forward to outgoing LLM calls.
+    Filters to x-* prefixed headers only."""
+    filtered = {k.lower(): v for k, v in headers.items() if k.lower().startswith('x-')}
+    _forwarded_headers.set(filtered)
+
+
+def get_forwarded_headers() -> Dict[str, str]:
+    """Get headers that should be forwarded to outgoing LLM calls."""
+    return _forwarded_headers.get({})
+
+
+def install_httpx_hook(client) -> None:
+    """Append an event hook to an httpx client that injects forwarded headers.
+
+    Works with OpenAI and Anthropic Python SDKs (both use httpx internally).
+    No-op when no headers are set (demo traffic).
+
+    Parameters
+    ----------
+    client : object
+        An OpenAI/Anthropic client instance, or a raw httpx.Client/AsyncClient.
+        For SDK clients the underlying transport lives at ``client._client``.
+    """
+    def _inject_headers(request):
+        headers = get_forwarded_headers()
+        for key, value in headers.items():
+            request.headers[key] = value
+
+    # OpenAI / Anthropic SDKs wrap an httpx client at client._client
+    if hasattr(client, '_client') and hasattr(client._client, 'event_hooks'):
+        client._client.event_hooks['request'].append(_inject_headers)
+    elif hasattr(client, 'event_hooks'):
+        # Raw httpx.Client / httpx.AsyncClient
+        client.event_hooks['request'].append(_inject_headers)
+    else:
+        warnings.warn(
+            f"install_httpx_hook: client of type {type(client).__name__} has no "
+            "recognized event_hooks attribute; x-* headers will not be forwarded",
+            stacklevel=2,
+        )

{copilotkit-0.1.88 → copilotkit-0.1.89}/copilotkit/integrations/fastapi.py

@@ -20,6 +20,7 @@ from ..exc import (
 )
 from ..action import ActionDict
 from ..html import generate_info_html
+from ..header_propagation import set_forwarded_headers
 logging.basicConfig(level=logging.ERROR)
 logger = logging.getLogger(__name__)
 
@@ -79,6 +80,9 @@ async def handler(request: Request, sdk: CopilotKitRemoteEndpoint):
     except: # pylint: disable=bare-except
         body = None
 
+    # Propagate x-aimock-* headers to outgoing LLM calls via ContextVar
+    set_forwarded_headers(dict(request.headers))
+
     path = request.path_params.get('path')
     method = request.method
     context = cast(

{copilotkit-0.1.88 → copilotkit-0.1.89}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "copilotkit"
-version = "0.1.88"
+version = "0.1.89"
 description = "CopilotKit python SDK"
 authors = ["Markus Ecker <markus.ecker@gmail.com>"]
 license = "MIT"