Merge branch 'main' into sql-support

Author: holtskinner

CONTRIBUTING.md

@@ -2,13 +2,6 @@
 
 We'd love to accept your patches and contributions to this project.
 
-## Before you begin
-
-### Review our community guidelines
-
-This project follows
-[Google's Open Source Community Guidelines](https://opensource.google/conduct/).
-
 ## Contribution process
 
 ### Code reviews
@@ -33,11 +26,3 @@ Here are some additional things to keep in mind during the process:
 
 - **Test your changes.** Before you submit a pull request, make sure that your changes work as expected.
 - **Be patient.** It may take some time for your pull request to be reviewed and merged.
-
----
-
-## For Google Employees
-
-Complete the following steps to register your GitHub account and be added as a contributor to this repository.
-
-1. Register your GitHub account at [go/GitHub](http://go/github).

tests/client/test_auth_middleware.py

@@ -12,9 +12,12 @@
     AgentCard,
     AuthorizationCodeOAuthFlow,
     In,
+    Message,
+    MessageSendParams,
     OAuth2SecurityScheme,
     OAuthFlows,
     OpenIdConnectSecurityScheme,
+    Role,
     SecurityScheme,
     SendMessageRequest,
 )
@@ -76,13 +79,13 @@ async def test_client_with_simple_interceptor():
         await client.send_message(
             request=SendMessageRequest(
                 id='1',
-                params={
-                    'message': {
-                        'messageId': 'msg1',
-                        'role': 'user',
-                        'parts': [],
-                    }
-                },
+                params=MessageSendParams(
+                    message=Message(
+                        messageId='msg1',
+                        role=Role.user,
+                        parts=[],
+                    )
+                ),
             )
         )
 
@@ -198,13 +201,13 @@ async def test_auth_interceptor_with_api_key():
         await client.send_message(
             request=SendMessageRequest(
                 id='1',
-                params={
-                    'message': {
-                        'messageId': 'msg1',
-                        'role': 'user',
-                        'parts': [],
-                    }
-                },
+                params=MessageSendParams(
+                    message=Message(
+                        messageId='msg1',
+                        role=Role.user,
+                        parts=[],
+                    )
+                ),
             ),
             context=context,
         )
@@ -284,13 +287,13 @@ async def test_auth_interceptor_with_oauth2_scheme():
         await client.send_message(
             request=SendMessageRequest(
                 id='oauth_test_1',
-                params={
-                    'message': {
-                        'messageId': 'msg-oauth',
-                        'role': 'user',
-                        'parts': [],
-                    }
-                },
+                params=MessageSendParams(
+                    message=Message(
+                        messageId='msg-oauth',
+                        role=Role.user,
+                        parts=[],
+                    )
+                ),
             ),
             context=context,
         )
@@ -366,13 +369,13 @@ async def test_auth_interceptor_with_oidc_scheme():
         await client.send_message(
             request=SendMessageRequest(
                 id='oidc_test_1',
-                params={
-                    'message': {
-                        'messageId': 'msg-oidc',
-                        'role': 'user',
-                        'parts': [],
-                    }
-                },
+                params=MessageSendParams(
+                    message=Message(
+                        messageId='msg-oidc',
+                        role=Role.user,
+                        parts=[],
+                    )
+                ),
             ),
             context=context,
         )