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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airbyte-internal-ops
-Version: 0.1.4
+Version: 0.1.6
 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
@@ -24,6 +24,7 @@ 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: google-cloud-storage<3.0,>=2.8.0
 Requires-Dist: jinja2<4.0,>=3.1.2
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: python-dotenv<2.0,>=1.0.0
@@ -95,3 +96,71 @@ AIRBYTE_CLOUD_TEST_WORKSPACE_ID="..."
 Once configured, use the `test_my_tools` prompt by typing "/test" into your agent and selecting the auto-complete option for the `test_my_tools` prompt.
 
 This prompt will step through all the tools, demoing their capabilities.
+
+## Usage Examples
+
+### Testing MCP Tools Locally
+
+Use the `mcp-tool-test` poe task to test tools directly:
+
+```bash
+# List connectors in a repo
+poe mcp-tool-test list_connectors_in_repo '{"repo_path": "/path/to/airbyte"}'
+
+# Get cloud connector version
+poe mcp-tool-test get_cloud_connector_version '{"workspace_id": "...", "actor_id": "..."}'
+```
+
+### Using Cloud SQL Proxy for Database Tools
+
+Some tools (like `list_org_connections_by_source_type_db`) require access to the Airbyte Cloud Prod DB Replica. To test these locally:
+
+1. Authenticate with GCP:
+   ```bash
+   gcloud auth login
+   gcloud auth application-default login
+   ```
+
+2. Start Cloud SQL Proxy:
+   ```bash
+   cloud-sql-proxy prod-ab-cloud-proj:us-west3:prod-pgsql-replica --port=15432
+   ```
+
+3. Run the tool with proxy environment variables:
+   ```bash
+   USE_CLOUD_SQL_PROXY=1 DB_PORT=15432 poe mcp-tool-test list_org_connections_by_source_type_db \
+     '{"organization_id": "...", "connector_canonical_name": "source-youtube-analytics"}'
+   ```
+
+## Instructions for Agents
+
+When working with a user to set up and use this MCP server, agents should follow these steps for authentication:
+
+### GCP Authentication Flow
+
+For tools that require GCP access (database queries, secret manager, etc.), guide the user through authentication:
+
+1. **Check if gcloud is installed.** If not, install it:
+   ```bash
+   curl -sSL https://sdk.cloud.google.com | bash -s -- --disable-prompts
+   ```
+
+2. **Run gcloud auth login with `--no-launch-browser`** to get a verification URL:
+   ```bash
+   gcloud auth login --no-launch-browser
+   ```
+   Send the verification URL to the user and ask them to complete sign-in and provide the verification code.
+
+3. **Set up Application Default Credentials (ADC)** for library access:
+   ```bash
+   gcloud auth application-default login --no-launch-browser
+   ```
+   Again, send the URL to the user and collect the verification code.
+
+4. **For database tools**, start Cloud SQL Proxy on an available port and use the `USE_CLOUD_SQL_PROXY` and `DB_PORT` environment variables.
+
+### Airbyte Cloud Authentication
+
+For tools that interact with Airbyte Cloud API, ensure the user has configured:
+- `AIRBYTE_CLOUD_CLIENT_ID` and `AIRBYTE_CLOUD_CLIENT_SECRET` in their `.env` file
+- For admin operations: `AIRBYTE_INTERNAL_ADMIN_FLAG=airbyte.io` and `AIRBYTE_INTERNAL_ADMIN_USER`

{airbyte_internal_ops-0.1.4.dist-info → airbyte_internal_ops-0.1.6.dist-info}/RECORD

@@ -1,6 +1,6 @@
-airbyte_ops_mcp/__init__.py,sha256=N6YUWYTrt2ObX7ha1o-9uuBbx9Ya44ucOwH01e9KHSQ,358
+airbyte_ops_mcp/__init__.py,sha256=HhzURuYr29_UIdMrnWYaZB8ENr_kFkBdm4uqeiIW3Vw,760
 airbyte_ops_mcp/_annotations.py,sha256=MO-SBDnbykxxHDESG7d8rviZZ4WlZgJKv0a8eBqcEzQ,1757
