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

acp/server/highlevel/agents/base.py

@@ -14,9 +14,7 @@ class Agent(BaseModel):
     input: Type[BaseModel] = Field(description="Model for input")
     output: Type[BaseModel] = Field(description="Model for output")
 
-    run_fn: Callable[[BaseModel, "Context"], Awaitable[BaseModel]] = Field(
-        exclude=True
-    )
+    run_fn: Callable[[BaseModel, "Context"], Awaitable[BaseModel]] = Field(exclude=True)
     destroy_fn: Callable[["Context"], Awaitable[None]] | None = Field(exclude=True)
 
     model_config = ConfigDict(extra="allow")

acp/server/highlevel/agents/templates.py

@@ -16,8 +16,6 @@ class AgentTemplate(BaseModel):
     input: Type[BaseModel] = Field(description="Model for run input")
     output: Type[BaseModel] = Field(description="Model for run output")
 
-    create_fn: Callable[[BaseModel, "Context"], Awaitable[Agent]] = Field(
-        exclude=True
-    )
+    create_fn: Callable[[BaseModel, "Context"], Awaitable[Agent]] = Field(exclude=True)
 
     model_config = ConfigDict(extra="allow")

acp/server/highlevel/context.py

@@ -124,9 +124,9 @@ class Context(BaseModel):
         Returns:
             The resource content as either text or bytes
         """
-        assert (
-            self._fastmcp is not None
-        ), "Context is not available outside of a request"
+        assert self._fastmcp is not None, (
+            "Context is not available outside of a request"
+        )
         return await self._fastmcp.read_resource(uri)
 
     async def log(

acp/server/highlevel/server.py

@@ -467,7 +467,7 @@ class Server:
         config: Type[BaseModel],
         input: Type[BaseModel],
         output: Type[BaseModel],
-        **kwargs
+        **kwargs,
     ) -> Callable:
         """Decorator to register an agent template.
 
@@ -494,7 +494,7 @@ class Server:
                 input=input,
                 output=output,
                 create_fn=func,
-                **kwargs
+                **kwargs,
             )
             self.add_agent_template(template)
             return func
@@ -513,7 +513,7 @@ class Server:
                     configSchema=template.config.model_json_schema(),
                     inputSchema=template.input.model_json_schema(),
                     outputSchema=template.output.model_json_schema(),
-                    **(template.model_extra if template.model_extra else {})
+                    **(template.model_extra if template.model_extra else {}),
                 )
                 for template in templates
             ]
@@ -529,7 +529,7 @@ class Server:
         description: str,
         input: Type[BaseModel],
         output: Type[BaseModel],
