From 354991f842f743f3d8ad5cd6505d190e2af518fc Mon Sep 17 00:00:00 2001
From: Jerome Romualdez <jerome.romualdez@scale.com>
Date: Mon, 29 Jun 2026 12:57:28 -0400
Subject: [PATCH] fix(packaging): publish only the tagged pypi package

Infer the package to upload from component release tags so a client release does not attempt to republish an immutable SDK wheel, and keep explicit package selection for manual dispatches.

Co-authored-by: Cursor <cursoragent@cursor.com>
---
 .github/workflows/publish-pypi.yml | 12 +++++++
 bin/publish-pypi                   | 50 +++++++++++++++++++++++++++---
 2 files changed, 57 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/publish-pypi.yml b/.github/workflows/publish-pypi.yml
index 8c6f3ad8a..b5ff5ca9b 100644
--- a/.github/workflows/publish-pypi.yml
+++ b/.github/workflows/publish-pypi.yml
@@ -4,6 +4,16 @@
 name: Publish PyPI
 on:
   workflow_dispatch:
+    inputs:
+      package:
+        description: Package to publish
+        required: true
+        default: all
+        type: choice
+        options:
+          - all
+          - agentex-client
+          - agentex-sdk
 
   release:
     types: [published]
@@ -33,3 +43,5 @@ jobs:
           # Back-compat fallback — used by bin/publish-pypi when the
           # dedicated tokens above are unset.
           PYPI_TOKEN: ${{ secrets.AGENTEX_PYPI_TOKEN || secrets.PYPI_TOKEN }}
+          # Manual dispatches can override tag-derived package selection.
+          PYPI_PACKAGE: ${{ inputs.package }}
diff --git a/bin/publish-pypi b/bin/publish-pypi
index 68d7df303..7107d194e 100644
--- a/bin/publish-pypi
+++ b/bin/publish-pypi
@@ -1,7 +1,8 @@
 #!/usr/bin/env bash
 
-# Publish slim (root) before heavy (adk/): heavy pins slim, so a slim-first
-# failure aborts before shipping a heavy that needs an unreleased client.
+# Publish only the package requested by the component tag. Manual dispatches can
+# set PYPI_PACKAGE=all to publish both packages; in that case publish slim
+# before heavy because the heavy package depends on the slim package.
 
 set -eux
 
@@ -9,6 +10,45 @@ rm -rf dist
 # --wheel: the heavy's cross-dir force-include can't build via sdist.
 uv build --all-packages --wheel
 
-# --check-url makes the per-component-tag double-trigger idempotent.
-uv publish --check-url https://pypi.org/simple/ --token="${AGENTEX_CLIENT_PYPI_TOKEN:-$PYPI_TOKEN}" dist/agentex_client-*.whl
-uv publish --check-url https://pypi.org/simple/ --token="${AGENTEX_PYPI_TOKEN:-$PYPI_TOKEN}" dist/agentex_sdk-*.whl
+publish_client() {
+  uv publish --check-url https://pypi.org/simple/ --token="${AGENTEX_CLIENT_PYPI_TOKEN:-${PYPI_TOKEN:-}}" dist/agentex_client-*.whl
+}
+
+publish_sdk() {
+  uv publish --check-url https://pypi.org/simple/ --token="${AGENTEX_PYPI_TOKEN:-${PYPI_TOKEN:-}}" dist/agentex_sdk-*.whl
+}
+
+package="${PYPI_PACKAGE:-}"
+
+if [ -z "$package" ]; then
+  tag_name="${GITHUB_REF_NAME:-}"
+  if [ -z "$tag_name" ] && [[ "${GITHUB_REF:-}" == refs/tags/* ]]; then
+    tag_name="${GITHUB_REF#refs/tags/}"
+  fi
+
+  case "$tag_name" in
+    agentex-client-v*) package="agentex-client" ;;
+    agentex-sdk-v*) package="agentex-sdk" ;;
+    *)
+      echo "Unable to infer package from tag '$tag_name'. Set PYPI_PACKAGE to one of: all, agentex-client, agentex-sdk." >&2
+      exit 1
+      ;;
+  esac
+fi
+
+case "$package" in
+  all)
+    publish_client
+    publish_sdk
+    ;;
+  agentex-client)
+    publish_client
+    ;;
+  agentex-sdk)
+    publish_sdk
+    ;;
+  *)
+    echo "Unknown PYPI_PACKAGE '$package'. Expected one of: all, agentex-client, agentex-sdk." >&2
+    exit 1
+    ;;
+esac