airbyte-internal-ops 0.6.1__py3-none-any.whl → 0.7.1__py3-none-any.whl

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/cli.py

@@ -7,21 +7,21 @@ import asyncclick as click
 import asyncer
 from jinja2 import Environment, PackageLoader, select_autoescape
 
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_ops.utils import (
+from airbyte_ops_mcp.connector_ops.utils import (
     Connector,  # type: ignore
 )
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_qa.checks import ENABLED_CHECKS
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_qa.consts import (
+from airbyte_ops_mcp.connector_qa.checks import ENABLED_CHECKS
+from airbyte_ops_mcp.connector_qa.consts import (
     CONNECTORS_QA_DOC_TEMPLATE_NAME,
 )
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_qa.models import (
+from airbyte_ops_mcp.connector_qa.models import (
     Check,
     CheckCategory,
     CheckResult,
     CheckStatus,
     Report,
 )
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_qa.utils import (
+from airbyte_ops_mcp.connector_qa.utils import (
     get_all_connectors_in_directory,
     remove_strict_encrypt_suffix,
 )
@@ -97,9 +97,7 @@ async def run(
     if connector_directory:
         connectors += get_all_connectors_in_directory(connector_directory)
 
-    connectors = sorted(
-        list(connectors), key=lambda connector: connector.technical_name
-    )
+    connectors = sorted(connectors, key=lambda connector: connector.technical_name)
 
     if not connectors:
         raise click.UsageError(

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/models.py

@@ -8,13 +8,12 @@ from dataclasses import dataclass
 from datetime import datetime
 from enum import Enum
 from pathlib import Path
-from typing import Dict, List, Optional
 
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_ops.utils import (
+from airbyte_ops_mcp.connector_ops.utils import (
     Connector,
     ConnectorLanguage,
 )  # type: ignore
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_qa import consts
+from airbyte_ops_mcp.connector_qa import consts
 
 ALL_LANGUAGES = [
     ConnectorLanguage.JAVA,
@@ -109,7 +108,7 @@ class Check(ABC):
         )
 
     @property
-    def applies_to_connector_languages(self) -> List[ConnectorLanguage]:
+    def applies_to_connector_languages(self) -> list[ConnectorLanguage]:
         """The connector languages that the QA check applies to
 
         Raises:
@@ -121,7 +120,7 @@ class Check(ABC):
         return ALL_LANGUAGES
 
     @property
-    def applies_to_connector_types(self) -> List[str]:
+    def applies_to_connector_types(self) -> list[str]:
         """The connector types that the QA check applies to
 
         Returns:
@@ -154,7 +153,7 @@ class Check(ABC):
         )
 
     @property
-    def applies_to_connector_support_levels(self) -> Optional[List[str]]:
+    def applies_to_connector_support_levels(self) -> list[str] | None:
         """The connector's support levels that the QA check applies to
 
         Returns:
@@ -163,7 +162,7 @@ class Check(ABC):
         return None
 
     @property
-    def applies_to_connector_cloud_usage(self) -> Optional[List[str]]:
+    def applies_to_connector_cloud_usage(self) -> list[str] | None:
         """The connector's cloud usage level that the QA check applies to
 
         Returns:
@@ -270,7 +269,7 @@ class Report:
         Returns:
             str: The connectors_report as a JSON string
         """
-        connectors_report: Dict[str, Dict] = {}
+        connectors_report: dict[str, dict] = {}
         for check_result in self.check_results:
             connector = check_result.connector
             connectors_report.setdefault(

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/utils.py

@@ -3,10 +3,10 @@
 from pathlib import Path
 from typing import Set
 
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_ops.utils import (
+from airbyte_ops_mcp.connector_ops.utils import (
     Connector,  # type: ignore
 )
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_qa import consts
+from airbyte_ops_mcp.connector_qa import consts
 
 
 def remove_strict_encrypt_suffix(connector_technical_name: str) -> str:

airbyte_ops_mcp/mcp/_guidance.py

@@ -4,6 +4,43 @@
 This module contains prompt text constants used by the MCP server's prompt templates.
 """
 
+# =============================================================================
+# Server Instructions
+# =============================================================================
+# This text is provided to AI agents via the MCP protocol's "instructions" field.
+# It helps agents understand when to use this server's tools, especially when
+# tool search is enabled. For more context, see:
+# - FastMCP docs: https://gofastmcp.com/servers/overview
+# - Claude tool search: https://www.anthropic.com/news/tool-use-improvements
+# =============================================================================
+
+MCP_SERVER_INSTRUCTIONS = """
+Airbyte internal operations server for connector management, cloud administration,
+and production database queries.
+
+Use this server for:
+- Publishing connector prereleases and managing version overrides/pins
+- Running connector regression tests (single-version and comparison modes)
+- Querying the Airbyte Cloud production database for workspace, connector, sync,
+  and connection diagnostics
+- Triggering and monitoring GitHub Actions CI workflows
+- Looking up Cloud Logging errors for debugging connector issues
+- Performing repository operations on the Airbyte monorepo (for example, listing
+  connectors in the repo or inspecting connector definitions)
+
+Requirements:
+- GCP credentials for database queries and Cloud Logging access
+- Airbyte Cloud credentials for cloud administration operations
+- GitHub token for workflow dispatch and repository operations
+- Local checkout of the Airbyte repository for repo tools (typically at `../airbyte`)
+
+Note: This server is for Airbyte internal use only.
+""".strip()
+
+# =============================================================================
+# Prompt Guidance Text
+# =============================================================================
+
 TEST_MY_TOOLS_GUIDANCE = """
 Test all available tools in this MCP server to confirm they are working properly.
 

airbyte_ops_mcp/mcp/cloud_connector_versions.py

@@ -112,6 +112,15 @@ def get_cloud_connector_version(
         Literal["source", "destination"],
         "The type of connector (source or destination)",
     ],
+    config_api_root: Annotated[
+        str | None,
+        Field(
+            description="Optional API root URL override for the Config API. "
+            "Defaults to Airbyte Cloud (https://cloud.airbyte.com/api/v1). "
+            "Use this to target local or self-hosted deployments.",
+            default=None,
+        ),
+    ] = None,
     *,
     ctx: Context,
 ) -> ConnectorVersionInfo:
@@ -137,7 +146,7 @@ def get_cloud_connector_version(
         version_data = api_client.get_connector_version(
             connector_id=actor_id,
             connector_type=actor_type,
-            config_api_root=constants.CLOUD_CONFIG_API_ROOT,
+            config_api_root=config_api_root or constants.CLOUD_CONFIG_API_ROOT,
             client_id=auth.client_id,
             client_secret=auth.client_secret,
             bearer_token=auth.bearer_token,
@@ -250,7 +259,16 @@ def set_cloud_connector_version_override(
             "Provides additional auditability for AI-driven operations.",
             default=None,
         ),
-    ],
+    ] = None,
+    config_api_root: Annotated[
+        str | None,
+        Field(
+            description="Optional API root URL override for the Config API. "
+            "Defaults to Airbyte Cloud (https://cloud.airbyte.com/api/v1). "
+            "Use this to target local or self-hosted deployments.",
+            default=None,
+        ),
+    ] = None,
     *,
     ctx: Context,
 ) -> VersionOverrideOperationResult:
@@ -363,6 +381,7 @@ def set_cloud_connector_version_override(
         enhanced_override_reason = " | ".join(audit_parts)
 
     # Resolve auth and get current version info
+    resolved_config_api_root = config_api_root or constants.CLOUD_CONFIG_API_ROOT
     try:
         auth = _resolve_cloud_auth(ctx)
 
@@ -370,7 +389,7 @@ def set_cloud_connector_version_override(
         current_version_data = api_client.get_connector_version(
             connector_id=actor_id,
             connector_type=actor_type,
-            config_api_root=constants.CLOUD_CONFIG_API_ROOT,
+            config_api_root=resolved_config_api_root,
             client_id=auth.client_id,
             client_secret=auth.client_secret,
             bearer_token=auth.bearer_token,
@@ -391,7 +410,7 @@ def set_cloud_connector_version_override(
         result = api_client.set_connector_version_override(
             connector_id=actor_id,
             connector_type=actor_type,
-            config_api_root=constants.CLOUD_CONFIG_API_ROOT,
+            config_api_root=resolved_config_api_root,
             client_id=auth.client_id,
             client_secret=auth.client_secret,
             workspace_id=resolved_workspace_id,
@@ -407,7 +426,7 @@ def set_cloud_connector_version_override(
         updated_version_data = api_client.get_connector_version(
             connector_id=actor_id,
             connector_type=actor_type,
-            config_api_root=constants.CLOUD_CONFIG_API_ROOT,
+            config_api_root=resolved_config_api_root,
             client_id=auth.client_id,
             client_secret=auth.client_secret,
             bearer_token=auth.bearer_token,
@@ -533,7 +552,16 @@ def set_workspace_connector_version_override(
             "Provides additional auditability for AI-driven operations.",
             default=None,
         ),
-    ],
+    ] = None,
+    config_api_root: Annotated[
+        str | None,
+        Field(
+            description="Optional API root URL override for the Config API. "
+            "Defaults to Airbyte Cloud (https://cloud.airbyte.com/api/v1). "
+            "Use this to target local or self-hosted deployments.",
+            default=None,
+        ),
+    ] = None,
     *,
     ctx: Context,
 ) -> WorkspaceVersionOverrideResult:
@@ -647,7 +675,7 @@ def set_workspace_connector_version_override(
             workspace_id=resolved_workspace_id,
             connector_name=connector_name,
             connector_type=connector_type,
-            config_api_root=constants.CLOUD_CONFIG_API_ROOT,
+            config_api_root=config_api_root or constants.CLOUD_CONFIG_API_ROOT,
             client_id=auth.client_id,
             client_secret=auth.client_secret,
             bearer_token=auth.bearer_token,
@@ -768,7 +796,16 @@ def set_organization_connector_version_override(
             "Provides additional auditability for AI-driven operations.",
             default=None,
         ),
-    ],
+    ] = None,
+    config_api_root: Annotated[
+        str | None,
+        Field(
+            description="Optional API root URL override for the Config API. "
+            "Defaults to Airbyte Cloud (https://cloud.airbyte.com/api/v1). "
+            "Use this to target local or self-hosted deployments.",
+            default=None,
+        ),
+    ] = None,
     *,
     ctx: Context,
 ) -> OrganizationVersionOverrideResult:
@@ -878,7 +915,7 @@ def set_organization_connector_version_override(
             organization_id=organization_id,
             connector_name=connector_name,
             connector_type=connector_type,
-            config_api_root=constants.CLOUD_CONFIG_API_ROOT,
+            config_api_root=config_api_root or constants.CLOUD_CONFIG_API_ROOT,
             client_id=auth.client_id,
             client_secret=auth.client_secret,
             bearer_token=auth.bearer_token,

airbyte_ops_mcp/mcp/server.py

@@ -22,12 +22,14 @@ from dotenv import load_dotenv
 from fastmcp import FastMCP
 from fastmcp_extensions import MCPServerConfigArg, mcp_server
 
+from airbyte_ops_mcp._sentry import init_sentry_tracking
 from airbyte_ops_mcp.constants import (
     HEADER_AIRBYTE_CLOUD_CLIENT_ID,
     HEADER_AIRBYTE_CLOUD_CLIENT_SECRET,
     MCP_SERVER_NAME,
     ServerConfigKey,
 )
+from airbyte_ops_mcp.mcp._guidance import MCP_SERVER_INSTRUCTIONS
 from airbyte_ops_mcp.mcp.cloud_connector_versions import (
     register_cloud_connector_version_tools,
 )
@@ -59,6 +61,7 @@ def _normalize_bearer_token(value: str) -> str | None:
 # Create the MCP server with built-in server info resource
 app = mcp_server(
     name=MCP_SERVER_NAME,
+    instructions=MCP_SERVER_INSTRUCTIONS,
     package_name="airbyte-internal-ops",
     advertised_properties={
         "docs_url": "https://github.com/airbytehq/airbyte-ops-mcp",
@@ -138,6 +141,7 @@ def main() -> None:
     suitable for direct MCP client connections.
     """
     _load_env()
+    init_sentry_tracking()
 
     print("=" * 60, flush=True, file=sys.stderr)
     print("Starting Airbyte Admin MCP server (stdio mode).", file=sys.stderr)
@@ -188,6 +192,7 @@ def main_http() -> None:
         MCP_HTTP_PORT: Port to listen on (default: 8082)
     """
     _load_env()
+    init_sentry_tracking()
 
     host = os.getenv("MCP_HTTP_HOST", DEFAULT_HTTP_HOST)
     port = _parse_port(os.getenv("MCP_HTTP_PORT"), DEFAULT_HTTP_PORT)