-airbyte_ops_mcp/constants.py,sha256=UXCILNepI7YFeyCwBG6Z2h6ckoYzeSvK3mi13wiTI-4,549
+airbyte_ops_mcp/constants.py,sha256=eDO262DXtlPbWENwpk9GTllmkz9S_xT7G4qX2-BcfkE,1743
 airbyte_ops_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airbyte_ops_mcp/_legacy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airbyte_ops_mcp/_legacy/airbyte_ci/README.md,sha256=qEYx4geDR8AEDjrcA303h7Nol-CMDLojxUyiGzQprM8,236
@@ -106,7 +106,7 @@ airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/__i
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/commands.py,sha256=omSpuwd7t8O-BuJI0YRTncfP1_r1n8bzFsGY_1ok11M,12294
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/consts.py,sha256=DGw8amiCwBcnFEtv1GmeNM-SQx04QRuITiifulA5bsM,1677
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/context.py,sha256=F1qKYh38JoA1k_BwL_Mt1LL82ErzJdGfJsrFHuLwoD0,13217
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/pipeline.py,sha256=koTXeGqYRekKBBsT29Gmo5H6QMfGpqc5saJdhWr7M1U,5268
+airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/pipeline.py,sha256=z5zef7RGdhFRDr5IdvxS57GSagwafv8hpW8U-Z-jaUQ,5068
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/reports.py,sha256=vl7AAX6ZxBk4KCEAJHzXqWntQ2k00abxGs5zDOWeRsU,9999
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/build_image/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/build_image/commands.py,sha256=W4goGOU8MqgLOdCneSGiO4uVQX4dCVEGU-UYyXekPl8,3518
@@ -126,10 +126,6 @@ airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/lis
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/migrate_to_inline_schemas/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/migrate_to_inline_schemas/commands.py,sha256=v-YHJDaR8MAT5VsguXxXbw2Gr5XHoiK9X1Sr75LkhMk,1001
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/migrate_to_inline_schemas/pipeline.py,sha256=GAPtRw9F1R_NtN1zKfOTAYezfovvr_IfTBjFJkg0yV8,14783
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/publish/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/publish/commands.py,sha256=2T8xxO0z9PPjvEbyDb1maLmN5GNDtb7Wj_T_Je1aW2A,9203
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/publish/context.py,sha256=95LXULM_1T-9ojfjSV8fzkJqpkmlLTWVmDlVnhZTXL8,6979
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/publish/pipeline.py,sha256=jVEw0HXYViHW7EVcQpOaUg64vwIFfs6Z1TFIsWtfIYY,43457
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/pull_request/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/pull_request/commands.py,sha256=ieLFxllbFHCgKwQnc196zodEN5Zo10SSH5zmEI7vZbE,2295
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/pull_request/pipeline.py,sha256=sJ6GLqJoYlEym5cshTxuqqvXD5_BFkREo-vdOJ3a8ks,3869
@@ -156,8 +152,6 @@ airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/metadata/comma
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/metadata/pipeline.py,sha256=J1Lk53KBshpoe11uI0VO3--xD59o5pozPx3vCKxvonI,10681
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/poetry/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/poetry/commands.py,sha256=ACHYsNDQ8kegCncZ-kb3jeN4T9if_vykwksiUDsFNpM,946
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/poetry/publish/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/poetry/publish/commands.py,sha256=cvmaqdXKXNvqizAAdcBfm36RCfTRr3-Zdoe30BZWvJ4,4234
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/steps/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/steps/base_image.py,sha256=yUw6xifY9lrKj3Lv5OpJoR95haz-XPBavSBfJjeH6uU,7382
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/steps/docker.py,sha256=inJ66IFQslQ6dSLdTOGgJow6IkjnZwzymgy36NLXduM,5235
@@ -165,7 +159,6 @@ airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/steps/gradle.p
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/steps/no_op.py,sha256=3NrvixYy092kKQ08aDFoEkc0ZGVdcdGU7qdIeJNV2dM,612
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/steps/poetry.py,sha256=ww0M-qYzl6fBjU8a4HhZ_Lz7mEa0Ku3eBAx0eCJ6rjM,1603
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/steps/pull_request.py,sha256=5PUJXmuZb6VgWcFJVlDZlQETeDwnwndSomo7ELg0TH8,2027
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/steps/python_registry.py,sha256=ozpe2t3l4zEA5oUQJB8z1-0ONCzeEY-8_KPSYVO087c,9560
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/test/__init__.py,sha256=AH4QS9r16-zwNGtTzbA9kbvZ2PaByYdw2SWYtWstoIM,708
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/test/commands.py,sha256=P5UlzG7hG4twSe7HKxWTMDb9qmh-7T4dsdTA7pyfYRk,5375
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/test/models.py,sha256=OXP-WtenvcEgU2ks5RhVxLkHbsFPmV_RwLqdiLtvA6s,1902
@@ -192,7 +185,6 @@ airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/models/steps.py,sha256=t7
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/models/contexts/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/models/contexts/click_pipeline_context.py,sha256=aPwt8XTzt2NOIU2ItAj0mLTZu4lmQwgwut7iOfM6lnI,5458
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/models/contexts/pipeline_context.py,sha256=68lfAR5NuLMqUgaHayJKkwXt-vqcucsKWq5L6PA4rqA,15640
-airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/models/contexts/python_registry_publish.py,sha256=wblo574PR0ScmVQ1Loap-GLM-_jvwl7KK3j2Z6bFzro,5058
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_qa/README.md,sha256=ajuFHew-gZPqGwoX3AAPbGvuysUyQJqTKmzvVEDJRWk,5146
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_qa/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airbyte_ops_mcp/_legacy/airbyte_ci/connector_qa/cli.py,sha256=-GpHJvlZDBU49XJmepBDqXlkRmf47K4GIpQYG0tVkKk,5152
@@ -352,38 +344,36 @@ 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=UI4RKMfFWOXHBtvh7TIw1v0BnV4rQQGuUgh8e9yvpos,19907
+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=iEkwromM8VByNa9V-2zBnAJZfr6CzCTiYSajqu4hSok,31996
+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/registry.py,sha256=OpON1p4_A-G-FSfIpr6UlKYTjcj_zyiprKOu7qxwuhc,5787
+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/connection_config_retriever/__init__.py,sha256=Xoi-YvARrNPhECdpwEDDkdwEpnvj8zuUlwULpf4iRrU,800
+airbyte_ops_mcp/connection_config_retriever/audit_logging.py,sha256=GjT4dVa0TtvGDmiBz9qwzcYCnSf9hTo7UM6l7ubUNE8,2846
+airbyte_ops_mcp/connection_config_retriever/retrieval.py,sha256=Bpbl1qoR0z1F8dgQOj-dB95UxqzGTXbkW4nHMPBIO-g,12112
+airbyte_ops_mcp/connection_config_retriever/secrets_resolution.py,sha256=12g0lZzhCzAPl4Iv4eMW6d76mvXjIBGspOnNhywzks4,3644
 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/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/connection_secret_retriever.py,sha256=x8JNG2byxnS5xNLnFK10LzsqeY9FU8ZECyReh1WOf0s,5864
 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
 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
