airbyte-internal-ops 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

airbyte_ops_mcp/prod_db_access/sql.py

@@ -131,6 +131,61 @@ SELECT_CONNECTIONS_BY_CONNECTOR_AND_ORG = sqlalchemy.text(
     """
 )
 
+# Query connections by DESTINATION connector type (no organization filter)
+SELECT_CONNECTIONS_BY_DESTINATION_CONNECTOR = sqlalchemy.text(
+    """
+    SELECT
+         connection.id AS connection_id,
+         connection.name AS connection_name,
+         connection.destination_id,
+         workspace.id AS workspace_id,
+         workspace.name AS workspace_name,
+         workspace.organization_id,
+         workspace.dataplane_group_id,
+         dataplane_group.name AS dataplane_name,
+         destination_actor.actor_definition_id AS destination_definition_id,
+         destination_actor.name AS destination_name
+    FROM connection
+    JOIN actor AS destination_actor
+      ON connection.destination_id = destination_actor.id
+    JOIN workspace
+      ON destination_actor.workspace_id = workspace.id
+    LEFT JOIN dataplane_group
+      ON workspace.dataplane_group_id = dataplane_group.id
+    WHERE
+         destination_actor.actor_definition_id = :connector_definition_id
+    LIMIT :limit
+    """
+)
+
+# Query connections by DESTINATION connector type, filtered by organization
+SELECT_CONNECTIONS_BY_DESTINATION_CONNECTOR_AND_ORG = sqlalchemy.text(
+    """
+    SELECT
+         connection.id AS connection_id,
+         connection.name AS connection_name,
+         connection.destination_id,
+         workspace.id AS workspace_id,
+         workspace.name AS workspace_name,
+         workspace.organization_id,
+         workspace.dataplane_group_id,
+         dataplane_group.name AS dataplane_name,
+         destination_actor.actor_definition_id AS destination_definition_id,
+         destination_actor.name AS destination_name
+    FROM connection
+    JOIN actor AS destination_actor
+      ON connection.destination_id = destination_actor.id
+    JOIN workspace
+      ON destination_actor.workspace_id = workspace.id
+    LEFT JOIN dataplane_group
+      ON workspace.dataplane_group_id = dataplane_group.id
+    WHERE
+         destination_actor.actor_definition_id = :connector_definition_id
+     AND workspace.organization_id = :organization_id
+    LIMIT :limit
+    """
+)
+
 # =============================================================================
 # Connector Version Queries
 # =============================================================================
@@ -441,3 +496,34 @@ SELECT_ORG_WORKSPACES = sqlalchemy.text(
          workspace.name
     """
 )
+
+# =============================================================================
+# Workspace Lookup by Email Domain
+# =============================================================================
+
+# Find workspaces by email domain
+# This is useful for identifying workspaces based on user email domains
+# (e.g., finding partner accounts like MotherDuck by searching for "motherduck.com")
+SELECT_WORKSPACES_BY_EMAIL_DOMAIN = sqlalchemy.text(
+    """
+    SELECT DISTINCT
+         workspace.organization_id,
+         workspace.id AS workspace_id,
+         workspace.name AS workspace_name,
+         workspace.slug,
+         workspace.email,
+         workspace.dataplane_group_id,
+         dataplane_group.name AS dataplane_name,
+         workspace.created_at
+    FROM workspace
+    LEFT JOIN dataplane_group
+      ON workspace.dataplane_group_id = dataplane_group.id
+    WHERE
+         workspace.email LIKE '%@' || :email_domain
+     AND workspace.tombstone = false
+    ORDER BY
+         workspace.organization_id,
+         workspace.name
+    LIMIT :limit
+    """
+)

airbyte_ops_mcp/{live_tests → regression_tests}/__init__.py

