Skip to content

Commit aedf78c

Browse files
committed
Fix Unit Tests from recent updates
1 parent 1fba1b4 commit aedf78c

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

tests/server/request_handlers/test_jsonrpc_handler.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -527,8 +527,8 @@ async def streaming_coro():
527527
json={
528528
'contextId': 'session-xyz',
529529
'id': 'task_123',
530+
'kind': 'task',
530531
'status': {'state': 'submitted'},
531-
'type': 'task',
532532
},
533533
),
534534
call(
@@ -537,13 +537,18 @@ async def streaming_coro():
537537
'artifacts': [
538538
{
539539
'artifactId': '11',
540-
'parts': [{'text': 'text', 'type': 'text'}],
540+
'parts': [
541+
{
542+
'kind': 'text',
543+
'text': 'text',
544+
}
545+
],
541546
}
542547
],
543548
'contextId': 'session-xyz',
544549
'id': 'task_123',
550+
'kind': 'task',
545551
'status': {'state': 'submitted'},
546-
'type': 'task',
547552
},
548553
),
549554
call(
@@ -552,13 +557,18 @@ async def streaming_coro():
552557
'artifacts': [
553558
{
554559
'artifactId': '11',
555-
'parts': [{'text': 'text', 'type': 'text'}],
560+
'parts': [
561+
{
562+
'kind': 'text',
563+
'text': 'text',
564+
}
565+
],
556566
}
557567
],
558568
'contextId': 'session-xyz',
559569
'id': 'task_123',
570+
'kind': 'task',
560571
'status': {'state': 'completed'},
561-
'type': 'task',
562572
},
563573
),
564574
]

0 commit comments

Comments
 (0)