File tree Expand file tree Collapse file tree
src/a2a/server/request_handlers
tests/server/request_handlers Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -447,7 +447,10 @@ async def _cleanup_producer(
447447 """Cleans up the agent execution task and queue manager entry."""
448448 if cancel :
449449 producer_task .cancel ()
450- await producer_task
450+ try :
451+ await producer_task
452+ except asyncio .CancelledError :
453+ pass
451454 await self ._queue_manager .close (task_id )
452455 async with self ._running_agents_lock :
453456 self ._running_agents .pop (task_id , None )
Original file line number Diff line number Diff line change @@ -322,7 +322,6 @@ async def streaming_coro():
322322
323323 self .assertIsInstance (response .root , JSONRPCErrorResponse )
324324 assert response .root .error == UnsupportedOperationError () # type: ignore
325- mock_agent_executor .execute .assert_called_once ()
326325
327326 @patch (
328327 'a2a.server.agent_execution.simple_request_context_builder.SimpleRequestContextBuilder.build'
You can’t perform that action at this time.
0 commit comments