Skip to content

Commit 526ee0b

Browse files
feat(server): add async context manager support to EventQueue (format)
1 parent dce25cf commit 526ee0b

2 files changed

Lines changed: 2 additions & 1 deletion

File tree

src/a2a/server/events/event_queue.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from typing_extensions import Self
88

9-
from a2a.types import (
9+
from a2a.types.a2a_pb2 import (
1010
Message,
1111
Task,
1212
TaskArtifactUpdateEvent,

tests/server/events/test_event_queue.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@ def test_constructor_invalid_max_queue_size() -> None:
7777
):
7878
EventQueue(max_queue_size=-10)
7979

80+
8081
@pytest.mark.asyncio
8182
async def test_event_queue_async_context_manager(
8283
event_queue: EventQueue,

0 commit comments

Comments
 (0)