agentstack-sdk 0.6.0rc4__py3-none-any.whl → 0.6.1rc1__py3-none-any.whl

agentstack_sdk/platform/__init__.py

@@ -8,6 +8,7 @@ from .file import *
 from .model_provider import *
 from .provider import *
 from .provider_build import *
+from .provider_discovery import *
 from .user import *
 from .user_feedback import *
 from .vector_store import *

agentstack_sdk/platform/provider_discovery.py

@@ -0,0 +1,57 @@
+# Copyright 2025 © BeeAI a Series of LF Projects, LLC
+# SPDX-License-Identifier: Apache-2.0
+from __future__ import annotations
+
+from enum import StrEnum
+from uuid import UUID
+
+import pydantic
+from a2a.types import AgentCard
+
+from agentstack_sdk.platform.client import PlatformClient, get_platform_client
+
+
+class DiscoveryState(StrEnum):
+    PENDING = "pending"
+    IN_PROGRESS = "in_progress"
+    COMPLETED = "completed"
+    FAILED = "failed"
+
+
+class ProviderDiscovery(pydantic.BaseModel):
+    id: UUID
+    created_at: pydantic.AwareDatetime
+    status: DiscoveryState
+    docker_image: str
+    created_by: UUID
+    agent_card: AgentCard | None = None
+    error_message: str | None = None
+
+    @staticmethod
+    async def create(
+        *,
+        docker_image: str,
+        client: PlatformClient | None = None,
+    ) -> ProviderDiscovery:
+        async with client or get_platform_client() as client:
+            return pydantic.TypeAdapter(ProviderDiscovery).validate_python(
+                (
+                    await client.post(
+                        url="/api/v1/providers/discovery",
+                        json={"docker_image": docker_image},
+                    )
+                )
+                .raise_for_status()
+                .json()
+            )
+
+    async def get(self: ProviderDiscovery | str, *, client: PlatformClient | None = None) -> ProviderDiscovery:
+        discovery_id = self if isinstance(self, str) else str(self.id)
+        async with client or get_platform_client() as client:
+            result = pydantic.TypeAdapter(ProviderDiscovery).validate_json(
+                (await client.get(url=f"/api/v1/providers/discovery/{discovery_id}")).raise_for_status().content
+            )
+        if isinstance(self, ProviderDiscovery):
+            self.__dict__.update(result.__dict__)
+            return self
+        return result

agentstack_sdk/server/app.py

