acp-sdk 0.0.4__py3-none-any.whl → 0.0.5__py3-none-any.whl

acp/server/lowlevel/server.py

@@ -566,9 +566,13 @@ class Server:
                 case (
                     RequestResponder(request=types.ClientRequest(root=req)) as responder
                 ):
-                    with responder:
-                        await self._handle_request(
-                            message, req, session, raise_exceptions
+                    async with responder:
+                        responder.task_group.start_soon(
+                            self._handle_request,
+                            message,
+                            req,
+                            session,
+                            raise_exceptions,
                         )
                 case types.ClientNotification(root=notify):
                     await self._handle_notification(notify)

acp/server/session.py

@@ -126,7 +126,7 @@ class ServerSession(
             case types.InitializeRequest(params=params):
                 self._initialization_state = InitializationState.Initializing
                 self._client_params = params
-                with responder:
+                async with responder:
                     await responder.respond(
                         types.ServerResult(
                             types.InitializeResult(

acp/shared/session.py

@@ -71,27 +71,27 @@ class RequestResponder(Generic[ReceiveRequestT, SendResultT]):
         self.request = request
         self._session = session
         self._completed = False
-        self._cancel_scope = anyio.CancelScope()
+        self.task_group = anyio.create_task_group()
         self._on_complete = on_complete
         self._entered = False  # Track if we're in a context manager
 
-    def __enter__(self) -> "RequestResponder[ReceiveRequestT, SendResultT]":
+    async def __aenter__(self) -> "RequestResponder[ReceiveRequestT, SendResultT]":
         """Enter the context manager, enabling request cancellation tracking."""
         self._entered = True
-        self._cancel_scope = anyio.CancelScope()
-        self._cancel_scope.__enter__()
+        self.task_group = anyio.create_task_group()
+        await self.task_group.__aenter__()
         return self
 
-    def __exit__(self, exc_type, exc_val, exc_tb) -> None:
+    async def __aexit__(self, exc_type, exc_val, exc_tb) -> None:
         """Exit the context manager, performing cleanup and notifying completion."""
         try:
+            if not self.task_group:
+                raise RuntimeError("No active cancel scope")
+            await self.task_group.__aexit__(exc_type, exc_val, exc_tb)
             if self._completed:
                 self._on_complete(self)
         finally:
             self._entered = False
-            if not self._cancel_scope:
-                raise RuntimeError("No active cancel scope")
-            self._cancel_scope.__exit__(exc_type, exc_val, exc_tb)
 
     async def respond(self, response: SendResultT | ErrorData) -> None:
         """Send a response for this request.
@@ -116,10 +116,10 @@ class RequestResponder(Generic[ReceiveRequestT, SendResultT]):
         """Cancel this request and mark it as completed."""
         if not self._entered:
             raise RuntimeError("RequestResponder must be used as a context manager")
-        if not self._cancel_scope:
+        if not self.task_group:
             raise RuntimeError("No active cancel scope")
 
-        self._cancel_scope.cancel()
+        self.task_group.cancel_scope.cancel()
         self._completed = True  # Mark as completed so it's removed from in_flight
         # Send an error response to indicate cancellation
         await self._session._send_response(
@@ -133,7 +133,9 @@ class RequestResponder(Generic[ReceiveRequestT, SendResultT]):
 
     @property
     def cancelled(self) -> bool:
-        return self._cancel_scope is not None and self._cancel_scope.cancel_called
+        return (
+            self.task_group is not None and self.task_group.cancel_scope.cancel_called
+        )
 
 
 class BaseSession(
@@ -203,6 +205,7 @@ class BaseSession(
         self,
         request: SendRequestT,
         result_type: type[ReceiveResultT],
+        request_id: RequestId | None = None,
     ) -> ReceiveResultT:
         """
         Sends a request and wait for a response. Raises an McpError if the
@@ -212,8 +215,9 @@ class BaseSession(
         instead.
         """
 
-        request_id = self._request_id
-        self._request_id = request_id + 1
+        if request_id is None:
+            request_id = self._request_id
+            self._request_id = request_id + 1
 
         response_stream, response_stream_reader = anyio.create_memory_object_stream[
             JSONRPCResponse | JSONRPCError

{acp_sdk-0.0.4.dist-info → acp_sdk-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acp-sdk
-Version: 0.0.4
+Version: 0.0.5
 Summary: Agent Communication Protocol SDK
 Project-URL: Homepage, https://github.com/i-am-bee/beeai
 Project-URL: Repository, https://github.com/i-am-bee/beeai

{acp_sdk-0.0.4.dist-info → acp_sdk-0.0.5.dist-info}/RECORD

@@ -12,7 +12,7 @@ acp/client/stdio.py,sha256=f4Q8pmXCHr14tMfyjWuMQ5AoT2F6rvnY_gcqLaa4BvY,4686
 acp/server/__init__.py,sha256=Y0b-MWN_eAL9rOye8OxUt9UUZguWBix4yEdJScXbYNU,95
 acp/server/__main__.py,sha256=Rf4EUZuiP6AqgMbsw9CvyA4nUmJcKTq1h54GWLAndNY,1328
 acp/server/models.py,sha256=XqJzNfbleQJ_Wj-fglgZrkQEK93EZ0dMg8DeANeHygQ,341
-acp/server/session.py,sha256=xRrrtTnXcZjlL6C1JYpZGQebJVzw4kKKeHWvs3HADTE,11438
+acp/server/session.py,sha256=W15GjzDsUMPKoKSR-vt2hRJm-WTef1OU_8aQhYiAoMU,11444
 acp/server/sse.py,sha256=hGdtFOkUlZJOoglI0UQYxz6KyZnaf2coHeBuHytCi7I,6716
 acp/server/stdio.py,sha256=QEscVVsH4h9G4e-NWsjl7_US0pCCPAzwR8-xfRbjads,2889
 acp/server/websocket.py,sha256=5sw6-64ES7rSLQ7UfY5aXSOnh16B3HWK2410kiDcxcU,2225
@@ -42,16 +42,16 @@ acp/server/highlevel/utilities/logging.py,sha256=07rXwvLf9Yd0EJCeu5bFlbXt-qr94Aq
 acp/server/highlevel/utilities/types.py,sha256=OR1JbgpRpGf19_-DOhwNr-OGPC-rCz1FLd8YBa4XqCc,1760
 acp/server/lowlevel/__init__.py,sha256=mS7slrVoWzHMYqDHDIJ7oRZxwPZG_I63UVQiJN311YY,93
 acp/server/lowlevel/helper_types.py,sha256=LmzGNx9FU258TKDyKjwfzH_KOnH4Mkk_k8i9EIFmN7Y,189
-acp/server/lowlevel/server.py,sha256=cRzDFKK_tw02eIvW6ie38NH7UXNaxZ2WPtoyX7iaoyI,22605
+acp/server/lowlevel/server.py,sha256=4kHco_HLGvjC0Gy28FR8jnMs2b7cVsfJ_bo1C9jY7CY,22751
 acp/shared/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 acp/shared/context.py,sha256=sc81CXNrn3OX_yX84nBfnGLvEHBUY9-rIpdb4S6mGus,350
 acp/shared/exceptions.py,sha256=gtaybYun_cRp7ZHoxV4iiorN8Dn_yUoj1-fpeMOnM2Y,316
 acp/shared/memory.py,sha256=RHEnuOVDkT8LrviEvUS5JGl0qs55DA0jcsiuOg82wa8,2802
 acp/shared/progress.py,sha256=enn_ZLnIPGS8107_rh7JgKvZXcXBUEv9STZdn84wIw0,1033
-acp/shared/session.py,sha256=IBVCFAUUfyx3TAzrhoCbLEsr7HCwPmfruVCmLHubcO8,15637
+acp/shared/session.py,sha256=VI60P5ecaYQtdllMHpd6uZ3yggTxyGVYRaCKXfpIsm8,15784
 acp/shared/version.py,sha256=si_GMtVN2H0Y3Rn4QNf8u2V0kZRTzzlMHuCWqDDDK1s,106
-acp_sdk-0.0.4.dist-info/METADATA,sha256=gzwK7QSN26Odr4ZWgbGM8twXZc5I-g6gM3BJKONqgQE,1932
-acp_sdk-0.0.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-acp_sdk-0.0.4.dist-info/entry_points.txt,sha256=zJgYH5TTBxQ1oBzEg-0RfEk09Te1W2LkqbyoLpFe9CA,42
-acp_sdk-0.0.4.dist-info/licenses/LICENSE,sha256=aLcXJ7ZsI7zZmq4n0iIQ6KYB3EJMtRJGLr7n514PE30,1100
-acp_sdk-0.0.4.dist-info/RECORD,,
+acp_sdk-0.0.5.dist-info/METADATA,sha256=hW9vaZ2xe2134yAnjqt_UMm3u7oAGwFIIcTgYqzgcEw,1932
+acp_sdk-0.0.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+acp_sdk-0.0.5.dist-info/entry_points.txt,sha256=zJgYH5TTBxQ1oBzEg-0RfEk09Te1W2LkqbyoLpFe9CA,42
+acp_sdk-0.0.5.dist-info/licenses/LICENSE,sha256=aLcXJ7ZsI7zZmq4n0iIQ6KYB3EJMtRJGLr7n514PE30,1100
+acp_sdk-0.0.5.dist-info/RECORD,,