acp-sdk 0.2.0__tar.gz → 0.2.2__tar.gz

{acp_sdk-0.2.0 → acp_sdk-0.2.2}/PKG-INFO

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

{acp_sdk-0.2.0 → acp_sdk-0.2.2}/examples/clients/advanced.py

@@ -13,7 +13,9 @@ async def example() -> None:
             # Additional client configuration
         )
     ) as client:
-        run = await client.run_sync(agent="echo", inputs=[Message(parts=[MessagePart(content="Howdy!")])])
+        run = await client.run_sync(
+            agent="echo", inputs=[Message(parts=[MessagePart(content="Howdy!", content_type="text/plain")])]
+        )
         print(run)
 
 

acp_sdk-0.2.2/examples/clients/session.py

@@ -0,0 +1,22 @@
+import asyncio
+
+from acp_sdk.client import Client
+from acp_sdk.models import (
+    Message,
+    MessagePart,
+)
+
+
+async def example() -> None:
+    async with Client(base_url="http://localhost:8000") as client, client.session() as session:
+        run = await session.run_sync(
+            agent="echo", inputs=[Message(parts=[MessagePart(content="Howdy!", content_type="text/plain")])]
+        )
+        run = await session.run_sync(
+            agent="echo", inputs=[Message(parts=[MessagePart(content="Howdy again!", content_type="text/plain")])]
+        )
+        print(run)
+
+
+if __name__ == "__main__":
+    asyncio.run(example())

{acp_sdk-0.2.0 → acp_sdk-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "acp-sdk"
-version = "0.2.0"
+version = "0.2.2"
 description = "Agent Communication Protocol SDK"
 license = "Apache-2.0"
 readme = "README.md"

{acp_sdk-0.2.0 → acp_sdk-0.2.2}/src/acp_sdk/client/client.py

@@ -39,16 +39,14 @@ class Client:
         base_url: httpx.URL | str = "",
         session_id: SessionId | None = None,
         client: httpx.AsyncClient | None = None,
+        instrument: bool = True,
     ) -> None:
         self.base_url = base_url
         self.session_id = session_id
 
-        self._client = self._init_client(client)
-
-    def _init_client(self, client: httpx.AsyncClient | None = None) -> httpx.AsyncClient:
-        client = client or httpx.AsyncClient(base_url=self.base_url)
-        HTTPXClientInstrumentor.instrument_client(client)
-        return client
+        self._client = client or httpx.AsyncClient(base_url=self.base_url)
+        if instrument:
+            HTTPXClientInstrumentor.instrument_client(self._client)
 
     async def __aenter__(self) -> Self:
         await self._client.__aenter__()
@@ -64,7 +62,7 @@ class Client:
 
     @asynccontextmanager
     async def session(self, session_id: SessionId | None = None) -> AsyncGenerator[Self]:
-        yield Client(client=self._client, session_id=session_id or uuid.uuid4())
+        yield Client(client=self._client, session_id=session_id or uuid.uuid4(), instrument=False)
 
     async def agents(self) -> AsyncIterator[Agent]:
         response = await self._client.get("/agents")
@@ -134,28 +132,28 @@ class Client:
         self._raise_error(response)
         return RunCancelResponse.model_validate(response.json())
 
-    async def run_resume_sync(self, *, run_id: RunId, await_: AwaitResume) -> Run:
+    async def run_resume_sync(self, *, run_id: RunId, await_resume: AwaitResume) -> Run:
         response = await self._client.post(
             f"/runs/{run_id}",
-            json=RunResumeRequest(await_=await_, mode=RunMode.SYNC).model_dump(),
+            json=RunResumeRequest(await_resume=await_resume, mode=RunMode.SYNC).model_dump(),
         )
         self._raise_error(response)
         return RunResumeResponse.model_validate(response.json())
 
-    async def run_resume_async(self, *, run_id: RunId, await_: AwaitResume) -> Run:
+    async def run_resume_async(self, *, run_id: RunId, await_resume: AwaitResume) -> Run:
         response = await self._client.post(
             f"/runs/{run_id}",
-            json=RunResumeRequest(await_=await_, mode=RunMode.ASYNC).model_dump(),
+            json=RunResumeRequest(await_resume=await_resume, mode=RunMode.ASYNC).model_dump(),
         )
         self._raise_error(response)
         return RunResumeResponse.model_validate(response.json())
 
-    async def run_resume_stream(self, *, run_id: RunId, await_: AwaitResume) -> AsyncIterator[Event]:
+    async def run_resume_stream(self, *, run_id: RunId, await_resume: AwaitResume) -> AsyncIterator[Event]:
         async with aconnect_sse(
             self._client,
             "POST",
             f"/runs/{run_id}",
-            json=RunResumeRequest(await_=await_, mode=RunMode.STREAM).model_dump(),
+            json=RunResumeRequest(await_resume=await_resume, mode=RunMode.STREAM).model_dump(),
         ) as event_source:
             async for event in self._validate_stream(event_source):
                 yield event

{acp_sdk-0.2.0 → acp_sdk-0.2.2}/src/acp_sdk/models/models.py

@@ -113,9 +113,9 @@ class MessageCompletedEvent(BaseModel):
     message: Message
 
 