-        **kwargs
+        **kwargs,
     ) -> Callable:
         """Decorator to register an agent.
 
@@ -554,7 +554,7 @@ class Server:
                 output=output,
                 run_fn=func,
                 destroy_fn=None,
-                **kwargs
+                **kwargs,
             )
             self.add_agent(agent=agent)
             return func
@@ -570,7 +570,7 @@ class Server:
                     description=agent.description,
                     inputSchema=agent.input.model_json_schema(),
                     outputSchema=agent.output.model_json_schema(),
-                    **(agent.model_extra if agent.model_extra else {})
+                    **(agent.model_extra if agent.model_extra else {}),
                 )
                 for agent in agents
             ]
@@ -613,7 +613,7 @@ class Server:
                 self._mcp_server.create_initialization_options(),
             )
 
-    async def run_sse_async(self) -> None:
+    async def run_sse_async(self, **uvicorn_kwargs) -> None:
         """Run the server using SSE transport."""
         from starlette.applications import Starlette
         from starlette.routing import Mount, Route
@@ -643,6 +643,7 @@ class Server:
             host=self.settings.host,
             port=self.settings.port,
             log_level=self.settings.log_level.lower(),
+            **uvicorn_kwargs,
         )
         server = uvicorn.Server(config)
         await server.serve()

acp/server/lowlevel/server.py

@@ -68,8 +68,10 @@ import contextvars
 import logging
 import warnings
 from collections.abc import Awaitable, Callable
+from contextlib import AsyncExitStack
 from typing import Any, Sequence
 
+import anyio
 from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from pydantic import AnyUrl
 
@@ -182,7 +184,7 @@ class Server:
         if types.ListAgentsRequest in self.request_handlers:
             agents_capability = types.AgentsCapability(
                 templates=types.ListAgentTemplatesRequest in self.request_handlers,
-                listChanged=notification_options.agents_changed
+                listChanged=notification_options.agents_changed,
             )
 
         # Set logging capabilities if handler exists
@@ -538,30 +540,41 @@ class Server:
         # in-process servers.
         raise_exceptions: bool = False,
     ):
-        with warnings.catch_warnings(record=True) as w:
-            async with ServerSession(
-                read_stream, write_stream, initialization_options
-            ) as session:
+        async with AsyncExitStack() as stack:
+            session = await stack.enter_async_context(
+                ServerSession(read_stream, write_stream, initialization_options)
+            )
+
+            async with anyio.create_task_group() as tg:
                 async for message in session.incoming_messages:
                     logger.debug(f"Received message: {message}")
 
-                    match message:
-                        case (
-                            RequestResponder(
-                                request=types.ClientRequest(root=req)
-                            ) as responder
-                        ):
-                            with responder:
-                                await self._handle_request(
-                                    message, req, session, raise_exceptions
-                                )
-                        case types.ClientNotification(root=notify):
-                            await self._handle_notification(notify)
-
-                    for warning in w:
-                        logger.info(
-                            f"Warning: {warning.category.__name__}: {warning.message}"
+                    tg.start_soon(
+                        self._handle_message, message, session, raise_exceptions
+                    )
+
+    async def _handle_message(
+        self,
+        message: RequestResponder[types.ClientRequest, types.ServerResult]
+        | types.ClientNotification
+        | Exception,
+        session: ServerSession,
+        raise_exceptions: bool = False,
+    ):
+        with warnings.catch_warnings(record=True) as w:
+            match message:
+                case (
+                    RequestResponder(request=types.ClientRequest(root=req)) as responder
+                ):
+                    with responder:
+                        await self._handle_request(
+                            message, req, session, raise_exceptions
                         )
+                case types.ClientNotification(root=notify):
+                    await self._handle_notification(notify)
+
+            for warning in w:
+                logger.info(f"Warning: {warning.category.__name__}: {warning.message}")
 
     async def _handle_request(
         self,
@@ -614,14 +627,12 @@ class Server:
             assert type(notify) in self.notification_handlers
 
             handler = self.notification_handlers[type(notify)]
-            logger.debug(
-                f"Dispatching notification of type " f"{type(notify).__name__}"
-            )
+            logger.debug(f"Dispatching notification of type {type(notify).__name__}")
 
             try:
                 await handler(notify)
             except Exception as err:
-                logger.error(f"Uncaught exception in notification handler: " f"{err}")
+                logger.error(f"Uncaught exception in notification handler: {err}")
 
 
 async def _ping_handler(request: types.PingRequest) -> types.ServerResult:

acp/shared/memory.py

@@ -20,9 +20,9 @@ MessageStream = tuple[
 
 
 @asynccontextmanager
-async def create_client_server_memory_streams() -> (
-    AsyncGenerator[tuple[MessageStream, MessageStream], None]
-):
+async def create_client_server_memory_streams() -> AsyncGenerator[
+    tuple[MessageStream, MessageStream], None
+]:
     """
     Creates a pair of bidirectional memory streams for client-server communication.
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acp-sdk
-Version: 0.0.1
+Version: 0.0.4
 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
@@ -17,7 +17,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Python: >=3.10
+Requires-Python: <4.0,>=3.10
 Requires-Dist: anyio>=4.5
 Requires-Dist: httpx-sse>=0.4
 Requires-Dist: httpx>=0.27

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

