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

{airbyte_internal_ops-0.1.5.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.5
+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.5.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,7 +344,7 @@ 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=rMiQFPGANSQzaBncVUh4oO8zdLafU4gxxqZB9o7_FAA,15714
+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
@@ -360,30 +352,28 @@ airbyte_ops_mcp/cli/_shared.py,sha256=jg-xMyGzTCGPqKd8VTfE_3kGPIyO_3Kx5sQbG4rPc0
 airbyte_ops_mcp/cli/app.py,sha256=SEdBpqFUG2O8zGV5ifwptxrLGFph_dLr66-MX9d69gQ,789
 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/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/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.py,sha256=7zi91xioVTx1q-bEleekZH2c2JnbzDQt_6zxdEwnLbg,2958
 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/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

@@ -34,6 +34,22 @@ 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 = GIT_DEFAULT_BRANCH,
@@ -245,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.
@@ -401,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:
@@ -417,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")
 
@@ -475,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/cli/registry.py

@@ -1,21 +1,36 @@
 # Copyright (c) 2025 Airbyte, Inc., all rights reserved.
 """CLI commands for connector registry operations.
 
+This module provides CLI wrappers for registry operations. The core logic
+lives in the `airbyte_ops_mcp.registry` capability module.
+
 Commands:
     airbyte-ops registry connector publish-prerelease - Publish connector prerelease
+    airbyte-ops registry connector publish - Publish connector (apply/rollback version override)
     airbyte-ops registry image inspect - Inspect Docker image on DockerHub
 """
 
 from __future__ import annotations
 
+from pathlib import Path
 from typing import Annotated
 
 from cyclopts import App, Parameter
 
 from airbyte_ops_mcp.cli._base import app
-from airbyte_ops_mcp.cli._shared import print_error, print_json, print_success
+from airbyte_ops_mcp.cli._shared import (
+    exit_with_error,
+    print_error,
+    print_json,
+    print_success,
+)
 from airbyte_ops_mcp.mcp.github import get_docker_image_info
 from airbyte_ops_mcp.mcp.prerelease import publish_connector_to_airbyte_registry
+from airbyte_ops_mcp.registry import (
+    ConnectorPublishResult,
+    PublishAction,
+    publish_connector,
+)
 
 # Create the registry sub-app
 registry_app = App(
@@ -65,6 +80,80 @@ def publish_prerelease(
     print_json(result.model_dump())
 
 
+@connector_app.command(name="publish")
+def publish(
+    name: Annotated[
+        str,
+        Parameter(help="Connector technical name (e.g., source-github)."),
+    ],
+    repo_path: Annotated[
+        Path,
+        Parameter(help="Path to the Airbyte monorepo. Defaults to current directory."),
+    ] = Path.cwd(),
+    apply_override: Annotated[
+        bool,
+        Parameter(
+            help="Apply a version override (promote RC to stable).",
+            negative="",  # Disable --no-apply-override
+        ),
+    ] = False,
+    rollback_override: Annotated[
+        bool,
+        Parameter(
+            help="Rollback a version override.",
+            negative="",  # Disable --no-rollback-override
+        ),
+    ] = False,
+    dry_run: Annotated[
+        bool,
+        Parameter(help="Show what would be published without making changes."),
+    ] = False,
+    prod: Annotated[
+        bool,
+        Parameter(
+            help="Target the production GCS bucket. Without this flag, operations target the dev bucket for safe testing.",
+            negative="",  # Disable --no-prod
+        ),
+    ] = False,
+) -> None:
+    """Publish a connector to the Airbyte registry.
+
+    This command handles connector publishing operations including applying
+    version overrides (promoting RC to stable) or rolling back version overrides.
+
+    By default, operations target the dev bucket (dev-airbyte-cloud-connector-metadata-service-2)
+    for safe testing. Use --prod to target the production bucket.
+    """
+    if apply_override and rollback_override:
+        exit_with_error("Cannot use both --apply-override and --rollback-override")
+
+    if not apply_override and not rollback_override:
+        exit_with_error("Must specify either --apply-override or --rollback-override")
+
+    # Map CLI flags to PublishAction
+    action: PublishAction = (
+        "apply-version-override" if apply_override else "rollback-version-override"
+    )
+
+    # Delegate to the capability module
+    if not repo_path.exists():
+        exit_with_error(f"Repository path not found: {repo_path}")
+
+    result: ConnectorPublishResult = publish_connector(
+        repo_path=repo_path,
+        connector_name=name,
+        action=action,
+        dry_run=dry_run,
+        use_prod=prod,
+    )
+
+    # Output result as JSON
+    print_json(result.model_dump())
+
+    if result.status == "failure":
+        exit_with_error(result.message or "Operation failed", code=1)
+
+
 @image_app.command(name="inspect")
 def inspect_image(
     image: Annotated[

airbyte_ops_mcp/connection_config_retriever/__init__.py

@@ -0,0 +1,26 @@
+# Copyright (c) 2025 Airbyte, Inc., all rights reserved.
+"""Connection config retriever module.
+
+This module provides functionality to retrieve unmasked connection configuration
+from Airbyte Cloud's internal database, including secret resolution from GCP
+Secret Manager and audit logging to GCP Cloud Logging.
+
+Refactored from: live_tests/_connection_retriever
+Original source: airbyte-platform-internal/tools/connection-retriever
+"""
+
+from airbyte_ops_mcp.connection_config_retriever.retrieval import (
+    ConnectionNotFoundError,
+    RetrievalMetadata,
+    TestingCandidate,
+    retrieve_objects,
+)
+from airbyte_ops_mcp.constants import ConnectionObject
+
+__all__ = [
+    "ConnectionNotFoundError",
+    "ConnectionObject",
+    "RetrievalMetadata",
+    "TestingCandidate",
+    "retrieve_objects",
+]

airbyte_ops_mcp/{live_tests/_connection_retriever → connection_config_retriever}/audit_logging.py

@@ -1,7 +1,8 @@
 # Copyright (c) 2025 Airbyte, Inc., all rights reserved.
-"""Audit logging for vendored connection-retriever.
+"""Audit logging for connection config retrieval.
 
-Vendored from: airbyte-platform-internal/tools/connection-retriever/src/connection_retriever/audit_logging.py
+Refactored from: live_tests/_connection_retriever/audit_logging.py
+Original source: airbyte-platform-internal/tools/connection-retriever/src/connection_retriever/audit_logging.py
 """
 
 from __future__ import annotations
@@ -12,12 +13,10 @@ from typing import TYPE_CHECKING, Any, Callable
 
 from google.cloud import logging as gcloud_logging
 
-from airbyte_ops_mcp.live_tests._connection_retriever.consts import (
-    GCP_PROJECT_NAME,
-)
+from airbyte_ops_mcp.constants import GCP_PROJECT_NAME
 
 if TYPE_CHECKING:
-    from airbyte_ops_mcp.live_tests._connection_retriever.retrieval import (
+    from airbyte_ops_mcp.connection_config_retriever.retrieval import (
         RetrievalMetadata,
     )