Skip to content

Commit 58a3aa2

Browse files
committed
Fix method name to message/stream
Change-Id: I5f0c9394d492abe61afa84d8ebb73b4f6dd34a1a
1 parent 80ae082 commit 58a3aa2

File tree

6 files changed

+12
-12
lines changed

6 files changed

+12
-12
lines changed

examples/langgraph/agent_executor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ async def on_message_stream(
5353
event_queue: EventQueue,
5454
task: Task | None,
5555
) -> None:
56-
"""Handler for 'message/sendStream' requests."""
56+
"""Handler for 'message/stream' requests."""
5757
params: MessageSendParams = request.params
5858
query = self._get_user_query(params)
5959

src/a2a/server/agent_execution/agent_executor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ async def on_message_stream(
2929
event_queue: EventQueue,
3030
task: Task | None,
3131
) -> None:
32-
"""Handler for 'message/sendStream' requests."""
32+
"""Handler for 'message/stream' requests."""
3333

3434
@abstractmethod
3535
async def on_cancel(

src/a2a/server/agent_execution/base_agent_executor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ async def on_message_stream(
2929
event_queue: EventQueue,
3030
task: Task | None,
3131
) -> None:
32-
"""Handler for 'message/sendStream' requests."""
32+
"""Handler for 'message/stream' requests."""
3333
event_queue.enqueue_event(A2AError(UnsupportedOperationError()))
3434

3535
async def on_cancel(

src/a2a/server/request_handlers/default_request_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ async def on_message_send_stream(
141141
self,
142142
request: SendStreamingMessageRequest,
143143
) -> AsyncGenerator[SendStreamingMessageResponse, None]:
144-
"""Default handler for 'message/sendStream'."""
144+
"""Default handler for 'message/stream'."""
145145
message_send_params: MessageSendParams = request.params
146146

147147
task_manager = TaskManager(

src/a2a/types.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1050,7 +1050,7 @@ class SendMessageRequest(BaseModel):
10501050

10511051
class SendStreamingMessageRequest(BaseModel):
10521052
"""
1053-
JSON-RPC request model for the 'message/sendStream' method.
1053+
JSON-RPC request model for the 'message/stream' method.
10541054
"""
10551055

10561056
id: str | int | None = None
@@ -1062,7 +1062,7 @@ class SendStreamingMessageRequest(BaseModel):
10621062
"""
10631063
Specifies the version of the JSON-RPC protocol. MUST be exactly "2.0".
10641064
"""
1065-
method: Literal['message/sendStream'] = 'message/sendStream'
1065+
method: Literal['message/stream'] = 'message/stream'
10661066
"""
10671067
A String containing the name of the method to be invoked.
10681068
"""
@@ -1278,7 +1278,7 @@ class SendMessageSuccessResponse(BaseModel):
12781278

12791279
class SendStreamingMessageSuccessResponse(BaseModel):
12801280
"""
1281-
JSON-RPC success response model for the 'message/sendStream' method.
1281+
JSON-RPC success response model for the 'message/stream' method.
12821282
"""
12831283

12841284
id: str | int | None = None
@@ -1347,5 +1347,5 @@ class SendStreamingMessageResponse(
13471347
):
13481348
root: JSONRPCErrorResponse | SendStreamingMessageSuccessResponse
13491349
"""
1350-
JSON-RPC response model for the 'message/sendStream' method.
1350+
JSON-RPC response model for the 'message/stream' method.
13511351
"""

tests/test_types.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -518,12 +518,12 @@ def test_send_subscribe_request() -> None:
518518
params = MessageSendParams(message=Message(**MINIMAL_MESSAGE_USER))
519519
req_data: dict[str, Any] = {
520520
'jsonrpc': '2.0',
521-
'method': 'message/sendStream',
521+
'method': 'message/stream',
522522
'params': params.model_dump(),
523523
'id': 5,
524524
}
525525
req = SendStreamingMessageRequest.model_validate(req_data)
526-
assert req.method == 'message/sendStream'
526+
assert req.method == 'message/stream'
527527
assert isinstance(req.params, MessageSendParams)
528528
assert req.params.message.role == Role.user
529529

@@ -874,13 +874,13 @@ def test_a2a_request_root_model() -> None:
874874
# SendStreamingMessageRequest case
875875
send_subs_req_data: dict[str, Any] = {
876876
'jsonrpc': '2.0',
877-
'method': 'message/sendStream',
877+
'method': 'message/stream',
878878
'params': send_params.model_dump(),
879879
'id': 1,
880880
}
881881
a2a_req_send_subs = A2ARequest.model_validate(send_subs_req_data)
882882
assert isinstance(a2a_req_send_subs.root, SendStreamingMessageRequest)
883-
assert a2a_req_send_subs.root.method == 'message/sendStream'
883+
assert a2a_req_send_subs.root.method == 'message/stream'
884884

885885
# GetTaskRequest case
886886
get_params = TaskQueryParams(id='t2')

0 commit comments

Comments
 (0)