@@ -18,8 +18,6 @@ from a2a.server.tasks import (
 from a2a.types import AgentInterface, TransportProtocol
 from fastapi import APIRouter, Depends, FastAPI
 from fastapi.applications import AppType
-from starlette.authentication import AuthenticationBackend
-from starlette.middleware.authentication import AuthenticationMiddleware
 from starlette.types import Lifespan
 
 from agentstack_sdk.server.agent import Agent, Executor
@@ -39,7 +37,6 @@ def create_app(
     dependencies: list[Depends] | None = None,  # pyright: ignore [reportGeneralTypeIssues]
     override_interfaces: bool = True,
     task_timeout: timedelta = timedelta(minutes=10),
-    auth_backend: AuthenticationBackend | None = None,
     **kwargs,
 ) -> FastAPI:
     queue_manager = queue_manager or InMemoryQueueManager()
@@ -78,10 +75,6 @@ def create_app(
         **kwargs,
     )
 
-    if auth_backend:
-        rest_app.add_middleware(AuthenticationMiddleware, backend=auth_backend)
-        jsonrpc_app.add_middleware(AuthenticationMiddleware, backend=auth_backend)
-
     rest_app.mount("/jsonrpc", jsonrpc_app)
     rest_app.include_router(APIRouter(lifespan=lifespan))
     return rest_app

agentstack_sdk/server/middleware/platform_auth_backend.py

@@ -7,6 +7,7 @@ from datetime import timedelta
 from urllib.parse import urljoin
 
 from a2a.auth.user import User
+from a2a.types import AgentCard, HTTPAuthSecurityScheme, SecurityScheme
 from async_lru import alru_cache
 from authlib.jose import JsonWebKey, JWTClaims, KeySet, jwt
 from authlib.jose.errors import JoseError
@@ -15,7 +16,6 @@ from fastapi.security import HTTPBearer
 from pydantic import Secret
 from starlette.authentication import (
     AuthCredentials,
-    AuthenticationBackend,
     AuthenticationError,
     BaseUser,
 )
@@ -23,7 +23,7 @@ from starlette.requests import HTTPConnection
 from typing_extensions import override
 
 from agentstack_sdk.platform import use_platform_client
-from agentstack_sdk.types import JsonValue
+from agentstack_sdk.types import JsonValue, SdkAuthenticationBackend
 
 logger = logging.getLogger(__name__)
 
@@ -70,7 +70,7 @@ async def discover_jwks() -> KeySet:
         raise RuntimeError(f"JWKS discovery failed for url {url}") from e
 
 
-class PlatformAuthBackend(AuthenticationBackend):
+class PlatformAuthBackend(SdkAuthenticationBackend):
     def __init__(self, public_url: str | None = None, skip_audience_validation: bool | None = None) -> None:
         self.skip_audience_validation: bool = (
             skip_audience_validation
@@ -129,3 +129,16 @@ class PlatformAuthBackend(AuthenticationBackend):
         except Exception as e:
             logger.error(f"Authentication error: {e}")
             raise AuthenticationError(f"Authentication failed: {e}") from e
+
+    @override
+    def update_card_security_schemes(self, agent_card: AgentCard) -> None:
+        agent_card.security_schemes = {
+            "platform_context_token": SecurityScheme(
+                HTTPAuthSecurityScheme(
+                    scheme="bearer",
+                    bearer_format="JWT",
+                    description="Platform context token, issued by the AgentStack server using POST /api/v1/context/{context_id}/token.",
+                )
+            ),
+        }
+        agent_card.security = [{"platform_context_token": []}]

agentstack_sdk/server/server.py

@@ -24,7 +24,7 @@ from fastapi.applications import AppType
 from fastapi.responses import PlainTextResponse
 from httpx import HTTPError, HTTPStatusError
 from pydantic import AnyUrl
-from starlette.authentication import AuthenticationBackend, AuthenticationError
+from starlette.authentication import AuthenticationError
 from starlette.middleware.authentication import AuthenticationMiddleware
 from starlette.requests import HTTPConnection
 from starlette.types import Lifespan
@@ -39,6 +39,7 @@ 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.types import SdkAuthenticationBackend
 from agentstack_sdk.util.logging import configure_logger as configure_logger_func
 from agentstack_sdk.util.logging import logger
 
@@ -131,7 +132,7 @@ class Server:
         factory: bool = False,
         h11_max_incomplete_event_size: int | None = None,
         self_registration_client_factory: Callable[[], PlatformClient] | None = None,
-        auth_backend: AuthenticationBackend | None = None,
+        auth_backend: SdkAuthenticationBackend | None = None,
     ) -> None:
         if self.server:
             raise RuntimeError("The server is already running")
@@ -201,9 +202,11 @@ class Server:
             push_sender=push_sender,
             task_timeout=task_timeout,
             request_context_builder=request_context_builder,
+            auth_backend=auth_backend,
         )
 
         if auth_backend:
+            auth_backend.update_card_security_schemes(self._agent.card)
 
             def on_error(connection: HTTPConnection, error: AuthenticationError) -> PlainTextResponse:
                 return PlainTextResponse("Unauthorized", status_code=401)

agentstack_sdk/types.py

@@ -1,8 +1,12 @@
 # Copyright 2025 © BeeAI a Series of LF Projects, LLC
 # SPDX-License-Identifier: Apache-2.0
 
+import abc
 from typing import TYPE_CHECKING, TypeAlias
 
+from a2a.types import AgentCard
+from starlette.authentication import AuthenticationBackend
+
 if TYPE_CHECKING:
     JsonValue: TypeAlias = list["JsonValue"] | dict[str, "JsonValue"] | str | bool | int | float | None
     JsonDict: TypeAlias = dict[str, JsonValue]
@@ -13,3 +17,8 @@ else:
 
     JsonValue = TypeAliasType("JsonValue", "Union[dict[str, JsonValue], list[JsonValue], str, int, float, bool, None]")  # noqa: UP007
     JsonDict = TypeAliasType("JsonDict", "dict[str, JsonValue]")
+
+
+class SdkAuthenticationBackend(AuthenticationBackend, abc.ABC):
+    @abc.abstractmethod
+    def update_card_security_schemes(self, agent_card: AgentCard) -> None: ...

agentstack_sdk/util/resource_context.py

@@ -1,8 +1,8 @@
 # Copyright 2025 © BeeAI a Series of LF Projects, LLC
 # SPDX-License-Identifier: Apache-2.0
-
 import contextlib
 import contextvars
+import re
 import typing
 
 
