agentstack-sdk 0.5.1rc3__py3-none-any.whl → 0.5.2__py3-none-any.whl

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/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/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/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/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/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)

{agentstack_sdk-0.5.1rc3.dist-info → agentstack_sdk-0.5.2.dist-info}/METADATA

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

{agentstack_sdk-0.5.1rc3.dist-info → agentstack_sdk-0.5.2.dist-info}/RECORD

@@ -37,7 +37,7 @@ agentstack_sdk/platform/__init__.py,sha256=jFlxr6P-6DJZVt1W8Xt1suISJpodD_EWX92Gm
 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
-agentstack_sdk/platform/context.py,sha256=4mKQ9quOEEvl2xsPDtXvJUW8mCYxq7wLrBUOcddtfzM,11435
+agentstack_sdk/platform/context.py,sha256=vYAQOgZUEelvhP4Sldzt4Y1CP4WVtEE42DSuAMGnn-0,12099
 agentstack_sdk/platform/file.py,sha256=lOqMwMKHKGWFy5P1huQ6DLRTiHU0KexX-s2qF3FIRxw,13062
 agentstack_sdk/platform/model_provider.py,sha256=hv5UbFUFQ23IlalH5W-nCTvnUCrxBtZF4UrkeNWnwFE,4588
 agentstack_sdk/platform/provider.py,sha256=1juAfYc39sVwZ8ZbYz1ohVnzzRacJx6Junvo-7GgVmY,9071
@@ -52,16 +52,16 @@ agentstack_sdk/server/__init__.py,sha256=vc06L5ILofytmLJC9hJUzA9LakUYUNYGiBoSXNs
 agentstack_sdk/server/agent.py,sha256=tprDvYwsSoImJkMCMVvHKzacH2pWtQUWu19qgWAyNMk,29170
 agentstack_sdk/server/app.py,sha256=1-g_6FBU4lnrBI2KZVnTvsfdKrHyChkeYmJ01iIrj4Y,3396
 agentstack_sdk/server/constants.py,sha256=huQAP351AeNt536tB-axJSseHMHfyDKwJdlackjLGnw,188
-agentstack_sdk/server/context.py,sha256=pAXdZ3CUESqUuquosPxIR9TOGAohqh01tiv6rhYRjD0,1915
+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/store/__init__.py,sha256=jr8otByt8HFY9OWptuWdpq4WHE7A1srCSRXZV066FrI,94
-agentstack_sdk/server/store/context_store.py,sha256=tXQWI3q1xDOJWySAFuv4-79xxDSu94tPBzGeuR8YUzU,833
-agentstack_sdk/server/store/memory_context_store.py,sha256=R6zyuKG2QCOQUyoFOj8q3XAfuwONMrFHLn2Ku9YoCW8,1697
-agentstack_sdk/server/store/platform_context_store.py,sha256=S7EjymNJ80jmYqtWw3trE82JCVJqsQTpUeVDrIo5vJA,2189
+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
@@ -70,7 +70,7 @@ agentstack_sdk/util/file.py,sha256=rm5b47tHTmJBsHlqGcUqFCP0Bz5HKXPHWW_h-IpG9pw,9
 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/utils.py,sha256=1Pe6VXcC0El1wKjzJ_vJj_9_lgOEiJ2EEVV-aoPj9Jk,1500
-agentstack_sdk-0.5.1rc3.dist-info/WHEEL,sha256=M6du7VZflc4UPsGphmOXHANdgk8zessdJG0DBUuoA-U,78
-agentstack_sdk-0.5.1rc3.dist-info/METADATA,sha256=-MLzznQ4puhsF43Gghf4-_QOVKmwVET5wqymRposcWE,4433
-agentstack_sdk-0.5.1rc3.dist-info/RECORD,,
+agentstack_sdk/util/utils.py,sha256=18qFqMRkX4g4eIpvIvLb4FZAs8Q8ojJrpm19oJleb-k,1593
+agentstack_sdk-0.5.2.dist-info/WHEEL,sha256=M6du7VZflc4UPsGphmOXHANdgk8zessdJG0DBUuoA-U,78
+agentstack_sdk-0.5.2.dist-info/METADATA,sha256=8Cqrmcj2-9kjceY1khGLT--8uR1sNxExl6-dmt5gLLY,4430
+agentstack_sdk-0.5.2.dist-info/RECORD,,