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

{airbyte_internal_ops-0.5.1.dist-info → airbyte_internal_ops-0.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airbyte-internal-ops
-Version: 0.5.1
+Version: 0.5.2
 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.5.1.dist-info → airbyte_internal_ops-0.5.2.dist-info}/RECORD

@@ -245,7 +245,7 @@ airbyte_ops_mcp/cli/__init__.py,sha256=XpL7FyVfgabfBF2JR7u7NwJ2krlYqjd_OwLcWf-Xc
 airbyte_ops_mcp/cli/_base.py,sha256=I8tWnyQf0ks4r3J8N8h-5GZxyn37T-55KsbuHnxYlcg,415
 airbyte_ops_mcp/cli/_shared.py,sha256=jg-xMyGzTCGPqKd8VTfE_3kGPIyO_3Kx5sQbG4rPc0Y,1311
 airbyte_ops_mcp/cli/app.py,sha256=SEdBpqFUG2O8zGV5ifwptxrLGFph_dLr66-MX9d69gQ,789
-airbyte_ops_mcp/cli/cloud.py,sha256=18SiVVMAibzPpGiEgKiga1BZ5x4QlLREOLd_VmNZwSo,45831
+airbyte_ops_mcp/cli/cloud.py,sha256=jac0FcG3UvNNeNNhqZR04uY9BCyVvins1kjO0LRZ86Y,45861
 airbyte_ops_mcp/cli/gh.py,sha256=koJPu0MDB6AW7mJq2z4dZV65ofvsZTkqoeitGF8KJR8,5364
 airbyte_ops_mcp/cli/registry.py,sha256=L4nDKhlegr31gSE-GUvDFSq10KgDz5kJuZXgLIxYIyg,9785
 airbyte_ops_mcp/cli/repo.py,sha256=G1hoQpH0XYhUH3FFOsia9xabGB0LP9o3XcwBuqvFVo0,16331
@@ -286,12 +286,12 @@ airbyte_ops_mcp/prod_db_access/sql.py,sha256=E7HQratZC5LbCXdvpQzZRd09SmCVnY_4Ru-
 airbyte_ops_mcp/registry/__init__.py,sha256=iEaPlt9GrnlaLbc__98TguNeZG8wuQu7S-_2QkhHcbA,858
 airbyte_ops_mcp/registry/models.py,sha256=B4L4TKr52wo0xs0CqvCBrpowqjShzVnZ5eTr2-EyhNs,2346
 airbyte_ops_mcp/registry/publish.py,sha256=VoPxsM2_0zJ829orzCRN-kjgcJtuBNyXgW4I9J680ro,12717
-airbyte_ops_mcp/regression_tests/__init__.py,sha256=8pwJIdz1Lb9oFV6UQ3DSjYKd8HCSqU8RpH5SDgEcEBA,1038
+airbyte_ops_mcp/regression_tests/__init__.py,sha256=qdveN866w50VmiDmTD0IW8Buo4UUn_1m9mKNS6hgj5U,1092
 airbyte_ops_mcp/regression_tests/cdk_secrets.py,sha256=iRjqqBS96KZoswfgT7ju-pE_pfbYoDy4PfrK-K8uyYs,3204
 airbyte_ops_mcp/regression_tests/ci_output.py,sha256=os69gcEhbomrGRAfnwPYAXhsAiJSWOuyW1gWZPx9-hw,15498
 airbyte_ops_mcp/regression_tests/config.py,sha256=dwWeY0tatdbwl9BqbhZ7EljoZDCtKmGO5fvOAIxeXmA,5873
 airbyte_ops_mcp/regression_tests/connection_fetcher.py,sha256=laGkxVfDl8_S-Azst28wijvPnOB-bmuuk2B_2e-0yjY,13170
-airbyte_ops_mcp/regression_tests/connection_secret_retriever.py,sha256=FhWNVWq7sON4nwUmVJv8BgXBOqg1YV4b5WuWyCzZ0LU,4695
+airbyte_ops_mcp/regression_tests/connection_secret_retriever.py,sha256=e2RvLEtAcpSHnYobP3wGNYGk8VNVI6MallX3cS0UPgA,5561
 airbyte_ops_mcp/regression_tests/connector_runner.py,sha256=OZzUa2aLh0sHaEARsDePOA-e3qEX4cvh3Jhnvi8S1rY,10130
 airbyte_ops_mcp/regression_tests/evaluation_modes.py,sha256=lAL6pEDmy_XCC7_m4_NXjt_f6Z8CXeAhMkc0FU8bm_M,1364
 airbyte_ops_mcp/regression_tests/http_metrics.py,sha256=busCYLb7qID0vv7yAtzARf__-HukpE3mAoc-pYI_gCQ,13256
