acp-sdk 0.2.4__py3-none-any.whl → 0.3.0__py3-none-any.whl

acp_sdk/client/client.py

@@ -38,17 +38,20 @@ class Client:
         self,
         *,
         base_url: httpx.URL | str = "",
+        timeout: httpx.Timeout | None = None,
         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 = client or httpx.AsyncClient(base_url=self.base_url)
+        self._session_id = session_id
+        self._client = client or httpx.AsyncClient(base_url=base_url, timeout=timeout)
         if instrument:
             HTTPXClientInstrumentor.instrument_client(self._client)
 
+    @property
+    def client(self) -> httpx.AsyncClient:
+        return self._client
+
     async def __aenter__(self) -> Self:
         await self._client.__aenter__()
         return self
@@ -85,7 +88,7 @@ class Client:
                 agent_name=agent,
                 inputs=inputs,
                 mode=RunMode.SYNC,
-                session_id=self.session_id,
+                session_id=self._session_id,
             ).model_dump_json(),
         )
         self._raise_error(response)
@@ -100,7 +103,7 @@ class Client:
                 agent_name=agent,
                 inputs=inputs,
                 mode=RunMode.ASYNC,
-                session_id=self.session_id,
+                session_id=self._session_id,
             ).model_dump_json(),
         )
         self._raise_error(response)
@@ -117,7 +120,7 @@ class Client:
                 agent_name=agent,
                 inputs=inputs,
                 mode=RunMode.STREAM,
-                session_id=self.session_id,
+                session_id=self._session_id,
             ).model_dump_json(),
         ) as event_source:
             async for event in self._validate_stream(event_source):
@@ -138,7 +141,7 @@ class Client:
     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_resume=await_resume, mode=RunMode.SYNC).model_dump(),
+            content=RunResumeRequest(await_resume=await_resume, mode=RunMode.SYNC).model_dump_json(),
         )
         self._raise_error(response)
         return RunResumeResponse.model_validate(response.json())
@@ -146,7 +149,7 @@ class Client:
     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_resume=await_resume, mode=RunMode.ASYNC).model_dump(),
+            content=RunResumeRequest(await_resume=await_resume, mode=RunMode.ASYNC).model_dump_json(),
         )
         self._raise_error(response)
         return RunResumeResponse.model_validate(response.json())
@@ -156,7 +159,7 @@ class Client:
             self._client,
             "POST",
             f"/runs/{run_id}",
-            json=RunResumeRequest(await_resume=await_resume, mode=RunMode.STREAM).model_dump(),
+            content=RunResumeRequest(await_resume=await_resume, mode=RunMode.STREAM).model_dump_json(),
         ) as event_source:
             async for event in self._validate_stream(event_source):
                 yield event
@@ -165,6 +168,9 @@ class Client:
         self,
         event_source: EventSource,
     ) -> AsyncIterator[Event]:
+        if event_source.response.is_error:
+            await event_source.response.aread()
+            self._raise_error(event_source.response)
         async for event in event_source.aiter_sse():
             event = TypeAdapter(Event).validate_json(event.data)
             yield event
@@ -176,4 +182,4 @@ class Client:
             raise ACPError(Error.model_validate(response.json()))
 
     def _set_session(self, run: Run) -> None:
-        self.session_id = run.session_id
+        self._session_id = run.session_id

acp_sdk/server/errors.py

@@ -32,7 +32,9 @@ def status_code_to_error_code(status_code: int) -> ErrorCode:
 
 async def acp_error_handler(request: Request, exc: ACPError, *, status_code: int | None = None) -> JSONResponse:
     error = exc.error
-    return JSONResponse(status_code=status_code or error_code_to_status_code(error.code), content=error.model_dump())
+    return JSONResponse(
+        status_code=status_code or error_code_to_status_code(error.code), content=error.model_dump_json()
+    )
 
 
 async def http_exception_handler(request: Request, exc: StarletteHTTPException) -> JSONResponse:

{acp_sdk-0.2.4.dist-info → acp_sdk-0.3.0.dist-info}/METADATA

@@ -1,23 +1,21 @@
 Metadata-Version: 2.4
 Name: acp-sdk
-Version: 0.2.4
+Version: 0.3.0
 Summary: Agent Communication Protocol SDK
 Author: IBM Corp.
 Maintainer-email: Tomas Pilar <thomas7pilar@gmail.com>
 License-Expression: Apache-2.0
 Requires-Python: <4.0,>=3.11
