We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fbcd4d3 commit a9d3001Copy full SHA for a9d3001
1 file changed
src/a2a/server/routes/jsonrpc_dispatcher.py
@@ -15,6 +15,7 @@
15
HTTP_EXTENSION_HEADER,
16
)
17
from a2a.server.context import ServerCallContext
18
+from a2a.server.events import Event
19
from a2a.server.jsonrpc_models import (
20
InternalError,
21
InvalidParamsError,
@@ -383,9 +384,9 @@ async def _process_streaming_request(
383
384
first_event = None
385
386
async def _wrap_stream(
- st: AsyncGenerator, first_evt: Any | None
387
+ st: AsyncGenerator, first_evt: Event | None
388
) -> AsyncGenerator[dict[str, Any], None]:
- def _map_event(evt: Any) -> dict[str, Any]:
389
+ def _map_event(evt: Event) -> dict[str, Any]:
390
stream_response = proto_utils.to_stream_response(evt)
391
result = MessageToDict(
392
stream_response, preserving_proto_field_name=False
0 commit comments