@@ -28,7 +28,14 @@ def resource_context(
             try:
                 yield resource
             finally:
-                contextvar.reset(token)
+                try:
+                    contextvar.reset(token)
+                except ValueError as e:
+                    if re.match(r"<Token var=.+ was created in a different Context", str(e)):
+                        # Ignore contextvar reset errors when the contextvar was created in a different Context
+                        pass
+                    else:
+                        raise
 
         return manager()
 

{agentstack_sdk-0.6.0rc4.dist-info → agentstack_sdk-0.6.1rc1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentstack-sdk
-Version: 0.6.0rc4
+Version: 0.6.1rc1
 Summary: Agent Stack SDK
 Author: IBM Corp.
 Requires-Dist: a2a-sdk==0.3.21
@@ -21,6 +21,7 @@ Requires-Dist: mcp>=1.12.3
 Requires-Dist: fastapi>=0.116.1
 Requires-Dist: authlib>=1.3.0
 Requires-Dist: async-lru>=2.0.4
+Requires-Dist: cachetools>=6.2.3
 Requires-Python: >=3.11, <3.14
 Description-Content-Type: text/markdown
 

{agentstack_sdk-0.6.0rc4.dist-info → agentstack_sdk-0.6.1rc1.dist-info}/RECORD

@@ -33,7 +33,7 @@ agentstack_sdk/a2a/extensions/ui/form_request.py,sha256=YeRiovV3EgqeBD8fHtu_b5qr
 agentstack_sdk/a2a/extensions/ui/settings.py,sha256=uZu_8u29gAu1H85twFb_LCRBO0jeaEiLgfamaFNUPGM,1814
 agentstack_sdk/a2a/extensions/ui/trajectory.py,sha256=G66rIpr2VTe1UvbK82EgLXwFtUC3QfbyN_uLQGUU8jA,2334
 agentstack_sdk/a2a/types.py,sha256=z1-OxRLIJgYDHwaBJ66MZNIqSTpE1zQ04BPGBGw9IC0,3582
-agentstack_sdk/platform/__init__.py,sha256=2yu0t7mtMPv14MWk1PFiU5t6xaZf9n4VhY6moKjdNFw,351
+agentstack_sdk/platform/__init__.py,sha256=B7aJGw0_llNCHJJyP0tFTzt6fcw4JUM2vXjbRRTWrbQ,385
 agentstack_sdk/platform/client.py,sha256=KDN3R1wyoo-ZUYl6ArjRJxre0yAqB0Y6NyKrQpdI268,4234
 agentstack_sdk/platform/common.py,sha256=p6w4l49NaFp7LsjEBDAwghX6FliNChOeO7IHP_-TnXA,750
 agentstack_sdk/platform/configuration.py,sha256=YfOasb6LxE9LN6sMZPEU07Q61F7uIhiU3DIQFRaqinU,1690
@@ -43,6 +43,7 @@ agentstack_sdk/platform/file.py,sha256=lOqMwMKHKGWFy5P1huQ6DLRTiHU0KexX-s2qF3FIR
 agentstack_sdk/platform/model_provider.py,sha256=hv5UbFUFQ23IlalH5W-nCTvnUCrxBtZF4UrkeNWnwFE,4588
 agentstack_sdk/platform/provider.py,sha256=1juAfYc39sVwZ8ZbYz1ohVnzzRacJx6Junvo-7GgVmY,9071
 agentstack_sdk/platform/provider_build.py,sha256=Vn2XDdMGgQWvChWJmg9nxYFVecp42A1jXzURkFgSEQQ,7317
+agentstack_sdk/platform/provider_discovery.py,sha256=mxf-8Rrjd6-r3M5L7-B9BWB8wwXOCzqzpo0cMMUH-3g,1865
 agentstack_sdk/platform/types.py,sha256=3OHERUoli6t9rlCiaZ2YfN26GXqMQzDFQx_UluCZVYA,1502
 agentstack_sdk/platform/user.py,sha256=pqkac4JZqagOYVvdm8mk48xHvVn6SPq2tbe5tXx_W4M,2090
 agentstack_sdk/platform/user_feedback.py,sha256=IfMzZsaFKpWCPz5MjppDsFLLMM-iVQUBGugz0_FmrqY,1305
@@ -51,27 +52,27 @@ agentstack_sdk/platform/vector_store.py,sha256=h98RKGuN1SvmfL_bxf3WCLvY1IeQ00Wf7
 agentstack_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentstack_sdk/server/__init__.py,sha256=vc06L5ILofytmLJC9hJUzA9LakUYUNYGiBoSXNs9hQc,152
 agentstack_sdk/server/agent.py,sha256=tprDvYwsSoImJkMCMVvHKzacH2pWtQUWu19qgWAyNMk,29170
-agentstack_sdk/server/app.py,sha256=1-g_6FBU4lnrBI2KZVnTvsfdKrHyChkeYmJ01iIrj4Y,3396
+agentstack_sdk/server/app.py,sha256=pLyf4CyG8PWYuNDUO0FE99XyDjjc9HFj8ii3PrWDGd8,3024
 agentstack_sdk/server/constants.py,sha256=huQAP351AeNt536tB-axJSseHMHfyDKwJdlackjLGnw,188
 agentstack_sdk/server/context.py,sha256=yBAk3RgBsat1nULzFs2OvF2UKKeNAT8yaSN_s-bcPTU,2712
 agentstack_sdk/server/dependencies.py,sha256=kyUwrIJe99n7UQmqGn_bo--YkPiAR6S9SoeOBwaeW68,5022
 agentstack_sdk/server/exceptions.py,sha256=jr8otByt8HFY9OWptuWdpq4WHE7A1srCSRXZV066FrI,94
 agentstack_sdk/server/middleware/__init__.py,sha256=pEJG3OupRq7IF3O70g2pvDU_EmDF0D3iAuGWXB-_Ipk,94
-agentstack_sdk/server/middleware/platform_auth_backend.py,sha256=h2P4jKJpXOmie1tFU2E5YW-c6-BksINmWEonNUQ5l3g,4986
-agentstack_sdk/server/server.py,sha256=twD5j4Re8rfg63CjcMLJg_tpPBKAkffnP22hTkVS5iY,15909
+agentstack_sdk/server/middleware/platform_auth_backend.py,sha256=fhSWM3Oj794qwRoKcCnAm6fFz8icjK2Ibg8qyCKaWHM,5606
+agentstack_sdk/server/server.py,sha256=OOEsBB_g5l4ALEepaK5ZZjtyYV65Kv_oLFBnPBZTLVs,16058
 agentstack_sdk/server/store/__init__.py,sha256=jr8otByt8HFY9OWptuWdpq4WHE7A1srCSRXZV066FrI,94
 agentstack_sdk/server/store/context_store.py,sha256=ZzXSlxaV_xaTEVDAO0oC0eCiNQve2ydjYUJK3oUAOXA,1060
 agentstack_sdk/server/store/memory_context_store.py,sha256=S604ATyk9wA2D_pZL-Q-I4x01mItKihLJmGwp7bumC4,2346
 agentstack_sdk/server/store/platform_context_store.py,sha256=bv5-sVJUVVblYIAjLsl0_VOqstjfAXWuQgWhewE_9OI,2614
 agentstack_sdk/server/telemetry.py,sha256=MJ5JGF3L_ef8eqx-yguJHgIhbpDtHRSfKm2QKFtlHmI,2014
 agentstack_sdk/server/utils.py,sha256=0s7N-kWEWnPuI2Ki4PZuX5uY48KFLCOHg-lnxRDVEg8,917
-agentstack_sdk/types.py,sha256=1GOt-l2xnJTS_6u37YNoMXkuckX2XwjQ1aH1iv6QRIQ,593
+agentstack_sdk/types.py,sha256=VTOV489DavWGT1qS5iDCSwY4lxa3PEFODeFk19OYHSw,865
 agentstack_sdk/util/__init__.py,sha256=dCOBqJYOJYvvQSjDZysZGkLxvZV-o9j367ze3jnX_Rc,126
 agentstack_sdk/util/file.py,sha256=rm5b47tHTmJBsHlqGcUqFCP0Bz5HKXPHWW_h-IpG9pw,9035
 agentstack_sdk/util/httpx.py,sha256=g6WKgkGCLc40wZA2CPYO_R2P-nav6QP1XmVlgkh9wYY,491
 agentstack_sdk/util/logging.py,sha256=hGLkjw-P3GefiUQmDmcz7BZQxfT2Y44XlLyAMNxZXMo,2136
-agentstack_sdk/util/resource_context.py,sha256=OmjEXvrLQA6nBkVSBt0n24hNNxJkucd-N32haxJ4Mno,1093
+agentstack_sdk/util/resource_context.py,sha256=AMvTxcG8zdMZplzFACFvgOgARp3xP9BQqz6eh0EyLqg,1458
 agentstack_sdk/util/utils.py,sha256=18qFqMRkX4g4eIpvIvLb4FZAs8Q8ojJrpm19oJleb-k,1593
-agentstack_sdk-0.6.0rc4.dist-info/WHEEL,sha256=M6du7VZflc4UPsGphmOXHANdgk8zessdJG0DBUuoA-U,78
-agentstack_sdk-0.6.0rc4.dist-info/METADATA,sha256=Ooq2LT9oNUCsmUZ7JE5gKg53fOcexsJbGdM68aRV6WQ,4433
-agentstack_sdk-0.6.0rc4.dist-info/RECORD,,
+agentstack_sdk-0.6.1rc1.dist-info/WHEEL,sha256=M6du7VZflc4UPsGphmOXHANdgk8zessdJG0DBUuoA-U,78
+agentstack_sdk-0.6.1rc1.dist-info/METADATA,sha256=gVOcXRyRSsmet1fMR4DnJNoTD7OYpjMdC3QjqmbAEZ4,4466
+agentstack_sdk-0.6.1rc1.dist-info/RECORD,,