agentstack-sdk 0.4.0__tar.gz → 0.4.1__tar.gz

{agentstack_sdk-0.4.0 → agentstack_sdk-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentstack-sdk
-Version: 0.4.0
+Version: 0.4.1
 Summary: Agent Stack SDK
 Author: IBM Corp.
 Requires-Dist: a2a-sdk==0.3.9

{agentstack_sdk-0.4.0 → agentstack_sdk-0.4.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentstack-sdk"
-version = "0.4.0"
+version = "0.4.1"
 description = "Agent Stack SDK"
 readme = "README.md"
 authors = [{ name = "IBM Corp." }]

{agentstack_sdk-0.4.0 → agentstack_sdk-0.4.1}/src/agentstack_sdk/a2a/extensions/services/mcp.py

@@ -3,9 +3,10 @@
 
 from __future__ import annotations
 
+import re
 from contextlib import asynccontextmanager
 from types import NoneType
-from typing import Annotated, Any, Literal, Self
+from typing import TYPE_CHECKING, Annotated, Any, Literal, Self
 
 import a2a.types
 import pydantic
@@ -14,6 +15,10 @@ from mcp.client.streamable_http import streamablehttp_client
 
 from agentstack_sdk.a2a.extensions.auth.oauth.oauth import OAuthExtensionServer
 from agentstack_sdk.a2a.extensions.base import BaseExtensionClient, BaseExtensionServer, BaseExtensionSpec
+from agentstack_sdk.util.logging import logger
+
+if TYPE_CHECKING:
+    from agentstack_sdk.server.context import RunContext
 
 _TRANSPORT_TYPES = Literal["streamable_http", "stdio"]
 
@@ -95,6 +100,23 @@ class MCPServiceExtensionMetadata(pydantic.BaseModel):
 
 
 class MCPServiceExtensionServer(BaseExtensionServer[MCPServiceExtensionSpec, MCPServiceExtensionMetadata]):
+    def handle_incoming_message(self, message: a2a.types.Message, context: RunContext):
+        from agentstack_sdk.platform import get_platform_client
+
+        super().handle_incoming_message(message, context)
+        if not self.data:
+            return
+
+        platform_url = str(get_platform_client().base_url)
+        for fullfilment in self.data.mcp_fulfillments.values():
+            if fullfilment.transport.type == "streamable_http":
+                try:
+                    fullfilment.transport.url = pydantic.AnyHttpUrl(
+                        re.sub(r"^http[s]?://{platform_url}", platform_url, str(fullfilment.transport.url))
+                    )
+                except Exception:
+                    logger.warning("Platform URL substitution failed", exc_info=True)
+
     def parse_client_metadata(self, message: a2a.types.Message) -> MCPServiceExtensionMetadata | None:
         metadata = super().parse_client_metadata(message)
         if metadata:

{agentstack_sdk-0.4.0 → agentstack_sdk-0.4.1}/src/agentstack_sdk/server/agent.py

@@ -41,9 +41,9 @@ from agentstack_sdk.a2a.types import ArtifactChunk, Metadata, RunYield, RunYield
 from agentstack_sdk.server.constants import _IMPLICIT_DEPENDENCY_PREFIX
 from agentstack_sdk.server.context import RunContext
 from agentstack_sdk.server.dependencies import extract_dependencies
-from agentstack_sdk.server.logging import logger
 from agentstack_sdk.server.store.context_store import ContextStore
 from agentstack_sdk.server.utils import cancel_task, close_queue
+from agentstack_sdk.util.logging import logger
 from agentstack_sdk.util.utils import extract_messages
 
 AgentFunction: TypeAlias = Callable[[], AsyncGenerator[RunYield, RunYieldResume]]

{agentstack_sdk-0.4.0 → agentstack_sdk-0.4.1}/src/agentstack_sdk/server/server.py

@@ -36,12 +36,12 @@ from agentstack_sdk.platform.client import PlatformClient
 from agentstack_sdk.platform.provider import Provider
 from agentstack_sdk.server.agent import Agent, AgentFactory
 from agentstack_sdk.server.agent import agent as agent_decorator
-from agentstack_sdk.server.logging import configure_logger as configure_logger_func
-from agentstack_sdk.server.logging import logger
 from agentstack_sdk.server.store.context_store import ContextStore
 from agentstack_sdk.server.store.memory_context_store import InMemoryContextStore
 from agentstack_sdk.server.telemetry import configure_telemetry as configure_telemetry_func
 from agentstack_sdk.server.utils import cancel_task
+from agentstack_sdk.util.logging import configure_logger as configure_logger_func
+from agentstack_sdk.util.logging import logger
 
 
 class Server:
@@ -83,6 +83,7 @@ class Server:
         request_context_builder: RequestContextBuilder | None = None,
         host: str = "127.0.0.1",
         port: int = 10000,
+        url: str | None = None,
         uds: str | None = None,
         fd: int | None = None,
         loop: Literal["none", "auto", "asyncio", "uvloop"] = "auto",
@@ -138,7 +139,8 @@ class Server:
 
         context_store = context_store or InMemoryContextStore()
         self._agent = self._agent_factory(context_store)
-        self._agent.card.url = f"http://{host}:{port}"
+        card_url = url and url.strip()
+        self._agent.card.url = card_url.rstrip("/") if card_url else f"http://{host}:{port}"
 
         self._self_registration_client = (
             self_registration_client_factory() if self_registration_client_factory else None