agentstack-sdk 0.6.0rc3__tar.gz → 0.6.1rc1__tar.gz

{agentstack_sdk-0.6.0rc3 → agentstack_sdk-0.6.1rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentstack-sdk
-Version: 0.6.0rc3
+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.0rc3 → agentstack_sdk-0.6.1rc1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentstack-sdk"
-version = "0.6.0-rc3"
+version = "0.6.1-rc1"
 description = "Agent Stack SDK"
 readme = "README.md"
 authors = [{ name = "IBM Corp." }]
@@ -19,16 +19,17 @@ dependencies = [
     "opentelemetry-sdk>=1.35.0",
     "tenacity>=9.1.2",
     "janus>=2.0.0",
-    "httpx",                                          # version determined by a2a-sdk
+    "httpx", # version determined by a2a-sdk
     "mcp>=1.12.3",
     "fastapi>=0.116.1",
     "authlib>=1.3.0",
     "async-lru>=2.0.4",
+    "cachetools>=6.2.3",
 ]
 
 [dependency-groups]
 dev = [
-    "beeai-framework[duckduckgo,wikipedia]>=0.1.75",
+    "beeai-framework[duckduckgo,wikipedia]>=0.1.76",
     "pyright>=1.1.403",
     "pytest>=8.4.1",
     "pytest-asyncio>=1.1.0",

{agentstack_sdk-0.6.0rc3 → agentstack_sdk-0.6.1rc1}/src/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-0.6.1rc1/src/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-0.6.0rc3 → agentstack_sdk-0.6.1rc1}/src/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-0.6.0rc3 → agentstack_sdk-0.6.1rc1}/src/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-0.6.0rc3 → agentstack_sdk-0.6.1rc1}/src/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-0.6.0rc3 → agentstack_sdk-0.6.1rc1}/src/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-0.6.0rc3 → agentstack_sdk-0.6.1rc1}/src/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()