agentstack-sdk 0.6.1rc1__tar.gz → 0.6.2__tar.gz

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentstack-sdk
-Version: 0.6.1rc1
+Version: 0.6.2
 Summary: Agent Stack SDK
 Author: IBM Corp.
 Requires-Dist: a2a-sdk==0.3.21
@@ -22,6 +22,8 @@ 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-Dist: opentelemetry-instrumentation-httpx>=0.60b1
+Requires-Dist: opentelemetry-instrumentation-openai>=0.52.3
 Requires-Python: >=3.11, <3.14
 Description-Content-Type: text/markdown
 

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentstack-sdk"
-version = "0.6.1-rc1"
+version = "0.6.2"
 description = "Agent Stack SDK"
 readme = "README.md"
 authors = [{ name = "IBM Corp." }]
@@ -25,12 +25,14 @@ dependencies = [
     "authlib>=1.3.0",
     "async-lru>=2.0.4",
     "cachetools>=6.2.3",
+    "opentelemetry-instrumentation-httpx>=0.60b1",
+    "opentelemetry-instrumentation-openai>=0.52.3",
 ]
 
 [dependency-groups]
 dev = [
     "beeai-framework[duckduckgo,wikipedia]>=0.1.76",
-    "pyright>=1.1.403",
+    "pyrefly>=0.52.0",
     "pytest>=8.4.1",
     "pytest-asyncio>=1.1.0",
     "pytest-httpx>=0.35.0",
@@ -64,7 +66,7 @@ lint.select = [
 lint.ignore = [
     "E501", # line lenght (annyoing)
     "N999", # invalid module name agentstack-server (yeah, we use a dash, deal with it)
-    "F403", # redundant with Pyright
+    "F403", # redundant with type checking
 ]
 force-exclude = true
 
@@ -74,7 +76,8 @@ asyncio_mode = "auto"
 asyncio_default_fixture_loop_scope = "function"
 addopts = "-v"
 
-[tool.pyright]
-ignore = ["tests/**", "examples/cli.py"]
-venvPath = "."
-venv = ".venv"
+[tool.pyrefly]
+project-includes = [
+    "**/*.py*",
+    "**/*.ipynb",
+]

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/a2a/extensions/auth/oauth/oauth.py

@@ -145,7 +145,7 @@ class OAuthExtensionClient(BaseExtensionClient[OAuthExtensionSpec, NoneType]):
         return A2AMessage(
             message_id=str(uuid.uuid4()),
             role=Role.user,
-            parts=[TextPart(text="Authorization completed")],  # type: ignore
+            parts=[TextPart(text="Authorization completed")],
             task_id=task_id,
             metadata={self.spec.URI: data.model_dump(mode="json")},
         )

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/a2a/extensions/base.py

@@ -51,11 +51,6 @@ class BaseExtensionSpec(abc.ABC, typing.Generic[ParamsT]):
     Description to be attached with the extension spec.
     """
 
-    Params: type[ParamsT]
-    """
-    Type of the extension params, attached to the agent card.
-    """
-
     def __init_subclass__(cls, **kwargs):
         super().__init_subclass__(**kwargs)
         cls.Params = _get_generic_args(cls, BaseExtensionSpec)[0]
@@ -72,7 +67,7 @@ class BaseExtensionSpec(abc.ABC, typing.Generic[ParamsT]):
         self.params = params
 
     @classmethod
-    def from_agent_card(cls, agent: AgentCard) -> typing.Self | None:
+    def from_agent_card(cls: type[BaseExtensionSpec], agent: AgentCard) -> typing.Self | None:
         """
         Client should construct an extension instance using this classmethod.
         """
@@ -120,7 +115,6 @@ ExtensionSpecT = typing.TypeVar("ExtensionSpecT", bound=BaseExtensionSpec[typing
 
 
 class BaseExtensionServer(abc.ABC, typing.Generic[ExtensionSpecT, MetadataFromClientT]):
-    MetadataFromClient: type[MetadataFromClientT]
     """
     Type of the extension metadata, attached to messages.
     """
@@ -182,7 +176,6 @@ class BaseExtensionServer(abc.ABC, typing.Generic[ExtensionSpecT, MetadataFromCl
 
 
 class BaseExtensionClient(abc.ABC, typing.Generic[ExtensionSpecT, MetadataFromServerT]):
-    MetadataFromServer: type[MetadataFromServerT]
     """
     Type of the extension metadata, attached to messages.
     """

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/a2a/extensions/common/form.py

@@ -1,6 +1,7 @@
 # Copyright 2025 © BeeAI a Series of LF Projects, LLC
 # SPDX-License-Identifier: Apache-2.0
 
+from collections.abc import Generator
 from typing import Literal
 
 from pydantic import BaseModel, Field, model_validator
@@ -137,7 +138,7 @@ FormFieldValue = (
 class FormResponse(BaseModel):
     values: dict[str, FormFieldValue]
 
-    def __iter__(self):
+    def __iter__(self) -> Generator[tuple[str, list[dict[str, str | None]] | list[str] | str | bool | None]]:
         for key, value in self.values.items():
             match value:
                 case FileFieldValue():

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/a2a/extensions/services/form.py

@@ -4,7 +4,7 @@
 
 from __future__ import annotations
 
-from typing import Self, TypeVar, cast
+from typing import Any, Self, TypeVar
 
 from pydantic import BaseModel, TypeAdapter
 from typing_extensions import TypedDict
@@ -38,17 +38,17 @@ T = TypeVar("T")
 
 
 class FormServiceExtensionServer(BaseExtensionServer[FormServiceExtensionSpec, FormServiceExtensionMetadata]):
-    def parse_initial_form(self, *, model: type[T] = FormResponse) -> T | None:
-        if self.data is None:
-            return None
-
-        initial_form = self.data.form_fulfillments.get("initial_form")
-
-        if initial_form is None:
-            return None
-        if model is FormResponse:
-            return cast(T, initial_form)
-        return TypeAdapter(model).validate_python(dict(initial_form))
-
-
-class FormServiceExtensionClient(BaseExtensionClient[FormServiceExtensionSpec, FormRender]): ...
+    def parse_initial_form(self, *, model: type[T] | None = None) -> T | None:
+        initial_form = getattr(self.data, "form_fulfillments", {}).get("initial_form")
+        return (
+            TypeAdapter(model).validate_python(dict(initial_form))
+            if initial_form is not None and model is not None
+            else initial_form
+        )
+
+
+class FormServiceExtensionClient(BaseExtensionClient[FormServiceExtensionSpec, FormRender]):
+    def fulfillment_metadata(self, *, form_fulfillments: dict[str, FormResponse]) -> dict[str, Any]:
+        return {
+            self.spec.URI: FormServiceExtensionMetadata(form_fulfillments=form_fulfillments).model_dump(mode="json")
+        }

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/a2a/extensions/services/mcp.py

@@ -12,7 +12,7 @@ import pydantic
 from a2a.server.agent_execution.context import RequestContext
 from a2a.types import Message as A2AMessage
 from mcp.client.stdio import StdioServerParameters, stdio_client
-from mcp.client.streamable_http import streamablehttp_client
+from mcp.client.streamable_http import streamablehttp_client  # pyrefly: ignore [deprecated] -- TODO: upgrade
 from typing_extensions import override
 
 from agentstack_sdk.a2a.extensions.auth.oauth.oauth import OAuthExtensionServer
@@ -160,6 +160,7 @@ class MCPServiceExtensionServer(BaseExtensionServer[MCPServiceExtensionSpec, MCP
             ):
                 yield (read, write)
         elif isinstance(transport, StreamableHTTPTransport):
+            # pyrefly: ignore [deprecated] -- TODO: upgrade
             async with streamablehttp_client(
                 url=transport.url,
                 headers=transport.headers,

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/a2a/extensions/tools/call.py

@@ -88,7 +88,7 @@ class ToolCallExtensionServer(BaseExtensionServer[ToolCallExtensionSpec, ToolCal
             match result.action:
                 case "accept":
                     return result
-                case "reject":
+                case _:
                     raise ToolCallRejectionError("User has rejected the tool call")
 
         else:

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/a2a/extensions/ui/form_request.py

@@ -1,9 +1,9 @@
-# Copyright 2025 © BeeAI a Series of LF Projects, LLC
+# Copyright 2026 © BeeAI a Series of LF Projects, LLC
 # SPDX-License-Identifier: Apache-2.0
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, TypeVar, cast
+from typing import TYPE_CHECKING, TypeVar, overload
 
 from a2a.server.agent_execution.context import RequestContext
 from a2a.types import Message as A2AMessage
@@ -34,19 +34,27 @@ class FormRequestExtensionServer(BaseExtensionServer[FormRequestExtensionSpec, F
         super().handle_incoming_message(message, run_context, request_context)
         self.context = run_context
 
-    async def request_form(self, *, form: FormRender, model: type[T] = FormResponse) -> T | None:
+    @overload
+    async def request_form(self, *, form: FormRender, model: None = None) -> FormResponse | None: ...
+    @overload
+    async def request_form(self, *, form: FormRender, model: type[T]) -> T | None: ...
+    async def request_form(self, *, form: FormRender, model: type[T] | None = None) -> T | FormResponse | None:
         message = await self.context.yield_async(
             InputRequired(message=AgentMessage(text=form.title, metadata={self.spec.URI: form}))
         )
-        return self.parse_form_response(message=message, model=model) if message else None
-
-    def parse_form_response(self, *, message: A2AMessage, model: type[T] = FormResponse) -> T | None:
-        form_response = self.parse_client_metadata(message)
-        if form_response is None:
-            return None
-        if model is FormResponse:
-            return cast(T, form_response)
-        return TypeAdapter(model).validate_python(dict(form_response))
+        return self.parse_form_response(message=message, model=model or FormResponse) if message else None
+
+    @overload
+    def parse_form_response(self, *, message: A2AMessage, model: None = None) -> FormResponse | None: ...
+    @overload
+    def parse_form_response(self, *, message: A2AMessage, model: type[T]) -> T | None: ...
+    def parse_form_response(self, *, message: A2AMessage, model: type[T] | None = None) -> T | FormResponse | None:
+        form_response: FormResponse | None = self.parse_client_metadata(message)
+        return (
+            TypeAdapter(model).validate_python(dict(iter(form_response)))
+            if form_response is not None and model is not None
+            else form_response
+        )
 
 
 class FormRequestExtensionClient(BaseExtensionClient[FormRequestExtensionSpec, FormRender]): ...

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/a2a/types.py

@@ -48,33 +48,37 @@ RunYieldResume: TypeAlias = Message | None
 
 class AgentArtifact(Artifact):
     artifact_id: str = Field(default_factory=lambda: str(uuid.uuid4()))
+    # pyrefly: ignore [bad-override] -- we intentionally broaden the base type
     parts: list[Part | TextPart | FilePart | DataPart]
 
     @model_validator(mode="after")
     def text_message_validate(self):
-        self.parts = [part if isinstance(part, Part) else Part(root=part) for part in self.parts]  # pyright: ignore [reportIncompatibleVariableOverride]
+        self.parts = [part if isinstance(part, Part) else Part(root=part) for part in self.parts]
         return self
 
 
 class ArtifactChunk(Artifact):
     last_chunk: bool = False
+    # pyrefly: ignore [bad-override] -- we intentionally broaden the base type
     parts: list[Part | TextPart | FilePart | DataPart]
 
     @model_validator(mode="after")
     def text_message_validate(self):
-        self.parts = [part if isinstance(part, Part) else Part(root=part) for part in self.parts]  # pyright: ignore [reportIncompatibleVariableOverride]
+        self.parts = [part if isinstance(part, Part) else Part(root=part) for part in self.parts]
         return self
 
 
 class AgentMessage(Message):
     message_id: str = Field(default_factory=lambda: str(uuid.uuid4()))
-    role: Literal[Role.agent] = Role.agent  # pyright: ignore [reportIncompatibleVariableOverride]
+    # pyrefly: ignore [bad-override] -- we treat this as read-only
+    role: Literal[Role.agent] = Role.agent
     text: str | None = Field(default=None, exclude=True)
+    # pyrefly: ignore [bad-override] -- we intentionally broaden the base type
     parts: list[Part | TextPart | FilePart | DataPart] = Field(default_factory=list)
 
     @model_validator(mode="after")
     def text_message_validate(self):
-        self.parts = [part if isinstance(part, Part) else Part(root=part) for part in self.parts]  # pyright: ignore [reportIncompatibleVariableOverride]
+        self.parts = [part if isinstance(part, Part) else Part(root=part) for part in self.parts]
         if self.parts and self.text is not None:
             raise ValueError("Message cannot have both parts and text")
         if self.text is not None:
@@ -84,7 +88,8 @@ class AgentMessage(Message):
 
 class InputRequired(TaskStatus):
     message: Message | None = None
-    state: Literal[TaskState.input_required] = TaskState.input_required  # pyright: ignore [reportIncompatibleVariableOverride]
+    # pyrefly: ignore [bad-override] -- we treat this as read-only
+    state: Literal[TaskState.input_required] = TaskState.input_required
     text: str | None = Field(default=None, exclude=True)
     parts: list[Part | TextPart | DataPart | FilePart] = Field(exclude=True, default_factory=list)
 
@@ -101,4 +106,5 @@ class InputRequired(TaskStatus):
 
 
 class AuthRequired(InputRequired):
-    state: Literal[TaskState.auth_required] = TaskState.auth_required  # pyright: ignore [reportIncompatibleVariableOverride]
+    # pyrefly: ignore [bad-override] -- we treat this as read-only
+    state: Literal[TaskState.auth_required] = TaskState.auth_required

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/platform/context.py

@@ -3,6 +3,7 @@
 
 from __future__ import annotations
 
+import builtins
 from collections.abc import AsyncIterator
 from typing import Literal
 from uuid import UUID, uuid4
@@ -181,7 +182,7 @@ class Context(pydantic.BaseModel):
     async def generate_token(
         self: Context | str,
         *,
-        providers: list[str] | list[Provider] | None = None,
+        providers: builtins.list[str] | builtins.list[Provider] | None = None,
         client: PlatformClient | None = None,
         grant_global_permissions: Permissions | None = None,
         grant_context_permissions: ContextPermissions | None = None,

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/platform/file.py

@@ -3,6 +3,7 @@
 
 from __future__ import annotations
 
+import builtins
 import typing
 from collections.abc import AsyncIterator
 from contextlib import asynccontextmanager
@@ -197,7 +198,7 @@ class File(pydantic.BaseModel):
     async def create_extraction(
         self: File | str,
         *,
-        formats: list[ExtractionFormatLiteral] | None = None,
+        formats: builtins.list[ExtractionFormatLiteral] | None = None,
         client: PlatformClient | None = None,
         context_id: str | None | Literal["auto"] = "auto",
     ) -> Extraction:

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/platform/model_provider.py

@@ -3,6 +3,7 @@
 
 from __future__ import annotations
 
+import builtins
 from enum import StrEnum
 
 import pydantic
@@ -44,6 +45,11 @@ class ModelCapability(StrEnum):
     EMBEDDING = "embedding"
 
 
+class ModelProviderState(StrEnum):
+    ONLINE = "online"
+    OFFLINE = "offline"
+
+
 class ModelProvider(pydantic.BaseModel):
     id: str
     name: str | None = None
@@ -54,6 +60,7 @@ class ModelProvider(pydantic.BaseModel):
     watsonx_space_id: str | None = None
     created_at: pydantic.AwareDatetime
     capabilities: set[ModelCapability]
+    state: ModelProviderState
 
     @staticmethod
     async def create(
@@ -110,7 +117,7 @@ class ModelProvider(pydantic.BaseModel):
         capability: ModelCapability = ModelCapability.LLM,
         suggested_models: tuple[str, ...] | None = None,
         client: PlatformClient | None = None,
-    ) -> list[ModelWithScore]:
+    ) -> builtins.list[ModelWithScore]:
         async with client or get_platform_client() as client:
             return pydantic.TypeAdapter(list[ModelWithScore]).validate_python(
                 (
@@ -124,7 +131,7 @@ class ModelProvider(pydantic.BaseModel):
             )
 
     @staticmethod
-    async def list(*, client: PlatformClient | None = None) -> list[ModelProvider]:
+    async def list(*, client: PlatformClient | None = None) -> builtins.list[ModelProvider]:
         async with client or get_platform_client() as client:
             return pydantic.TypeAdapter(list[ModelProvider]).validate_python(
                 (await client.get(url="/api/v1/model_providers")).raise_for_status().json()["items"]

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/server/agent.py

@@ -54,7 +54,7 @@ from agentstack_sdk.util.logging import logger
 AgentFunction: TypeAlias = Callable[[], AsyncGenerator[RunYield, RunYieldResume]]
 AgentFunctionFactory: TypeAlias = Callable[[RequestContext, ContextStore], AbstractAsyncContextManager[AgentFunction]]
 
-OriginalFnType = TypeVar("OriginalFnType", bound=Callable[..., Any])  # pyright: ignore[reportExplicitAny]
+OriginalFnType = TypeVar("OriginalFnType", bound=Callable[..., Any])
 
 
 class AgentExecuteFn(typing.Protocol):
@@ -121,7 +121,7 @@ def agent(
     """
 
     capabilities = capabilities.model_copy(deep=True) if capabilities else AgentCapabilities(streaming=True)
-    detail = detail or AgentDetail()  # pyright: ignore [reportCallIssue]
+    detail = detail or AgentDetail()
 
     def decorator(fn: OriginalFnType) -> AgentFactory:
         def agent_factory(modify_dependencies: Callable[[dict[str, Depends]], None]):
@@ -339,7 +339,7 @@ class AgentRun:
                     else initialize_deps_exceptions[0]
                 )
 
-            self.run_context._store = await self._context_store.create(  # pyright: ignore[reportPrivateUsage]
+            self.run_context._store = await self._context_store.create(
                 context_id=self.run_context.context_id,
                 initialized_dependencies=list(dependency_args.values()),
             )
@@ -360,8 +360,8 @@ class AgentRun:
         )
 
     async def _run_agent_function(self, initial_message: Message) -> None:
-        yield_queue = self.run_context._yield_queue  # pyright: ignore[reportPrivateUsage]
-        yield_resume_queue = self.run_context._yield_resume_queue  # pyright: ignore[reportPrivateUsage]
+        yield_queue = self.run_context._yield_queue
+        yield_resume_queue = self.run_context._yield_resume_queue
 
         try:
             async with self._dependencies_lifespan(initial_message) as dependency_args:
@@ -565,8 +565,9 @@ class Executor(AgentExecutor):
         assert task_id and context_id
 
         async def cleanup_fn():
+            nonlocal task_id
             await asyncio.sleep(self._task_timeout.total_seconds())
-            if not (run := self._running_tasks.get(task_id)):
+            if task_id is None or not (run := self._running_tasks.get(task_id)):
                 return
             try:
                 while not run.done:

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/server/app.py

@@ -16,8 +16,9 @@ from a2a.server.tasks import (
     TaskStore,
 )
 from a2a.types import AgentInterface, TransportProtocol
-from fastapi import APIRouter, Depends, FastAPI
+from fastapi import APIRouter, FastAPI
 from fastapi.applications import AppType
+from fastapi.params import Depends
 from starlette.types import Lifespan
 
 from agentstack_sdk.server.agent import Agent, Executor
@@ -34,7 +35,7 @@ def create_app(
     push_sender: PushNotificationSender | None = None,
     request_context_builder: RequestContextBuilder | None = None,
     lifespan: Lifespan[AppType] | None = None,
-    dependencies: list[Depends] | None = None,  # pyright: ignore [reportGeneralTypeIssues]
+    dependencies: list[Depends] | None = None,
     override_interfaces: bool = True,
     task_timeout: timedelta = timedelta(minutes=10),
     **kwargs,

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/server/context.py

@@ -33,14 +33,10 @@ class RunContext(BaseModel, arbitrary_types_allowed=True):
         await self._store.store(data)
 
     @overload
-    async def load_history(
-        self, load_history_items: Literal[False] = False
-    ) -> AsyncGenerator[Message | Artifact, None]:
-        yield ...  # type: ignore
+    def load_history(self, load_history_items: Literal[False] = False) -> AsyncGenerator[Message | Artifact, None]: ...
 
     @overload
-    async def load_history(self, load_history_items: Literal[True]) -> AsyncGenerator[ContextHistoryItem, None]:
-        yield ...  # type: ignore
+    def load_history(self, load_history_items: Literal[True]) -> AsyncGenerator[ContextHistoryItem, None]: ...
 
     async def load_history(
         self, load_history_items: bool = False

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/server/middleware/platform_auth_backend.py

@@ -63,7 +63,7 @@ async def discover_jwks() -> KeySet:
     try:
         async with use_platform_client() as client:
             response = await client.get("/.well-known/jwks")
-            return JsonWebKey.import_key_set(response.raise_for_status().json())  # pyright: ignore[reportAny]
+            return JsonWebKey.import_key_set(response.raise_for_status().json())
     except Exception as e:
         url = "{platform_url}/.well-known/jwks"
         logger.warning(f"JWKS discovery failed for url {url}: {e}")

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/server/server.py

@@ -56,13 +56,12 @@ class Server:
         self._all_configured_variables: set[str] = set()
 
     @functools.wraps(agent_decorator)
-    def agent(*args, **kwargs) -> Callable:
-        self, other_args = args[0], args[1:]  # Must hide self due to pyright issues
+    def agent(self, *args, **kwargs) -> Callable:
         if self._agent_factory:
             raise ValueError("Server can have only one agent.")
 
         def decorator(fn: Callable) -> Callable:
-            self._agent_factory = agent_decorator(*other_args, **kwargs)(fn)  # pyright: ignore [reportArgumentType]
+            self._agent_factory = agent_decorator(*args, **kwargs)(fn)
             return fn
 
         return decorator
@@ -158,6 +157,8 @@ class Server:
 
         from agentstack_sdk.server.app import create_app
 
+        self_registration = False if self._production_mode else self_registration
+
         @asynccontextmanager
         async def _lifespan_fn(app: FastAPI) -> AsyncGenerator[None, None]:
             async with self._self_registration_client or nullcontext():
@@ -165,7 +166,8 @@ class Server:
                 reload_task = asyncio.create_task(self._reload_variables_periodically()) if self_registration else None
 
                 try:
-                    async with lifespan_fn(app) if lifespan_fn else nullcontext():  # pyright: ignore [reportArgumentType]
+                    # pyrefly: ignore [bad-argument-type] -- probably bug in Pyrefly
+                    async with lifespan_fn(app) if lifespan_fn else nullcontext():
                         yield
                 finally:
                     if register_task:
@@ -275,7 +277,7 @@ class Server:
 
     @functools.wraps(serve)
     def run(*args, **kwargs) -> None:
-        self = args[0]  # Must hide self due to pyright issues
+        self = args[0]  # TODO(typing): resolve without hiding `self`
         asyncio.run(self.serve(**kwargs))
 
     @property

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/server/store/context_store.py

@@ -17,13 +17,10 @@ if TYPE_CHECKING:
 
 
 class ContextStoreInstance(Protocol):
-    async def load_history(
+    def load_history(
         self, load_history_items: bool = False
-    ) -> AsyncIterator[ContextHistoryItem | Message | Artifact]:
-        yield ...  # type: ignore
-
+    ) -> AsyncIterator[ContextHistoryItem | Message | Artifact]: ...
     async def store(self, data: Message | Artifact) -> None: ...
-
     async def delete_history_from_id(self, from_id: UUID) -> None: ...
 
 

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/server/telemetry.py

@@ -9,6 +9,7 @@ from opentelemetry.exporter.otlp.proto.http._log_exporter import OTLPLogExporter
 from opentelemetry.exporter.otlp.proto.http.metric_exporter import OTLPMetricExporter
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
 from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler
 from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
 from opentelemetry.sdk.metrics import MeterProvider
@@ -31,6 +32,20 @@ def configure_telemetry(app: FastAPI) -> None:
 
     FastAPIInstrumentor.instrument_app(app)
 
+    httpxclient_instrumentor = HTTPXClientInstrumentor()
+    if httpxclient_instrumentor:
+        httpxclient_instrumentor.instrument()
+
+    try:
+        import openai  # noqa: F401
+        from opentelemetry.instrumentation.openai import OpenAIInstrumentor
+
+        openai_instrumentor = OpenAIInstrumentor()
+        if openai_instrumentor:
+            openai_instrumentor.instrument()
+    except ModuleNotFoundError:
+        pass
+
     resource = Resource(attributes={SERVICE_NAME: "agentstack-sdk-a2a-server", SERVICE_VERSION: __version__})
 
     # Traces

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/util/file.py

@@ -90,7 +90,7 @@ class LoadedFileWithBytes(LoadedFile):
         return self._content
 
     @cached_property
-    def text(self) -> str:  # pyright: ignore [reportIncompatibleMethodOverride]
+    def text(self) -> str:
         return self._content.decode(self._encoding)
 
     def read(self) -> bytes:

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/util/logging.py

@@ -54,8 +54,8 @@ def configure_logger(level: int | str | None = None) -> None:
     if level is not None:
         level = level if isinstance(level, int) else logging.getLevelNamesMapping()[level.upper()]
         logging.config.dictConfig(
-            {
-                **LOGGING_CONFIG,
+            LOGGING_CONFIG
+            | {
                 "loggers": {"root": {"level": getLevelName(level), "handlers": ["console"]}},
             }
         )

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/util/resource_context.py

@@ -18,7 +18,9 @@ def resource_context(
     factory: typing.Callable[P, T],
     default_factory: typing.Callable[[], T],
 ) -> tuple[typing.Callable[[], T], typing.Callable[P, contextlib.AbstractContextManager[T]]]:
-    contextvar: contextvars.ContextVar[T] = contextvars.ContextVar(f"resource_context({factory.__name__})")
+    contextvar: contextvars.ContextVar[T] = contextvars.ContextVar(
+        f"resource_context({getattr(factory, '__name__', '<unknown>')})"
+    )
 
     def use_resource(*args: P.args, **kwargs: P.kwargs):
         @contextlib.contextmanager

{agentstack_sdk-0.6.1rc1 → agentstack_sdk-0.6.2}/src/agentstack_sdk/util/utils.py

@@ -4,18 +4,16 @@ import json
 import re
 from collections.abc import AsyncIterator
 from datetime import UTC, datetime
-from typing import Any, TypeVar, cast
+from typing import Any, TypeVar
 
 import httpx
 from httpx import HTTPStatusError
 
 T = TypeVar("T")
-V = TypeVar("V")
 
 
-def filter_dict(map: dict[str, T | V], value_to_exclude: V = None) -> dict[str, T]:
-    """Remove entries with unwanted values (None by default) from dictionary."""
-    return {key: cast(T, value) for key, value in map.items() if value is not value_to_exclude}
+def filter_dict(d: dict):
+    return {k: v for k, v in d.items() if v is not None}
 
 
 async def parse_stream(response: httpx.Response) -> AsyncIterator[dict[str, Any]]: