acp-sdk 0.7.2__py3-none-any.whl → 0.8.0__py3-none-any.whl

acp_sdk/client/client.py

@@ -23,11 +23,12 @@ from acp_sdk.models import (
     AwaitResume,
     Error,
     Event,
+    PingResponse,
     Run,
     RunCancelResponse,
-    RunCreatedEvent,
     RunCreateRequest,
     RunCreateResponse,
+    RunEventsListResponse,
     RunId,
     RunMode,
     RunResumeRequest,
@@ -121,9 +122,10 @@ class Client:
         return Agent(**response.model_dump())
 
     async def ping(self) -> bool:
-        response = await self._client.get("/healthcheck")
+        response = await self._client.get("/ping")
         self._raise_error(response)
-        return response.json() == "OK"
+        PingResponse.model_validate(response.json())
+        return
 
     async def run_sync(self, input: Input, *, agent: AgentName) -> Run:
         response = await self._client.post(
@@ -137,7 +139,6 @@ class Client:
         )
         self._raise_error(response)
         response = RunCreateResponse.model_validate(response.json())
-        self._set_session(response)
         return Run(**response.model_dump())
 
     async def run_async(self, input: Input, *, agent: AgentName) -> Run:
@@ -152,7 +153,6 @@ class Client:
         )
         self._raise_error(response)
         response = RunCreateResponse.model_validate(response.json())
-        self._set_session(response)
         return Run(**response.model_dump())
 
     async def run_stream(self, input: Input, *, agent: AgentName) -> AsyncIterator[Event]:
@@ -168,8 +168,6 @@ class Client:
             ).model_dump_json(),
         ) as event_source:
             async for event in self._validate_stream(event_source):
-                if isinstance(event, RunCreatedEvent):
-                    self._set_session(event.run)
                 yield event
 
     async def run_status(self, *, run_id: RunId) -> Run:
@@ -177,6 +175,13 @@ class Client:
         self._raise_error(response)
         return Run.model_validate(response.json())
 
+    async def run_events(self, *, run_id: RunId) -> AsyncIterator[Event]:
+        response = await self._client.get(f"/runs/{run_id}/events")
+        self._raise_error(response)
+        response = RunEventsListResponse.model_validate(response.json())
+        for event in response.events:
+            yield event
+
     async def run_cancel(self, *, run_id: RunId) -> Run:
         response = await self._client.post(f"/runs/{run_id}/cancel")
         self._raise_error(response)
@@ -227,6 +232,3 @@ class Client:
             response.raise_for_status()
         except httpx.HTTPError:
             raise ACPError(Error.model_validate(response.json()))
-
-    def _set_session(self, run: Run) -> None:
-        self._session_id = run.session_id

acp_sdk/models/models.py

@@ -1,5 +1,5 @@
 import uuid
-from datetime import datetime
+from datetime import datetime, timezone
 from enum import Enum
 from typing import Any, Literal, Optional, Union
 
@@ -95,11 +95,17 @@ class Artifact(MessagePart):
 
 class Message(BaseModel):
     parts: list[MessagePart]
+    created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    completed_at: datetime | None = Field(default_factory=lambda: datetime.now(timezone.utc))
 
     def __add__(self, other: "Message") -> "Message":
         if not isinstance(other, Message):
             raise TypeError(f"Cannot concatenate Message with {type(other).__name__}")
