agentstack-sdk 0.5.1rc3__tar.gz → 0.5.2rc2__tar.gz

{agentstack_sdk-0.5.1rc3 → agentstack_sdk-0.5.2rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentstack-sdk
-Version: 0.5.1rc3
+Version: 0.5.2rc2
 Summary: Agent Stack SDK
 Author: IBM Corp.
 Requires-Dist: a2a-sdk==0.3.21

{agentstack_sdk-0.5.1rc3 → agentstack_sdk-0.5.2rc2}/pyproject.toml

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

{agentstack_sdk-0.5.1rc3 → agentstack_sdk-0.5.2rc2}/src/agentstack_sdk/platform/context.py

@@ -5,25 +5,29 @@ from __future__ import annotations
 
 from collections.abc import AsyncIterator
 from typing import Literal
-from uuid import UUID
+from uuid import UUID, uuid4
 
 import pydantic
 from a2a.types import Artifact, Message
-from pydantic import AwareDatetime, BaseModel, SerializeAsAny
+from pydantic import AwareDatetime, BaseModel, Field, SerializeAsAny, computed_field
 
 from agentstack_sdk.platform.client import PlatformClient, get_platform_client
 from agentstack_sdk.platform.common import PaginatedResult
 from agentstack_sdk.platform.provider import Provider
 from agentstack_sdk.platform.types import Metadata, MetadataPatch
-from agentstack_sdk.util.utils import filter_dict
+from agentstack_sdk.util.utils import filter_dict, utc_now
 
 
 class ContextHistoryItem(BaseModel):
-    id: UUID
+    id: UUID = Field(default_factory=uuid4)
     data: Artifact | Message
-    created_at: AwareDatetime
-    context_id: UUID
-    kind: Literal["message", "artifact"]
+    created_at: AwareDatetime = Field(default_factory=utc_now)
+    context_id: str
+
+    @computed_field
+    @property
+    def kind(self) -> Literal["message", "artifact"]:
+        return getattr(self.data, "kind", "artifact")
 
 
 class ContextToken(pydantic.BaseModel):
@@ -49,9 +53,6 @@ class Permissions(ContextPermissions):
     provider_variables: set[Literal["read", "write", "*"]] = set()
 
     contexts: set[Literal["read", "write", "*"]] = set()
-    mcp_providers: set[Literal["read", "write", "*"]] = set()
-    mcp_tools: set[Literal["read", "*"]] = set()
-    mcp_proxy: set[Literal["*"]] = set()
 
     connectors: set[Literal["read", "write", "proxy", "*"]] = set()
 
@@ -238,6 +239,21 @@ class Context(pydantic.BaseModel):
                 )
             ).raise_for_status()
 
