Skip to content

Commit a9d3001

Browse files
committed
fix
1 parent fbcd4d3 commit a9d3001

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

src/a2a/server/routes/jsonrpc_dispatcher.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
HTTP_EXTENSION_HEADER,
1616
)
1717
from a2a.server.context import ServerCallContext
18+
from a2a.server.events import Event
1819
from a2a.server.jsonrpc_models import (
1920
InternalError,
2021
InvalidParamsError,
@@ -383,9 +384,9 @@ async def _process_streaming_request(
383384
first_event = None
384385

385386
async def _wrap_stream(
386-
st: AsyncGenerator, first_evt: Any | None
387+
st: AsyncGenerator, first_evt: Event | None
387388
) -> AsyncGenerator[dict[str, Any], None]:
388-
def _map_event(evt: Any) -> dict[str, Any]:
389+
def _map_event(evt: Event) -> dict[str, Any]:
389390
stream_response = proto_utils.to_stream_response(evt)
390391
result = MessageToDict(
391392
stream_response, preserving_proto_field_name=False

0 commit comments

Comments
 (0)