+Requires-Dist: fastapi[standard]>=0.115.8
+Requires-Dist: httpx-sse>=0.4.0
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: janus>=2.0.0
 Requires-Dist: opentelemetry-api>=1.31.1
+Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.31.1
+Requires-Dist: opentelemetry-instrumentation-fastapi>=0.52b1
+Requires-Dist: opentelemetry-instrumentation-httpx>=0.52b1
+Requires-Dist: opentelemetry-sdk>=1.31.1
 Requires-Dist: pydantic>=2.11.1
-Provides-Extra: client
-Requires-Dist: httpx-sse>=0.4.0; extra == 'client'
-Requires-Dist: httpx>=0.28.1; extra == 'client'
-Requires-Dist: opentelemetry-instrumentation-httpx>=0.52b1; extra == 'client'
-Provides-Extra: server
-Requires-Dist: fastapi[standard]>=0.115.8; extra == 'server'
-Requires-Dist: janus>=2.0.0; extra == 'server'
-Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.31.1; extra == 'server'
-Requires-Dist: opentelemetry-instrumentation-fastapi>=0.52b1; extra == 'server'
-Requires-Dist: opentelemetry-sdk>=1.31.1; extra == 'server'
 Description-Content-Type: text/markdown
 
 # Agent Communication Protocol SDK for Python

{acp_sdk-0.2.4.dist-info → acp_sdk-0.3.0.dist-info}/RECORD

@@ -3,7 +3,7 @@ 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=5aA078pJxedVSGhhJ-NTTfjc19atSkdTyfR8LO6jgyM,6301
+acp_sdk/client/client.py,sha256=lEUs0Oc7MZbKkTF2E6e8Wn3dTW5cMVj6fD8TbuEOMDk,6584
 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=Z8qgURpz18uPXpYDYmlbMQ3Du9-hrJUE3GVPf7ufdUY,3994
@@ -13,13 +13,13 @@ acp_sdk/server/agent.py,sha256=fGky5MIuknw-Gy-THqhWLt9I9-gUyNIar8qEAvZb3uQ,6195
 acp_sdk/server/app.py,sha256=qLeQ1STgX8_7cRtG-7Be6V5vS9ooBq6O89cQryqBdII,5903
 acp_sdk/server/bundle.py,sha256=VchPUW8URH83-qJo9kRQsYCBcuLKbxGXHB3sw9CK0cc,6178
 acp_sdk/server/context.py,sha256=MgnLV6qcDIhc_0BjW7r4Jj1tHts4ZuwpdTGIBnz2Mgo,1036
-acp_sdk/server/errors.py,sha256=fWlgVsQ5hs_AXwzc-wvy6QgoDWEMRUBlSrfJfhHHMyE,2085
+acp_sdk/server/errors.py,sha256=IGtpPpb2ChtYvCac8kf_P-RkcY71gBvX0yq97uZWa-w,2104
 acp_sdk/server/logging.py,sha256=Oc8yZigCsuDnHHPsarRzu0RX3NKaLEgpELM2yovGKDI,411
 acp_sdk/server/server.py,sha256=-eT3fmnEsBUN44Spi2EP2eV0l4RAlKa8bzqxnhz16SM,5399
 acp_sdk/server/session.py,sha256=0cDr924HC5x2bBNbK9NSKVHAt5A_mi5dK8P4jP_ugq0,629
 acp_sdk/server/telemetry.py,sha256=1BUxNg-xL_Vqgs27PDWNc3HikrQW2lidAtT_FKlp_Qk,1833
 acp_sdk/server/types.py,sha256=1bqMCjwZM3JzvJ1h4aBHWzjbldMQ45HqcezBD6hUoYo,175
 acp_sdk/server/utils.py,sha256=EfrF9VCyVk3AM_ao-BIB9EzGbfTrh4V2Bz-VFr6f6Sg,351
-acp_sdk-0.2.4.dist-info/METADATA,sha256=JCfcw8ZueUe0WzIBTrfQMcn4yRPIShxt0woTWN0M9sk,3463
-acp_sdk-0.2.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-acp_sdk-0.2.4.dist-info/RECORD,,
+acp_sdk-0.3.0.dist-info/METADATA,sha256=7GnGZ92jFahL6u1NS9lKMg04V6zFjyprgwBUQota1iE,3265
+acp_sdk-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+acp_sdk-0.3.0.dist-info/RECORD,,