@@ -306,7 +306,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.5.1.dist-info/METADATA,sha256=6FSpbOvu29YiilQjtOkOgmjW66Pxx-AlPciUPTmM8G0,5731
-airbyte_internal_ops-0.5.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-airbyte_internal_ops-0.5.1.dist-info/entry_points.txt,sha256=WxP0l7bRFss4Cr5uQqVj9mTEKwnRKouNuphXQF0lotA,171
-airbyte_internal_ops-0.5.1.dist-info/RECORD,,
+airbyte_internal_ops-0.5.2.dist-info/METADATA,sha256=vKolAxrXmB-dgddkjeQxYNZZL0ID11L51nID5LeMkMk,5731
+airbyte_internal_ops-0.5.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+airbyte_internal_ops-0.5.2.dist-info/entry_points.txt,sha256=WxP0l7bRFss4Cr5uQqVj9mTEKwnRKouNuphXQF0lotA,171
+airbyte_internal_ops-0.5.2.dist-info/RECORD,,

airbyte_ops_mcp/cli/cloud.py

@@ -65,6 +65,7 @@ from airbyte_ops_mcp.regression_tests.connection_fetcher import (
     save_connection_data_to_files,
 )
 from airbyte_ops_mcp.regression_tests.connection_secret_retriever import (
+    SecretRetrievalError,
     enrich_config_with_secrets,
     should_use_secret_retriever,
 )
@@ -847,13 +848,14 @@ def regression_test(
                     retrieval_reason="Regression test with USE_CONNECTION_SECRET_RETRIEVER=true",
                 )
                 print_success("Successfully retrieved unmasked secrets from database")
-            except Exception as e:
-                print_error(f"Failed to retrieve unmasked secrets: {e}")
+            except SecretRetrievalError as e:
+                write_github_output("success", False)
+                write_github_output("error", str(e))
                 exit_with_error(
-                    f"Failed to retrieve unmasked secrets: {e}\n"
-                    f"Unset USE_CONNECTION_SECRET_RETRIEVER or verify that the "
-                    f"{ENV_GCP_PROD_DB_ACCESS_CREDENTIALS} environment variable is set "
-                    f"with valid database credentials and that the Cloud SQL Proxy is running."
+                    f"{e}\n\n"
+                    f"This connection cannot be used for regression testing. "
+                    f"Please use a connection from a non-EU workspace, or use GSM-based "
+                    f"integration test credentials instead (by omitting --connection-id)."
                 )
 
         config_file, catalog_file = save_connection_data_to_files(

airbyte_ops_mcp/regression_tests/__init__.py

@@ -10,6 +10,7 @@ from airbyte_ops_mcp.regression_tests.connection_fetcher import (
     fetch_connection_data,
 )
 from airbyte_ops_mcp.regression_tests.connection_secret_retriever import (
+    SecretRetrievalError,
     enrich_config_with_secrets,
     is_secret_retriever_enabled,
     retrieve_unmasked_config,
@@ -27,6 +28,7 @@ __all__ = [
     "ConnectionData",
     "ConnectorUnderTest",
     "ExecutionResult",
+    "SecretRetrievalError",
     "TargetOrControl",
     "enrich_config_with_secrets",
     "fetch_connection_data",

airbyte_ops_mcp/regression_tests/connection_secret_retriever.py

@@ -107,9 +107,18 @@ def retrieve_unmasked_config(
     return None
 
 
+class SecretRetrievalError(Exception):
+    """Raised when secret retrieval fails.
+
+    This exception is raised when USE_CONNECTION_SECRET_RETRIEVER is enabled
+    but secrets cannot be retrieved (e.g., EU data residency restrictions).
+    """
+
+
 def enrich_config_with_secrets(
     connection_data: ConnectionData,
     retrieval_reason: str = "MCP live tests",
+    raise_on_failure: bool = True,
 ) -> ConnectionData:
     """Enrich connection data with unmasked secrets from internal retriever.
 
@@ -120,10 +129,16 @@ def enrich_config_with_secrets(
     Args:
         connection_data: The connection data to enrich.
         retrieval_reason: Reason for retrieval (for audit logging).
+        raise_on_failure: If True (default), raise SecretRetrievalError when
+            secrets cannot be retrieved. If False, return the original
+            connection_data with masked secrets (legacy behavior).
 
     Returns:
-        A new ConnectionData with unmasked config, or the original if
-        retrieval fails or is not available.
+        A new ConnectionData with unmasked config.
+
+    Raises:
+        SecretRetrievalError: If raise_on_failure is True and secrets cannot
+            be retrieved (e.g., due to EU data residency restrictions).
     """
     unmasked_config = retrieve_unmasked_config(
         connection_id=connection_data.connection_id,
@@ -131,10 +146,15 @@ def enrich_config_with_secrets(
     )
 
     if unmasked_config is None:
-        logger.info(
-            f"Could not retrieve unmasked config for {connection_data.connection_id}, "
-            "using masked config from Cloud API"
+        error_msg = (
+            "Could not retrieve unmasked secrets for connection "
+            f"{connection_data.connection_id}. This may be due to EU data "
+            "residency restrictions or database connectivity issues. "
+            "The connection's credentials cannot be used for regression testing."
         )
+        logger.warning(error_msg)
+        if raise_on_failure:
+            raise SecretRetrievalError(error_msg)
         return connection_data
 
     logger.info(