Merge branch 'main' into auto-update-a2a-types-0a9f629e801d4ae89f94991fc28afe9429c91cbc

Author: lkawka

CHANGELOG.md

@@ -1,5 +1,24 @@
 # Changelog
 
+## [0.3.10](https://github.com/a2aproject/a2a-python/compare/v0.3.9...v0.3.10) (2025-10-21)
+
+
+### Features
+
+* add `get_artifact_text()` helper method ([9155888](https://github.com/a2aproject/a2a-python/commit/9155888d258ca4d047002997e6674f3f15a67232))
+* Add a `ClientFactory.connect()` method for easy client creation ([d585635](https://github.com/a2aproject/a2a-python/commit/d5856359034f4d3d1e4578804727f47a3cd7c322))
+
+
+### Bug Fixes
+
+* change `MAX_CONTENT_LENGTH` (for file attachment) in json-rpc to be larger size (10mb) ([#518](https://github.com/a2aproject/a2a-python/issues/518)) ([5b81385](https://github.com/a2aproject/a2a-python/commit/5b813856b4b4e07510a4ef41980d388e47c73b8e))
+* correct `new_artifact` methods signature ([#503](https://github.com/a2aproject/a2a-python/issues/503)) ([ee026aa](https://github.com/a2aproject/a2a-python/commit/ee026aa356042b9eb212eee59fa5135b280a3077))
+
+
+### Code Refactoring
+
+* **utils:** move part helpers to their own file ([9155888](https://github.com/a2aproject/a2a-python/commit/9155888d258ca4d047002997e6674f3f15a67232))
+
 ## [0.3.9](https://github.com/a2aproject/a2a-python/compare/v0.3.8...v0.3.9) (2025-10-15)
 
 

scripts/checkout_experimental_types.sh

@@ -0,0 +1,98 @@
+#!/bin/bash
+
+# Exit immediately if a command exits with a non-zero status.
+# Treat unset variables as an error.
+set -euo pipefail
+
+A2A_SPEC_REPO="https://github.com/a2aproject/A2A.git" # URL for the A2A spec repo.
+A2A_SPEC_BRANCH="main" # Name of the branch with experimental changes.
+FEATURE_BRANCH="experimental-types" # Name of the feature branch to create.
+ROOT_DIR=$(git rev-parse --show-toplevel)
+
+usage() {
+  cat <<EOF
+Usage: $0 [OPTIONS]
+
+Creates a new feature branch with types generated from unmerged A2A spec changes.
+
+This script clones the A2A spec repository, checks out a specific branch,
+and creates a new local feature branch from it. 
+
+The script requires uv and buf to be installed.
+
+OPTIONS:
+  -r, --spec-repo       URL for the A2A spec repository.
+                        (Default: "$A2A_SPEC_REPO")
+
+  -b, --spec-branch     Name of the branch with the experimental changes.
+                        (Default: "$A2A_SPEC_BRANCH")
+
+  -f, --feature-branch  Name of the new feature branch to create.
+                        (Default: "$FEATURE_BRANCH")
+
+  -h, --help            Display this help message and exit.
+
+EXAMPLE:
+  # Run with all default settings:
+  $0
+
+  # Run with custom settings:
+  $0  -r "https://github.com/spec-fork/A2A.git" -b "spec-change" -f "my-branch"
+EOF
+}
+
+# Handle command-line arguments.
+while [[ $# -gt 0 ]]; do
+  case $1 in
+    -h|--help)
+      usage
+      exit 0
+      ;;
+    -r|--spec-repo)
+      A2A_SPEC_REPO="$2"
+      shift 2
+      ;;
+    -b|--spec-branch)
+      A2A_SPEC_BRANCH="$2"
+      shift 2
+      ;;
+    -f|--feature-branch)
+      FEATURE_BRANCH="$2"
+      shift 2
+      ;;
+    *)
+      echo "Error: Unknown option '$1'" >&2
+      usage
+      exit 1
+      ;;
+  esac
+done
+
+
+TMP_WORK_DIR=$(mktemp -d)
+echo "Created a temporary working directory: $TMP_WORK_DIR"
+trap 'rm -rf -- "$TMP_WORK_DIR"' EXIT
+cd $TMP_WORK_DIR
+
+echo "Cloning the \"$A2A_SPEC_REPO\" repository..."
+git clone $A2A_SPEC_REPO spec_repo
+cd spec_repo
+
+echo "Checking out the \"$A2A_SPEC_BRANCH\" branch..."
+git checkout "$A2A_SPEC_BRANCH"
+
+echo "Invoking the generate_types.sh script..."
+GENERATED_FILE="$ROOT_DIR/src/a2a/types.py"
+$ROOT_DIR/scripts/generate_types.sh "$GENERATED_FILE" --input-file "$TMP_WORK_DIR/spec_repo/specification/json/a2a.json"
+
+
+echo "Running buf generate..."
+cd "$ROOT_DIR"
+buf generate
+uv run "$ROOT_DIR/scripts/grpc_gen_post_processor.py"
+
+
+echo "Committing generated types file to the \"$FEATURE_BRANCH\" branch..."
+git checkout -b "$FEATURE_BRANCH"
+git add "$GENERATED_FILE" "$ROOT_DIR/src/a2a/grpc"
+git commit -m "Experimental types"

scripts/generate_types.sh

@@ -4,21 +4,44 @@
 # Treat unset variables as an error.
 set -euo pipefail
 
-# Check if an output file path was provided as an argument.
-if [ -z "$1" ]; then
-  echo "Error: Output file path must be provided as the first argument." >&2
+REMOTE_URL="https://raw.githubusercontent.com/a2aproject/A2A/refs/heads/main/specification/json/a2a.json"
+
+GENERATED_FILE=""
+INPUT_FILE=""
+
+# Parse command-line arguments
+while [[ $# -gt 0 ]]; do
+  case "$1" in
+    --input-file)
+      INPUT_FILE="$2"
+      shift 2
+      ;;
+    *)
+      GENERATED_FILE="$1"
+      shift 1
+      ;;
+  esac
+done
+
+if [ -z "$GENERATED_FILE" ]; then
+  echo "Error: Output file path must be provided." >&2
+  echo "Usage: $0 [--input-file <path>] <output-file-path>"
   exit 1
 fi
 
-REMOTE_URL="https://raw.githubusercontent.com/a2aproject/A2A/refs/heads/main/specification/json/a2a.json"
-GENERATED_FILE="$1"
-
 echo "Running datamodel-codegen..."
-echo "  - Source URL: $REMOTE_URL"
+declare -a source_args
+if [ -n "$INPUT_FILE" ]; then
+  echo "  - Source File: $INPUT_FILE"
+  source_args=("--input" "$INPUT_FILE")
+else
+  echo "  - Source URL: $REMOTE_URL"
+  source_args=("--url" "$REMOTE_URL")
+fi
 echo "  - Output File: $GENERATED_FILE"
 
 uv run datamodel-codegen \
-  --url "$REMOTE_URL" \
+  "${source_args[@]}" \
   --input-file-type jsonschema \
   --output "$GENERATED_FILE" \
   --target-python-version 3.10 \

src/a2a/server/apps/jsonrpc/jsonrpc_app.py

@@ -92,7 +92,7 @@
         Response = Any
         HTTP_413_REQUEST_ENTITY_TOO_LARGE = Any
 
-MAX_CONTENT_LENGTH = 1_000_000
+MAX_CONTENT_LENGTH = 10_000_000
 
 
 class StarletteUserProxy(A2AUser):

src/a2a/utils/__init__.py

@@ -1,6 +1,7 @@
 """Utility functions for the A2A Python SDK."""
 
 from a2a.utils.artifact import (
+    get_artifact_text,
     new_artifact,
     new_data_artifact,
     new_text_artifact,
@@ -18,13 +19,15 @@
     create_task_obj,
 )
 from a2a.utils.message import (
-    get_data_parts,
-    get_file_parts,
     get_message_text,
-    get_text_parts,
     new_agent_parts_message,
     new_agent_text_message,
 )
+from a2a.utils.parts import (
+    get_data_parts,
+    get_file_parts,
+    get_text_parts,
+)
 from a2a.utils.task import (
     completed_task,
     new_task,
@@ -41,6 +44,7 @@
     'build_text_artifact',
     'completed_task',
     'create_task_obj',
+    'get_artifact_text',
     'get_data_parts',
     'get_file_parts',
     'get_message_text',

src/a2a/utils/artifact.py

@@ -5,10 +5,13 @@
 from typing import Any
 
 from a2a.types import Artifact, DataPart, Part, TextPart
+from a2a.utils.parts import get_text_parts
 
 
 def new_artifact(
-    parts: list[Part], name: str, description: str = ''
+    parts: list[Part],
+    name: str,
+    description: str | None = None,
 ) -> Artifact:
     """Creates a new Artifact object.
 
@@ -31,7 +34,7 @@ def new_artifact(
 def new_text_artifact(
     name: str,
     text: str,
-    description: str = '',
+    description: str | None = None,
 ) -> Artifact:
     """Creates a new Artifact object containing only a single TextPart.
 
@@ -53,7 +56,7 @@ def new_text_artifact(
 def new_data_artifact(
     name: str,
     data: dict[str, Any],
-    description: str = '',
+    description: str | None = None,
 ) -> Artifact:
     """Creates a new Artifact object containing only a single DataPart.
 
@@ -70,3 +73,16 @@ def new_data_artifact(
         name,
         description,
     )
+
+
+def get_artifact_text(artifact: Artifact, delimiter: str = '\n') -> str:
+    """Extracts and joins all text content from an Artifact's parts.
+
+    Args:
+        artifact: The `Artifact` object.
+        delimiter: The string to use when joining text from multiple TextParts.
+
+    Returns:
+        A single string containing all text content, or an empty string if no text parts are found.
+    """
+    return delimiter.join(get_text_parts(artifact.parts))

src/a2a/utils/message.py

@@ -2,18 +2,13 @@
 
 import uuid
 
-from typing import Any
-
 from a2a.types import (
-    DataPart,
-    FilePart,
-    FileWithBytes,
-    FileWithUri,
     Message,
     Part,
     Role,
     TextPart,
 )
+from a2a.utils.parts import get_text_parts
 
 
 def new_agent_text_message(
@@ -64,42 +59,6 @@ def new_agent_parts_message(
     )
 
 
-def get_text_parts(parts: list[Part]) -> list[str]:
-    """Extracts text content from all TextPart objects in a list of Parts.
-
-    Args:
-        parts: A list of `Part` objects.
-
-    Returns:
-        A list of strings containing the text content from any `TextPart` objects found.
-    """
-    return [part.root.text for part in parts if isinstance(part.root, TextPart)]
-
-
-def get_data_parts(parts: list[Part]) -> list[dict[str, Any]]:
-    """Extracts dictionary data from all DataPart objects in a list of Parts.
-
-    Args:
-        parts: A list of `Part` objects.
-
-    Returns:
-        A list of dictionaries containing the data from any `DataPart` objects found.
-    """
-    return [part.root.data for part in parts if isinstance(part.root, DataPart)]
-
-
-def get_file_parts(parts: list[Part]) -> list[FileWithBytes | FileWithUri]:
-    """Extracts file data from all FilePart objects in a list of Parts.
-
-    Args:
-        parts: A list of `Part` objects.
-
-    Returns:
-        A list of `FileWithBytes` or `FileWithUri` objects containing the file data from any `FilePart` objects found.
-    """
-    return [part.root.file for part in parts if isinstance(part.root, FilePart)]
-
-
 def get_message_text(message: Message, delimiter: str = '\n') -> str:
     """Extracts and joins all text content from a Message's parts.
 

src/a2a/utils/parts.py

@@ -0,0 +1,48 @@
+"""Utility functions for creating and handling A2A Parts objects."""
+
+from typing import Any
+
+from a2a.types import (
+    DataPart,
+    FilePart,
+    FileWithBytes,
+    FileWithUri,
+    Part,
+    TextPart,
+)
+
+
+def get_text_parts(parts: list[Part]) -> list[str]:
+    """Extracts text content from all TextPart objects in a list of Parts.
+
+    Args:
+        parts: A list of `Part` objects.
+
+    Returns:
+        A list of strings containing the text content from any `TextPart` objects found.
+    """
+    return [part.root.text for part in parts if isinstance(part.root, TextPart)]
+
+
+def get_data_parts(parts: list[Part]) -> list[dict[str, Any]]:
+    """Extracts dictionary data from all DataPart objects in a list of Parts.
+
+    Args:
+        parts: A list of `Part` objects.
+
+    Returns:
+        A list of dictionaries containing the data from any `DataPart` objects found.
+    """
+    return [part.root.data for part in parts if isinstance(part.root, DataPart)]
+
+
+def get_file_parts(parts: list[Part]) -> list[FileWithBytes | FileWithUri]:
+    """Extracts file data from all FilePart objects in a list of Parts.
+
+    Args:
+        parts: A list of `Part` objects.
+
+    Returns:
+        A list of `FileWithBytes` or `FileWithUri` objects containing the file data from any `FilePart` objects found.
+    """
+    return [part.root.file for part in parts if isinstance(part.root, FilePart)]

tests/README.md

@@ -2,7 +2,7 @@
 
 1. Run the tests
     ```bash
-    uv run pytest -v -s client/test_client.py
+    uv run pytest -v -s client/test_client_factory.py
     ```
 
 In case of failures, you can cleanup the cache: