Merge branch 'main' into make-fastapi-package-optional

Author: darkhaniop

.github/actions/spelling/allow.txt

@@ -9,17 +9,22 @@ AServer
 AServers
 AService
 AStarlette
+AUser
+DSNs
 EUR
 GBP
+GVsb
 INR
 JPY
 JSONRPCt
 Llm
+POSTGRES
 RUF
 aconnect
 adk
 agentic
 aio
+aiomysql
 aproject
 autouse
 backticks
@@ -29,21 +34,31 @@ coc
 codegen
 coro
 datamodel
+drivername
 dunders
 euo
+excinfo
+fetchrow
+fetchval
 genai
 getkwargs
 gle
+initdb
 inmemory
+isready
 kwarg
 langgraph
 lifecycles
 linting
 lstrips
 mockurl
+notif
 oauthoidc
 oidc
 opensource
+otherurl
+postgres
+postgresql
 protoc
 pyi
 pypistats

.github/actions/spelling/expect.txt

@@ -9,7 +9,18 @@ jobs:
   test:
     name: Test with Python ${{ matrix.python-version }}
     runs-on: ubuntu-latest
+
     if: github.repository == 'a2aproject/a2a-python'
+    services:
+      postgres:
+        image: postgres:15-alpine
+        env:
+          POSTGRES_USER: postgres
+          POSTGRES_PASSWORD: postgres
+          POSTGRES_DB: a2a_test
+        ports:
+          - 5432:5432
+
     strategy:
       matrix:
         python-version: ['3.10', '3.13']
@@ -20,13 +31,19 @@ jobs:
         uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
+      - name: Set postgres for tests
+        run: |
+          sudo apt-get update && sudo apt-get install -y postgresql-client
+          PGPASSWORD=postgres psql -h localhost -p 5432 -U postgres -d a2a_test -f ${{ github.workspace }}/docker/postgres/init.sql
+          export POSTGRES_TEST_DSN="postgresql+asyncpg://postgres:postgres@localhost:5432/a2a_test"
+
       - name: Install uv
         uses: astral-sh/setup-uv@v6
       - name: Add uv to PATH
         run: |
           echo "$HOME/.cargo/bin" >> $GITHUB_PATH
       - name: Install dependencies
-        run: uv sync --dev
+        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

.github/workflows/unit-tests.yml

@@ -27,6 +27,8 @@ You can install the A2A SDK using either `uv` or `pip`.
 
 ### Using `uv`
 
+When you're working within a uv project or a virtual environment managed by uv, the preferred way to add packages is using uv add.
+
 ```bash
 uv add a2a-sdk
 ```
@@ -37,9 +39,25 @@ To include the optional HTTP server components (FastAPI, Starlette), install the
 uv add a2a-sdk[http-server]
 ```
 
+To install with database support:
+
+```bash
+# PostgreSQL support
+uv add "a2a-sdk[postgresql]"
+
+# MySQL support
+uv add "a2a-sdk[mysql]"
+
+# SQLite support
+uv add "a2a-sdk[sqlite]"
+
+# All database drivers
+uv add "a2a-sdk[sql]"
+```
+
 ### Using `pip`
 
-You can install `a2a-sdk` using pip, the standard Python package installer.
+If you prefer to use pip, the standard Python package installer, you can install `a2a-sdk` as follows
 
 ```bash
 pip install a2a-sdk
@@ -51,6 +69,22 @@ To include the optional HTTP server components (FastAPI, Starlette), install the
 pip install a2a-sdk[http-server]
 ```
 
+To install with database support:
+
+```bash
+# PostgreSQL support
+pip install "a2a-sdk[postgresql]"
+
+# MySQL support
+pip install "a2a-sdk[mysql]"
+
+# SQLite support
+pip install "a2a-sdk[sqlite]"
+
+# All database drivers
+pip install "a2a-sdk[sql]"
+```
+
 ## Examples
 
 ### [Helloworld Example](https://github.com/a2aproject/a2a-samples/tree/main/samples/python/agents/helloworld)
@@ -70,7 +104,7 @@ pip install a2a-sdk[http-server]
    uv run test_client.py
    ```
 
-3. You can validate your agent using the agent inspector. Follow the instructions at the [a2a-inspector](https://github.com/a2aproject/a2a-inspector) repo. 
+3. You can validate your agent using the agent inspector. Follow the instructions at the [a2a-inspector](https://github.com/a2aproject/a2a-inspector) repo.
 
 You can also find more Python samples [here](https://github.com/a2aproject/a2a-samples/tree/main/samples/python) and JavaScript samples [here](https://github.com/a2aproject/a2a-samples/tree/main/samples/js).
 

README.md

@@ -0,0 +1,8 @@
+-- Create a dedicated user for the application
+CREATE USER a2a WITH PASSWORD 'a2a_password';
+
+-- Create the tasks database
+CREATE DATABASE a2a_tasks;
+
+GRANT ALL PRIVILEGES ON DATABASE a2a_test TO a2a;
+

docker/postgres/init.sql

@@ -3,10 +3,10 @@ name = "a2a-sdk"
 dynamic = ["version"]
 description = "A2A Python SDK"
 readme = "README.md"
-license = { file = "LICENSE" }
+license = "Apache-2.0"
 authors = [{ name = "Google LLC", email = "googleapis-packages@google.com" }]
 requires-python = ">=3.10"
-keywords = ["A2A", "A2A SDK", "A2A Protocol", "Agent2Agent"]
+keywords = ["A2A", "A2A SDK", "A2A Protocol", "Agent2Agent", "Agent 2 Agent"]
 dependencies = [
   "httpx>=0.28.1",
   "httpx-sse>=0.4.0",
@@ -35,6 +35,10 @@ classifiers = [
 
 [project.optional-dependencies]
 http-server = ["fastapi>=0.115.2", "sse-starlette", "starlette"]
+postgresql = ["sqlalchemy[asyncio,postgresql-asyncpg]>=2.0.0"]
+mysql = ["sqlalchemy[asyncio,aiomysql]>=2.0.0"]
+sqlite = ["sqlalchemy[asyncio,aiosqlite]>=2.0.0"]
+sql = ["sqlalchemy[asyncio,postgresql-asyncpg,aiomysql,aiosqlite]>=2.0.0"]
 
 [project.urls]
 homepage = "https://a2aproject.github.io/A2A/"