acp-sdk 0.1.0rc5__py3-none-any.whl → 0.1.0rc6__py3-none-any.whl

acp_sdk/server/server.py

@@ -19,19 +19,17 @@ class Server:
         """Decorator to register an agent."""
 
         def decorator(fn: Callable) -> Callable:
-            # check agent's function signature
             signature = inspect.signature(fn)
             parameters = list(signature.parameters.values())
 
-            # validate agent's function
-            if inspect.isasyncgenfunction(fn):
-                if len(parameters) != 2:
-                    raise TypeError(
-                        "The agent generator function must have one 'input' argument and one 'context' argument"
-                    )
-            else:
-                if len(parameters) != 2:
-                    raise TypeError("The agent function must have one 'input' argument and one 'context' argument")
+            if len(parameters) == 0:
+                raise TypeError("The agent function must have at least 'input' argument")
+            if len(parameters) > 2:
+                raise TypeError("The agent function must have only 'input' and 'context' arguments")
+            if len(parameters) == 2 and parameters[1].name != "context":
+                raise TypeError("The second argument of the agent function must be 'context'")
+
+            has_context_param = len(parameters) == 2
 
             agent: Agent
             if inspect.isasyncgenfunction(fn):
@@ -47,7 +45,9 @@ class Server:
 
                     async def run(self, input: Message, context: Context) -> AsyncGenerator[RunYield, RunYieldResume]:
                         try:
-                            gen: AsyncGenerator[RunYield, RunYieldResume] = fn(input, context)
+                            gen: AsyncGenerator[RunYield, RunYieldResume] = (
+                                fn(input, context) if has_context_param else fn(input)
+                            )
                             value = None
                             while True:
                                 value = yield await gen.asend(value)
@@ -67,7 +67,7 @@ class Server:
                         return description or fn.__doc__ or ""
 
                     async def run(self, input: Message, context: Context) -> Coroutine[RunYield]:
-                        return await fn(input, context)
+                        return await (fn(input, context) if has_context_param else fn(input))
 
                 agent = DecoratedAgent()
             elif inspect.isgeneratorfunction(fn):
@@ -82,7 +82,7 @@ class Server:
                         return description or fn.__doc__ or ""
 
                     def run(self, input: Message, context: Context) -> Generator[RunYield, RunYieldResume]:
-                        yield from fn(input, context)
+                        yield from (fn(input, context) if has_context_param else fn(input))
 
                 agent = DecoratedAgent()
             else:
@@ -97,7 +97,7 @@ class Server:
                         return description or fn.__doc__ or ""
 
                     def run(self, input: Message, context: Context) -> RunYield:
-                        return fn(input, context)
+                        return fn(input, context) if has_context_param else fn(input)
 
                 agent = DecoratedAgent()
 
@@ -109,7 +109,9 @@ class Server:
     def register(self, *agents: Agent) -> None:
         self.agents.extend(agents)
 
-    def run(self, configure_logger: bool = True, configure_telemetry: bool = False, **kwargs: dict[str, Any]) -> None:
+    def __call__(
+        self, configure_logger: bool = True, configure_telemetry: bool = False, **kwargs: dict[str, Any]
+    ) -> None:
         import uvicorn
 
         if configure_logger:

{acp_sdk-0.1.0rc5.dist-info → acp_sdk-0.1.0rc6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acp-sdk
-Version: 0.1.0rc5
+Version: 0.1.0rc6
 Summary: Agent Communication Protocol SDK
 Requires-Python: <4.0,>=3.11
 Requires-Dist: opentelemetry-api>=1.31.1

{acp_sdk-0.1.0rc5.dist-info → acp_sdk-0.1.0rc6.dist-info}/RECORD

@@ -13,10 +13,10 @@ acp_sdk/server/bundle.py,sha256=ZVkIOwWfYgi-VyGYSMKr9WrG_WzYbej-LD-YPUWQdeg,4982
 acp_sdk/server/context.py,sha256=MgnLV6qcDIhc_0BjW7r4Jj1tHts4ZuwpdTGIBnz2Mgo,1036
 acp_sdk/server/errors.py,sha256=fWlgVsQ5hs_AXwzc-wvy6QgoDWEMRUBlSrfJfhHHMyE,2085
 acp_sdk/server/logging.py,sha256=Oc8yZigCsuDnHHPsarRzu0RX3NKaLEgpELM2yovGKDI,411
-acp_sdk/server/server.py,sha256=xSS_TjxQ5xi_X9RJDNjFsU0JEytLlCCSqbp6sBVXXhY,4442
+acp_sdk/server/server.py,sha256=H-oh-cRpo9NR7_Mdt7ECRXrGnk8mfAAb7pp94XLGoAs,4637
 acp_sdk/server/telemetry.py,sha256=EwmtUrWMYid7XHiX76V1J6CigJPa2NrzEPOX0fBoY3o,1838
 acp_sdk/server/types.py,sha256=2yJPkfUzjVIhHmc0SegGTMqDROe2uFgycb-7CATvYVw,161
 acp_sdk/server/utils.py,sha256=EfrF9VCyVk3AM_ao-BIB9EzGbfTrh4V2Bz-VFr6f6Sg,351
-acp_sdk-0.1.0rc5.dist-info/METADATA,sha256=laguwvWACy9Yslwj_Wh2p25rfEciRu217HEGGXTIOKg,2147
-acp_sdk-0.1.0rc5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-acp_sdk-0.1.0rc5.dist-info/RECORD,,
+acp_sdk-0.1.0rc6.dist-info/METADATA,sha256=t5zWJsU-C67HQNzCFyo561s7epz4KQvsFjlqUIkF9xw,2147
+acp_sdk-0.1.0rc6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+acp_sdk-0.1.0rc6.dist-info/RECORD,,