airbyte-internal-ops 0.9.1__py3-none-any.whl → 0.10.0__py3-none-any.whl

{airbyte_internal_ops-0.9.1.dist-info → airbyte_internal_ops-0.10.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airbyte-internal-ops
-Version: 0.9.1
+Version: 0.10.0
 Summary: MCP and API interfaces that let the agents do the admin work
 Author-email: Aaron Steers <aj@airbyte.io>
 Keywords: admin,airbyte,api,mcp

{airbyte_internal_ops-0.9.1.dist-info → airbyte_internal_ops-0.10.0.dist-info}/RECORD

@@ -220,6 +220,7 @@ airbyte_ops_mcp/airbyte_repo/__init__.py,sha256=3yEpCOop1h33UuCtU7-1UB3kun2Qsxtg
 airbyte_ops_mcp/airbyte_repo/bump_version.py,sha256=rtXYv105BhElj7twQmy20VjNsj6vCb0Ulrzmc4Z-1cE,15384
 airbyte_ops_mcp/airbyte_repo/changelog_fix.py,sha256=sT94FkH5YuPc_NZCoMr9IUydcD4CO-i5ifGm8FOuI9g,13890
 airbyte_ops_mcp/airbyte_repo/list_connectors.py,sha256=UI4RKMfFWOXHBtvh7TIw1v0BnV4rQQGuUgh8e9yvpos,19907
+airbyte_ops_mcp/airbyte_repo/progressive_rollout.py,sha256=axBXWDCxfho5hOHIjbnQKWQNrDJR4kKuOdVQwJoqTTs,9459
 airbyte_ops_mcp/airbyte_repo/utils.py,sha256=TXlOAfhiu_hVRNjCxB4PRPVDhTWCU5lYmgqz4QG_-EA,3201
 airbyte_ops_mcp/cli/__init__.py,sha256=XpL7FyVfgabfBF2JR7u7NwJ2krlYqjd_OwLcWf-Xc7s,114
 airbyte_ops_mcp/cli/_base.py,sha256=I8tWnyQf0ks4r3J8N8h-5GZxyn37T-55KsbuHnxYlcg,415
@@ -227,13 +228,13 @@ airbyte_ops_mcp/cli/_shared.py,sha256=jg-xMyGzTCGPqKd8VTfE_3kGPIyO_3Kx5sQbG4rPc0
 airbyte_ops_mcp/cli/app.py,sha256=xxlyJzlFvZuy-qKAhTPPmzj-t0o7MSqQ7aPHonfT1vU,790
 airbyte_ops_mcp/cli/cloud.py,sha256=jac0FcG3UvNNeNNhqZR04uY9BCyVvins1kjO0LRZ86Y,45861
 airbyte_ops_mcp/cli/gh.py,sha256=koJPu0MDB6AW7mJq2z4dZV65ofvsZTkqoeitGF8KJR8,5364
-airbyte_ops_mcp/cli/local.py,sha256=kJe3rO6AWdXyjvrlcyP5MC_wIGu9cm1DaCNuO2gxkhU,40806
+airbyte_ops_mcp/cli/local.py,sha256=amxtJ7aY1nrOs36zuWkV7jZVMyHNLT7HDUvyyOXF8mM,44459
 airbyte_ops_mcp/cli/registry.py,sha256=CVm15_6YWpLD99DZ8f8y68npXJw01Ckde7Ty3o89YNE,16631
 airbyte_ops_mcp/cloud_admin/__init__.py,sha256=cqE96Q10Kp6elhH9DAi6TVsIwSUy3sooDLLrxTaktGk,816
-airbyte_ops_mcp/cloud_admin/api_client.py,sha256=pzdnZpIxOaXJsF_vQuw9_goiDbqwgEQxFwZwsoupY40,57459
+airbyte_ops_mcp/cloud_admin/api_client.py,sha256=2DNYx-GXOTCuP9-rEkZFTM8MIR3MDnXSaqs_z1DsTwQ,70361
 airbyte_ops_mcp/cloud_admin/auth.py,sha256=qE2Aqe0qbZB755KscL65s54Jz78-F-X5a8fXKsrYEOQ,3749
 airbyte_ops_mcp/cloud_admin/connection_config.py,sha256=9opGQer-cGMJANmm-LFLMwvMCNu3nzxa2n2XHkZj9Fw,4899
-airbyte_ops_mcp/cloud_admin/models.py,sha256=yldPAL9Lce83U3JN-f6qPc2X5LPnLCYqgnXrM1DRq98,4799
+airbyte_ops_mcp/cloud_admin/models.py,sha256=Wn6O48Yp2QB3jsknmw-mG-LJ6GdBIyAg06MGLC1kOjM,8299
 airbyte_ops_mcp/connection_config_retriever/__init__.py,sha256=Xoi-YvARrNPhECdpwEDDkdwEpnvj8zuUlwULpf4iRrU,800
 airbyte_ops_mcp/connection_config_retriever/audit_logging.py,sha256=QdOG9984NXeMaKeJnFUZ4oCOmqi37PBRG2NRBBjrZQQ,2753
 airbyte_ops_mcp/connection_config_retriever/retrieval.py,sha256=s6yeCyrboWkUd6KdaheEo87x-rLtQNTL8XeR8O9z2HI,12160
@@ -268,21 +269,22 @@ airbyte_ops_mcp/mcp/_guidance.py,sha256=n0dLggQXNi2PKZ0eFDqQx5YIjhQpMnI3xcjouub9
 airbyte_ops_mcp/mcp/cloud_connector_versions.py,sha256=dRKrgsRGadQieTSvBrc5HYI6TTDzVAT6-4tiOHNcDHU,36244
 airbyte_ops_mcp/mcp/connector_analysis.py,sha256=OC4KrOSkMkKPkOisWnSv96BDDE5TQYHq-Jxa2vtjJpo,298
 airbyte_ops_mcp/mcp/connector_qa.py,sha256=aImpqdnqBPDrz10BS0owsV4kuIU2XdalzgbaGZsbOL0,258
+airbyte_ops_mcp/mcp/connector_rollout.py,sha256=KYQhRCcpJCquhX_aGNjhT9WKT5lqDpNKNA1Wki-7zPU,29020
 airbyte_ops_mcp/mcp/gcp_logs.py,sha256=QCDQHmsxQHJ26BB0sxkBgKXr7Ja9wVFkdpY6423H-xo,2677
 airbyte_ops_mcp/mcp/github_actions.py,sha256=G0NmjNVWpVtLXdQnhX7qJwPFkfEX5gBXf55xt0FpJJ0,11752
 airbyte_ops_mcp/mcp/github_repo_ops.py,sha256=YHeuN7Xc_L3xkJ-F3l1t1TIPW2j2CjecBDbb0DUvZO8,5306
 airbyte_ops_mcp/mcp/metadata.py,sha256=fwGW97WknR5lfKcQnFtK6dU87aA6TmLj1NkKyqDAV9g,270
 airbyte_ops_mcp/mcp/prerelease.py,sha256=jSrAwk95vZLfwYFn5Menb-ziAMTUZnzAMHIJAitd9x8,9553
-airbyte_ops_mcp/mcp/prod_db_queries.py,sha256=EdZyrccrM5tHKEtjks4598hzbZ62FuDCCHTjjFT4Wt0,47545
+airbyte_ops_mcp/mcp/prod_db_queries.py,sha256=iesP8H2hCmkdjsmLgYOrwqNQeQwbcScQc4tr4cfs8es,56334
 airbyte_ops_mcp/mcp/prompts.py,sha256=v4bguskw7hSsISkseACzKQm5QwrIXmiwbs27oclXTE8,1591
 airbyte_ops_mcp/mcp/registry.py,sha256=PW-VYUj42qx2pQ_apUkVaoUFq7VgB9zEU7-aGrkSCCw,290
 airbyte_ops_mcp/mcp/regression_tests.py,sha256=zwdQ-ymUhWtVcIjwiNIZAC151GKhuxi55HDi4S91RnI,17046
-airbyte_ops_mcp/mcp/server.py,sha256=ID7wZIds8e-qcGAhsaK0vTJDQQ4YNSdYKTQGQWL3Evw,7280
+airbyte_ops_mcp/mcp/server.py,sha256=7VO-awh4gh51EiYCkUvEyQd4r1gl6XRJ4-I-JybkEiA,7405
 airbyte_ops_mcp/prod_db_access/__init__.py,sha256=5pxouMPY1beyWlB0UwPnbaLTKTHqU6X82rbbgKY2vYU,1069
 airbyte_ops_mcp/prod_db_access/db_engine.py,sha256=VUqEWZtharJUR-Cri_pMwtGh1C4Neu4s195mbEXlm-w,9190
 airbyte_ops_mcp/prod_db_access/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-airbyte_ops_mcp/prod_db_access/queries.py,sha256=PyZ0HU-E6jDhSgQ94XRz4wzh61vCr0ACxfMoGjeTKUU,22312
-airbyte_ops_mcp/prod_db_access/sql.py,sha256=E7HQratZC5LbCXdvpQzZRd09SmCVnY_4Ru-HluWaQ_c,38308
+airbyte_ops_mcp/prod_db_access/queries.py,sha256=OMnIejwVoQSaTV1It1btzui6Zmrvihh3d6KNZgyyE7I,27741
+airbyte_ops_mcp/prod_db_access/sql.py,sha256=J89eGlrG_0NPtpAuVVKTBS9tG3LKGj0DK-zTGUEzvFA,49526
 airbyte_ops_mcp/registry/__init__.py,sha256=iEaPlt9GrnlaLbc__98TguNeZG8wuQu7S-_2QkhHcbA,858
 airbyte_ops_mcp/registry/_gcs_util.py,sha256=3dL7iqpyoSPfl2nGBPp73TqrqGWI4Ks6egKpQXDnw24,2838
 airbyte_ops_mcp/registry/connector_stubs.py,sha256=pBzaQIOgCfBvHuy3_N45amU9Do2ALDpU8q5HnZySPTQ,5807
@@ -308,7 +310,7 @@ airbyte_ops_mcp/regression_tests/regression/comparators.py,sha256=MJkLZEKHivgrG0
 airbyte_ops_mcp/regression_tests/validation/__init__.py,sha256=MBEwGOoNuqT4_oCahtoK62OKWIjUCfWa7vZTxNj_0Ek,1532
 airbyte_ops_mcp/regression_tests/validation/catalog_validators.py,sha256=jqqVAMOk0mtdPgwu4d0hA0ZEjtsNh5gapvGydRv3_qk,12553
 airbyte_ops_mcp/regression_tests/validation/record_validators.py,sha256=RjauAhKWNwxMBTu0eNS2hMFNQVs5CLbQU51kp6FOVDk,7432
-airbyte_internal_ops-0.9.1.dist-info/METADATA,sha256=_EJMWK1bcz3g2o27Zo1c3uHqCDsgwEvzI7QD9hdm1tY,6001
-airbyte_internal_ops-0.9.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-airbyte_internal_ops-0.9.1.dist-info/entry_points.txt,sha256=WxP0l7bRFss4Cr5uQqVj9mTEKwnRKouNuphXQF0lotA,171
-airbyte_internal_ops-0.9.1.dist-info/RECORD,,
+airbyte_internal_ops-0.10.0.dist-info/METADATA,sha256=eP4lUCpuijAH7aJIkDW4n3mZOIoMqq8nvkF3IU5EqyM,6002
+airbyte_internal_ops-0.10.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+airbyte_internal_ops-0.10.0.dist-info/entry_points.txt,sha256=WxP0l7bRFss4Cr5uQqVj9mTEKwnRKouNuphXQF0lotA,171
+airbyte_internal_ops-0.10.0.dist-info/RECORD,,

airbyte_ops_mcp/airbyte_repo/progressive_rollout.py

@@ -0,0 +1,279 @@
+# Copyright (c) 2025 Airbyte, Inc., all rights reserved.
+"""Progressive rollout setup utilities for Airbyte connectors.
+
+This module provides pure Python functionality to configure a connector's
+metadata.yaml for progressive rollout of a release candidate (RC) version.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from pathlib import Path
+
+import semver
+import yaml
+
+from airbyte_ops_mcp.airbyte_repo.bump_version import (
+    InvalidVersionError,
+    get_connector_path,
+    get_current_version,
+)
+from airbyte_ops_mcp.airbyte_repo.list_connectors import METADATA_FILE_NAME
+
+
+@dataclass
+class ProgressiveRolloutSetupResult:
+    """Result of a progressive rollout setup operation."""
+
+    connector: str
+    previous_version: str
+    rc_version: str
+    pin_version: str
+    files_modified: list[str]
+    dry_run: bool
+
+
+@dataclass
+class ProgressiveRolloutCleanupResult:
+    """Result of a progressive rollout cleanup operation."""
+
+    connector: str
+    previous_version: str
+    new_version: str
+    files_modified: list[str]
+    dry_run: bool
+    changes_made: list[str]
+
+
+def calculate_rc_version(
+    current_version: str,
+    rc_version: str | None = None,
+) -> str:
+    """Calculate the RC version for progressive rollout.
+
+    If rc_version is provided, validates and returns it.
+    Otherwise, bumps the minor version and adds -rc.1 suffix.
+
+    Args:
+        current_version: Current version string (e.g., "0.0.46")
+        rc_version: Explicit RC version (e.g., "0.1.0-rc.1")
+
+    Returns:
+        RC version string
+
+    Raises:
+        InvalidVersionError: If version format is invalid
+    """
+    if rc_version is not None:
+        if not rc_version.endswith("-rc.1") and "-rc." not in rc_version:
+            raise InvalidVersionError(
+                f"RC version must contain '-rc.' suffix, got: {rc_version}"
+            )
+        return rc_version
+
+    try:
+        version = semver.Version.parse(current_version)
+    except ValueError as e:
+        raise InvalidVersionError(
+            f"Cannot parse current version: {current_version}"
+        ) from e
+
+    bumped = version.bump_minor()
+    return f"{bumped}-rc.1"
+
+
+def setup_progressive_rollout(
+    repo_path: str | Path,
+    connector_name: str,
+    rc_version: str | None = None,
+    pin_version: str | None = None,
+    dry_run: bool = False,
+) -> ProgressiveRolloutSetupResult:
+    """Configure a connector's metadata.yaml for progressive rollout.
+
+    This function modifies metadata.yaml to:
+    1. Set dockerImageTag to the RC version
+    2. Add registryOverrides.cloud.dockerImageTag to pin Cloud users to current version
+    3. Add registryOverrides.oss.dockerImageTag to pin OSS users to current version
+    4. Add releases.rolloutConfiguration.enableProgressiveRollout: true
+
+    Args:
+        repo_path: Path to the Airbyte monorepo
+        connector_name: Technical name of the connector (e.g., "source-github")
+        rc_version: Explicit RC version (default: bumps minor and adds -rc.1)
+        pin_version: Version to pin Cloud/OSS users to (default: current version)
+        dry_run: If True, don't actually modify files
+
+    Returns:
+        ProgressiveRolloutSetupResult with details of the operation
+
+    Raises:
+        ConnectorNotFoundError: If connector doesn't exist
+        VersionNotFoundError: If current version cannot be found
+        InvalidVersionError: If version format is invalid
+    """
+    repo_path = Path(repo_path)
+    connector_path = get_connector_path(repo_path, connector_name)
+    metadata_file = connector_path / METADATA_FILE_NAME
+
+    current_version = get_current_version(connector_path)
+    calculated_rc_version = calculate_rc_version(current_version, rc_version)
+    calculated_pin_version = pin_version if pin_version else current_version
+
+    with open(metadata_file) as f:
+        metadata = yaml.safe_load(f)
+
+    data = metadata.get("data", {})
+
+    data["dockerImageTag"] = calculated_rc_version
+
+    if "registryOverrides" not in data:
+        data["registryOverrides"] = {}
+
+    if "cloud" not in data["registryOverrides"]:
+        data["registryOverrides"]["cloud"] = {}
+    data["registryOverrides"]["cloud"]["dockerImageTag"] = calculated_pin_version
+
+    if "oss" not in data["registryOverrides"]:
+        data["registryOverrides"]["oss"] = {}
+    data["registryOverrides"]["oss"]["dockerImageTag"] = calculated_pin_version
+
+    if "releases" not in data:
+        data["releases"] = {}
+    if "rolloutConfiguration" not in data["releases"]:
+        data["releases"]["rolloutConfiguration"] = {}
+    data["releases"]["rolloutConfiguration"]["enableProgressiveRollout"] = True
+
+    metadata["data"] = data
+
+    files_modified: list[str] = []
+
+    if not dry_run:
+        with open(metadata_file, "w") as f:
+            yaml.dump(metadata, f, default_flow_style=False, sort_keys=False)
+        files_modified.append(str(metadata_file.relative_to(repo_path)))
+    else:
+        files_modified.append(str(metadata_file.relative_to(repo_path)))
+
+    return ProgressiveRolloutSetupResult(
+        connector=connector_name,
+        previous_version=current_version,
+        rc_version=calculated_rc_version,
+        pin_version=calculated_pin_version,
+        files_modified=files_modified,
+        dry_run=dry_run,
+    )
+
+
+def extract_ga_version(rc_version: str) -> str:
+    """Extract the GA version from an RC version string.
+
+    Args:
+        rc_version: RC version string (e.g., "1.0.0-rc.1")
+
+    Returns:
+        GA version string (e.g., "1.0.0")
+
+    Raises:
+        InvalidVersionError: If version format is invalid
+    """
+    if "-rc." in rc_version:
+        return rc_version.split("-rc.")[0]
+    if "-preview." in rc_version:
+        return rc_version.split("-preview.")[0]
+    return rc_version
+
+
+def cleanup_progressive_rollout(
+    repo_path: str | Path,
+    connector_name: str,
+    ga_version: str | None = None,
+    dry_run: bool = False,
+) -> ProgressiveRolloutCleanupResult:
+    """Clean up a connector's metadata.yaml after progressive rollout completion.
+
+    This function reverses the changes made by setup_progressive_rollout:
+    1. Updates dockerImageTag to the GA version (removes -rc.X suffix)
+    2. Removes registryOverrides.cloud.dockerImageTag (if it was a pin)
+    3. Removes registryOverrides.oss.dockerImageTag (if it was a pin)
+    4. Sets releases.rolloutConfiguration.enableProgressiveRollout to false
+
+    Args:
+        repo_path: Path to the Airbyte monorepo
+        connector_name: Technical name of the connector (e.g., "source-github")
+        ga_version: Explicit GA version (default: extracts from current RC version)
+        dry_run: If True, don't actually modify files
+
+    Returns:
+        ProgressiveRolloutCleanupResult with details of the operation
+
+    Raises:
+        ConnectorNotFoundError: If connector doesn't exist
+        VersionNotFoundError: If current version cannot be found
+        InvalidVersionError: If version format is invalid
+    """
+    repo_path = Path(repo_path)
+    connector_path = get_connector_path(repo_path, connector_name)
+    metadata_file = connector_path / METADATA_FILE_NAME
+
+    current_version = get_current_version(connector_path)
+
+    new_version = ga_version or extract_ga_version(current_version)
+
+    with open(metadata_file) as f:
+        metadata = yaml.safe_load(f)
+
+    data = metadata.get("data", {})
+    changes_made: list[str] = []
+
+    if data.get("dockerImageTag") != new_version:
+        data["dockerImageTag"] = new_version
+        changes_made.append(f"Updated dockerImageTag to {new_version}")
+
+    registry_overrides = data.get("registryOverrides", {})
+    cloud_overrides = registry_overrides.get("cloud", {})
+    oss_overrides = registry_overrides.get("oss", {})
+
+    if "dockerImageTag" in cloud_overrides:
+        del cloud_overrides["dockerImageTag"]
+        changes_made.append("Removed registryOverrides.cloud.dockerImageTag")
+        if not cloud_overrides:
+            del registry_overrides["cloud"]
+            changes_made.append("Removed empty registryOverrides.cloud section")
+
+    if "dockerImageTag" in oss_overrides:
+        del oss_overrides["dockerImageTag"]
+        changes_made.append("Removed registryOverrides.oss.dockerImageTag")
+        if not oss_overrides:
+            del registry_overrides["oss"]
+            changes_made.append("Removed empty registryOverrides.oss section")
+
+    if not registry_overrides and "registryOverrides" in data:
+        del data["registryOverrides"]
+        changes_made.append("Removed empty registryOverrides section")
+
+    releases = data.get("releases", {})
+    rollout_config = releases.get("rolloutConfiguration", {})
+    if rollout_config.get("enableProgressiveRollout"):
+        rollout_config["enableProgressiveRollout"] = False
+        changes_made.append("Set enableProgressiveRollout to false")
+
+    metadata["data"] = data
+
+    files_modified: list[str] = []
+
+    if not dry_run:
+        with open(metadata_file, "w") as f:
+            yaml.dump(metadata, f, default_flow_style=False, sort_keys=False)
+        files_modified.append(str(metadata_file.relative_to(repo_path)))
+    else:
+        files_modified.append(str(metadata_file.relative_to(repo_path)))
+
+    return ProgressiveRolloutCleanupResult(
+        connector=connector_name,
+        previous_version=current_version,
+        new_version=new_version,
+        files_modified=files_modified,
+        dry_run=dry_run,
+        changes_made=changes_made,
+    )

airbyte_ops_mcp/cli/local.py

@@ -44,6 +44,10 @@ from airbyte_ops_mcp.airbyte_repo.list_connectors import (
     _detect_connector_language,
     get_connectors_with_local_cdk,
 )
+from airbyte_ops_mcp.airbyte_repo.progressive_rollout import (
+    cleanup_progressive_rollout,
+    setup_progressive_rollout,
+)
 from airbyte_ops_mcp.cli._base import app
 from airbyte_ops_mcp.cli._shared import error_console, exit_with_error, print_json
 from airbyte_ops_mcp.connector_ops.utils import Connector
@@ -1187,3 +1191,117 @@ def enterprise_stub_sync(
     print_json(
         synced_stubs if all_connectors else synced_stubs[0] if synced_stubs else {}
     )
+
+
+@connector_app.command(name="setup-progressive-rollout")
+def setup_progressive_rollout_cmd(
+    name: Annotated[
+        str,
+        Parameter(help="Connector technical name (e.g., source-github)."),
+    ],
+    repo_path: Annotated[
+        str,
+        Parameter(help="Absolute path to the Airbyte monorepo."),
+    ],
+    rc_version: Annotated[
+        str | None,
+        Parameter(
+            help="Explicit RC version (e.g., 1.0.0-rc.1). Default: bumps minor and adds -rc.1."
+        ),
+    ] = None,
+    pin_version: Annotated[
+        str | None,
+        Parameter(help="Version to pin Cloud/OSS users to. Default: current version."),
+    ] = None,
+    dry_run: Annotated[
+        bool,
+        Parameter(help="Show what would be changed without modifying files."),
+    ] = False,
+) -> None:
+    """Configure a connector's metadata.yaml for progressive rollout.
+
+    Sets up the connector for progressive rollout by:
+    - Setting dockerImageTag to the RC version
+    - Adding registryOverrides to pin Cloud/OSS users to current stable version
+    - Enabling releases.rolloutConfiguration.enableProgressiveRollout
+    """
+    result = setup_progressive_rollout(
+        repo_path=repo_path,
+        connector_name=name,
+        rc_version=rc_version,
+        pin_version=pin_version,
+        dry_run=dry_run,
+    )
+
+    output = {
+        "connector": result.connector,
+        "previous_version": result.previous_version,
+        "rc_version": result.rc_version,
+        "pin_version": result.pin_version,
+        "files_modified": result.files_modified,
+        "dry_run": result.dry_run,
+    }
+    print_json(output)
+
+    _write_github_step_outputs(
+        {
+            "connector": result.connector,
+            "previous_version": result.previous_version,
+            "rc_version": result.rc_version,
+        }
+    )
+
+
+@connector_app.command(name="cleanup-progressive-rollout")
+def cleanup_progressive_rollout_cmd(
+    name: Annotated[
+        str,
+        Parameter(help="Connector technical name (e.g., source-github)."),
+    ],
+    repo_path: Annotated[
+        str,
+        Parameter(help="Absolute path to the Airbyte monorepo."),
+    ],
+    ga_version: Annotated[
+        str | None,
+        Parameter(
+            help="Explicit GA version. Default: extracts from current RC version."
+        ),
+    ] = None,
+    dry_run: Annotated[
+        bool,
+        Parameter(help="Show what would be changed without modifying files."),
+    ] = False,
+) -> None:
+    """Clean up a connector's metadata.yaml after progressive rollout completion.
+
+    Reverses the changes made by setup-progressive-rollout:
+    - Updates dockerImageTag to the GA version (removes -rc.X suffix)
+    - Removes registryOverrides.cloud.dockerImageTag
+    - Removes registryOverrides.oss.dockerImageTag
+    - Sets enableProgressiveRollout to false
+    """
+    result = cleanup_progressive_rollout(
+        repo_path=repo_path,
+        connector_name=name,
+        ga_version=ga_version,
+        dry_run=dry_run,
+    )
+
+    output = {
+        "connector": result.connector,
+        "previous_version": result.previous_version,
+        "new_version": result.new_version,
+        "files_modified": result.files_modified,
+        "changes_made": result.changes_made,
+        "dry_run": result.dry_run,
+    }
+    print_json(output)
+
+    _write_github_step_outputs(
+        {
+            "connector": result.connector,
+            "previous_version": result.previous_version,
+            "new_version": result.new_version,
+        }
+    )