@@ -399,15 +389,24 @@ airbyte_ops_mcp/mcp/cloud_connector_versions.py,sha256=G8mVzhvSCmrTEqDseV57-wwL0
 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=4KTGCfct69WTCCrxZ23l-Nrz3iKZA5tuieNM_Aa4aTs,17408
+airbyte_ops_mcp/mcp/github_repo_ops.py,sha256=pE6RyATBfUKBOffQ6iyNJhHfCLjwq0pzVmmFYUg39UQ,5375
+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/prod_db_queries.py,sha256=L0p7LlF7GV5mXoH_2GwgQp04sXL2QA5v-RiuLsTb-yA,12614
 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=oJOrwZP7hAwx2pjChZUZYv-aA4NITWiCRiXFiKVOf8E,2843
-airbyte_ops_mcp/mcp/server_info.py,sha256=VcNLn6CSy2UcgM2V1ep9-JGwV63p3cK1G14BFHXEn0U,2402
-airbyte_internal_ops-0.1.4.dist-info/METADATA,sha256=5vi9tCrTgfHK0NnN7hG2jkdRsuBalql3pCVO8nVh2RE,2866
-airbyte_internal_ops-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-airbyte_internal_ops-0.1.4.dist-info/entry_points.txt,sha256=eUgJ9xIy9PlR-CgRbqRMsh1NVp6jz08v9bul9vCYlU4,111
-airbyte_internal_ops-0.1.4.dist-info/RECORD,,
+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/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_ops_mcp/__init__.py

