Merge branch 'sql-support' of https://github.com/a2aproject/a2a-python into sql-support

Author: holtskinner

.github/workflows/unit-tests.yml

@@ -43,7 +43,7 @@ jobs:
         run: |
           echo "$HOME/.cargo/bin" >> $GITHUB_PATH
       - name: Install dependencies
-        run: uv sync --dev  --extra sql
+        run: uv sync --dev --extra sql
       - name: Run tests and check coverage
         run: uv run pytest --cov=a2a --cov-report=xml --cov-fail-under=90
       - name: Show coverage summary in log

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,
         )