File tree Expand file tree Collapse file tree 1 file changed +31
-28
lines changed Expand file tree Collapse file tree 1 file changed +31
-28
lines changed Original file line number Diff line number Diff line change 12
12
AgentCard ,
13
13
AuthorizationCodeOAuthFlow ,
14
14
In ,
15
+ Message ,
16
+ MessageSendParams ,
15
17
OAuth2SecurityScheme ,
16
18
OAuthFlows ,
17
19
OpenIdConnectSecurityScheme ,
20
+ Role ,
18
21
SecurityScheme ,
19
22
SendMessageRequest ,
20
23
)
@@ -76,13 +79,13 @@ async def test_client_with_simple_interceptor():
76
79
await client .send_message (
77
80
request = SendMessageRequest (
78
81
id = '1' ,
79
- params = {
80
- ' message' : {
81
- ' messageId' : 'msg1' ,
82
- ' role' : ' user' ,
83
- ' parts' : [],
84
- }
85
- } ,
82
+ params = MessageSendParams (
83
+ message = Message (
84
+ messageId = 'msg1' ,
85
+ role = Role . user ,
86
+ parts = [],
87
+ )
88
+ ) ,
86
89
)
87
90
)
88
91
@@ -198,13 +201,13 @@ async def test_auth_interceptor_with_api_key():
198
201
await client .send_message (
199
202
request = SendMessageRequest (
200
203
id = '1' ,
201
- params = {
202
- ' message' : {
203
- ' messageId' : 'msg1' ,
204
- ' role' : ' user' ,
205
- ' parts' : [],
206
- }
207
- } ,
204
+ params = MessageSendParams (
205
+ message = Message (
206
+ messageId = 'msg1' ,
207
+ role = Role . user ,
208
+ parts = [],
209
+ )
210
+ ) ,
208
211
),
209
212
context = context ,
210
213
)
@@ -284,13 +287,13 @@ async def test_auth_interceptor_with_oauth2_scheme():
284
287
await client .send_message (
285
288
request = SendMessageRequest (
286
289
id = 'oauth_test_1' ,
287
- params = {
288
- ' message' : {
289
- ' messageId' : 'msg-oauth' ,
290
- ' role' : ' user' ,
291
- ' parts' : [],
292
- }
293
- } ,
290
+ params = MessageSendParams (
291
+ message = Message (
292
+ messageId = 'msg-oauth' ,
293
+ role = Role . user ,
294
+ parts = [],
295
+ )
296
+ ) ,
294
297
),
295
298
context = context ,
296
299
)
@@ -366,13 +369,13 @@ async def test_auth_interceptor_with_oidc_scheme():
366
369
await client .send_message (
367
370
request = SendMessageRequest (
368
371
id = 'oidc_test_1' ,
369
- params = {
370
- ' message' : {
371
- ' messageId' : 'msg-oidc' ,
372
- ' role' : ' user' ,
373
- ' parts' : [],
374
- }
375
- } ,
372
+ params = MessageSendParams (
373
+ message = Message (
374
+ messageId = 'msg-oidc' ,
375
+ role = Role . user ,
376
+ parts = [],
377
+ )
378
+ ) ,
376
379
),
377
380
context = context ,
378
381
)
You can’t perform that action at this time.
0 commit comments