@@ -5,17 +5,17 @@ This module provides tools for testing Airbyte connectors against live data
 without using Dagger. It uses Docker SDK directly for container orchestration.
 """
 
-from airbyte_ops_mcp.live_tests.connection_fetcher import (
+from airbyte_ops_mcp.regression_tests.connection_fetcher import (
     ConnectionData,
     fetch_connection_data,
 )
-from airbyte_ops_mcp.live_tests.connection_secret_retriever import (
+from airbyte_ops_mcp.regression_tests.connection_secret_retriever import (
     enrich_config_with_secrets,
     is_secret_retriever_enabled,
     retrieve_unmasked_config,
     should_use_secret_retriever,
 )
-from airbyte_ops_mcp.live_tests.models import (
+from airbyte_ops_mcp.regression_tests.models import (
     Command,
     ConnectorUnderTest,
     ExecutionResult,

airbyte_ops_mcp/{live_tests → regression_tests}/cdk_secrets.py

@@ -5,7 +5,7 @@ This module uses the PyAirbyte GoogleGSMSecretManager to retrieve
 integration test secrets from Google Secret Manager for connectors.
 
 Usage:
-    from airbyte_ops_mcp.live_tests.cdk_secrets import get_first_config_from_secrets
+    from airbyte_ops_mcp.regression_tests.cdk_secrets import get_first_config_from_secrets
 
     # Fetch the first config for a connector
     config = get_first_config_from_secrets("source-github")

airbyte_ops_mcp/{live_tests → regression_tests}/connection_secret_retriever.py

@@ -11,8 +11,8 @@ The secret retriever requires:
 - Cloud SQL Proxy running to internal Postgres (or CI environment)
 
 Usage:
-    from airbyte_ops_mcp.live_tests.connection_fetcher import fetch_connection_data
-    from airbyte_ops_mcp.live_tests.connection_secret_retriever import (
+    from airbyte_ops_mcp.regression_tests.connection_fetcher import fetch_connection_data
+    from airbyte_ops_mcp.regression_tests.connection_secret_retriever import (
         enrich_config_with_secrets,
         should_use_secret_retriever,
     )
@@ -42,7 +42,7 @@ from airbyte_ops_mcp.connection_config_retriever import (
 from airbyte_ops_mcp.gcp_auth import ensure_adc_credentials
 
 if TYPE_CHECKING:
-    from airbyte_ops_mcp.live_tests.connection_fetcher import ConnectionData
+    from airbyte_ops_mcp.regression_tests.connection_fetcher import ConnectionData
 
 logger = logging.getLogger(__name__)
 

airbyte_ops_mcp/{live_tests → regression_tests}/connector_runner.py

@@ -15,7 +15,7 @@ import uuid
 from pathlib import Path
 from typing import TYPE_CHECKING
 
-from airbyte_ops_mcp.live_tests.models import (
+from airbyte_ops_mcp.regression_tests.models import (
     Command,
     ExecutionInputs,
     ExecutionResult,

airbyte_ops_mcp/{live_tests → regression_tests}/message_cache/__init__.py

@@ -8,7 +8,9 @@ Based on airbyte-ci implementation:
 https://github.com/airbytehq/airbyte/tree/master/airbyte-ci/connectors/live-tests/src/live_tests/commons/backends
 """
 
-from airbyte_ops_mcp.live_tests.message_cache.duckdb_cache import DuckDbMessageCache
+from airbyte_ops_mcp.regression_tests.message_cache.duckdb_cache import (
+    DuckDbMessageCache,
+)
 
 __all__ = [
     "DuckDbMessageCache",

airbyte_ops_mcp/{live_tests → regression_tests}/regression/__init__.py

@@ -8,7 +8,7 @@ Based on airbyte-ci implementation:
 https://github.com/airbytehq/airbyte/blob/master/airbyte-ci/connectors/live-tests/src/live_tests/regression_tests/test_read.py
 """
 
-from airbyte_ops_mcp.live_tests.regression.comparators import (
+from airbyte_ops_mcp.regression_tests.regression.comparators import (
     ComparisonResult,
     RecordDiff,
     StreamComparisonResult,

airbyte_ops_mcp/{live_tests → regression_tests}/schema_generation.py

@@ -19,7 +19,9 @@ from airbyte_protocol.models import AirbyteMessage
 from airbyte_protocol.models import Type as AirbyteMessageType
 from genson import SchemaBuilder
 
-from airbyte_ops_mcp.live_tests.obfuscation import convert_obfuscated_record_to_typed
+from airbyte_ops_mcp.regression_tests.obfuscation import (
+    convert_obfuscated_record_to_typed,
+)
 
 logger = logging.getLogger(__name__)
 

airbyte_ops_mcp/{live_tests → regression_tests}/validation/__init__.py

@@ -8,7 +8,7 @@ Based on airbyte-ci validation tests:
 https://github.com/airbytehq/airbyte/tree/master/airbyte-ci/connectors/live-tests/src/live_tests/validation_tests
 """
 
-from airbyte_ops_mcp.live_tests.validation.catalog_validators import (
+from airbyte_ops_mcp.regression_tests.validation.catalog_validators import (
     ValidationResult,
     validate_additional_properties_is_true,
     validate_catalog,
@@ -20,7 +20,7 @@ from airbyte_ops_mcp.live_tests.validation.catalog_validators import (
     validate_schemas_are_valid_json_schema,
     validate_streams_have_sync_modes,
 )
-from airbyte_ops_mcp.live_tests.validation.record_validators import (
+from airbyte_ops_mcp.regression_tests.validation.record_validators import (
     validate_primary_keys_in_records,
     validate_records_conform_to_schema,
     validate_state_messages_emitted,

airbyte_ops_mcp/{live_tests → regression_tests}/validation/record_validators.py

@@ -15,10 +15,12 @@ import jsonschema
 from airbyte_protocol.models import AirbyteMessage, AirbyteStateType
 from airbyte_protocol.models import Type as AirbyteMessageType
 
-from airbyte_ops_mcp.live_tests.validation.catalog_validators import ValidationResult
+from airbyte_ops_mcp.regression_tests.validation.catalog_validators import (
+    ValidationResult,
+)
 
 if TYPE_CHECKING:
-    from airbyte_ops_mcp.live_tests.models import ExecutionResult
+    from airbyte_ops_mcp.regression_tests.models import ExecutionResult
 
 
 def validate_records_conform_to_schema(