@aws/nx-plugin 1.0.0-rc.17 → 1.0.0-rc.19

package/src/utils/agent-connection/files/py-core-gateway/agentcore_gateway_mcp_client.py.template

@@ -1,78 +0,0 @@
-import re
-
-import boto3
-import httpx
-from mcp.client.streamable_http import streamablehttp_client
-from strands.tools.mcp.mcp_client import MCPClient
-
-from .auth import SigV4HTTPXAuth
-from .session_context import get_current_session_id
-
-SESSION_HEADER = "X-Amzn-Bedrock-AgentCore-Runtime-Session-Id"
-
-
-class _SessionHeaderAuth(httpx.Auth):
-    """Stamps the AgentCore session header from the current async context."""
-
-    requires_request_body = True
-    requires_response_body = True
-
-    def __init__(self, inner: httpx.Auth | None = None):
-        self._inner = inner
-
-    def auth_flow(self, request: httpx.Request):  # type: ignore[override]
-        sid = get_current_session_id()
-        if sid:
-            request.headers[SESSION_HEADER] = sid
-        if self._inner is None:
-            yield request
-            return
-        yield from self._inner.auth_flow(request)
-
-
-def _parse_region_from_gateway_url(gateway_url: str) -> str:
-    match = re.search(r"\.bedrock-agentcore\.([^.]+)\.amazonaws\.com", gateway_url)
-    if not match:
-        raise ValueError(
-            f"Cannot determine region from gateway URL '{gateway_url}'. "
-            "Pass region explicitly."
-        )
-    return match.group(1)
-
-
-class AgentCoreGatewayMCPClient:
-    """Factory for MCP clients that connect to an AgentCore Gateway. Forwards
-    the current async-context session via the AgentCore session header, which
-    the gateway propagates to downstream MCP server targets.
-    """
-
-    @staticmethod
-    def with_iam_auth(
-        gateway_url: str,
-        region: str | None = None,
-    ) -> MCPClient:
-        """Create a gateway MCP client authenticated with IAM SigV4."""
-        resolved_region = region or _parse_region_from_gateway_url(gateway_url)
-        credentials = boto3.Session(region_name=resolved_region).get_credentials()
-        return _create_client(
-            gateway_url,
-            _SessionHeaderAuth(
-                SigV4HTTPXAuth(credentials, "bedrock-agentcore", resolved_region),
-            ),
-        )
-
-    @staticmethod
-    def without_auth(gateway_url: str) -> MCPClient:
-        """Plain-HTTP gateway client, for the local gateway started by serve-local."""
-        return _create_client(gateway_url, _SessionHeaderAuth())
-
-
-def _create_client(gateway_url: str, auth: httpx.Auth) -> MCPClient:
-    return MCPClient(
-        lambda: streamablehttp_client(
-            gateway_url,
-            auth=auth,
-            timeout=120,
-            terminate_on_close=False,
-        )
-    )

package/src/utils/agent-connection/files/py-core-mcp/agentcore_mcp_client.py.template

@@ -1,89 +0,0 @@
-from collections.abc import Callable
-
-import boto3
-import httpx
-from mcp.client.streamable_http import streamablehttp_client
-from strands.tools.mcp.mcp_client import MCPClient
-
-from .auth import SigV4HTTPXAuth
-from .session_context import get_current_session_id
-
-SESSION_HEADER = "X-Amzn-Bedrock-AgentCore-Runtime-Session-Id"
-
-
-class _SessionHeaderAuth(httpx.Auth):
-    """Stamps the AgentCore session header from the current async context."""
-
-    requires_request_body = True
-    requires_response_body = True
-
-    def __init__(self, inner: httpx.Auth | None = None):
-        self._inner = inner
-
-    def auth_flow(self, request: httpx.Request):  # type: ignore[override]
-        sid = get_current_session_id()
-        if sid:
-            request.headers[SESSION_HEADER] = sid
-        if self._inner is None:
-            yield request
-            return
-        yield from self._inner.auth_flow(request)
-
-
-class AgentCoreMCPClient:
-    """Factory for MCP clients that forward the current async-context session."""
-
-    @staticmethod
-    def _build_url(agent_runtime_arn: str) -> tuple[str, str]:
-        """Extract region from ARN and construct the invocation URL.
-
-        ARN format: arn:partition:service:region:account-id:resource
-        """
-        region = agent_runtime_arn.split(":")[3]
-        encoded_arn = agent_runtime_arn.replace(":", "%3A").replace("/", "%2F")
-        url = (
-            f"https://bedrock-agentcore.{region}.amazonaws.com/runtimes/"
-            f"{encoded_arn}/invocations?qualifier=DEFAULT"
-        )
-        return region, url
-
-    @staticmethod
-    def _create(url: str, auth: httpx.Auth | None = None) -> MCPClient:
-        return MCPClient(
-            lambda: streamablehttp_client(
-                url,
-                auth=_SessionHeaderAuth(auth),
-                timeout=120,
-                terminate_on_close=False,
-            )
-        )
-
-    @staticmethod
-    def with_iam_auth(agent_runtime_arn: str) -> MCPClient:
-        """SigV4-authenticated client for a Bedrock AgentCore runtime."""
-        region, url = AgentCoreMCPClient._build_url(agent_runtime_arn)
-        credentials = boto3.Session(region_name=region).get_credentials()
-        return AgentCoreMCPClient._create(
-            url, SigV4HTTPXAuth(credentials, "bedrock-agentcore", region)
-        )
-
-    @staticmethod
-    def with_jwt_auth(
-        agent_runtime_arn: str, access_token_provider: Callable[[], str]
-    ) -> MCPClient:
-        """Bearer-authenticated client for a Bedrock AgentCore runtime."""
-        _, url = AgentCoreMCPClient._build_url(agent_runtime_arn)
-        return MCPClient(
-            lambda: streamablehttp_client(
-                url,
-                auth=_SessionHeaderAuth(),
-                headers={"Authorization": f"Bearer {access_token_provider()}"},
-                timeout=120,
-                terminate_on_close=False,
-            )
-        )
-
-    @staticmethod
-    def without_auth(url: str) -> MCPClient:
-        """Plain-HTTP client — for local dev."""
-        return AgentCoreMCPClient._create(url)