airbyte-internal-ops 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

airbyte_ops_mcp/registry/models.py

@@ -0,0 +1,63 @@
+# Copyright (c) 2025 Airbyte, Inc., all rights reserved.
+"""Pydantic models for registry connector publish operations.
+
+This module defines the data models used for connector publish operations
+including applying and rolling back version overrides.
+"""
+
+from __future__ import annotations
+
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+
+class ConnectorMetadata(BaseModel):
+    """Connector metadata from metadata.yaml.
+
+    This model represents the essential metadata about a connector
+    read from its metadata.yaml file in the Airbyte monorepo.
+    """
+
+    name: str = Field(description="The connector technical name")
+    docker_repository: str = Field(description="The Docker repository")
+    docker_image_tag: str = Field(description="The Docker image tag/version")
+    support_level: str | None = Field(
+        default=None, description="The support level (certified, community, etc.)"
+    )
+    definition_id: str | None = Field(
+        default=None, description="The connector definition ID"
+    )
+
+
+class ConnectorPublishResult(BaseModel):
+    """Result of a connector publish operation.
+
+    This model provides detailed information about the outcome of a
+    connector publish operation (apply or rollback version override).
+    """
+
+    connector: str = Field(description="The connector technical name")
+    version: str = Field(description="The connector version")
+    action: Literal["apply-version-override", "rollback-version-override"] = Field(
+        description="The action performed"
+    )
+    status: Literal["success", "failure", "dry-run"] = Field(
+        description="The status of the operation"
+    )
+    docker_image: str | None = Field(
+        default=None, description="The Docker image name if applicable"
+    )
+    registry_updated: bool = Field(
+        default=False, description="Whether the registry was updated"
+    )
+    message: str | None = Field(default=None, description="Additional status message")
+
+    def __str__(self) -> str:
+        """Return a string representation of the publish result."""
+        status_prefix = "dry-run" if self.status == "dry-run" else self.status
+        return f"[{status_prefix}] {self.connector}:{self.version} - {self.action}"
+
+
+# Type alias for publish actions
+PublishAction = Literal["apply-version-override", "rollback-version-override"]

airbyte_ops_mcp/registry/publish.py

