acp-sdk 0.2.3__py3-none-any.whl → 0.2.5__py3-none-any.whl

acp_sdk/client/client.py

@@ -9,6 +9,7 @@ from httpx_sse import EventSource, aconnect_sse
 from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
 from pydantic import TypeAdapter
 
+from acp_sdk.instrumentation import get_tracer
 from acp_sdk.models import (
     ACPError,
     Agent,
@@ -37,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
@@ -62,7 +66,9 @@ 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(), instrument=False)
+        session_id = session_id or uuid.uuid4()
+        with get_tracer().start_as_current_span("session", attributes={"acp.session": str(session_id)}):
+            yield Client(client=self._client, session_id=session_id, instrument=False)
 
     async def agents(self) -> AsyncIterator[Agent]:
         response = await self._client.get("/agents")
@@ -82,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)
@@ -97,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)
@@ -114,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):
@@ -135,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())
@@ -143,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())
@@ -153,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
@@ -162,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
@@ -173,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/instrumentation.py

@@ -0,0 +1,7 @@
+from opentelemetry import trace
+
+from acp_sdk.version import __version__
+
+
+def get_tracer() -> trace.Tracer:
+    return trace.get_tracer("acp-sdk", __version__)

acp_sdk/server/bundle.py

@@ -5,6 +5,7 @@ from concurrent.futures import ThreadPoolExecutor
 
 from pydantic import ValidationError
 
+from acp_sdk.instrumentation import get_tracer
 from acp_sdk.models import (
     ACPError,
     AnyModel,
@@ -30,7 +31,6 @@ from acp_sdk.models import (
 )
 from acp_sdk.server.agent import Agent
 from acp_sdk.server.logging import logger
-from acp_sdk.server.telemetry import get_tracer
 
 
 class RunBundle:
@@ -127,6 +127,8 @@ class RunBundle:
                         await_resume = await self.await_()
                         await self.emit(RunInProgressEvent(run=self.run))
                         run_logger.info("Run resumed")
+                    elif isinstance(next, Error):
+                        raise ACPError(error=next)
                     else:
                         try:
                             generic = AnyModel.model_validate(next)

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/server/telemetry.py

@@ -51,7 +51,3 @@ def configure_telemetry() -> None:
     processor = BatchLogRecordProcessor(OTLPLogExporter())
     logger_provider.add_log_record_processor(processor)
     root_logger.addHandler(LoggingHandler(logger_provider=logger_provider))
-
-
-def get_tracer() -> trace.Tracer:
-    return trace.get_tracer("acp-sdk", __version__)

{acp_sdk-0.2.3.dist-info → acp_sdk-0.2.5.dist-info}/METADATA

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

{acp_sdk-0.2.3.dist-info → acp_sdk-0.2.5.dist-info}/RECORD

@@ -1,8 +1,9 @@
 acp_sdk/__init__.py,sha256=tXdAUM9zcmdSKCAkVrOCrGcXcuVS-yuvQUoQwTe9pek,98
+acp_sdk/instrumentation.py,sha256=JqSyvILN3sGAfOZrmckQq4-M_4_5alyPn95DK0o5lfA,161
 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=N3kkxtSFlz_BzDcnq4xQCGDiu66EWzTzZ9HTpzNEOBk,6113
+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
@@ -10,15 +11,15 @@ acp_sdk/models/schemas.py,sha256=Kj7drJSR8d-N3KHzu_qTnLdagrMtAyhid5swluuhHTw,645
 acp_sdk/server/__init__.py,sha256=mxBBBFaZuMEUENRMLwp1XZkuLeT9QghcFmNvjnqvAAU,377
 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=89Iu6YjsaqGDmLcP4ScLPIP6oi08Lf2SFQr8WBQ9TK8,6078
+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=WIEHK8syOTG9SyWi3Y-cos7CsCF5-IHGiyL9bCaUN0E,1921
+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.3.dist-info/METADATA,sha256=LwqB-ZDr3k6tz5lzjD3rEzWz-h5Ccu4W9AI5-EpACGQ,3463
-acp_sdk-0.2.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-acp_sdk-0.2.3.dist-info/RECORD,,
+acp_sdk-0.2.5.dist-info/METADATA,sha256=ovtFlDH-TAiF1iW3-elh0-69d9A5h-Poaw1yheSrFvM,3463
+acp_sdk-0.2.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+acp_sdk-0.2.5.dist-info/RECORD,,