agentenv-framework-protocol 0.1.341__tar.gz → 0.1.342__tar.gz

{agentenv_framework_protocol-0.1.341 → agentenv_framework_protocol-0.1.342}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: agentenv-framework-protocol
-Version: 0.1.341
+Version: 0.1.342
 Summary: Open data-plane protocol and server SDK for agent environments.
 Project-URL: Homepage, https://www.agentenvframework.com
 Project-URL: Documentation, https://www.agentenvframework.com/docs
@@ -724,8 +724,9 @@ multi-operation custom extensions, and advanced ASGI-lifespan plus
 common SDK-operation override hooks.
 
 The agent examples make real OpenAI-compatible model calls. Set
-`LITELLM_API_KEY` and, when needed, `LITELLM_BASE_URL`; their typed agent config
-selects the model and system prompt for each deployment. Tests inject a fake
+`LITELLM_API_KEY` and, when needed, `LITELLM_BASE_URL`; they also send the headers in
+`LITELLM_EXTRA_HEADERS`, a JSON object AgentEnv sets when the endpoint needs any. Their typed
+agent config selects the model and system prompt for each deployment. Tests inject a fake
 model client, so the example suite remains offline and deterministic.
 
 ### Object transfer

{agentenv_framework_protocol-0.1.341 → agentenv_framework_protocol-0.1.342}/README.md

@@ -679,8 +679,9 @@ multi-operation custom extensions, and advanced ASGI-lifespan plus
 common SDK-operation override hooks.
 
 The agent examples make real OpenAI-compatible model calls. Set
-`LITELLM_API_KEY` and, when needed, `LITELLM_BASE_URL`; their typed agent config
-selects the model and system prompt for each deployment. Tests inject a fake
+`LITELLM_API_KEY` and, when needed, `LITELLM_BASE_URL`; they also send the headers in
+`LITELLM_EXTRA_HEADERS`, a JSON object AgentEnv sets when the endpoint needs any. Their typed
+agent config selects the model and system prompt for each deployment. Tests inject a fake
 model client, so the example suite remains offline and deterministic.
 
 ### Object transfer

{agentenv_framework_protocol-0.1.341 → agentenv_framework_protocol-0.1.342}/examples/reference_model.py

@@ -42,13 +42,18 @@ class OpenAICompatibleModel:
             raise RuntimeError(
                 "LITELLM_BASE_URL must be set before running the agent"
             )
+        # The endpoint's headers are for the endpoint AgentEnv configured, never one this client was pointed at.
+        configured = os.getenv("LITELLM_BASE_URL") or ""
+        headers = (json.loads(os.getenv("LITELLM_EXTRA_HEADERS") or "{}")
+                   if base_url.rstrip("/") == configured.rstrip("/") else {})
         base_url = base_url.rstrip("/")
         endpoint = (
             f"{base_url}/chat/completions"
             if base_url.endswith("/v1")
             else f"{base_url}/v1/chat/completions"
         )
-        return endpoint, {"Authorization": f"Bearer {api_key}"}
+        own = {k: v for k, v in headers.items() if k.lower() != "authorization"}
+        return endpoint, {**own, "Authorization": f"Bearer {api_key}"}
 
     async def complete(
         self,

{agentenv_framework_protocol-0.1.341 → agentenv_framework_protocol-0.1.342}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentenv-framework-protocol"
-version = "0.1.341"
+version = "0.1.342"
 description = "Open data-plane protocol and server SDK for agent environments."
 readme = "README.md"
 license = "Apache-2.0"

{agentenv_framework_protocol-0.1.341 → agentenv_framework_protocol-0.1.342}/tests/test_a2a_agent_examples.py

@@ -64,9 +64,24 @@ StreamingAgent = _streaming.StreamingAgent
 
 
 @pytest.mark.asyncio
-async def test_reference_model_forwards_generic_metadata(
+@pytest.mark.parametrize("endpoint_headers, configured_url, sent_headers", [
+    (None, "https://models.example/v1", {"Authorization": "Bearer request-scoped-key"}),
+    ('{"X-Bill-To": "acme", "authorization": "Bearer theirs"}', "https://models.example/v1",
+     {"X-Bill-To": "acme", "Authorization": "Bearer request-scoped-key"}),
+    ('{"X-Bill-To": "acme"}', "https://models.example/v1/", {"X-Bill-To": "acme", "Authorization": "Bearer request-scoped-key"}),
+    ('{"X-Bill-To": "acme"}', "https://configured.example/v1", {"Authorization": "Bearer request-scoped-key"}),
+], ids=["no-endpoint-headers", "endpoint-headers", "trailing-slash", "another-endpoint"])
+async def test_reference_model_forwards_generic_metadata_and_the_endpoints_headers(
     monkeypatch: pytest.MonkeyPatch,
+    endpoint_headers: str | None,
+    configured_url: str,
+    sent_headers: dict[str, str],
 ) -> None:
+    monkeypatch.setenv("LITELLM_BASE_URL", configured_url)
+    if endpoint_headers is None:
+        monkeypatch.delenv("LITELLM_EXTRA_HEADERS", raising=False)
+    else:
+        monkeypatch.setenv("LITELLM_EXTRA_HEADERS", endpoint_headers)
     captured: dict[str, Any] = {}
     class Response:
         def raise_for_status(self) -> None:
@@ -107,7 +122,7 @@ async def test_reference_model_forwards_generic_metadata(
         "client": {"timeout": 120},
         "request": {
             "url": "https://models.example/v1/chat/completions",
-            "headers": {"Authorization": "Bearer request-scoped-key"},
+            "headers": sent_headers,
             "json": {
                 "model": "test-model",
                 "messages": [{"role": "user", "content": "hello"}],