@@ -0,0 +1,368 @@
+# Copyright (c) 2025 Airbyte, Inc., all rights reserved.
+"""Core logic for registry connector publish operations.
+
+This module provides the core functionality for publishing connectors
+to the Airbyte registry, including applying and rolling back version overrides.
+"""
+
+from __future__ import annotations
+
+import json
+import os
+from pathlib import Path
+
+import yaml
+from google.cloud import storage
+from google.oauth2 import service_account
+
+from airbyte_ops_mcp.registry.models import (
+    ConnectorMetadata,
+    ConnectorPublishResult,
+    PublishAction,
+)
+
+CONNECTOR_PATH_PREFIX = "airbyte-integrations/connectors"
+METADATA_FILE_NAME = "metadata.yaml"
+METADATA_FOLDER = "metadata"
+LATEST_GCS_FOLDER_NAME = "latest"
+RELEASE_CANDIDATE_GCS_FOLDER_NAME = "release_candidate"
+
+# Bucket names for metadata service
+PROD_METADATA_SERVICE_BUCKET_NAME = "prod-airbyte-cloud-connector-metadata-service"
+DEV_METADATA_SERVICE_BUCKET_NAME = "dev-airbyte-cloud-connector-metadata-service-2"
+
+# Default to dev bucket for safety - use --prod flag to target production
+DEFAULT_METADATA_SERVICE_BUCKET_NAME = DEV_METADATA_SERVICE_BUCKET_NAME
+
+
+def _get_gcs_client() -> storage.Client:
+    """Get a GCS storage client using credentials from environment."""
+    gcs_creds = os.environ.get("GCS_CREDENTIALS")
+    if not gcs_creds:
+        raise ValueError(
+            "GCS_CREDENTIALS environment variable is required for registry operations"
+        )
+    service_account_info = json.loads(gcs_creds)
+    credentials = service_account.Credentials.from_service_account_info(
+        service_account_info
+    )
+    return storage.Client(credentials=credentials)
+
+
+def _get_bucket_name(use_prod: bool = False) -> str:
+    """Get the metadata service bucket name.
+
+    Args:
+        use_prod: If True, use the production bucket. Otherwise use dev bucket.
+                  Can be overridden by METADATA_SERVICE_BUCKET_NAME env var.
+
+    Returns:
+        The bucket name to use for GCS operations.
+    """
+    # Environment variable takes precedence if set
+    env_bucket = os.environ.get("METADATA_SERVICE_BUCKET_NAME")
+    if env_bucket:
+        return env_bucket
+
+    # Otherwise use prod or dev based on flag
+    if use_prod:
+        return PROD_METADATA_SERVICE_BUCKET_NAME
+    return DEV_METADATA_SERVICE_BUCKET_NAME
+
+
+def get_connector_metadata(repo_path: Path, connector_name: str) -> ConnectorMetadata:
+    """Read connector metadata from metadata.yaml.
+
+    Args:
+        repo_path: Path to the Airbyte monorepo.
+        connector_name: The connector technical name (e.g., 'source-github').
+
+    Returns:
+        ConnectorMetadata object with the connector's metadata.
+
+    Raises:
+        FileNotFoundError: If the connector directory or metadata file doesn't exist.
+    """
+    connector_dir = repo_path / CONNECTOR_PATH_PREFIX / connector_name
+    if not connector_dir.exists():
+        raise FileNotFoundError(f"Connector directory not found: {connector_dir}")
+
+    metadata_file = connector_dir / METADATA_FILE_NAME
+    if not metadata_file.exists():
+        raise FileNotFoundError(f"Metadata file not found: {metadata_file}")
+
+    with open(metadata_file) as f:
+        metadata = yaml.safe_load(f)
+
+    data = metadata.get("data", {})
+    return ConnectorMetadata(
+        name=connector_name,
+        docker_repository=data.get("dockerRepository", f"airbyte/{connector_name}"),
+        docker_image_tag=data.get("dockerImageTag", "unknown"),
+        support_level=data.get("supportLevel"),
+        definition_id=data.get("definitionId"),
+    )
+
+
+def is_release_candidate(version: str) -> bool:
+    """Check if a version string is a release candidate.
+
+    Args:
+        version: The version string to check.
+
+    Returns:
+        True if the version is a release candidate (format: X.Y.Z-rc.N), False otherwise.
+    """
+    return "-rc." in version
+
+
+def strip_rc_suffix(version: str) -> str:
+    """Strip the release candidate suffix from a version string.
+
+    Args:
+        version: The version string (e.g., '1.2.3-rc.1').
+
+    Returns:
+        The base version without RC suffix (e.g., '1.2.3').
+        Returns the original version if no RC suffix is present.
+    """
+    if "-rc." in version:
+        return version.split("-rc.")[0]
+    return version
+
+
+def publish_connector(
+    repo_path: Path,
+    connector_name: str,
+    action: PublishAction,
+    dry_run: bool = False,
+    use_prod: bool = False,
+) -> ConnectorPublishResult:
+    """Publish a connector to the Airbyte registry.
+
+    This function handles both applying version overrides (from RC to stable)
+    and rolling back version overrides.
+
+    Args:
+        repo_path: Path to the Airbyte monorepo.
+        connector_name: The connector technical name (e.g., 'source-github').
+        action: The publish action to perform ('apply-version-override' or 'rollback-version-override').
+        dry_run: If True, show what would be published without making changes.
+        use_prod: If True, target the production GCS bucket. Otherwise use dev bucket (default).
+
+    Returns:
+        ConnectorPublishResult with the operation outcome.
+
+    Raises:
+        FileNotFoundError: If the connector or metadata file doesn't exist.
+    """
+    metadata = get_connector_metadata(repo_path, connector_name)
+
+    if action == "apply-version-override":
+        return _apply_version_override(metadata, dry_run, use_prod)
+    else:
+        return _rollback_version_override(metadata, dry_run, use_prod)
+
+
+def _apply_version_override(
+    metadata: ConnectorMetadata, dry_run: bool, use_prod: bool = False
+) -> ConnectorPublishResult:
+    """Apply a version override to promote an RC to stable.
+
+    This copies the release candidate metadata to the 'latest' path in GCS,
+    then deletes the release candidate metadata.
+
+    Requires GCS_CREDENTIALS environment variable to be set.
+    """
+    version = metadata.docker_image_tag
+    docker_repo = metadata.docker_repository
+
+    if not is_release_candidate(version):
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=version,
+            action="apply-version-override",
+            status="failure",
+            docker_image=f"{docker_repo}:{version}",
+            registry_updated=False,
+            message=f"Version '{version}' is not a release candidate. "
+            "Expected format: X.Y.Z-rc.N",
+        )
+
+    target_version = strip_rc_suffix(version)
+
+    if dry_run:
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=target_version,
+            action="apply-version-override",
+            status="dry-run",
+            docker_image=f"{docker_repo}:{target_version}",
+            registry_updated=False,
+            message=f"Would apply version override for {metadata.name}: {version} -> {target_version}",
+        )
+
+    bucket_name = _get_bucket_name(use_prod)
+    storage_client = _get_gcs_client()
+    bucket = storage_client.bucket(bucket_name)
+
+    gcp_connector_dir = f"{METADATA_FOLDER}/{docker_repo}"
+    version_path = f"{gcp_connector_dir}/{version}/{METADATA_FILE_NAME}"
+    rc_path = (
+        f"{gcp_connector_dir}/{RELEASE_CANDIDATE_GCS_FOLDER_NAME}/{METADATA_FILE_NAME}"
+    )
+    latest_path = f"{gcp_connector_dir}/{LATEST_GCS_FOLDER_NAME}/{METADATA_FILE_NAME}"
+
+    version_blob = bucket.blob(version_path)
+    rc_blob = bucket.blob(rc_path)
+    latest_blob = bucket.blob(latest_path)
+
+    if not version_blob.exists():
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=version,
+            action="apply-version-override",
+            status="failure",
+            docker_image=f"{docker_repo}:{version}",
+            registry_updated=False,
+            message=f"Version metadata file not found: {version_path}",
+        )
+
+    if not rc_blob.exists():
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=version,
+            action="apply-version-override",
+            status="failure",
+            docker_image=f"{docker_repo}:{version}",
+            registry_updated=False,
+            message=f"Release candidate metadata file not found: {rc_path}",
+        )
+
+    version_blob.reload()
+    rc_blob.reload()
+    if rc_blob.md5_hash != version_blob.md5_hash:
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=version,
+            action="apply-version-override",
+            status="failure",
+            docker_image=f"{docker_repo}:{version}",
+            registry_updated=False,
+            message=f"RC metadata hash does not match version metadata hash. "
+            f"Unsafe to promote. RC: {rc_path}, Version: {version_path}",
+        )
+
+    bucket.copy_blob(rc_blob, bucket, latest_blob.name)
+    rc_blob.delete()
+
+    return ConnectorPublishResult(
+        connector=metadata.name,
+        version=target_version,
+        action="apply-version-override",
+        status="success",
+        docker_image=f"{docker_repo}:{target_version}",
+        registry_updated=True,
+        message=f"Applied version override for {metadata.name}: {version} -> {target_version}. "
+        f"Copied RC to latest and deleted RC metadata.",
+    )
+
+
+def _rollback_version_override(
+    metadata: ConnectorMetadata, dry_run: bool, use_prod: bool = False
+) -> ConnectorPublishResult:
+    """Rollback a version override by deleting the RC metadata from GCS.
+
+    This deletes both the release candidate metadata and the versioned metadata
+    after verifying their hashes match.
+
+    Requires GCS_CREDENTIALS environment variable to be set.
+    """
+    version = metadata.docker_image_tag
+    docker_repo = metadata.docker_repository
+
+    if not is_release_candidate(version):
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=version,
+            action="rollback-version-override",
+            status="failure",
+            docker_image=f"{docker_repo}:{version}",
+            registry_updated=False,
+            message=f"Version '{version}' is not a release candidate. "
+            "Expected format: X.Y.Z-rc.N",
+        )
+
+    if dry_run:
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=version,
+            action="rollback-version-override",
+            status="dry-run",
+            docker_image=f"{docker_repo}:{version}",
+            registry_updated=False,
+            message=f"Would rollback version override for {metadata.name} (current: {version})",
+        )
+
+    bucket_name = _get_bucket_name(use_prod)
+    storage_client = _get_gcs_client()
+    bucket = storage_client.bucket(bucket_name)
+
+    gcp_connector_dir = f"{METADATA_FOLDER}/{docker_repo}"
+    version_path = f"{gcp_connector_dir}/{version}/{METADATA_FILE_NAME}"
+    rc_path = (
+        f"{gcp_connector_dir}/{RELEASE_CANDIDATE_GCS_FOLDER_NAME}/{METADATA_FILE_NAME}"
+    )
+
+    version_blob = bucket.blob(version_path)
+    rc_blob = bucket.blob(rc_path)
+
+    if not version_blob.exists():
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=version,
+            action="rollback-version-override",
+            status="failure",
+            docker_image=f"{docker_repo}:{version}",
+            registry_updated=False,
+            message=f"Version metadata file not found: {version_path}",
+        )
+
+    if not rc_blob.exists():
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=version,
+            action="rollback-version-override",
+            status="failure",
+            docker_image=f"{docker_repo}:{version}",
+            registry_updated=False,
+            message=f"Release candidate metadata file not found: {rc_path}",
+        )
+
+    version_blob.reload()
+    rc_blob.reload()
+    if rc_blob.md5_hash != version_blob.md5_hash:
+        return ConnectorPublishResult(
+            connector=metadata.name,
+            version=version,
+            action="rollback-version-override",
+            status="failure",
+            docker_image=f"{docker_repo}:{version}",
+            registry_updated=False,
+            message=f"RC metadata hash does not match version metadata hash. "
+            f"Unsafe to delete. RC: {rc_path}, Version: {version_path}",
+        )
+
+    rc_blob.delete()
+    version_blob.delete()
+
+    return ConnectorPublishResult(
+        connector=metadata.name,
+        version=version,
+        action="rollback-version-override",
+        status="success",
+        docker_image=f"{docker_repo}:{version}",
+        registry_updated=True,
+        message=f"Rolled back version override for {metadata.name}. "
+        f"Deleted RC and version metadata from GCS.",
+    )

airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/publish/__init__.py

@@ -1,3 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#

airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/publish/commands.py

@@ -1,242 +0,0 @@
-#
-# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-#
-from typing import Callable, Dict, Iterable, List
-
-import asyncclick as click
-from consts import (
-    DEFAULT_PYTHON_PACKAGE_REGISTRY_CHECK_URL,
-    DEFAULT_PYTHON_PACKAGE_REGISTRY_URL,
-    ContextState,
-)
-from pipelines import main_logger
-from pipelines.airbyte_ci.connectors.pipeline import run_connectors_pipelines
-from pipelines.airbyte_ci.connectors.publish.context import (
-    PublishConnectorContext,
-    RolloutMode,
-)
-from pipelines.airbyte_ci.connectors.publish.pipeline import (
-    reorder_contexts,
-    run_connector_promote_pipeline,
-    run_connector_publish_pipeline,
-    run_connector_rollback_pipeline,
-)
-from pipelines.cli.click_decorators import click_ci_requirements_option
-from pipelines.cli.confirm_prompt import confirm
-from pipelines.cli.dagger_pipeline_command import DaggerPipelineCommand
-from pipelines.cli.secrets import wrap_gcp_credentials_in_secret, wrap_in_secret
-from pipelines.helpers.connectors.modifed import ConnectorWithModifiedFiles
-from pipelines.helpers.utils import fail_if_missing_docker_hub_creds
-from pipelines.models.secrets import Secret
-
-ROLLOUT_MODE_TO_PIPELINE_FUNCTION: Dict[RolloutMode, Callable] = {
-    RolloutMode.PUBLISH: run_connector_publish_pipeline,
-    RolloutMode.PROMOTE: run_connector_promote_pipeline,
-    RolloutMode.ROLLBACK: run_connector_rollback_pipeline,
-}
-
-
-# Third-party connectors can't be published with this pipeline, skip them.
-# This is not the same as partner connectors. Partner connectors use our tech stack and can
-# be published just fine. Third-party connectors are in their own subdirectory.
-def filter_out_third_party_connectors(
-    selected_connectors_with_modified_files: Iterable[ConnectorWithModifiedFiles],
-) -> List[ConnectorWithModifiedFiles]:
-    """
-    Return the list of connectors filtering out the connectors stored in connectors/third-party directory.
-    """
-    filtered_connectors = []
-    for connector in selected_connectors_with_modified_files:
-        if connector.is_third_party:
-            main_logger.info(
-                f"Skipping third party connector {connector.technical_name} from the list of connectors"
-            )
-        else:
-            filtered_connectors.append(connector)
-    return filtered_connectors
-
-
-@click.command(
-    cls=DaggerPipelineCommand, help="Publish all images for the selected connectors."
-)
-@click_ci_requirements_option()
-@click.option(
-    "--pre-release/--main-release",
-    help="Use this flag if you want to publish pre-release images.",
-    default=True,
-    type=bool,
-)
-@click.option(
-    "--spec-cache-gcs-credentials",
-    help="The service account key to upload files to the GCS bucket hosting spec cache.",
-    type=click.STRING,
-    required=True,
-    envvar="SPEC_CACHE_GCS_CREDENTIALS",
-    callback=wrap_gcp_credentials_in_secret,
-)
-@click.option(
-    "--spec-cache-bucket-name",
-    help="The name of the GCS bucket where specs will be cached.",
-    type=click.STRING,
-    required=True,
-    envvar="SPEC_CACHE_BUCKET_NAME",
-)
-@click.option(
-    "--metadata-service-gcs-credentials",
-    help="The service account key to upload files to the GCS bucket hosting the metadata files.",
-    type=click.STRING,
-    required=True,
-    envvar="METADATA_SERVICE_GCS_CREDENTIALS",
-    callback=wrap_gcp_credentials_in_secret,
-)
-@click.option(
-    "--metadata-service-bucket-name",
-    help="The name of the GCS bucket where metadata files will be uploaded.",
-    type=click.STRING,
-    required=True,
-    envvar="METADATA_SERVICE_BUCKET_NAME",
-)
-@click.option(
-    "--slack-webhook",
-    help="The Slack webhook URL to send notifications to.",
-    type=click.STRING,
-    envvar="SLACK_WEBHOOK",
-)
-@click.option(
-    "--python-registry-token",
-    help="Access token for python registry",
-    type=click.STRING,
-    envvar="PYTHON_REGISTRY_TOKEN",
-    callback=wrap_in_secret,
-)
-@click.option(
-    "--python-registry-url",
-    help="Which python registry url to publish to. If not set, the default pypi is used. For test pypi, use https://test.pypi.org/legacy/",
-    type=click.STRING,
-    default=DEFAULT_PYTHON_PACKAGE_REGISTRY_URL,
-    envvar="PYTHON_REGISTRY_URL",
-)
-@click.option(
-    "--python-registry-check-url",
-    help="Which url to check whether a certain version is published already. If not set, the default pypi is used. For test pypi, use https://test.pypi.org/pypi/",
-    type=click.STRING,
-    default=DEFAULT_PYTHON_PACKAGE_REGISTRY_CHECK_URL,
-    envvar="PYTHON_REGISTRY_CHECK_URL",
-)
-@click.option(
-    "--promote-release-candidate",
-    help="Promote a release candidate to a main release.",
-    type=click.BOOL,
-    default=False,
-    is_flag=True,
-)
-@click.option(
-    "--rollback-release-candidate",
-    help="Rollback a release candidate to a previous version.",
-    type=click.BOOL,
-    default=False,
-    is_flag=True,
-)
-@click.pass_context
-async def publish(
-    ctx: click.Context,
-    pre_release: bool,
-    spec_cache_gcs_credentials: Secret,
-    spec_cache_bucket_name: str,
-    metadata_service_bucket_name: str,
-    metadata_service_gcs_credentials: Secret,
-    slack_webhook: str,
-    python_registry_token: Secret,
-    python_registry_url: str,
-    python_registry_check_url: str,
-    promote_release_candidate: bool,
-    rollback_release_candidate: bool,
-) -> bool:
-    if promote_release_candidate and rollback_release_candidate:
-        raise click.UsageError(
-            "You can't promote and rollback a release candidate at the same time."
-        )
-    elif promote_release_candidate:
-        rollout_mode = RolloutMode.PROMOTE
-    elif rollback_release_candidate:
-        rollout_mode = RolloutMode.ROLLBACK
-    else:
-        rollout_mode = RolloutMode.PUBLISH
-
-    ctx.obj["selected_connectors_with_modified_files"] = (
-        filter_out_third_party_connectors(
-            ctx.obj["selected_connectors_with_modified_files"]
-        )
-    )
-    if not ctx.obj["selected_connectors_with_modified_files"]:
-        return True
-
-    if ctx.obj["is_local"]:
-        confirm(
-            "Publishing from a local environment is not recommended and requires to be logged in Airbyte's DockerHub registry, do you want to continue?",
-            abort=True,
-        )
-
-    fail_if_missing_docker_hub_creds(ctx)
-
-    publish_connector_contexts = reorder_contexts(
-        [
-            PublishConnectorContext(
-                connector=connector,
-                pre_release=pre_release,
-                spec_cache_gcs_credentials=spec_cache_gcs_credentials,
-                spec_cache_bucket_name=spec_cache_bucket_name,
-                metadata_service_gcs_credentials=metadata_service_gcs_credentials,
-                metadata_bucket_name=metadata_service_bucket_name,
-                docker_hub_username=Secret(
-                    "docker_hub_username", ctx.obj["secret_stores"]["in_memory"]
-                ),
-                docker_hub_password=Secret(
-                    "docker_hub_password", ctx.obj["secret_stores"]["in_memory"]
-                ),
-                slack_webhook=slack_webhook,
-                ci_report_bucket=ctx.obj["ci_report_bucket_name"],
-                report_output_prefix=ctx.obj["report_output_prefix"],
-                is_local=ctx.obj["is_local"],
-                git_branch=ctx.obj["git_branch"],
-                git_revision=ctx.obj["git_revision"],
-                diffed_branch=ctx.obj["diffed_branch"],
-                git_repo_url=ctx.obj["git_repo_url"],
-                gha_workflow_run_url=ctx.obj.get("gha_workflow_run_url"),
-                dagger_logs_url=ctx.obj.get("dagger_logs_url"),
-                pipeline_start_timestamp=ctx.obj.get("pipeline_start_timestamp"),
-                ci_context=ctx.obj.get("ci_context"),
-                ci_gcp_credentials=ctx.obj["ci_gcp_credentials"],
-                pull_request=ctx.obj.get("pull_request"),
-                s3_build_cache_access_key_id=ctx.obj.get(
-                    "s3_build_cache_access_key_id"
-                ),
-                s3_build_cache_secret_key=ctx.obj.get("s3_build_cache_secret_key"),
-                use_local_cdk=ctx.obj.get("use_local_cdk"),
-                use_cdk_ref=ctx.obj.get("use_cdk_ref"),
-                python_registry_token=python_registry_token,
-                python_registry_url=python_registry_url,
-                python_registry_check_url=python_registry_check_url,
-                rollout_mode=rollout_mode,
-                ci_github_access_token=ctx.obj.get("ci_github_access_token"),
-            )
-            for connector in ctx.obj["selected_connectors_with_modified_files"]
-        ]
-    )
-    main_logger.warn(
-        "Concurrency is forced to 1. For stability reasons we disable parallel publish pipelines."
-    )
-    ctx.obj["concurrency"] = 1
-
-    ran_publish_connector_contexts = await run_connectors_pipelines(
-        publish_connector_contexts,
-        ROLLOUT_MODE_TO_PIPELINE_FUNCTION[rollout_mode],
-        f"{rollout_mode.value} connectors",
-        ctx.obj["concurrency"],
-        ctx.obj["dagger_logs_path"],
-        ctx.obj["execute_timeout"],
-    )
-    return all(
-        context.state is ContextState.SUCCESSFUL
-        for context in ran_publish_connector_contexts
-    )