@@ -1,4 +1,19 @@
-"""Airbyte Admin MCP - MCP and API interfaces that let the agents do the admin work."""
+"""Airbyte Admin MCP - MCP and API interfaces that let the agents do the admin work.
+
+.. include:: ../../../README.md
+"""
+
+from airbyte_ops_mcp import (
+    airbyte_repo,
+    cli,
+    cloud_admin,
+    connection_config_retriever,
+    constants,
+    live_tests,
+    mcp,
+    prod_db_access,
+    registry,
+)
 
 __version__ = "0.1.0"
 
@@ -13,4 +28,17 @@ def get_version() -> str:
     return __version__
 
 
-__all__ = ["__version__", "get_version", "hello"]
+__all__ = [
+    "__version__",
+    "airbyte_repo",
+    "cli",
+    "cloud_admin",
+    "connection_config_retriever",
+    "constants",
+    "get_version",
+    "hello",
+    "live_tests",
+    "mcp",
+    "prod_db_access",
+    "registry",
+]

airbyte_ops_mcp/_legacy/airbyte_ci/connector_pipelines/airbyte_ci/connectors/pipeline.py

@@ -16,7 +16,6 @@ from connector_ops.utils import ConnectorLanguage  # type: ignore
 from consts import ContextState
 from dagger import Config
 from pipelines.airbyte_ci.connectors.context import ConnectorContext
-from pipelines.airbyte_ci.connectors.publish.context import PublishConnectorContext
 from pipelines.airbyte_ci.connectors.test.context import ConnectorTestContext
 from pipelines.airbyte_ci.steps.no_op import NoOpStep
 from pipelines.dagger.actions.system import docker
@@ -55,7 +54,6 @@ async def context_to_step_result(context: PipelineContext) -> StepResult:
 async def run_report_complete_pipeline(
     dagger_client: dagger.Client,
     contexts: List[ConnectorContext]
-    | List[PublishConnectorContext]
     | List[PipelineContext]
     | List[ConnectorTestContext],
 ) -> None:
@@ -87,18 +85,14 @@ async def run_report_complete_pipeline(
 
 
 async def run_connectors_pipelines(
-    contexts: List[ConnectorContext]
-    | List[PublishConnectorContext]
-    | List[ConnectorTestContext],
+    contexts: List[ConnectorContext] | List[ConnectorTestContext],
     connector_pipeline: Callable,
     pipeline_name: str,
     concurrency: int,
     dagger_logs_path: Optional[Path],
     execute_timeout: Optional[int],
     *args: Any,
-) -> (
-    List[ConnectorContext] | List[PublishConnectorContext] | List[ConnectorTestContext]
-):
+) -> List[ConnectorContext] | List[ConnectorTestContext]:
     """Run a connector pipeline for all the connector contexts."""
 
     default_connectors_semaphore = anyio.Semaphore(concurrency)

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):
@@ -27,9 +34,25 @@ class ConnectorLanguage(StrEnum):
     MANIFEST_ONLY = "manifest-only"
 
 
+class ConnectorType(StrEnum):
+    """Connector types (source or destination)."""
+
+    SOURCE = "source"
+    DESTINATION = "destination"
+
+
+class ConnectorSubtype(StrEnum):
+    """Connector subtypes based on data source category."""
+
+    API = "api"
+    DATABASE = "database"
+    FILE = "file"
+    CUSTOM = "custom"
+
+
 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 +63,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 +181,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,