+    async def delete_history_from_id(
+        self: Context | str,
+        *,
+        from_id: UUID | str,
+        client: PlatformClient | None = None,
+    ) -> None:
+        """Delete all history items from a specific item onwards (inclusive)"""
+        target_context_id = self if isinstance(self, str) else self.id
+        async with client or get_platform_client() as platform_client:
+            _ = (
+                await platform_client.delete(
+                    url=f"/api/v1/contexts/{target_context_id}/history", params={"from_id": str(from_id)}
+                )
+            ).raise_for_status()
+
     async def list_history(
         self: Context | str,
         *,

{agentstack_sdk-0.5.1rc3 → agentstack_sdk-0.5.2rc2}/src/agentstack_sdk/server/context.py

@@ -1,14 +1,16 @@
 # Copyright 2025 © BeeAI a Series of LF Projects, LLC
 # SPDX-License-Identifier: Apache-2.0
 
-
-from collections.abc import AsyncIterator
+from collections.abc import AsyncGenerator
+from typing import Literal, overload
+from uuid import UUID
 
 import janus
 from a2a.types import Artifact, Message, MessageSendConfiguration, Task
 from pydantic import BaseModel, PrivateAttr
 
 from agentstack_sdk.a2a.types import RunYield, RunYieldResume
+from agentstack_sdk.platform.context import ContextHistoryItem
 from agentstack_sdk.server.store.context_store import ContextStoreInstance
 
 
@@ -30,12 +32,29 @@ class RunContext(BaseModel, arbitrary_types_allowed=True):
             data = data.model_copy(deep=True, update={"context_id": self.context_id, "task_id": self.task_id})
         await self._store.store(data)
 
-    async def load_history(self) -> AsyncIterator[Message | Artifact]:
+    @overload
+    async def load_history(
+        self, load_history_items: Literal[False] = False
+    ) -> AsyncGenerator[Message | Artifact, None]:
+        yield ...  # type: ignore
+
+    @overload
+    async def load_history(self, load_history_items: Literal[True]) -> AsyncGenerator[ContextHistoryItem, None]:
+        yield ...  # type: ignore
+
+    async def load_history(
+        self, load_history_items: bool = False
+    ) -> AsyncGenerator[ContextHistoryItem | Message | Artifact]:
         if not self._store:
             raise RuntimeError("Context store is not initialized")
-        async for item in self._store.load_history():
+        async for item in self._store.load_history(load_history_items=load_history_items):
             yield item
 
+    async def delete_history_from_id(self, from_id: UUID) -> None:
+        if not self._store:
+            raise RuntimeError("Context store is not initialized")
+        await self._store.delete_history_from_id(from_id)
+
     def yield_sync(self, value: RunYield) -> RunYieldResume:
         self._yield_queue.sync_q.put(value)
         return self._yield_resume_queue.sync_q.get()

{agentstack_sdk-0.5.1rc3 → agentstack_sdk-0.5.2rc2}/src/agentstack_sdk/server/store/context_store.py

@@ -6,19 +6,26 @@ from __future__ import annotations
 import abc
 from collections.abc import AsyncIterator
 from typing import TYPE_CHECKING, Protocol
+from uuid import UUID
 
 from a2a.types import Artifact, Message
 
+from agentstack_sdk.platform.context import ContextHistoryItem
+
 if TYPE_CHECKING:
     from agentstack_sdk.server.dependencies import Dependency, Depends
 
 
 class ContextStoreInstance(Protocol):
-    async def load_history(self) -> AsyncIterator[Message | Artifact]:
+    async def load_history(
+        self, load_history_items: bool = False
+    ) -> AsyncIterator[ContextHistoryItem | Message | Artifact]:
         yield ...  # type: ignore
 
     async def store(self, data: Message | Artifact) -> None: ...
 
+    async def delete_history_from_id(self, from_id: UUID) -> None: ...
+
 
 class ContextStore(abc.ABC):
     def modify_dependencies(self, dependencies: dict[str, Depends]) -> None:

{agentstack_sdk-0.5.1rc3 → agentstack_sdk-0.5.2rc2}/src/agentstack_sdk/server/store/memory_context_store.py

@@ -3,10 +3,12 @@
 
 from collections.abc import AsyncIterator
 from datetime import timedelta
+from uuid import UUID
 
 from a2a.types import Artifact, Message
 from cachetools import TTLCache
 
+from agentstack_sdk.platform.context import ContextHistoryItem
 from agentstack_sdk.server.dependencies import Dependency
 from agentstack_sdk.server.store.context_store import ContextStore, ContextStoreInstance
 
@@ -14,14 +16,28 @@ from agentstack_sdk.server.store.context_store import ContextStore, ContextStore
 class MemoryContextStoreInstance(ContextStoreInstance):
     def __init__(self, context_id: str):
         self.context_id = context_id
-        self._history: list[Message | Artifact] = []
+        self._history: list[ContextHistoryItem] = []
 
-    async def load_history(self) -> AsyncIterator[Message | Artifact]:
-        for message in self._history.copy():
-            yield message.model_copy(deep=True)
+    async def load_history(
+        self, load_history_items: bool = False
+    ) -> AsyncIterator[ContextHistoryItem | Message | Artifact]:
+        for item in self._history.copy():
+            if load_history_items:
+                yield item.model_copy(deep=True)
+            else:
+                yield item.data.model_copy(deep=True)
 
     async def store(self, data: Message | Artifact) -> None:
-        self._history.append(data.model_copy(deep=True))
+        self._history.append(ContextHistoryItem(data=data.model_copy(deep=True), context_id=self.context_id))
+
+    async def delete_history_from_id(self, from_id: UUID) -> None:
+        # Does not allow to delete from an artifact onwards
+        index = next(
+            (i for i, item in enumerate(self._history) if item.id == from_id),
+            None,
+        )
+        if index is not None:
+            self._history = self._history[:index]
 
 
 class InMemoryContextStore(ContextStore):

{agentstack_sdk-0.5.1rc3 → agentstack_sdk-0.5.2rc2}/src/agentstack_sdk/server/store/platform_context_store.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from collections.abc import AsyncIterator
+from uuid import UUID
 
 from a2a.types import Artifact, Message
 
@@ -9,7 +10,7 @@ from agentstack_sdk.a2a.extensions.services.platform import (
     PlatformApiExtensionServer,
     PlatformApiExtensionSpec,
 )
-from agentstack_sdk.platform.context import Context
+from agentstack_sdk.platform.context import Context, ContextHistoryItem
 from agentstack_sdk.server.constants import _IMPLICIT_DEPENDENCY_PREFIX
 from agentstack_sdk.server.dependencies import Dependency, Depends
 from agentstack_sdk.server.store.context_store import ContextStore, ContextStoreInstance
@@ -38,11 +39,20 @@ class PlatformContextStoreInstance(ContextStoreInstance):
         self._context_id = context_id
         self._platform_extension = platform_extension
 
-    async def load_history(self) -> AsyncIterator[Message | Artifact]:
+    async def load_history(
+        self, load_history_items: bool = False
+    ) -> AsyncIterator[ContextHistoryItem | Message | Artifact]:
         async with self._platform_extension.use_client():
             async for history_item in Context.list_all_history(self._context_id):
-                yield history_item.data
+                if load_history_items:
+                    yield history_item
+                else:
+                    yield history_item.data
 
     async def store(self, data: Message | Artifact) -> None:
         async with self._platform_extension.use_client():
             await Context.add_history_item(self._context_id, data=data)
+
+    async def delete_history_from_id(self, from_id: UUID) -> None:
+        async with self._platform_extension.use_client():
+            await Context.delete_history_from_id(self._context_id, from_id=from_id)

{agentstack_sdk-0.5.1rc3 → agentstack_sdk-0.5.2rc2}/src/agentstack_sdk/util/utils.py

@@ -3,6 +3,7 @@
 import json
 import re
 from collections.abc import AsyncIterator
+from datetime import UTC, datetime
 from typing import Any, TypeVar, cast
 
 import httpx
@@ -40,3 +41,7 @@ def extract_messages(exc: BaseException) -> list[tuple[str, str]]:
         return [(exc_type, msg) for e in exc.exceptions for exc_type, msg in extract_messages(e)]
     else:
         return [(type(exc).__name__, str(exc))]
+
+
+def utc_now() -> datetime:
+    return datetime.now(UTC)