airbyte-internal-ops 0.4.2__py3-none-any.whl → 0.5.1__py3-none-any.whl

airbyte_ops_mcp/cloud_admin/models.py

@@ -71,3 +71,59 @@ class VersionOverrideOperationResult(BaseModel):
         if self.success:
             return f"✓ {self.message}"
         return f"✗ {self.message}"
+
+
+class WorkspaceVersionOverrideResult(BaseModel):
+    """Result of a workspace-level version override operation.
+
+    This model provides detailed information about the outcome of a workspace-level
+    version pinning or unpinning operation.
+    """
+
+    success: bool = Field(description="Whether the operation succeeded")
+    message: str = Field(description="Human-readable message describing the result")
+    workspace_id: str = Field(description="The workspace ID")
+    connector_name: str = Field(
+        description="The connector name (e.g., 'source-github')"
+    )
+    connector_type: Literal["source", "destination"] = Field(
+        description="The type of connector (source or destination)"
+    )
+    version: str | None = Field(
+        default=None,
+        description="The version that was pinned (None if cleared or failed)",
+    )
+
+    def __str__(self) -> str:
+        """Return a string representation of the operation result."""
+        if self.success:
+            return f"✓ {self.message}"
+        return f"✗ {self.message}"
+
+
+class OrganizationVersionOverrideResult(BaseModel):
+    """Result of an organization-level version override operation.
+
+    This model provides detailed information about the outcome of an organization-level
+    version pinning or unpinning operation.
+    """
+
+    success: bool = Field(description="Whether the operation succeeded")
+    message: str = Field(description="Human-readable message describing the result")
+    organization_id: str = Field(description="The organization ID")
+    connector_name: str = Field(
+        description="The connector name (e.g., 'source-github')"
+    )
+    connector_type: Literal["source", "destination"] = Field(
+        description="The type of connector (source or destination)"
+    )
+    version: str | None = Field(
+        default=None,
+        description="The version that was pinned (None if cleared or failed)",
+    )
+
+    def __str__(self) -> str:
+        """Return a string representation of the operation result."""
+        if self.success:
+            return f"✓ {self.message}"
+        return f"✗ {self.message}"

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