airbyte-internal-ops 0.5.0__py3-none-any.whl → 0.5.2__py3-none-any.whl

airbyte_ops_mcp/cloud_admin/api_client.py

@@ -132,7 +132,7 @@ def _get_access_token(
 
 def get_user_id_by_email(
     email: str,
-    api_root: str,
+    config_api_root: str,
     client_id: str | None = None,
     client_secret: str | None = None,
     bearer_token: str | None = None,
@@ -141,7 +141,7 @@ def get_user_id_by_email(
 
     Args:
         email: The user's email address
-        api_root: The API root URL
+        config_api_root: The Config API root URL (e.g., CLOUD_CONFIG_API_ROOT)
         client_id: The Airbyte Cloud client ID (required if no bearer_token)
         client_secret: The Airbyte Cloud client secret (required if no bearer_token)
         bearer_token: Pre-existing bearer token (takes precedence over client credentials)
@@ -154,7 +154,7 @@ def get_user_id_by_email(
     """
     access_token = _get_access_token(client_id, client_secret, bearer_token)
 
-    endpoint = f"{api_root}/users/list_instance_admin"
+    endpoint = f"{config_api_root}/users/list_instance_admin"
     response = requests.post(
         endpoint,
         json={},
@@ -196,7 +196,7 @@ def resolve_connector_version_id(
     actor_definition_id: str,
     connector_type: Literal["source", "destination"],
     version: str,
-    api_root: str,
+    config_api_root: str,
     client_id: str | None = None,
     client_secret: str | None = None,
     bearer_token: str | None = None,
@@ -207,7 +207,7 @@ def resolve_connector_version_id(
         actor_definition_id: The actor definition ID
         connector_type: Either "source" or "destination"
         version: The version string (e.g., "0.1.47-preview.abe7cb4")
-        api_root: The API root URL
+        config_api_root: The Config API root URL (e.g., CLOUD_CONFIG_API_ROOT)
         client_id: The Airbyte Cloud client ID (required if no bearer_token)
         client_secret: The Airbyte Cloud client secret (required if no bearer_token)
         bearer_token: Pre-existing bearer token (takes precedence over client credentials)
@@ -220,7 +220,7 @@ def resolve_connector_version_id(
     """
     access_token = _get_access_token(client_id, client_secret, bearer_token)
 
-    endpoint = f"{api_root}/actor_definition_versions/resolve"
+    endpoint = f"{config_api_root}/actor_definition_versions/resolve"
     payload = {
         "actorDefinitionId": actor_definition_id,
         "actorType": connector_type,
@@ -270,7 +270,7 @@ def _get_scoped_configuration_context(
     actor_definition_id: str,
     scope_type: _ScopeType,
     scope_id: str,
-    api_root: str,
+    config_api_root: str,
     access_token: str,
 ) -> dict[str, Any] | None:
     """Get the active scoped configuration for a single scope level.
@@ -282,7 +282,7 @@ def _get_scoped_configuration_context(
         actor_definition_id: The actor definition ID for the connector
         scope_type: The scope type enum (ACTOR, WORKSPACE, or ORGANIZATION)
         scope_id: The ID for the scope (connector_id, workspace_id, or organization_id)
-        api_root: The API root URL
+        config_api_root: The Config API root URL (e.g., CLOUD_CONFIG_API_ROOT)
         access_token: Pre-authenticated access token
 
     Returns:
@@ -291,7 +291,7 @@ def _get_scoped_configuration_context(
     Raises:
         PyAirbyteInputError: If the API request fails
     """
-    endpoint = f"{api_root}/scoped_configuration/get_context"
+    endpoint = f"{config_api_root}/scoped_configuration/get_context"
     context_payload = {
         "config_key": _ScopedConfigKey.CONNECTOR_VERSION.value,
         "resource_type": _ResourceType.ACTOR_DEFINITION.value,
@@ -333,7 +333,7 @@ def get_all_scoped_configuration_contexts(
     actor_definition_id: str,
     workspace_id: str,
     organization_id: str,
-    api_root: str,
+    config_api_root: str,
     client_id: str | None = None,
     client_secret: str | None = None,
     bearer_token: str | None = None,
@@ -349,7 +349,7 @@ def get_all_scoped_configuration_contexts(
         actor_definition_id: The actor definition ID for the connector
         workspace_id: The workspace ID (required - must always check workspace scope)
         organization_id: The organization ID (required - must always check org scope)
-        api_root: The API root URL
+        config_api_root: The Config API root URL (e.g., CLOUD_CONFIG_API_ROOT)
         client_id: The Airbyte Cloud client ID (required if no bearer_token)
         client_secret: The Airbyte Cloud client secret (required if no bearer_token)
         bearer_token: Pre-existing bearer token (takes precedence over client credentials)
@@ -381,7 +381,7 @@ def get_all_scoped_configuration_contexts(
             actor_definition_id=actor_definition_id,
             scope_type=scope_type_enum,
             scope_id=scope_id,
-            api_root=api_root,
+            config_api_root=config_api_root,
             access_token=access_token,
         )
         if active_config:
@@ -393,7 +393,7 @@ def get_all_scoped_configuration_contexts(
 def get_connector_version(
     connector_id: str,
     connector_type: Literal["source", "destination"],
-    api_root: str,
+    config_api_root: str,
     client_id: str | None = None,
     client_secret: str | None = None,
     bearer_token: str | None = None,
@@ -408,7 +408,7 @@ def get_connector_version(
     Args:
         connector_id: The ID of the deployed connector (source or destination)
         connector_type: Either "source" or "destination"
-        api_root: The API root URL
+        config_api_root: The Config API root URL (e.g., CLOUD_CONFIG_API_ROOT)
         client_id: The Airbyte Cloud client ID (required if no bearer_token)
         client_secret: The Airbyte Cloud client secret (required if no bearer_token)
         bearer_token: Pre-existing bearer token (takes precedence over client credentials)
@@ -427,13 +427,13 @@ def get_connector_version(
     access_token = _get_access_token(client_id, client_secret, bearer_token)
 
     # Determine endpoint based on connector type
-    # api_root already includes /v1
+    # config_api_root already includes /v1
     if connector_type == "source":
-        endpoint = f"{api_root}/actor_definition_versions/get_for_source"
+        endpoint = f"{config_api_root}/actor_definition_versions/get_for_source"
         payload = {"sourceId": connector_id}
         definition_id_key = "sourceDefinitionId"
     else:
-        endpoint = f"{api_root}/actor_definition_versions/get_for_destination"
+        endpoint = f"{config_api_root}/actor_definition_versions/get_for_destination"
         payload = {"destinationId": connector_id}
         definition_id_key = "destinationDefinitionId"
 
@@ -456,7 +456,7 @@ def get_connector_version(
                 "connector_type": connector_type,
                 "endpoint": endpoint,
                 "payload": payload,
-                "api_root": api_root,
+                "config_api_root": config_api_root,
                 "status_code": response.status_code,
                 "response": response.text,
             },
@@ -474,7 +474,7 @@ def get_connector_version(
     # If workspace_id is provided, also get detailed scoped configuration context
     if workspace_id:
         # Get actor_definition_id from the connector info
-        get_endpoint = f"{api_root}/{connector_type}s/get"
+        get_endpoint = f"{config_api_root}/{connector_type}s/get"
         get_payload: dict[str, str] = {f"{connector_type}Id": connector_id}
 
         get_response = requests.post(
@@ -494,7 +494,7 @@ def get_connector_version(
 
             if actor_definition_id:
                 # Get organization_id from workspace
-                workspace_endpoint = f"{api_root}/workspaces/get"
+                workspace_endpoint = f"{config_api_root}/workspaces/get"
                 workspace_response = requests.post(
                     workspace_endpoint,
                     json={"workspaceId": workspace_id},
@@ -517,7 +517,7 @@ def get_connector_version(
                         actor_definition_id=actor_definition_id,
                         workspace_id=workspace_id,
                         organization_id=organization_id,
-                        api_root=api_root,
+                        config_api_root=config_api_root,
                         bearer_token=access_token,
                     )
 
@@ -527,7 +527,7 @@ def get_connector_version(
 def set_connector_version_override(
     connector_id: str,
     connector_type: Literal["source", "destination"],
-    api_root: str,
+    config_api_root: str,
     client_id: str | None = None,
     client_secret: str | None = None,
     workspace_id: str | None = None,
@@ -554,7 +554,7 @@ def set_connector_version_override(
     Args:
         connector_id: The ID of the deployed connector
         connector_type: Either "source" or "destination"
-        api_root: The API root URL
+        config_api_root: The Config API root URL (e.g., CLOUD_CONFIG_API_ROOT)
         client_id: The Airbyte Cloud client ID (required if no bearer_token)
         client_secret: The Airbyte Cloud client secret (required if no bearer_token)
         workspace_id: The workspace ID
@@ -590,7 +590,7 @@ def set_connector_version_override(
     if unset:
         # To unset, we need to delete the scoped configuration
         # First, get the actor_definition_id for the connector
-        get_endpoint: str = f"{api_root}/{connector_type}s/get"
+        get_endpoint: str = f"{config_api_root}/{connector_type}s/get"
         get_payload: dict[str, str] = {f"{connector_type}Id": connector_id}
         definition_id_key = f"{connector_type}DefinitionId"
 
@@ -623,7 +623,7 @@ def set_connector_version_override(
             actor_definition_id=actor_definition_id,
             scope_type=_ScopeType.ACTOR,
             scope_id=connector_id,
-            api_root=api_root,
+            config_api_root=config_api_root,
             access_token=access_token,
         )
 
@@ -632,7 +632,7 @@ def set_connector_version_override(
             return False
 
         # Delete the active configuration
-        delete_endpoint = f"{api_root}/scoped_configuration/delete"
+        delete_endpoint = f"{config_api_root}/scoped_configuration/delete"
         delete_payload = {"scopedConfigurationId": active_config["id"]}
 
         response = requests.post(
@@ -662,7 +662,7 @@ def set_connector_version_override(
     else:
         # Set a new override
         # First, get the actor_definition_id for the connector
-        get_endpoint = f"{api_root}/{connector_type}s/get"
+        get_endpoint = f"{config_api_root}/{connector_type}s/get"
         get_payload: dict[str, str] = {f"{connector_type}Id": connector_id}
         definition_id_key = f"{connector_type}DefinitionId"
 
@@ -695,7 +695,7 @@ def set_connector_version_override(
             actor_definition_id=actor_definition_id,
             connector_type=connector_type,
             version=version,
-            api_root=api_root,
+            config_api_root=config_api_root,
             client_id=client_id,
             client_secret=client_secret,
             bearer_token=bearer_token,
@@ -703,7 +703,7 @@ def set_connector_version_override(
 
         # Get organization_id from workspace info for comprehensive scope checking
         # This is REQUIRED - we must always check all three scopes
-        workspace_endpoint = f"{api_root}/workspaces/get"
+        workspace_endpoint = f"{config_api_root}/workspaces/get"
         workspace_payload = {"workspaceId": workspace_id}
         workspace_response = requests.post(
             workspace_endpoint,
@@ -745,7 +745,7 @@ def set_connector_version_override(
             actor_definition_id=actor_definition_id,
             workspace_id=workspace_id,
             organization_id=organization_id,
-            api_root=api_root,
+            config_api_root=config_api_root,
             client_id=client_id,
             client_secret=client_secret,
             bearer_token=bearer_token,
@@ -836,7 +836,7 @@ def set_connector_version_override(
                     },
                 )
 
-            delete_endpoint = f"{api_root}/scoped_configuration/delete"
+            delete_endpoint = f"{config_api_root}/scoped_configuration/delete"
             delete_payload = {"scopedConfigurationId": actor_config["id"]}
 
             delete_response = requests.post(
@@ -889,14 +889,14 @@ def set_connector_version_override(
             )
         origin = get_user_id_by_email(
             email=user_email,
-            api_root=api_root,
+            config_api_root=config_api_root,
             client_id=client_id,
             client_secret=client_secret,
             bearer_token=bearer_token,
         )
 
         # Create the override with correct schema
-        endpoint = f"{api_root}/scoped_configuration/create"
+        endpoint = f"{config_api_root}/scoped_configuration/create"
 
         # Build payload with explicit string values for auditability
         # (enum.value ensures we log exactly what we send)
@@ -968,7 +968,7 @@ def set_workspace_connector_version_override(
     workspace_id: str,
     connector_name: str,
     connector_type: Literal["source", "destination"],
-    api_root: str,
+    config_api_root: str,
     client_id: str | None = None,
     client_secret: str | None = None,
     version: str | None = None,
@@ -988,7 +988,7 @@ def set_workspace_connector_version_override(
         workspace_id: The workspace ID
         connector_name: The connector name (e.g., 'source-github')
         connector_type: Either "source" or "destination"
-        api_root: The API root URL
+        config_api_root: The Config API root URL (e.g., CLOUD_CONFIG_API_ROOT)
         client_id: The Airbyte Cloud client ID (required if no bearer_token)
         client_secret: The Airbyte Cloud client secret (required if no bearer_token)
         version: The version to pin to (e.g., "0.1.0"), or None to unset
@@ -1026,7 +1026,7 @@ def set_workspace_connector_version_override(
             actor_definition_id=actor_definition_id,
             scope_type=_ScopeType.WORKSPACE,
             scope_id=workspace_id,
-            api_root=api_root,
+            config_api_root=config_api_root,
             access_token=access_token,
         )
 
@@ -1052,7 +1052,7 @@ def set_workspace_connector_version_override(
             )
 
         # Delete the configuration
-        delete_endpoint = f"{api_root}/scoped_configuration/delete"
+        delete_endpoint = f"{config_api_root}/scoped_configuration/delete"
         delete_payload = {"scopedConfigurationId": active_config["id"]}
 
         response = requests.post(
@@ -1085,7 +1085,7 @@ def set_workspace_connector_version_override(
         actor_definition_id=actor_definition_id,
         connector_type=connector_type,
         version=version,
-        api_root=api_root,
+        config_api_root=config_api_root,
         bearer_token=access_token,
     )
 
@@ -1094,7 +1094,7 @@ def set_workspace_connector_version_override(
         actor_definition_id=actor_definition_id,
         scope_type=_ScopeType.WORKSPACE,
         scope_id=workspace_id,
-        api_root=api_root,
+        config_api_root=config_api_root,
         access_token=access_token,
     )
 
@@ -1123,7 +1123,7 @@ def set_workspace_connector_version_override(
             and api_scope_id == workspace_id
         ):
             # Delete existing workspace-level config before creating new one
-            delete_endpoint = f"{api_root}/scoped_configuration/delete"
+            delete_endpoint = f"{config_api_root}/scoped_configuration/delete"
             delete_payload = {"scopedConfigurationId": existing_config["id"]}
 
             delete_response = requests.post(
@@ -1150,12 +1150,12 @@ def set_workspace_connector_version_override(
         )
     origin = get_user_id_by_email(
         email=user_email,
-        api_root=api_root,
+        config_api_root=config_api_root,
         bearer_token=access_token,
     )
 
     # Create the override
-    endpoint = f"{api_root}/scoped_configuration/create"
+    endpoint = f"{config_api_root}/scoped_configuration/create"
     payload: dict[str, Any] = {
         "config_key": _ScopedConfigKey.CONNECTOR_VERSION.value,
         "resource_type": _ResourceType.ACTOR_DEFINITION.value,
@@ -1202,7 +1202,7 @@ def set_organization_connector_version_override(
     organization_id: str,
     connector_name: str,
     connector_type: Literal["source", "destination"],
-    api_root: str,
+    config_api_root: str,
     client_id: str | None = None,
     client_secret: str | None = None,
     version: str | None = None,
@@ -1222,7 +1222,7 @@ def set_organization_connector_version_override(
         organization_id: The organization ID
         connector_name: The connector name (e.g., 'source-github')
         connector_type: Either "source" or "destination"
-        api_root: The API root URL
+        config_api_root: The Config API root URL (e.g., CLOUD_CONFIG_API_ROOT)
         client_id: The Airbyte Cloud client ID (required if no bearer_token)
         client_secret: The Airbyte Cloud client secret (required if no bearer_token)
         version: The version to pin to (e.g., "0.1.0"), or None to unset
@@ -1260,7 +1260,7 @@ def set_organization_connector_version_override(
             actor_definition_id=actor_definition_id,
             scope_type=_ScopeType.ORGANIZATION,
             scope_id=organization_id,
-            api_root=api_root,
+            config_api_root=config_api_root,
             access_token=access_token,
         )
 
@@ -1288,7 +1288,7 @@ def set_organization_connector_version_override(
             )
 
         # Delete the configuration
-        delete_endpoint = f"{api_root}/scoped_configuration/delete"
+        delete_endpoint = f"{config_api_root}/scoped_configuration/delete"
         delete_payload = {"scopedConfigurationId": active_config["id"]}
 
         response = requests.post(
@@ -1321,7 +1321,7 @@ def set_organization_connector_version_override(
         actor_definition_id=actor_definition_id,
         connector_type=connector_type,
         version=version,
-        api_root=api_root,
+        config_api_root=config_api_root,
         bearer_token=access_token,
     )
 
@@ -1330,7 +1330,7 @@ def set_organization_connector_version_override(
         actor_definition_id=actor_definition_id,
         scope_type=_ScopeType.ORGANIZATION,
         scope_id=organization_id,
-        api_root=api_root,
+        config_api_root=config_api_root,
         access_token=access_token,
     )
 
@@ -1359,7 +1359,7 @@ def set_organization_connector_version_override(
             and api_scope_id == organization_id
         ):
             # Delete existing organization-level config before creating new one
-            delete_endpoint = f"{api_root}/scoped_configuration/delete"
+            delete_endpoint = f"{config_api_root}/scoped_configuration/delete"
             delete_payload = {"scopedConfigurationId": existing_config["id"]}
 
             delete_response = requests.post(
@@ -1386,12 +1386,12 @@ def set_organization_connector_version_override(
         )
     origin = get_user_id_by_email(
         email=user_email,
-        api_root=api_root,
+        config_api_root=config_api_root,
         bearer_token=access_token,
     )
 
     # Create the override
-    endpoint = f"{api_root}/scoped_configuration/create"
+    endpoint = f"{config_api_root}/scoped_configuration/create"
     payload: dict[str, Any] = {
         "config_key": _ScopedConfigKey.CONNECTOR_VERSION.value,
         "resource_type": _ResourceType.ACTOR_DEFINITION.value,

airbyte_ops_mcp/constants.py

@@ -122,6 +122,64 @@ class OrganizationAliasEnum(StrEnum):
         return alias_mapping[org_id]
 
 
+# =============================================================================
+# Workspace ID Aliases
+# =============================================================================
+
+
+class WorkspaceAliasEnum(StrEnum):
+    """Workspace ID aliases that can be used in place of UUIDs.
+
+    Each member's name is the alias (e.g., "@devin-ai-sandbox") and its value
+    is the actual workspace UUID. Use `WorkspaceAliasEnum.resolve()` to
+    resolve aliases to actual IDs.
+    """
+
+    DEVIN_AI_SANDBOX = "266ebdfe-0d7b-4540-9817-de7e4505ba61"
+    """The Devin AI sandbox workspace for testing and development.
+
+    Alias: @devin-ai-sandbox
+    """
+
+    @classmethod
+    def resolve(cls, workspace_id: str | None) -> str | None:
+        """Resolve a workspace ID alias to its actual UUID.
+
+        Accepts either an alias string (e.g., "@devin-ai-sandbox") or a
+        WorkspaceAliasEnum enum member, and returns the actual UUID.
+
+        Returns:
+            The resolved workspace ID (UUID), or None if input is None.
+            If the input doesn't start with "@", it is returned unchanged.
+
+        Raises:
+            PyAirbyteInputError: If the input starts with "@" but is not a recognized alias.
+        """
+        if workspace_id is None:
+            return None
+
+        # Handle WorkspaceAliasEnum enum members directly
+        if isinstance(workspace_id, cls):
+            return workspace_id.value
+
+        # If it doesn't look like an alias, return as-is (assume it's a UUID)
+        if not workspace_id.startswith("@"):
+            return workspace_id
+
+        # Handle alias strings or raise an error if invalid
+        alias_mapping = {
+            "@devin-ai-sandbox": cls.DEVIN_AI_SANDBOX.value,
+        }
+        if workspace_id not in alias_mapping:
+            raise PyAirbyteInputError(
+                message=f"Unknown workspace alias: {workspace_id}",
+                context={
+                    "valid_aliases": list(alias_mapping.keys()),
+                },
+            )
+        return alias_mapping[workspace_id]
+
+
 CONNECTION_RETRIEVER_PG_CONNECTION_DETAILS_SECRET_ID = (
     "projects/587336813068/secrets/CONNECTION_RETRIEVER_PG_CONNECTION_DETAILS"
 )

airbyte_ops_mcp/{_legacy/airbyte_ci/metadata_service/docker_hub.py → docker_hub.py}

@@ -1,6 +1,13 @@
 #
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 #
+"""DockerHub API client with authentication, pagination, and retry support.
+
+Provides utilities for interacting with the DockerHub API, including fetching
+image tags and digests, checking image existence, and finding the latest version.
+Supports authenticated requests to avoid rate limits and includes retry logic
+for handling API flakiness.
+"""
 
 import os
 import time
@@ -95,17 +102,16 @@ def get_docker_hub_tags_and_digests(
             for result in json_response.get("results", [])
         }
     )
-    if paginate:
-        if next_page_url := json_response.get("next"):
-            tags_and_digests.update(
-                get_docker_hub_tags_and_digests(
-                    image_name,
-                    retries=retries,
-                    wait_sec=wait_sec,
-                    next_page_url=next_page_url,
-                    tags_and_digests=tags_and_digests,
-                )
+    if paginate and (next_page_url := json_response.get("next")):
+        tags_and_digests.update(
+            get_docker_hub_tags_and_digests(
+                image_name,
+                retries=retries,
+                wait_sec=wait_sec,
+                next_page_url=next_page_url,
+                tags_and_digests=tags_and_digests,
             )
+        )
     return tags_and_digests