agentenv-framework-protocol 0.1.292__tar.gz → 0.1.293__tar.gz

{agentenv_framework_protocol-0.1.292 → agentenv_framework_protocol-0.1.293}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: agentenv-framework-protocol
-Version: 0.1.292
+Version: 0.1.293
 Summary: Open data-plane protocol and server SDK for agent environments.
 Project-URL: Homepage, https://www.agentenvframework.com
 Project-URL: Documentation, https://www.agentenvframework.com/docs
@@ -345,6 +345,11 @@ A successful `TaskResult` must contain at least one text, file, or data part;
 the framework rejects empty successes rather than emitting an ungradeable task.
 The SDK does not retry tasks.
 
+A client that gives up on a task sends `tasks/cancel`. The framework marks the
+task canceled and cancels `run()`, which gets `asyncio.CancelledError` at its
+next `await`. A process `run()` started keeps running unless `run()` stops it,
+so kill it before re-raising.
+
 `enable(..., description="...")` is reserved for declarations carrying
 configuration or metadata. It preserves the agent-specific extension prose
 published in the Agent Card. The versioned SDK definition provides a generic

{agentenv_framework_protocol-0.1.292 → agentenv_framework_protocol-0.1.293}/README.md

@@ -301,6 +301,11 @@ A successful `TaskResult` must contain at least one text, file, or data part;
 the framework rejects empty successes rather than emitting an ungradeable task.
 The SDK does not retry tasks.
 
+A client that gives up on a task sends `tasks/cancel`. The framework marks the
+task canceled and cancels `run()`, which gets `asyncio.CancelledError` at its
+next `await`. A process `run()` started keeps running unless `run()` stops it,
+so kill it before re-raising.
+
 `enable(..., description="...")` is reserved for declarations carrying
 configuration or metadata. It preserves the agent-specific extension prose
 published in the Agent Card. The versioned SDK definition provides a generic

{agentenv_framework_protocol-0.1.292 → agentenv_framework_protocol-0.1.293}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentenv-framework-protocol"
-version = "0.1.292"
+version = "0.1.293"
 description = "Open data-plane protocol and server SDK for agent environments."
 readme = "README.md"
 license = "Apache-2.0"

{agentenv_framework_protocol-0.1.292 → agentenv_framework_protocol-0.1.293}/src/agentenv_protocol/a2a_agent/framework.py

@@ -1226,12 +1226,12 @@ class _StandardExecutor:
                 "streaming TaskStatusUpdateEvent must be non-terminal and final=False"
             )
 
-    async def cancel(self, _context: Any, _event_queue: Any) -> None:
-        """Satisfy the upstream executor interface without offering cancellation."""
-        from a2a.types import UnsupportedOperationError
-        from a2a.utils.errors import ServerError
+    async def cancel(self, context: Any, event_queue: Any) -> None:
+        """Mark the task canceled. The SDK then cancels the coroutine running it, so ``agent.run`` gets
+        ``CancelledError`` at its next ``await`` and should stop whatever it started, such as a CLI process."""
+        from a2a.server.tasks import TaskUpdater
 
-        raise ServerError(error=UnsupportedOperationError())
+        await TaskUpdater(event_queue, context.task_id, context.context_id).cancel()
 
 
 def _from_a2a_part(part: Any) -> Any:

{agentenv_framework_protocol-0.1.292 → agentenv_framework_protocol-0.1.293}/tests/test_a2a_agent.py

@@ -2908,3 +2908,76 @@ def test_extension_cannot_shadow_framework_route(
 
     with pytest.raises(ValueError, match=rf"conflicts with {owner} route"):
         Agent().create_app()
+
+
+def _send_in_context(client: TestClient, text: str, message_id: str) -> str:
+    response = client.post(
+        "/a2a",
+        json={
+            "jsonrpc": "2.0",
+            "id": message_id,
+            "method": "message/send",
+            "params": {
+                "message": {
+                    "kind": "message",
+                    "messageId": message_id,
+                    "role": "user",
+                    "contextId": "context-1",
+                    "parts": [{"kind": "text", "text": text}],
+                },
+                "configuration": {"blocking": False},
+            },
+        },
+    )
+    return response.json()["result"]["id"]
+
+
+def _task_rpc(client: TestClient, method: str, task_id: str) -> dict[str, Any]:
+    return client.post(
+        "/a2a",
+        json={"jsonrpc": "2.0", "id": method, "method": method, "params": {"id": task_id}},
+    ).json()
+
+
+def _within(seconds: float, condition) -> bool:
+    deadline = time.monotonic() + seconds
+    while not condition():
+        if time.monotonic() > deadline:
+            return False
+        time.sleep(0.01)
+    return True
+
+
+def test_cancel_stops_a_running_task_and_frees_its_context() -> None:
+    ran: list[str] = []
+    cancelled: list[str] = []
+
+    @a2a_agent(identity=AgentIdentity(name="slow", description="test", version="1"))
+    class Agent(AgentEnvAgent):
+        async def run(self, request: TaskRequest) -> TaskResult:
+            text = request.parts[0].text
+            ran.append(text)
+            if text == "hang":
+                try:
+                    await asyncio.Event().wait()
+                except asyncio.CancelledError:
+                    cancelled.append(text)
+                    raise
+            return TaskResult.text(f"did {text}")
+
+    def state(client: TestClient, task_id: str) -> str:
+        return _task_rpc(client, "tasks/get", task_id)["result"]["status"]["state"]
+
+    with TestClient(Agent().create_app()) as client:
+        hung = _send_in_context(client, "hang", "message-1")
+        assert _within(5, lambda: ran == ["hang"])
+
+        canceled = _task_rpc(client, "tasks/cancel", hung)
+        assert canceled["result"]["status"]["state"] == "canceled"
+
+        following = _send_in_context(client, "next", "message-2")
+        assert _within(5, lambda: state(client, following) == "completed")
+        assert state(client, hung) == "canceled"
+
+    assert cancelled == ["hang"]
+    assert ran == ["hang", "next"]