-class AwaitEvent(BaseModel):
-    type: Literal["await"] = "await"
-    await_request: AwaitRequest | None = None
+class RunAwaitingEvent(BaseModel):
+    type: Literal["run.awaiting"] = "run.awaiting"
+    run: Run
 
 
 class GenericEvent(BaseModel):
@@ -155,7 +155,7 @@ Event = Union[
     ArtifactEvent,
     MessagePartEvent,
     MessageCompletedEvent,
-    AwaitEvent,
+    RunAwaitingEvent,
     GenericEvent,
     RunCancelledEvent,
     RunFailedEvent,

{acp_sdk-0.2.0 → acp_sdk-0.2.2}/src/acp_sdk/models/schemas.py

@@ -1,4 +1,4 @@
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel
 
 from acp_sdk.models.models import Agent, AgentName, AwaitResume, Message, Run, RunMode, SessionId
 
@@ -23,11 +23,9 @@ class RunCreateResponse(Run):
 
 
 class RunResumeRequest(BaseModel):
-    await_: AwaitResume = Field(alias="await")
+    await_resume: AwaitResume
     mode: RunMode
 
-    model_config = ConfigDict(populate_by_name=True)
-
 
 class RunResumeResponse(Run):
     pass

{acp_sdk-0.2.0 → acp_sdk-0.2.2}/src/acp_sdk/server/app.py

@@ -144,7 +144,7 @@ def create_app(*agents: Agent) -> FastAPI:
     @app.post("/runs/{run_id}")
     async def resume_run(run_id: RunId, request: RunResumeRequest) -> RunResumeResponse:
         bundle = find_run_bundle(run_id)
-        await bundle.resume(request.await_)
+        await bundle.resume(request.await_resume)
         match request.mode:
             case RunMode.STREAM:
                 return StreamingResponse(

{acp_sdk-0.2.0 → acp_sdk-0.2.2}/src/acp_sdk/server/bundle.py

@@ -7,7 +7,6 @@ from pydantic import ValidationError
 
 from acp_sdk.models import (
     AnyModel,
-    AwaitEvent,
     AwaitRequest,
     AwaitResume,
     Error,
@@ -17,6 +16,7 @@ from acp_sdk.models import (
     MessageCreatedEvent,
     MessagePartEvent,
     Run,
+    RunAwaitingEvent,
     RunCancelledEvent,
     RunCompletedEvent,
     RunCreatedEvent,
@@ -120,15 +120,7 @@ class RunBundle:
                     elif isinstance(next, AwaitRequest):
                         self.run.await_request = next
                         self.run.status = RunStatus.AWAITING
-                        await self.emit(
-                            AwaitEvent.model_validate(
-                                {
-                                    "run_id": self.run.run_id,
-                                    "type": "await",
-                                    "await": next,
-                                }
-                            )
-                        )
+                        await self.emit(RunAwaitingEvent(run=self.run))
                         run_logger.info("Run awaited")
                         await_resume = await self.await_()
                         await self.emit(RunInProgressEvent(run=self.run))

{acp_sdk-0.2.0 → acp_sdk-0.2.2}/tests/e2e/test_suites/test_runs.py

@@ -66,7 +66,7 @@ async def test_run_resume_sync(server: Server, client: Client) -> None:
     assert run.status == RunStatus.AWAITING
     assert run.await_request is not None
 
-    run = await client.run_resume_sync(run_id=run.run_id, await_=AwaitResume())
+    run = await client.run_resume_sync(run_id=run.run_id, await_resume=AwaitResume())
     assert run.status == RunStatus.COMPLETED
 
 
@@ -76,7 +76,7 @@ async def test_run_resume_async(server: Server, client: Client) -> None:
     assert run.status == RunStatus.AWAITING
     assert run.await_request is not None
 
-    run = await client.run_resume_async(run_id=run.run_id, await_=AwaitResume())
+    run = await client.run_resume_async(run_id=run.run_id, await_resume=AwaitResume())
     assert run.status == RunStatus.IN_PROGRESS
 
 
@@ -86,7 +86,7 @@ async def test_run_resume_stream(server: Server, client: Client) -> None:
     assert run.status == RunStatus.AWAITING
     assert run.await_request is not None
 
-    event_stream = [event async for event in client.run_resume_stream(run_id=run.run_id, await_=AwaitResume())]
+    event_stream = [event async for event in client.run_resume_stream(run_id=run.run_id, await_resume=AwaitResume())]
     assert isinstance(event_stream[0], RunInProgressEvent)
     assert isinstance(event_stream[-1], RunCompletedEvent)
 

acp_sdk-0.2.0/examples/clients/session.py

@@ -1,18 +0,0 @@
-import asyncio
-
-from acp_sdk.client import Client
-from acp_sdk.models import (
-    Message,
-    MessagePart,
-)
-
-
-async def example() -> None:
-    async with Client(base_url="http://localhost:8000") as client, client.session() as session:
-        run = await session.run_sync(agent="echo", inputs=[Message(parts=[MessagePart(content="Howdy!")])])
-        run = await session.run_sync(agent="echo", inputs=[Message(parts=[MessagePart(content="Howdy again!")])])
-        print(run)
-
-
-if __name__ == "__main__":
-    asyncio.run(example())