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

{airbyte_internal_ops-0.1.6.dist-info → airbyte_internal_ops-0.1.7.dist-info}/METADATA

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

@@ -367,7 +367,7 @@ airbyte_ops_mcp/live_tests/__init__.py,sha256=qJac67dt6DQCqif39HqeiG3Tr9xrxfP-al
 airbyte_ops_mcp/live_tests/ci_output.py,sha256=NQATOGid0OCbLEl2NOtGK4cHLL5OxXhjmtanBjIlCyE,11369
 airbyte_ops_mcp/live_tests/config.py,sha256=dwWeY0tatdbwl9BqbhZ7EljoZDCtKmGO5fvOAIxeXmA,5873
 airbyte_ops_mcp/live_tests/connection_fetcher.py,sha256=5wIiA0VvCFNEc-fr6Po18gZMX3E5fyPOGf2SuVOqv5U,12799
-airbyte_ops_mcp/live_tests/connection_secret_retriever.py,sha256=x8JNG2byxnS5xNLnFK10LzsqeY9FU8ZECyReh1WOf0s,5864
+airbyte_ops_mcp/live_tests/connection_secret_retriever.py,sha256=C3yGoE8JSlkypKlpGYoI-5oczFexHyl5Z3avkDKyqQc,6445
 airbyte_ops_mcp/live_tests/connector_runner.py,sha256=fGE_TCii9zhC3pbyBupJ3JVkuxOWB59Q1DgigcF3q04,9707
 airbyte_ops_mcp/live_tests/evaluation_modes.py,sha256=lAL6pEDmy_XCC7_m4_NXjt_f6Z8CXeAhMkc0FU8bm_M,1364
 airbyte_ops_mcp/live_tests/http_metrics.py,sha256=oTD7f2MnQOvx4plOxHop2bInQ0-whvuToSsrC7TIM-M,12469
@@ -399,14 +399,14 @@ airbyte_ops_mcp/mcp/registry.py,sha256=PW-VYUj42qx2pQ_apUkVaoUFq7VgB9zEU7-aGrkSC
 airbyte_ops_mcp/mcp/server.py,sha256=7zi91xioVTx1q-bEleekZH2c2JnbzDQt_6zxdEwnLbg,2958
 airbyte_ops_mcp/mcp/server_info.py,sha256=4yNBA_N_vUyLwVJqp7abyFuzZkcnv6-ck_Beb2SXqTE,2426
 airbyte_ops_mcp/prod_db_access/__init__.py,sha256=5pxouMPY1beyWlB0UwPnbaLTKTHqU6X82rbbgKY2vYU,1069
-airbyte_ops_mcp/prod_db_access/db_engine.py,sha256=Pfc2wASTlinaRep9GwRpI9NInbFpIowJL3sLov3MyrU,4371
+airbyte_ops_mcp/prod_db_access/db_engine.py,sha256=ia1KcuQOXi3Qhy_MnxYmccCBJ4rAt_d4nVDjcyzje6o,4289
 airbyte_ops_mcp/prod_db_access/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airbyte_ops_mcp/prod_db_access/queries.py,sha256=bZZntlTXzvI2uSAFMkOfb_LcfCOlxyTuOddRVw4yU2M,8676
 airbyte_ops_mcp/prod_db_access/sql.py,sha256=zHPucNuMlfxz3aU8vYo1ziiGk0lIncG9XmblEoRDd4c,12725
 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_internal_ops-0.1.6.dist-info/METADATA,sha256=T4zoKmfLQHIBSZ0L-KIZql-OdczU1DJLB3rj3OlMJmQ,5282
-airbyte_internal_ops-0.1.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-airbyte_internal_ops-0.1.6.dist-info/entry_points.txt,sha256=eUgJ9xIy9PlR-CgRbqRMsh1NVp6jz08v9bul9vCYlU4,111
-airbyte_internal_ops-0.1.6.dist-info/RECORD,,
+airbyte_internal_ops-0.1.7.dist-info/METADATA,sha256=lIb2MpFriSDAUAYqwerfgA6a3zz3YOrwzDY17cyIZno,5282
+airbyte_internal_ops-0.1.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+airbyte_internal_ops-0.1.7.dist-info/entry_points.txt,sha256=eUgJ9xIy9PlR-CgRbqRMsh1NVp6jz08v9bul9vCYlU4,111
+airbyte_internal_ops-0.1.7.dist-info/RECORD,,

airbyte_ops_mcp/live_tests/connection_secret_retriever.py

@@ -32,7 +32,9 @@ from __future__ import annotations
 
 import logging
 import os
+import tempfile
 from dataclasses import replace
+from pathlib import Path
 from typing import TYPE_CHECKING
 
 from airbyte_ops_mcp.connection_config_retriever import (
@@ -57,18 +59,27 @@ def _ensure_gcp_credentials_env() -> None:
     """Ensure GCP credentials are available via standard env var.
 
     If GOOGLE_APPLICATION_CREDENTIALS is not set but GCP_PROD_DB_ACCESS_CREDENTIALS is,
-    copy the value to GOOGLE_APPLICATION_CREDENTIALS. This provides a fallback
-    for internal employees who use GCP_PROD_DB_ACCESS_CREDENTIALS as their standard
-    credential path for prod database access.
+    write the JSON credentials to a temp file and set GOOGLE_APPLICATION_CREDENTIALS
+    to point to that file. This provides a fallback for internal employees who use
+    GCP_PROD_DB_ACCESS_CREDENTIALS as their standard credential source for prod
+    database access.
+
+    Note: GOOGLE_APPLICATION_CREDENTIALS must be a file path, not JSON content.
+    The GCP_PROD_DB_ACCESS_CREDENTIALS env var contains the JSON content directly,
+    so we write it to a temp file first.
 
     This function is idempotent and safe to call multiple times.
     """
     if ENV_GOOGLE_APPLICATION_CREDENTIALS not in os.environ:
         gsm_creds = os.getenv(ENV_GCP_PROD_DB_ACCESS_CREDENTIALS)
         if gsm_creds:
-            os.environ[ENV_GOOGLE_APPLICATION_CREDENTIALS] = gsm_creds
+            # Write credentials to a temp file since GOOGLE_APPLICATION_CREDENTIALS
+            # expects a file path, not JSON content
+            creds_file = Path(tempfile.gettempdir()) / "gcp_prod_db_creds.json"
+            creds_file.write_text(gsm_creds)
+            os.environ[ENV_GOOGLE_APPLICATION_CREDENTIALS] = str(creds_file)
             logger.debug(
-                f"Using {ENV_GCP_PROD_DB_ACCESS_CREDENTIALS} as fallback for "
+                f"Wrote {ENV_GCP_PROD_DB_ACCESS_CREDENTIALS} to {creds_file} and set "
                 f"{ENV_GOOGLE_APPLICATION_CREDENTIALS}"
             )
 

airbyte_ops_mcp/prod_db_access/db_engine.py

@@ -19,9 +19,8 @@ from google.cloud import secretmanager
 from google.cloud.sql.connector import Connector
 from google.cloud.sql.connector.enums import IPTypes
 
-# Secret ID for database connection details
-CONNECTION_RETRIEVER_PG_CONNECTION_DETAILS_SECRET_ID = (
-    "projects/587336813068/secrets/CONNECTION_RETRIEVER_PG_CONNECTION_DETAILS"
+from airbyte_ops_mcp.constants import (
+    CONNECTION_RETRIEVER_PG_CONNECTION_DETAILS_SECRET_ID,
 )
 
 PG_DRIVER = "pg8000"