-        return Message(parts=self.parts + other.parts)
+        return Message(
+            parts=self.parts + other.parts,
+            created_at=min(self.created_at, other.created_at),
+            completed_at=max(self.completed_at, other.completed_at),
+        )
 
     def __str__(self) -> str:
         return "".join(
@@ -134,7 +140,7 @@ class Message(BaseModel):
                 parts[-1] = join(parts[-1], part)
             else:
                 parts.append(part)
-        return Message(parts=parts)
+        return Message(parts=parts, created_at=self.created_at, completed_at=self.completed_at)
 
 
 AgentName = str
@@ -185,6 +191,8 @@ class Run(BaseModel):
     await_request: AwaitRequest | None = None
     output: list[Message] = []
     error: Error | None = None
+    created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
+    finished_at: datetime | None = None
 
 
 class MessageCreatedEvent(BaseModel):

acp_sdk/models/schemas.py

@@ -1,6 +1,10 @@
 from pydantic import BaseModel
 
-from acp_sdk.models.models import Agent, AgentName, AwaitResume, Message, Run, RunMode, SessionId
+from acp_sdk.models.models import Agent, AgentName, AwaitResume, Event, Message, Run, RunMode, SessionId
+
+
+class PingResponse(BaseModel):
+    pass
 
 
 class AgentsListResponse(BaseModel):
@@ -37,3 +41,7 @@ class RunReadResponse(Run):
 
 class RunCancelResponse(Run):
     pass
+
+
+class RunEventsListResponse(BaseModel):
+    events: list[Event]

acp_sdk/server/app.py

@@ -21,6 +21,7 @@ from acp_sdk.models import (
     RunCancelResponse,
     RunCreateRequest,
     RunCreateResponse,
+    RunEventsListResponse,
     RunId,
     RunMode,
     RunReadResponse,
@@ -29,6 +30,7 @@ from acp_sdk.models import (
     SessionId,
 )
 from acp_sdk.models.errors import ACPError
+from acp_sdk.models.schemas import PingResponse
 from acp_sdk.server.agent import Agent
 from acp_sdk.server.bundle import RunBundle
 from acp_sdk.server.errors import (
@@ -104,15 +106,15 @@ def create_app(
         agent = find_agent(name)
         return AgentModel(name=agent.name, description=agent.description, metadata=agent.metadata)
 
-    @app.get("/healthcheck")
-    async def healthcheck() -> str:
-        return "OK"
+    @app.get("/ping")
+    async def ping() -> PingResponse:
+        return PingResponse()
 
     @app.post("/runs")
     async def create_run(request: RunCreateRequest) -> RunCreateResponse:
         agent = find_agent(request.agent_name)
 
-        session = sessions.get(request.session_id, Session()) if request.session_id else Session()
+        session = sessions.get(request.session_id, Session(id=request.session_id)) if request.session_id else Session()
         nonlocal executor
         bundle = RunBundle(
             agent=agent,
@@ -155,6 +157,11 @@ def create_app(
         bundle = find_run_bundle(run_id)
         return bundle.run
 
+    @app.get("/runs/{run_id}/events")
+    async def list_run_events(run_id: RunId) -> RunEventsListResponse:
+        bundle = find_run_bundle(run_id)
+        return RunEventsListResponse(events=bundle.events)
+
     @app.post("/runs/{run_id}")
     async def resume_run(run_id: RunId, request: RunResumeRequest) -> RunResumeResponse:
         bundle = find_run_bundle(run_id)

acp_sdk/server/bundle.py

@@ -2,6 +2,7 @@ import asyncio
 import logging
 from collections.abc import AsyncGenerator
 from concurrent.futures import ThreadPoolExecutor
+from datetime import datetime, timezone
 
 from pydantic import BaseModel, ValidationError
 
@@ -43,6 +44,7 @@ class RunBundle:
         self.history = history
 
         self.stream_queue: asyncio.Queue[Event] = asyncio.Queue()
+        self.events: list[Event] = []
 
         self.await_queue: asyncio.Queue[AwaitResume] = asyncio.Queue(maxsize=1)
         self.await_or_terminate_event = asyncio.Event()
@@ -58,7 +60,9 @@ class RunBundle:
             self.stream_queue.task_done()
 
     async def emit(self, event: Event) -> None:
-        await self.stream_queue.put(event)
+        freeze = event.model_copy(deep=True)
+        self.events.append(freeze)
+        await self.stream_queue.put(freeze)
 
     async def await_(self) -> AwaitResume:
         await self.stream_queue.put(None)
@@ -92,7 +96,9 @@ class RunBundle:
             async def flush_message() -> None:
                 nonlocal in_message
                 if in_message:
-                    await self.emit(MessageCompletedEvent(message=self.run.output[-1]))
+                    message = self.run.output[-1]
+                    message.completed_at = datetime.now(timezone.utc)
+                    await self.emit(MessageCompletedEvent(message=message))
                     in_message = False
 
             try:
@@ -114,7 +120,7 @@ class RunBundle:
                         if isinstance(next, str):
                             next = MessagePart(content=next)
                         if not in_message:
-                            self.run.output.append(Message(parts=[]))
+                            self.run.output.append(Message(parts=[], completed_at=None))
                             in_message = True
                             await self.emit(MessageCreatedEvent(message=self.run.output[-1]))
                         self.run.output[-1].parts.append(next)
@@ -149,10 +155,12 @@ class RunBundle:
             except StopAsyncIteration:
                 await flush_message()
                 self.run.status = RunStatus.COMPLETED
+                self.run.finished_at = datetime.now(timezone.utc)
                 await self.emit(RunCompletedEvent(run=self.run))
                 run_logger.info("Run completed")
             except asyncio.CancelledError:
                 self.run.status = RunStatus.CANCELLED
+                self.run.finished_at = datetime.now(timezone.utc)
                 await self.emit(RunCancelledEvent(run=self.run))
                 run_logger.info("Run cancelled")
             except Exception as e:
@@ -161,6 +169,7 @@ class RunBundle:
                 else:
                     self.run.error = Error(code=ErrorCode.SERVER_ERROR, message=str(e))
                 self.run.status = RunStatus.FAILED
+                self.run.finished_at = datetime.now(timezone.utc)
                 await self.emit(RunFailedEvent(run=self.run))
                 run_logger.exception("Run failed")
                 raise

acp_sdk/server/session.py

@@ -7,8 +7,8 @@ from acp_sdk.server.bundle import RunBundle
 
 
 class Session:
-    def __init__(self) -> None:
-        self.id: SessionId = uuid.uuid4()
+    def __init__(self, id: SessionId | None = None) -> None:
+        self.id: SessionId = id or uuid.uuid4()
         self.bundles: list[RunBundle] = []
 
     def append(self, bundle: RunBundle) -> None:

{acp_sdk-0.7.2.dist-info → acp_sdk-0.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acp-sdk
-Version: 0.7.2
+Version: 0.8.0
 Summary: Agent Communication Protocol SDK
 Author: IBM Corp.
 Maintainer-email: Tomas Pilar <thomas7pilar@gmail.com>

{acp_sdk-0.7.2.dist-info → acp_sdk-0.8.0.dist-info}/RECORD

@@ -3,25 +3,25 @@ acp_sdk/instrumentation.py,sha256=JqSyvILN3sGAfOZrmckQq4-M_4_5alyPn95DK0o5lfA,16
 acp_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 acp_sdk/version.py,sha256=Niy83rgvigB4hL_rR-O4ySvI7dj6xnqkyOe_JTymi9s,73
 acp_sdk/client/__init__.py,sha256=Bca1DORrswxzZsrR2aUFpATuNG2xNSmYvF1Z2WJaVbc,51
-acp_sdk/client/client.py,sha256=WJNKGCml3A3zrOdSOIC8cS5s-ySfWLDWSuvAr9y25x8,8504
+acp_sdk/client/client.py,sha256=GnkbhCoEghI9A0_Otmzb18V4ot1uNz0hEXNjIAPNHOE,8599
 acp_sdk/client/types.py,sha256=_H6zYt-2OHOOYRtssRnbDIiwmgsl2-KIXc9lb-mJLFA,133
 acp_sdk/client/utils.py,sha256=2jhJyrPJmVFRoDJh0q_JMqOMlC3IxCh-6HXed-PIZS8,924
 acp_sdk/models/__init__.py,sha256=numSDBDT1QHx7n_Y3Deb5VOvKWcUBxbOEaMwQBSRHxc,151
 acp_sdk/models/errors.py,sha256=rEyaMVvQuBi7fwWe_d0PGGySYsD3FZTluQ-SkC0yhAs,444
-acp_sdk/models/models.py,sha256=aA_ylU-Z0ZAlqPB8j-JfqJ2q516pgvTIyZ0wbfY8-UU,6654
-acp_sdk/models/schemas.py,sha256=OQ2bEfwFFZ1D9mhpgzfPBCcnPNvbl4qRfCWChH2wCro,644
+acp_sdk/models/models.py,sha256=6OEkTMdMsl3_IbiFmFG2ChEqY_KYTf3Bi_MW5p168GQ,7183
+acp_sdk/models/schemas.py,sha256=_ah7_zHsQJGxDXvnzsBvASdRsQHVphFQ7Sum6A04iRw,759
 acp_sdk/server/__init__.py,sha256=mxBBBFaZuMEUENRMLwp1XZkuLeT9QghcFmNvjnqvAAU,377
 acp_sdk/server/agent.py,sha256=wvwpi83osmW7zQWxVnzyVMXIvzOswAfhKWHscVIldhA,6245
-acp_sdk/server/app.py,sha256=tMWbtJgEEqIYe6gtJiVRCzt0OTTATsLLZt51SfBIqLc,6806
-acp_sdk/server/bundle.py,sha256=P3DfmzQOewsK8jr8yG8_RsGSrXQI4PsZCQbspVxKlus,6552
+acp_sdk/server/app.py,sha256=1S1mxECioL5NeGjGo5C8u94x7Ybvj79L_Aauu2352vA,7117
+acp_sdk/server/bundle.py,sha256=5Rq6E4WgmuwPQ4u-sViaHeu5loCLEUgojWav9LAghX4,7036
 acp_sdk/server/context.py,sha256=MgnLV6qcDIhc_0BjW7r4Jj1tHts4ZuwpdTGIBnz2Mgo,1036
 acp_sdk/server/errors.py,sha256=GSO8yYIqEeX8Y4Lz86ks35dMTHiQiXuOrLYYx0eXsbI,2110
 acp_sdk/server/logging.py,sha256=Oc8yZigCsuDnHHPsarRzu0RX3NKaLEgpELM2yovGKDI,411
 acp_sdk/server/server.py,sha256=3b_nhbDR_VUz9lneJow4Jpw-6cF2xf2k94grYmfbm1E,8144
-acp_sdk/server/session.py,sha256=fCRPHc2sQrz4G0n25I1S4LB7mI1wo0yYQJ1V9WRWH3g,627
+acp_sdk/server/session.py,sha256=ekz1o6Sy1tQZlpaoS_VgbvFuUQh2qpiHG71mvBdvhgc,662
 acp_sdk/server/telemetry.py,sha256=1BUxNg-xL_Vqgs27PDWNc3HikrQW2lidAtT_FKlp_Qk,1833
 acp_sdk/server/types.py,sha256=teBNRWSks8XP1SCQKGEtbNWQahVD3RAOPnysTxcQPxI,292
 acp_sdk/server/utils.py,sha256=y-DDWv_QI25OJJYP5cni2FzfolKXtH3S2SYOm6OL_gc,1835
-acp_sdk-0.7.2.dist-info/METADATA,sha256=-g7nf-ZOtPPNVh46fhjUjYljp6Lodnu4qSwD_vSJqbI,1651
-acp_sdk-0.7.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-acp_sdk-0.7.2.dist-info/RECORD,,
+acp_sdk-0.8.0.dist-info/METADATA,sha256=Inaa0idYypuECgh9JWdUVxIwJDRhslH7-YwAW6lupMg,1651
+acp_sdk-0.8.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+acp_sdk-0.8.0.dist-info/RECORD,,