airbyte-internal-ops 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

{airbyte_internal_ops-0.1.3.dist-info → airbyte_internal_ops-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airbyte-internal-ops
-Version: 0.1.3
+Version: 0.1.5
 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
@@ -10,18 +10,20 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Requires-Python: >=3.11
-Requires-Dist: airbyte-cdk<7.0,>=6.0.0
+Requires-Python: <3.13,>=3.11
+Requires-Dist: airbyte-cdk<8.0,>=7.3.9
 Requires-Dist: airbyte-protocol-models-pdv2>=0.13.0
-Requires-Dist: airbyte>=0.1.0
+Requires-Dist: airbyte>=0.34.0
 Requires-Dist: asyncer<1.0,>=0.0.4
 Requires-Dist: click<9.0,>=8.1.3
+Requires-Dist: cloud-sql-python-connector[pg8000]<2.0,>=1.7.0
 Requires-Dist: cyclopts<5.0,>=4.0.0
 Requires-Dist: docker<7.0,>=6.0
 Requires-Dist: dpath<3.0,>=2.1.5
 Requires-Dist: fastmcp<3.0,>=2.12.1
 Requires-Dist: gitpython<4.0,>=3.1.29
+Requires-Dist: google-cloud-logging<4.0,>=3.9.0
+Requires-Dist: google-cloud-secret-manager<3.0,>=2.18.0
 Requires-Dist: jinja2<4.0,>=3.1.2
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: python-dotenv<2.0,>=1.0.0
@@ -30,6 +32,7 @@ Requires-Dist: requests<3.0,>=2.31.0
 Requires-Dist: rich<14.0,>=13.0.0
 Requires-Dist: semver<4.0,>=3.0.1
 Requires-Dist: slack-sdk<4.0,>=3.36.0
+Requires-Dist: sqlalchemy<3.0,>=2.0.0
 Requires-Dist: toml<1.0,>=0.10.2
 Description-Content-Type: text/markdown
 

{airbyte_internal_ops-0.1.3.dist-info → airbyte_internal_ops-0.1.5.dist-info}/RECORD

@@ -138,7 +138,7 @@ airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/tes
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/test/context.py,sha256=ounJj-ZjLBTtZx088t6u_UKy_aHJZmMmfXskSTbWjFM,9789
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/test/pipeline.py,sha256=QaxeepQxl5PX5xitTt7pbt2eIkQJTgAqmmxL2nFqvvI,3020
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/test/steps/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/test/steps/common.py,sha256=e4SWZVYIbUiTnTSbamMJLLIDlzrTbh78a1KcXwkpsv0,42952
+airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/test/steps/common.py,sha256=Gv9kFNgteTqGV92TgQxzlUefEi6NL0VNZy31V1l7DwM,42948
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/test/steps/java_connectors.py,sha256=UOwoLiSpVdDSGFhrV__XTbJgF4eQj-MyS3tezej09qA,5774
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/test/steps/manifest_only_connectors.py,sha256=aVD-FYW_5yDryZUXIjgm-H_jJetiAfmxRwiPQV4Fnlc,9529
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/test/steps/python_connectors.py,sha256=BJKZLovnfA1_vYNcEC2ko8xeD0FhSLorhccVCx7q7RY,15755
@@ -352,42 +352,62 @@ airbyte_ops_mcp/_legacy/airbyte_ci/metadata_service/templates/render.py,sha256=g
 airbyte_ops_mcp/_legacy/airbyte_ci/metadata_service/validators/metadata_validator.py,sha256=xh3Q7XgXcFaHS4Fa9A6anDecudaEXQX-SOetjgm5L4Q,14653
 airbyte_ops_mcp/airbyte_repo/__init__.py,sha256=3yEpCOop1h33UuCtU7-1UB3kun2QsxtgvsfVEpuqN6s,1572
 airbyte_ops_mcp/airbyte_repo/bump_version.py,sha256=rtXYv105BhElj7twQmy20VjNsj6vCb0Ulrzmc4Z-1cE,15384
-airbyte_ops_mcp/airbyte_repo/list_connectors.py,sha256=IyYLtcBDsRmxe0SGJd3VHsO_0iaiTjsNMV6BqBUtZPI,14634
-airbyte_ops_mcp/airbyte_repo/utils.py,sha256=IZ7bJJz3I0BBfSxRfINY193VylMeg2GEXe0A1ZCiR80,3030
+airbyte_ops_mcp/airbyte_repo/list_connectors.py,sha256=rMiQFPGANSQzaBncVUh4oO8zdLafU4gxxqZB9o7_FAA,15714
+airbyte_ops_mcp/airbyte_repo/utils.py,sha256=TXlOAfhiu_hVRNjCxB4PRPVDhTWCU5lYmgqz4QG_-EA,3201
 airbyte_ops_mcp/cli/__init__.py,sha256=XpL7FyVfgabfBF2JR7u7NwJ2krlYqjd_OwLcWf-Xc7s,114
 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=7lx5FE65hpBBxsNDNouoAkounCUqxy1xZE-xlDpom9Q,20735
+airbyte_ops_mcp/cli/cloud.py,sha256=BMFYs5bTEgdOhxwzBrtSyYMKaHhXnMM_SGzK2hFDPBY,32076
 airbyte_ops_mcp/cli/gh.py,sha256=91b1AxFXvHQCFyXhrrym-756ZjnMCqvxFdmwCtma1zI,2046
 airbyte_ops_mcp/cli/registry.py,sha256=wHyfiysASuy-HGvLJIiU8TRguaiuqRaXQP9QJ-LC7bk,2940
-airbyte_ops_mcp/cli/repo.py,sha256=2LurE6AvRpdgYpiRZuvA33bI1-e7_e0QsFbEO7Bba_g,15780
+airbyte_ops_mcp/cli/repo.py,sha256=G1hoQpH0XYhUH3FFOsia9xabGB0LP9o3XcwBuqvFVo0,16331
 airbyte_ops_mcp/cloud_admin/__init__.py,sha256=cqE96Q10Kp6elhH9DAi6TVsIwSUy3sooDLLrxTaktGk,816
 airbyte_ops_mcp/cloud_admin/api_client.py,sha256=4vZv1J4S2Q8ETl6gIB20X1X6KHTVV-bx__b2Ax8oqyc,17358
 airbyte_ops_mcp/cloud_admin/auth.py,sha256=j45pRR8fg6CLwVdn7Uu5KW_kTz_CjRP6ZJGUzqHj_Dk,2558
+airbyte_ops_mcp/cloud_admin/connection_config.py,sha256=UtbIwuB7CA3WJr9oYRwlKDsjciqd_9ewWdml2f8DuXw,4887
 airbyte_ops_mcp/cloud_admin/models.py,sha256=YZ3FbEW-tZa50khKTTl4Bzvy_LsGyyQd6qcpXo62jls,2670
-airbyte_ops_mcp/live_tests/__init__.py,sha256=udSsiz8WNI69WSn6Ib9uGB1LEblL6ElVHaY8hWGMHLc,525
+airbyte_ops_mcp/live_tests/__init__.py,sha256=qJac67dt6DQCqif39HqeiG3Tr9xrxfP-ala8HsLZKis,1020
 airbyte_ops_mcp/live_tests/ci_output.py,sha256=NQATOGid0OCbLEl2NOtGK4cHLL5OxXhjmtanBjIlCyE,11369
-airbyte_ops_mcp/live_tests/connection_fetcher.py,sha256=ffMD6bGWW4a1MA83Fs-tXLyOnT6aO9dvm1Kk9FT9ocA,6662
+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=XBuPfmA9GnNCHlJdDocwiB_UIx8F0D6OHBNNIDPrfk0,5869
 airbyte_ops_mcp/live_tests/connector_runner.py,sha256=fGE_TCii9zhC3pbyBupJ3JVkuxOWB59Q1DgigcF3q04,9707
-airbyte_ops_mcp/live_tests/http_metrics.py,sha256=ryZANWz4CXT24j7PxeE7r3gDuRbcEGR9ezLrav_E32U,9701
+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
 airbyte_ops_mcp/live_tests/models.py,sha256=brtAT9oO1TwjFcP91dFcu0XcUNqQb-jf7di1zkoVEuo,8782
+airbyte_ops_mcp/live_tests/obfuscation.py,sha256=JanpCLj6M9-_Zto6PABzNaY3OA93Frq3YpJ1411QNtQ,4395
+airbyte_ops_mcp/live_tests/schema_generation.py,sha256=VQfn2WbsMptfjO_ub709FYSwwwvjOuWy2Jut7o5ThIs,5308
+airbyte_ops_mcp/live_tests/_connection_retriever/__init__.py,sha256=3Mm9Lauqf4TcdeXhujbXXitl_A7_3iUbc0iQqd6gfog,1125
+airbyte_ops_mcp/live_tests/_connection_retriever/audit_logging.py,sha256=DbziKNbZppb9YBly9x3WGWtPhSTkg9MIreIr7mUBkl8,2823
+airbyte_ops_mcp/live_tests/_connection_retriever/consts.py,sha256=k1FJk9Gyc9fWwuzh9TFUU-XpbohhbsrGC5EEJaIPy2c,1021
+airbyte_ops_mcp/live_tests/_connection_retriever/db_access.py,sha256=riTQ7OjwJhZqPQqjwT3WLxbx3TenI2PfgpUpuwoVsd0,2686
+airbyte_ops_mcp/live_tests/_connection_retriever/retrieval.py,sha256=wehmHbEoai_ab9lyobzq6ifMzacJEZv5mwJh53tOi60,12443
+airbyte_ops_mcp/live_tests/_connection_retriever/secrets_resolution.py,sha256=TnqhzkpIaM3aHV34MMw65i1-jVdtRAk9Qqefz7DZJP8,4384
 airbyte_ops_mcp/live_tests/commons/__init__.py,sha256=lNew_sAL4c8dPy3gMFbGC5_FuUX1P6QzGULbqS2H4M0,104
+airbyte_ops_mcp/live_tests/message_cache/__init__.py,sha256=h6G_c73k2_OR8otg1YYV42rfKgKG7eiMIRiNxfo5fzU,545
+airbyte_ops_mcp/live_tests/message_cache/duckdb_cache.py,sha256=LVj8sCA1pgJimT8NFQJzbsW8dEzIX5hB280sFtkP3xA,15355
+airbyte_ops_mcp/live_tests/regression/__init__.py,sha256=SsgBKULvowwYeeS5GmrtyyVzghS9-cfZwdxskWCcRBg,851
+airbyte_ops_mcp/live_tests/regression/comparators.py,sha256=MJkLZEKHivgrG0-3Y8BzLtnXlOb16t9z-reC9lhyObQ,15967
+airbyte_ops_mcp/live_tests/validation/__init__.py,sha256=To16AfC5k4xwtKI4Ep8FZrdl1DexesVKe002reWwmGA,1520
+airbyte_ops_mcp/live_tests/validation/catalog_validators.py,sha256=jqqVAMOk0mtdPgwu4d0hA0ZEjtsNh5gapvGydRv3_qk,12553
+airbyte_ops_mcp/live_tests/validation/record_validators.py,sha256=-7Ir2LWGCrtadK2JLuBgppSyk0RFJX6Nsy0lrabtwrs,7411
 airbyte_ops_mcp/mcp/__init__.py,sha256=QqkNkxzdXlg-W03urBAQ3zmtOKFPf35rXgO9ceUjpng,334
 airbyte_ops_mcp/mcp/_guidance.py,sha256=48tQSnDnxqXtyGJxxgjz0ZiI814o_7Fj7f6R8jpQ7so,2375
-airbyte_ops_mcp/mcp/_mcp_utils.py,sha256=LTekWdqvyzLd_2VFeUZ1Qd_ibIDJ91ZNPS0mEnXnqBo,9011
+airbyte_ops_mcp/mcp/_mcp_utils.py,sha256=Im3J3yCtXPbW_Kp-oh5b11Z2187V3-Zsos5LBgdvQrc,9111
 airbyte_ops_mcp/mcp/cloud_connector_versions.py,sha256=G8mVzhvSCmrTEqDseV57-wwL0s3oWdydgyvgMr027tU,10443
 airbyte_ops_mcp/mcp/connector_analysis.py,sha256=OC4KrOSkMkKPkOisWnSv96BDDE5TQYHq-Jxa2vtjJpo,298
 airbyte_ops_mcp/mcp/connector_qa.py,sha256=aImpqdnqBPDrz10BS0owsV4kuIU2XdalzgbaGZsbOL0,258
 airbyte_ops_mcp/mcp/github.py,sha256=5ZPsSTy4-gummS96xGoG-n2RwCgyg3-UWAvmEmxd5x4,7686
 airbyte_ops_mcp/mcp/github_repo_ops.py,sha256=D7yDtqMISFqaUzqnyA0dLE_6j6G3wHrNz8Byo8ajR8E,4929
+airbyte_ops_mcp/mcp/live_tests.py,sha256=KnxZLuUNmm_3Clt0DU8H9rJ01zOKefnL_wqdSCMDjkE,17992
 airbyte_ops_mcp/mcp/metadata.py,sha256=fwGW97WknR5lfKcQnFtK6dU87aA6TmLj1NkKyqDAV9g,270
 airbyte_ops_mcp/mcp/prerelease.py,sha256=2Mr0LdCLhEc9Q7CEtmganJXHGHCLCXODKlkSapLsSsY,9484
 airbyte_ops_mcp/mcp/prompts.py,sha256=6opN4ZweQxfSdtoK0gL6wTrlxkRvxTQvH1VTmAuhoBE,1645
 airbyte_ops_mcp/mcp/registry.py,sha256=PW-VYUj42qx2pQ_apUkVaoUFq7VgB9zEU7-aGrkSCCw,290
-airbyte_ops_mcp/mcp/server.py,sha256=RtyQfCsxiCvcVHpm6w7Ozt9-77j0KrUYROdryrmhCFo,2673
-airbyte_ops_mcp/mcp/server_info.py,sha256=VcNLn6CSy2UcgM2V1ep9-JGwV63p3cK1G14BFHXEn0U,2402
-airbyte_internal_ops-0.1.3.dist-info/METADATA,sha256=ZDywWwtt2Gj5UjG1mV7AaFcgGpDJIEa9I3UP2aj9sv0,2706
-airbyte_internal_ops-0.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-airbyte_internal_ops-0.1.3.dist-info/entry_points.txt,sha256=eUgJ9xIy9PlR-CgRbqRMsh1NVp6jz08v9bul9vCYlU4,111
-airbyte_internal_ops-0.1.3.dist-info/RECORD,,
+airbyte_ops_mcp/mcp/server.py,sha256=oJOrwZP7hAwx2pjChZUZYv-aA4NITWiCRiXFiKVOf8E,2843
+airbyte_ops_mcp/mcp/server_info.py,sha256=4yNBA_N_vUyLwVJqp7abyFuzZkcnv6-ck_Beb2SXqTE,2426
+airbyte_internal_ops-0.1.5.dist-info/METADATA,sha256=XmVAvFOU6d8fO_atjptKKYgjPPQeBYspi2pupKXHxiY,2866
+airbyte_internal_ops-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+airbyte_internal_ops-0.1.5.dist-info/entry_points.txt,sha256=eUgJ9xIy9PlR-CgRbqRMsh1NVp6jz08v9bul9vCYlU4,111
+airbyte_internal_ops-0.1.5.dist-info/RECORD,,

airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/test/steps/common.py

@@ -976,7 +976,7 @@ class LiveTests(Step):
                 # Add GCP credentials from the environment and point google to their location (also required for connection-retriever)
                 .with_new_file(
                     "/tmp/credentials.json",
-                    contents=os.getenv("GCP_INTEGRATION_TESTER_CREDENTIALS", ""),
+                    contents=os.getenv("GCP_PROD_DB_ACCESS_CREDENTIALS", ""),
                 )
                 .with_env_variable(
                     "GOOGLE_APPLICATION_CREDENTIALS", "/tmp/credentials.json"

airbyte_ops_mcp/airbyte_repo/list_connectors.py

@@ -7,15 +7,22 @@ by comparing git diffs between branches.
 
 from __future__ import annotations
 
+import logging
 import re
 import subprocess
 from dataclasses import dataclass
 from enum import StrEnum
 from functools import lru_cache
 from pathlib import Path
+from typing import Any
+
+import yaml
 
 CONNECTOR_PATH_PREFIX = "airbyte-integrations/connectors"
 METADATA_FILE_NAME = "metadata.yaml"
+GIT_DEFAULT_BRANCH = "origin/master"
+
+logger = logging.getLogger(__name__)
 
 
 class ConnectorLanguage(StrEnum):
@@ -29,7 +36,7 @@ class ConnectorLanguage(StrEnum):
 
 def get_modified_connectors(
     repo_path: str | Path,
-    base_ref: str = "origin/main",
+    base_ref: str = GIT_DEFAULT_BRANCH,
     head_ref: str = "HEAD",
 ) -> list[str]:
     """Get list of connector IDs that have been modified.
@@ -40,14 +47,14 @@ def get_modified_connectors(
 
     Args:
         repo_path: Path to the Airbyte monorepo
-        base_ref: Base git reference to compare against (default: "origin/main")
+        base_ref: Base git reference to compare against (default: "origin/master")
         head_ref: Head git reference to compare (default: "HEAD")
 
     Returns:
         List of connector technical names (e.g., ["source-faker", "destination-postgres"])
 
     Example:
-        >>> connectors = get_changed_connectors("/path/to/airbyte", "origin/main")
+        >>> connectors = get_changed_connectors("/path/to/airbyte", "origin/master")
         >>> print(connectors)
         ['source-faker', 'destination-postgres']
     """
@@ -158,6 +165,39 @@ def get_all_connectors(repo_path: str | Path) -> set[str]:
     return {p.name for p in connectors_dir.iterdir() if p.is_dir()}
 
 
+def get_connector_metadata(
+    repo_path: str | Path,
+    connector_name: str,
+) -> dict[str, Any] | None:
+    """Get metadata for a specific connector.
+
+    Args:
+        repo_path: Path to the Airbyte monorepo
+        connector_name: Technical name of the connector (e.g., "source-faker")
+
+    Returns:
+        The connector's metadata dict (the 'data' section), or None if not found
+
+    Example:
+        >>> metadata = get_connector_metadata("/path/to/airbyte", "source-faker")
+        >>> metadata.get("supportLevel")
+        'certified'
+    """
+    repo_path = Path(repo_path)
+    connector_dir = repo_path / CONNECTOR_PATH_PREFIX / connector_name
+    metadata_file = connector_dir / METADATA_FILE_NAME
+
+    if not metadata_file.exists():
+        return None
+
+    try:
+        with open(metadata_file) as f:
+            metadata = yaml.safe_load(f)
+        return metadata.get("data", {})
+    except Exception:
+        return None
+
+
 def get_connectors_by_language(
     repo_path: str | Path,
     language: ConnectorLanguage,
@@ -408,7 +448,7 @@ def list_connectors(
 
     # Apply modified filter
     if modified is not None:
-        base = base_ref if base_ref is not None else "origin/main"
+        base = base_ref if base_ref is not None else GIT_DEFAULT_BRANCH
         head = head_ref if head_ref is not None else "HEAD"
         changed_set = set(get_modified_connectors(repo_path, base, head))
         if modified:

airbyte_ops_mcp/airbyte_repo/utils.py

@@ -81,9 +81,11 @@ def resolve_diff_range(
 
     # Determine base_ref and head_ref based on PR detection
     if pr_number is not None:
-        # PR detected - use origin/main vs HEAD (assumes CI checked out the PR)
-        # TODO: In future, use GitHub API to get PR diff directly
-        base_ref = "origin/main"
+        # PR detected - use origin/{base_branch} vs HEAD (assumes CI checked out the PR)
+        # Use GITHUB_BASE_REF if available (set by GitHub Actions for PRs)
+        # This handles repos with different default branches (main, master, etc.)
+        base_branch = os.getenv("GITHUB_BASE_REF", "master")
+        base_ref = f"origin/{base_branch}"
         head_ref = "HEAD"
     else:
         # No PR detected - fallback to HEAD~1 vs HEAD (post-merge use case)