@@ -205,6 +261,106 @@ def get_connectors_by_language(
     return connectors_by_language
 
 
+def get_connectors_by_type(
+    repo_path: str | Path,
+    connector_type: ConnectorType,
+) -> set[str]:
+    """Get set of all connector IDs for a specific type (source or destination).
+
+    This function reads connector directories to determine which connectors
+    match the specified type based on their metadata.yaml or name prefix.
+
+    Args:
+        repo_path: Path to the Airbyte monorepo
+        connector_type: Type to filter by (source or destination)
+
+    Returns:
+        Set of connector technical names matching the specified type
+
+    Example:
+        >>> source_connectors = get_connectors_by_type(
+        ...     "/path/to/airbyte", ConnectorType.SOURCE
+        ... )
+        >>> "source-postgres" in source_connectors
+        True
+    """
+    repo_path = Path(repo_path)
+    connectors_dir = repo_path / CONNECTOR_PATH_PREFIX
+
+    if not connectors_dir.exists():
+        raise ValueError(f"Connectors directory not found: {connectors_dir}")
+
+    type_value = connector_type.value
+    connectors_by_type = set()
+
+    for connector_dir in connectors_dir.iterdir():
+        if not connector_dir.is_dir():
+            continue
+
+        connector_name = connector_dir.name
+
+        # First try to get type from metadata.yaml
+        metadata = get_connector_metadata(repo_path, connector_name)
+        if metadata:
+            metadata_type = metadata.get("connectorType")
+            if metadata_type == type_value:
+                connectors_by_type.add(connector_name)
+                continue
+
+        # Fallback to name prefix detection
+        if connector_name.startswith(f"{type_value}-"):
+            connectors_by_type.add(connector_name)
+
+    return connectors_by_type
+
+
+def get_connectors_by_subtype(
+    repo_path: str | Path,
+    connector_subtype: ConnectorSubtype,
+) -> set[str]:
+    """Get set of all connector IDs for a specific subtype (api, database, file, etc.).
+
+    This function reads connector metadata.yaml files to determine which connectors
+    match the specified subtype.
+
+    Args:
+        repo_path: Path to the Airbyte monorepo
+        connector_subtype: Subtype to filter by (api, database, file, custom)
+
+    Returns:
+        Set of connector technical names matching the specified subtype
+
+    Example:
+        >>> database_connectors = get_connectors_by_subtype(
+        ...     "/path/to/airbyte", ConnectorSubtype.DATABASE
+        ... )
+        >>> "source-postgres" in database_connectors
+        True
+    """
+    repo_path = Path(repo_path)
+    connectors_dir = repo_path / CONNECTOR_PATH_PREFIX
+
+    if not connectors_dir.exists():
+        raise ValueError(f"Connectors directory not found: {connectors_dir}")
+
+    subtype_value = connector_subtype.value
+    connectors_by_subtype = set()
+
+    for connector_dir in connectors_dir.iterdir():
+        if not connector_dir.is_dir():
+            continue
+
+        connector_name = connector_dir.name
+        metadata = get_connector_metadata(repo_path, connector_name)
+
+        if metadata:
+            metadata_subtype = metadata.get("connectorSubtype")
+            if metadata_subtype == subtype_value:
+                connectors_by_subtype.add(connector_name)
+
+    return connectors_by_subtype
+
+
 @lru_cache(maxsize=1024)
 def _detect_connector_language(connector_dir: Path, connector_name: str) -> str | None:
     """Detect the language of a connector based on its file structure.
@@ -361,6 +517,8 @@ def list_connectors(
     modified: bool | None = None,
     language_filter: set[str] | None = None,
     language_exclude: set[str] | None = None,
+    connector_type: str | None = None,
+    connector_subtype: str | None = None,
     base_ref: str | None = None,
     head_ref: str | None = None,
 ) -> ConnectorListResult:
@@ -377,6 +535,8 @@ def list_connectors(
             False=not-modified only, None=all)
         language_filter: Set of languages to include (python, java, low-code, manifest-only)
         language_exclude: Set of languages to exclude (mutually exclusive with language_filter)
+        connector_type: Filter by connector type (source, destination, or None for all)
+        connector_subtype: Filter by connector subtype (api, database, file, custom, or None for all)
         base_ref: Base git reference for modification detection (default: "origin/main")
         head_ref: Head git reference for modification detection (default: "HEAD")
 
@@ -408,7 +568,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:
@@ -435,6 +595,18 @@ def list_connectors(
             excluded = get_connectors_by_language(repo_path, ConnectorLanguage(lang))
             result -= excluded
 
+    # Apply connector type filter
+    if connector_type is not None:
+        type_set = get_connectors_by_type(repo_path, ConnectorType(connector_type))
+        result &= type_set
+
+    # Apply connector subtype filter
+    if connector_subtype is not None:
+        subtype_set = get_connectors_by_subtype(
+            repo_path, ConnectorSubtype(connector_subtype)
+        )
+        result &= subtype_set
+
     return ConnectorListResult(
         connectors=sorted(result),
         count=len(result),

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)