diff --git a/agent_sdks/python/README.md b/agent_sdks/python/README.md index 63309fe81..6eb2ff538 100644 --- a/agent_sdks/python/README.md +++ b/agent_sdks/python/README.md @@ -75,6 +75,24 @@ directory: uv run pyink . ``` +## Releasing the SDK + +To release a new version of the SDK, follow these steps: + +1. Update the version in `src/a2ui/version.py`. +2. Run the release script from the `agent_sdks/python` directory: + + ```bash + ./release.sh + ``` + +The script will build the package, upload it to the Artifact Registry, and trigger the release pipeline. + +## Tracking the release + +1. Check the release status through Sponge: go/spng2?q=PROJECT%3Aa2ui%2Fpypi +2. You will see logs for the jobs involved in your release. A successful release is confirmed when the final job, named "publisher", completes successfully. + ## Disclaimer Important: The sample code provided is for demonstration purposes and diff --git a/agent_sdks/python/release.sh b/agent_sdks/python/release.sh new file mode 100755 index 000000000..8482ebcd8 --- /dev/null +++ b/agent_sdks/python/release.sh @@ -0,0 +1,66 @@ +#!/bin/bash +# Copyright 2026 Google LLC +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + + +set -e # Exit on error +#set -x # Echo commands + +PACKAGE_NAME="a2ui-agent-sdk" +REPOSITORY="a2ui--pypi" +PROJECT="oss-exit-gate-prod" +LOCATION="us" +REPOSITORY_URL="https://us-python.pkg.dev/${PROJECT}/${REPOSITORY}" +GCS_URI="gs://oss-exit-gate-prod-projects-bucket/a2ui/pypi/manifests" + +echo "--- Installing helper packages ---" +uv tool install twine --with keyrings.google-artifactregistry-auth --with keyring +uv tool install keyring --with keyrings.google-artifactregistry-auth + +echo "--- Building the package ---" +rm -rf dist +uv build + +echo "--- Uploading the package ---" +twine --version +twine check dist/* + +version=$(uv run python -c "import a2ui; print(a2ui.__version__)") + +# Authenticate with Google Cloud +if ! gcloud auth application-default print-access-token --quiet > /dev/null; then + gcloud auth application-default login +fi + +# Check if the version already exists in the staging repository +if gcloud artifacts versions describe "$version" --package=$PACKAGE_NAME --repository=$REPOSITORY --location=$LOCATION --project=$PROJECT > /dev/null 2>&1; then + echo "Version $version already exists in Artifact Registry. Skip the release." + echo "Hint: If you intended to release a new version, please update 'src/a2ui/version.py'." + exit 0 +fi + +twine upload --repository-url $REPOSITORY_URL dist/* +echo "Version $version uploaded to Artifact Registry." + +echo "--- Creating manifest.json ---" +MANIFEST_FILE="manifest.json" +echo '{ "publish_all": true }' > $MANIFEST_FILE + +echo "--- Uploading manifest to GCS to trigger OSS Exit Gate ---" +MANIFEST_NAME="manifest-${version}-$(date +%Y%m%d%H%M%S).json" +gcloud storage cp $MANIFEST_FILE "${GCS_URI}/${MANIFEST_NAME}" +rm -rf $MANIFEST_FILE + +echo "Manifest ${MANIFEST_NAME} uploaded." +echo "--- Build script finished ---" diff --git a/agent_sdks/python/src/a2ui/version.py b/agent_sdks/python/src/a2ui/version.py index c915fb471..0e6380f40 100644 --- a/agent_sdks/python/src/a2ui/version.py +++ b/agent_sdks/python/src/a2ui/version.py @@ -21,4 +21,4 @@ # Switch to VCS-based versioning which derives version # from Git tags automatically with the hatch-vcs plugin # when we start using Git tags. -__version__ = "0.1.2" +__version__ = "0.2.2" diff --git a/samples/agent/adk/orchestrator/agent.py b/samples/agent/adk/orchestrator/agent.py index 483eafecb..455c740b8 100644 --- a/samples/agent/adk/orchestrator/agent.py +++ b/samples/agent/adk/orchestrator/agent.py @@ -98,12 +98,12 @@ async def intercept( SubagentRouteManager.get_route_to_subagent_name(sid, context.state) for sid in surface_ids_to_check ]) - + filtered_surfaces = {} - for i, surface_id in enumerate(surface_ids_to_check): + for i, surface_id in enumerate(surface_ids_to_check): if owner_agents[i] == agent_card.name: filtered_surfaces[surface_id] = current_surfaces[surface_id] - + message["metadata"]["a2uiClientDataModel"]["surfaces"] = filtered_surfaces logger.info( f"Stripped data model for {agent_card.name}. " @@ -111,7 +111,9 @@ async def intercept( ) else: message["metadata"]["a2uiClientDataModel"]["surfaces"] = {} - logger.warning("No agent card or name provided. Stripped all surfaces from data model.") + logger.warning( + "No agent card or name provided. Stripped all surfaces from data model." + ) return request_payload, http_kwargs