@@ -17,13 +17,13 @@ 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
 acp/server/highlevel/__init__.py,sha256=zpNfKzPtHeuOT74iaoT1938XpXurH82E_NJEg1n-p4E,247
-acp/server/highlevel/context.py,sha256=QOKqXceInNAwwhBSkyxrELY12dqmLHX33E-LFUo9OeU,5645
+acp/server/highlevel/context.py,sha256=1vEoiNhJQ4khbS1Ln5FAFj6rqPsXfiUB47-nDzYK0Is,5645
 acp/server/highlevel/exceptions.py,sha256=FQhduu4gNAGIfpW9Twx4LsN8UR2KONdxmubZDbXNDKg,506
-acp/server/highlevel/server.py,sha256=io0Ldmkao-hxlho1BdJCSmV9GGwB80vnl8IyZkbtxAw,24185
+acp/server/highlevel/server.py,sha256=t82jPgqCm7YZfcxIaeCkLDXVXxUgpPRJJ2e6MtA2baA,24239
 acp/server/highlevel/agents/__init__.py,sha256=iahXO9_-1Fm0_6HcaFwcgiHTvY3v83GAVzWvVaTFOog,155
 acp/server/highlevel/agents/agent_manager.py,sha256=FYRXE-RFWWlkbci8GfF1XS_n5CouGXSdvCw0jBbfcHo,3767
-acp/server/highlevel/agents/base.py,sha256=Q_DM7WE0yWxyqkDvJDP-Kyo4BfA_3_mYhvSqW3K43Ik,714
-acp/server/highlevel/agents/templates.py,sha256=ORDbsX6h6UUU9zkMSPj1P8CCOfWOeH41Q2sBKJvrE4U,775
+acp/server/highlevel/agents/base.py,sha256=Ets5AYbkB8gsRVMKaNjVRBonw6sOtAqmxIAti6emlNE,700
+acp/server/highlevel/agents/templates.py,sha256=gDfWrsO6UjRLAGaJQYxv_KTp0or5a8SBQvXCYjb4oww,761
 acp/server/highlevel/prompts/__init__.py,sha256=4BsMxoYolpoxg74xkkkzCFL8vvdkLVJ5cIPNs1ND1Jo,99
 acp/server/highlevel/prompts/base.py,sha256=m5-CB-Jtu5V5fowml-7IQZDpqa7UmKRHc7Ex2tskmQk,5554
 acp/server/highlevel/prompts/manager.py,sha256=UjGCo9w1RBcQ8Eo9aVEvLOrzSI9T1r7jEbXuvuklIuc,1489
@@ -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=jg6GuNBe3P0zbjA9AOExKC1vMtyVi0zT2FygjXGwJmw,22286
+acp/server/lowlevel/server.py,sha256=cRzDFKK_tw02eIvW6ie38NH7UXNaxZ2WPtoyX7iaoyI,22605
 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=Dg04QJiLCe8n0oUVvJuGjnF37M4nvgpZsFsQ-n-lgTE,2804
+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/version.py,sha256=si_GMtVN2H0Y3Rn4QNf8u2V0kZRTzzlMHuCWqDDDK1s,106
-acp_sdk-0.0.1.dist-info/METADATA,sha256=MVhFq8yZuFG5Fey1oDKWR616L8ec_0V6r7BDpBUlGqU,1927
-acp_sdk-0.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-acp_sdk-0.0.1.dist-info/entry_points.txt,sha256=zJgYH5TTBxQ1oBzEg-0RfEk09Te1W2LkqbyoLpFe9CA,42
-acp_sdk-0.0.1.dist-info/licenses/LICENSE,sha256=aLcXJ7ZsI7zZmq4n0iIQ6KYB3EJMtRJGLr7n514PE30,1100
-acp_sdk-0.0.1.dist-info/RECORD,,
+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,,