File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -485,14 +485,14 @@ async def test_send_message_streaming_comment_success(
485485 id = 'stream_id_123' ,
486486 jsonrpc = '2.0' ,
487487 result = create_text_message_object (
488- content = 'First part ' , role = Role .agent
488+ content = 'First part' , role = Role .agent
489489 ),
490490 )
491491 mock_stream_response_2 = SendMessageSuccessResponse (
492492 id = 'stream_id_123' ,
493493 jsonrpc = '2.0' ,
494494 result = create_text_message_object (
495- content = 'second part ' , role = Role .agent
495+ content = 'Second part' , role = Role .agent
496496 ),
497497 )
498498
@@ -521,16 +521,8 @@ async def test_send_message_streaming_comment_success(
521521 ]
522522
523523 assert len (results ) == 2
524- assert isinstance (results [0 ], Message )
525- assert (
526- results [0 ].model_dump ()
527- == mock_stream_response_1 .result .model_dump ()
528- )
529- assert isinstance (results [1 ], Message )
530- assert (
531- results [1 ].model_dump ()
532- == mock_stream_response_2 .result .model_dump ()
533- )
524+ assert results [0 ] == mock_stream_response_1 .result
525+ assert results [1 ] == mock_stream_response_2 .result
534526
535527 @pytest .mark .asyncio
536528 async def test_send_request_http_status_error (
You can’t perform that action at this time.
0 commit comments