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

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/checks/assets.py

@@ -5,10 +5,10 @@ from __future__ import annotations
 import xml.etree.ElementTree as ET
 from typing import TYPE_CHECKING
 
-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.models import (
+from airbyte_ops_mcp.connector_qa.models import (
     Check,
     CheckCategory,
     CheckResult,
@@ -16,7 +16,6 @@ from airbyte_ops_mcp._legacy.airbyte_ci.connector_qa.models import (
 
 if TYPE_CHECKING:
     from pathlib import Path
-    from typing import Tuple
 
 DEFAULT_AIRBYTE_ICON = """<svg xmlns="http://www.w3.org/2000/svg" width="250" height="250" fill="none">
     <path fill="#e8e8ed" fill-rule="evenodd" d="M95.775 53.416c20.364-22.88 54.087-29.592 81.811-16.385 36.836 17.549 50.274 62.252 30.219 96.734l-45.115 77.487a18.994 18.994 0 0 1-11.536 8.784 19.12 19.12 0 0 1-14.412-1.878l54.62-93.829c14.55-25.027 4.818-57.467-21.888-70.24-20.038-9.583-44.533-4.795-59.336 11.685a50.008 50.008 0 0 0-12.902 32.877 49.989 49.989 0 0 0 16.664 37.87l-31.887 54.875a18.917 18.917 0 0 1-4.885 5.534 19.041 19.041 0 0 1-6.647 3.255 19.13 19.13 0 0 1-7.395.482 19.087 19.087 0 0 1-7.018-2.365l34.617-59.575a68.424 68.424 0 0 1-10.524-23.544l-21.213 36.579a18.994 18.994 0 0 1-11.535 8.784A19.123 19.123 0 0 1 33 158.668l54.856-94.356a70.296 70.296 0 0 1 7.919-10.896Zm63.314 30.034c13.211 7.577 17.774 24.427 10.13 37.54l-52.603 90.251a18.997 18.997 0 0 1-11.536 8.784 19.122 19.122 0 0 1-14.412-1.878l48.843-84.024a27.778 27.778 0 0 1-10.825-4.847 27.545 27.545 0 0 1-7.783-8.907 27.344 27.344 0 0 1-3.307-11.326 27.293 27.293 0 0 1 1.776-11.66 27.454 27.454 0 0 1 6.533-9.846 27.703 27.703 0 0 1 10.087-6.222 27.858 27.858 0 0 1 23.097 2.135Zm-19.134 16.961a8.645 8.645 0 0 0-2.232 2.529h-.003a8.565 8.565 0 0 0 .632 9.556 8.68 8.68 0 0 0 4.097 2.915 8.738 8.738 0 0 0 5.036.163 8.692 8.692 0 0 0 4.279-2.642 8.59 8.59 0 0 0 2.079-4.558 8.563 8.563 0 0 0-.821-4.938 8.645 8.645 0 0 0-3.444-3.652 8.72 8.72 0 0 0-6.586-.86 8.7 8.7 0 0 0-3.037 1.487Z" clip-rule="evenodd"/>
@@ -32,10 +31,10 @@ class CheckConnectorIconIsAvailable(AssetsCheck):
     description = "Each connector must have an icon available in at the root of the connector code directory. It must be an SVG file named `icon.svg` and must be a square."
     requires_metadata = False
 
-    def _check_is_valid_svg(self, icon_path: Path) -> Tuple[bool, str | None]:
+    def _check_is_valid_svg(self, icon_path: Path) -> tuple[bool, str | None]:
         try:
             # Ensure the file has an .svg extension
-            if not icon_path.suffix.lower() == ".svg":
+            if icon_path.suffix.lower() != ".svg":
                 return False, "Icon file is not a SVG file"
 
             # Parse the file as XML

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/checks/documentation/__init__.py

@@ -1,5 +1,5 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_qa.checks.documentation.documentation import (
+from airbyte_ops_mcp.connector_qa.checks.documentation.documentation import (
     CheckChangelogEntry,
     CheckChangelogSectionContent,
     CheckDocumentationExists,

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/checks/documentation/documentation.py

@@ -3,16 +3,16 @@ import abc
 import textwrap
 from difflib import get_close_matches, ndiff
 from threading import Thread
-from typing import List
+from typing import ClassVar
 
 import requests  # type: ignore
 from pydash.objects import get  # type: ignore
 
-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.models import (
+from airbyte_ops_mcp.connector_qa.models import (
     Check,
     CheckCategory,
     CheckResult,
@@ -37,7 +37,7 @@ class DocumentationCheck(Check):
 
 class CheckMigrationGuide(DocumentationCheck):
     name = "Breaking changes must be accompanied by a migration guide"
-    description = "When a breaking change is introduced, we check that a migration guide is available. It should be stored under `./docs/integrations/<connector-type>s/<connector-name>-migrations.md`.\nThis document should contain a section for each breaking change, in order of the version descending. It must explain users which action to take to migrate to the new version."
+    description = "When a breaking change is introduced, we check that a migration guide is available. It should be stored under `./docs/integrations/<connector-type>s/<connector-name>-migrations.md`.\nThis document should contain a section for each breaking change, in order of the version descending. It must explain users which action to take to migrate to the new version.\nSee the Breaking Changes Policy for full requirements: https://docs.airbyte.com/platform/connector-development/connector-breaking-changes"
 
     def _run(self, connector: Connector) -> CheckResult:
         breaking_changes = get(connector.metadata, "releases.breakingChanges")
@@ -65,7 +65,7 @@ class CheckMigrationGuide(DocumentationCheck):
             first_line = migration_guide_content.splitlines()[0]
         except IndexError:
             first_line = migration_guide_content
-        if not first_line == expected_title:
+        if first_line != expected_title:
             return self.create_check_result(
                 connector=connector,
                 passed=False,
@@ -134,19 +134,19 @@ class CheckDocumentationContent(DocumentationCheck):
     For now, we check documentation structure for sources with sl >= 300.
     """
 
-    applies_to_connector_languages = [
+    applies_to_connector_languages: ClassVar[list] = [
         ConnectorLanguage.PYTHON,
         ConnectorLanguage.LOW_CODE,
     ]
     applies_to_connector_ab_internal_sl = 300
-    applies_to_connector_types = ["source"]
+    applies_to_connector_types: ClassVar[list[str]] = ["source"]
 
 
 class CheckDocumentationLinks(CheckDocumentationContent):
     name = "Links used in connector documentation are valid"
     description = "The user facing connector documentation should update invalid links in connector documentation. For links that are used as example and return 404 status code, use `example: ` before link to skip it."
 
-    def validate_links(self, connector: Connector) -> List[str]:
+    def validate_links(self, connector: Connector) -> list[str]:
         errors = []
         threads = []
 
@@ -246,7 +246,7 @@ class CheckDocumentationHeadersOrder(CheckDocumentationContent):
         ]
         return not_required
 
-    def check_headers(self, connector: Connector) -> List[str]:
+    def check_headers(self, connector: Connector) -> list[str]:
         """
         test_docs_structure gets all top-level headers from source documentation file and check that the order is correct.
         The order of the headers should follow our standard template connectors_qa/checks/documentation/templates/template.md.j2,
@@ -356,9 +356,15 @@ class CheckPrerequisitesSectionDescribesRequiredFieldsFromSpec(
     )
 
     PREREQUISITES = "Prerequisites"
-    CREDENTIALS_KEYWORDS = ["account", "auth", "credentials", "access", "client"]
+    CREDENTIALS_KEYWORDS: ClassVar[list[str]] = [
+        "account",
+        "auth",
+        "credentials",
+        "access",
+        "client",
+    ]
 
-    def check_prerequisites(self, connector: Connector) -> List[str]:
+    def check_prerequisites(self, connector: Connector) -> list[str]:
         actual_connector_spec = connector.connector_spec_file_content
         if not actual_connector_spec:
             return []
@@ -383,7 +389,7 @@ class CheckPrerequisitesSectionDescribesRequiredFieldsFromSpec(
         ) or actual_connector_spec.get("connection_specification")
         required_titles, has_credentials = required_titles_from_spec(spec)  # type: ignore
 
-        missing_fields: List[str] = []
+        missing_fields: list[str] = []
         for title in required_titles:
             if title.lower() not in section_text:
                 missing_fields.append(title)
@@ -458,7 +464,7 @@ class CheckSection(CheckDocumentationContent):
     def header(self) -> str:
         """The name of header for validating content"""
 
-    def check_section(self, connector: Connector) -> List[str]:
+    def check_section(self, connector: Connector) -> list[str]:
         documentation = DocumentationContent(connector=connector)
 
         if self.header not in documentation.headers:
@@ -466,7 +472,7 @@ class CheckSection(CheckDocumentationContent):
                 return [f"Documentation does not have {self.header} section."]
             return []
 
-        errors: List[str] = []
+        errors: list[str] = []
 
         expected = TemplateContent(connector.name_from_metadata).section(self.header)[
             self.expected_section_index
@@ -532,19 +538,14 @@ class CheckSourceSectionContent(CheckDocumentationContent):
         template = TemplateContent("CONNECTOR_NAME_FROM_METADATA").section(
             "CONNECTOR_NAME_FROM_METADATA"
         )
-        if template is None:
-            template_content = (
-                ""  # Provide default empty template if section is missing
-            )
-        else:
-            template_content = template[0]  # type: ignore
+        template_content = "" if template is None else template[0]
 
         return generate_description(
             "section_content_description.md.j2",
             {"header": "CONNECTOR_NAME_FROM_METADATA", "template": template_content},
         )
 
-    def check_source_follows_template(self, connector: Connector) -> List[str]:
+    def check_source_follows_template(self, connector: Connector) -> list[str]:
         documentation = DocumentationContent(connector=connector)
 
         if connector.name_from_metadata not in documentation.headers:
@@ -635,7 +636,7 @@ class CheckTutorialsSectionContent(CheckSection):
 class CheckChangelogSectionContent(CheckSection):
     header = "Changelog"
 
-    def check_section(self, connector: Connector) -> List[str]:
+    def check_section(self, connector: Connector) -> list[str]:
         documentation = DocumentationContent(connector=connector)
 
         if self.header not in documentation.headers:

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

@@ -2,12 +2,12 @@
 
 import re
 from pathlib import Path
-from typing import Dict, List
+from typing import ClassVar
 
 from jinja2 import Environment, FileSystemLoader
 from markdown_it.tree import SyntaxTreeNode
 
-from airbyte_ops_mcp._legacy.airbyte_ci.connector_ops.utils import (
+from airbyte_ops_mcp.connector_ops.utils import (
     Connector,  # type: ignore
 )
 
@@ -26,10 +26,10 @@ class SectionLines:
 class SectionContent:
     def __init__(self, header: str):
         self.header = header
-        self._content: List[str] = []
+        self._content: list[str] = []
 
     @property
-    def content(self) -> List[str]:
+    def content(self) -> list[str]:
         return self._content
 
     @content.setter
@@ -42,7 +42,7 @@ class SectionContent:
 
 class Content:
     HEADING = "heading"
-    supported_header_levels = ["h1", "h2", "h3", "h4"]
+    supported_header_levels: ClassVar[list[str]] = ["h1", "h2", "h3", "h4"]
 
     def __init__(self) -> None:
         self.content = self._content()
@@ -69,10 +69,10 @@ class Content:
 
         return headers
 
-    def _header_line_map(self) -> Dict[str, list[SectionLines]]:
+    def _header_line_map(self) -> dict[str, list[SectionLines]]:
         headers = []
         starts = []
-        header_line_map: Dict[str, list[SectionLines]] = {}
+        header_line_map: dict[str, list[SectionLines]] = {}
 
         for n in self.node:  # type: ignore
             if n.type == self.HEADING:

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/checks/metadata.py

@@ -1,24 +1,25 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 
 from datetime import datetime, timedelta
+from typing import ClassVar
 
 import toml
 
-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._legacy.airbyte_ci.connector_qa.models import (
+from airbyte_ops_mcp.connector_qa import consts
+from airbyte_ops_mcp.connector_qa.models import (
     Check,
     CheckCategory,
     CheckResult,
 )
-from airbyte_ops_mcp._legacy.airbyte_ci.metadata_service.validators.metadata_validator import (
+from airbyte_ops_mcp.metadata_validator import (
     PRE_UPLOAD_VALIDATORS,
     ValidatorOptions,
     validate_and_load,
-)  # type: ignore
+)
 
 
 class MetadataCheck(Check):
@@ -107,7 +108,7 @@ class CheckConnectorLanguageTag(MetadataCheck):
 class CheckConnectorCDKTag(MetadataCheck):
     name = "Python connectors must have a CDK tag in metadata"
     description = f"Python connectors must have a CDK tag in their metadata. It must be set in the `tags` field in {consts.METADATA_FILE_NAME}. The values can be `cdk:low-code`, `cdk:python`, or `cdk:file`."
-    applies_to_connector_languages = [
+    applies_to_connector_languages: ClassVar[list] = [
         ConnectorLanguage.PYTHON,
         ConnectorLanguage.LOW_CODE,
     ]
@@ -138,12 +139,11 @@ class CheckConnectorCDKTag(MetadataCheck):
                 and "file-based" in cdk_deps.get("extras", [])
             ):
                 return self.CDKTag.FILE
-        if setup_py_file.exists():
-            if (
-                "airbyte-cdk[file-based]"
-                in (connector.code_directory / consts.SETUP_PY_FILE_NAME).read_text()
-            ):
-                return self.CDKTag.FILE
+        if setup_py_file.exists() and (
+            "airbyte-cdk[file-based]"
+            in (connector.code_directory / consts.SETUP_PY_FILE_NAME).read_text()
+        ):
+            return self.CDKTag.FILE
         return self.CDKTag.PYTHON
 
     def _run(self, connector: Connector) -> CheckResult:
@@ -179,7 +179,7 @@ class ValidateBreakingChangesDeadlines(MetadataCheck):
     """
 
     name = "Breaking change deadline should be a week in the future"
-    description = "If the connector version has a breaking change, the deadline field must be set to at least a week in the future."
+    description = "If the connector version has a breaking change, the deadline field must be set to at least a week in the future. See the Breaking Changes Policy for full requirements: https://docs.airbyte.com/platform/connector-development/connector-breaking-changes"
     runs_on_released_connectors = False
     minimum_days_until_deadline = 7
 
@@ -238,8 +238,8 @@ class ValidateBreakingChangesDeadlines(MetadataCheck):
 class CheckConnectorMaxSecondsBetweenMessagesValue(MetadataCheck):
     name = "Certified source connector must have a value filled out for maxSecondsBetweenMessages in metadata"
     description = "Certified source connectors must have a value filled out for `maxSecondsBetweenMessages` in metadata. This value represents the maximum number of seconds we could expect between messages for API connectors. And it's used by platform to tune connectors heartbeat timeout. The value must be set in the 'data' field in connector's `metadata.yaml` file."
-    applies_to_connector_types = ["source"]
-    applies_to_connector_support_levels = ["certified"]
+    applies_to_connector_types: ClassVar[list[str]] = ["source"]
+    applies_to_connector_support_levels: ClassVar[list[str]] = ["certified"]
 
     def _run(self, connector: Connector) -> CheckResult:
         max_seconds_between_messages = connector.metadata.get(

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/checks/packaging.py

@@ -1,15 +1,17 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 
+from typing import ClassVar
+
 import semver
 import toml
 from pydash.objects import get  # type: ignore
 
-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._legacy.airbyte_ci.connector_qa.models import (
+from airbyte_ops_mcp.connector_qa import consts
+from airbyte_ops_mcp.connector_qa.models import (
     Check,
     CheckCategory,
     CheckResult,
@@ -25,7 +27,7 @@ class CheckConnectorUsesPoetry(PackagingCheck):
     description = "Connectors must use [Poetry](https://python-poetry.org/) for dependency management. This is to ensure that all connectors use a dependency management tool which locks dependencies and ensures reproducible installs."
     requires_metadata = False
     runs_on_released_connectors = False
-    applies_to_connector_languages = [
+    applies_to_connector_languages: ClassVar[list] = [
         ConnectorLanguage.PYTHON,
         ConnectorLanguage.LOW_CODE,
     ]
@@ -56,11 +58,11 @@ class CheckConnectorUsesPoetry(PackagingCheck):
 class CheckPublishToPyPiIsDeclared(PackagingCheck):
     name = "Python connectors must have PyPi publishing declared."
     description = f"Python connectors must have [PyPi](https://pypi.org/) publishing enabled in their `{consts.METADATA_FILE_NAME}` file. This is declared by setting `remoteRegistries.pypi.enabled` to `true` in {consts.METADATA_FILE_NAME}. This is to ensure that all connectors can be published to PyPi and can be used in `PyAirbyte`."
-    applies_to_connector_languages = [
+    applies_to_connector_languages: ClassVar[list] = [
         ConnectorLanguage.PYTHON,
         ConnectorLanguage.LOW_CODE,
     ]
-    applies_to_connector_types = ["source"]
+    applies_to_connector_types: ClassVar[list[str]] = ["source"]
 
     def _run(self, connector: Connector) -> CheckResult:
         publish_to_pypi_is_enabled = get(
@@ -80,7 +82,7 @@ class CheckPublishToPyPiIsDeclared(PackagingCheck):
 class CheckManifestOnlyConnectorBaseImage(PackagingCheck):
     name = "Manifest-only connectors must use `source-declarative-manifest` as their base image"
     description = "Manifest-only connectors must use `airbyte/source-declarative-manifest` as their base image."
-    applies_to_connector_languages = [ConnectorLanguage.MANIFEST_ONLY]
+    applies_to_connector_languages: ClassVar[list] = [ConnectorLanguage.MANIFEST_ONLY]
 
     def _run(self, connector: Connector) -> CheckResult:
         base_image = get(connector.metadata, "connectorBuildOptions.baseImage")
@@ -125,7 +127,7 @@ class CheckConnectorLicense(PackagingCheck):
 class CheckConnectorLicenseMatchInPyproject(PackagingCheck):
     name = f"Connector license in {consts.METADATA_FILE_NAME} and {consts.PYPROJECT_FILE_NAME} file must match"
     description = f"Connectors license in {consts.METADATA_FILE_NAME} and {consts.PYPROJECT_FILE_NAME} file must match. This is to ensure that all connectors are consistently licensed."
-    applies_to_connector_languages = [
+    applies_to_connector_languages: ClassVar[list] = [
         ConnectorLanguage.PYTHON,
         ConnectorLanguage.LOW_CODE,
     ]
@@ -174,7 +176,7 @@ class CheckConnectorLicenseMatchInPyproject(PackagingCheck):
 class CheckConnectorVersionMatchInPyproject(PackagingCheck):
     name = f"Connector version in {consts.METADATA_FILE_NAME} and {consts.PYPROJECT_FILE_NAME} file must match"
     description = f"Connector version in {consts.METADATA_FILE_NAME} and {consts.PYPROJECT_FILE_NAME} file must match. This is to ensure that connector release is consistent."
-    applies_to_connector_languages = [
+    applies_to_connector_languages: ClassVar[list] = [
         ConnectorLanguage.PYTHON,
         ConnectorLanguage.LOW_CODE,
     ]

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/checks/security.py

@@ -1,16 +1,16 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 
 from pathlib import Path
-from typing import Iterable, Optional, Set, Tuple
+from typing import ClassVar, Iterable
 
 from pydash.objects import get  # type: ignore
 
-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._legacy.airbyte_ci.connector_qa.models import (
+from airbyte_ops_mcp.connector_qa import consts
+from airbyte_ops_mcp.connector_qa.models import (
     Check,
     CheckCategory,
     CheckResult,
@@ -31,7 +31,7 @@ class CheckConnectorUsesHTTPSOnly(SecurityCheck):
 
     ignore_comment = "# ignore-https-check"  # Define the ignore comment pattern
 
-    ignored_directories_for_https_checks = {
+    ignored_directories_for_https_checks: ClassVar[set[str]] = {
         ".venv",
         "tests",
         "unit_tests",
@@ -46,7 +46,7 @@ class CheckConnectorUsesHTTPSOnly(SecurityCheck):
         "htmlcov",
     }
 
-    ignored_file_name_pattern_for_https_checks = {
+    ignored_file_name_pattern_for_https_checks: ClassVar[set[str]] = {
         "*Test.java",
         "*.jar",
         "*.pyc",
@@ -56,7 +56,7 @@ class CheckConnectorUsesHTTPSOnly(SecurityCheck):
         "expected_records.json",
     }
 
-    ignored_url_prefixes = {
+    ignored_url_prefixes: ClassVar[set[str]] = {
         "http://json-schema.org",
         "http://localhost",
     }
@@ -64,9 +64,9 @@ class CheckConnectorUsesHTTPSOnly(SecurityCheck):
     @staticmethod
     def _read_all_files_in_directory(
         directory: Path,
-        ignored_directories: Optional[Set[str]] = None,
-        ignored_filename_patterns: Optional[Set[str]] = None,
-    ) -> Iterable[Tuple[Path, str]]:
+        ignored_directories: set[str] | None = None,
+        ignored_filename_patterns: set[str] | None = None,
+    ) -> Iterable[tuple[Path, str]]:
         ignored_directories = (
             ignored_directories if ignored_directories is not None else set()
         )
@@ -78,21 +78,17 @@ class CheckConnectorUsesHTTPSOnly(SecurityCheck):
 
         for path in directory.rglob("*"):
             ignore_directory = any(
-                [
-                    ignored_directory in path.parts
-                    for ignored_directory in ignored_directories
-                ]
+                ignored_directory in path.parts
+                for ignored_directory in ignored_directories
             )
             ignore_filename = any(
-                [
-                    path.match(ignored_filename_pattern)
-                    for ignored_filename_pattern in ignored_filename_patterns
-                ]
+                path.match(ignored_filename_pattern)
+                for ignored_filename_pattern in ignored_filename_patterns
             )
             ignore = ignore_directory or ignore_filename
             if path.is_file() and not ignore:
                 try:
-                    for line in open(path):
+                    for line in path.read_text().splitlines():
                         yield path, line
                 except UnicodeDecodeError:
                     continue
@@ -146,7 +142,7 @@ class CheckConnectorUsesHTTPSOnly(SecurityCheck):
 class CheckConnectorUsesPythonBaseImage(SecurityCheck):
     name = f"Python connectors must not use a {consts.DOCKERFILE_NAME} and must declare their base image in {consts.METADATA_FILE_NAME} file"
     description = f"Connectors must use our Python connector base image (`{consts.AIRBYTE_PYTHON_CONNECTOR_BASE_IMAGE_NAME}`), declared through the `connectorBuildOptions.baseImage` in their `{consts.METADATA_FILE_NAME}`.\nThis is to ensure that all connectors use a base image which is maintained and has security updates."
-    applies_to_connector_languages = [
+    applies_to_connector_languages: ClassVar[list] = [
         ConnectorLanguage.PYTHON,
         ConnectorLanguage.LOW_CODE,
         ConnectorLanguage.MANIFEST_ONLY,

airbyte_ops_mcp/{_legacy/airbyte_ci/connector_qa → connector_qa}/checks/version.py

@@ -1,16 +1,16 @@
 # Copyright (c) 2025 Airbyte, Inc., all rights reserved.
 
-from typing import Any, Dict
+from typing import Any
 
 import requests  # type: ignore
 import semver
 import yaml  # type: ignore
 
-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._legacy.airbyte_ci.connector_qa.models import (
+from airbyte_ops_mcp.connector_qa import consts
+from airbyte_ops_mcp.connector_qa.models import (
     Check,
     CheckCategory,
     CheckResult,
@@ -47,7 +47,7 @@ class CheckVersionIncrement(Check):
         # TODO: don't run if only files changed are in the bypass list or running in the context of the master branch
         return True
 
-    def _get_master_metadata(self, connector: Connector) -> Dict[str, Any] | None:
+    def _get_master_metadata(self, connector: Connector) -> dict[str, Any] | None:
         """Get the metadata from the master branch or None if unable to retrieve."""
         # TODO: test out if this works on the private airbyte-enterprise repo - consider using git-based approach
         github_url_prefix = "https://raw.githubusercontent.com/airbytehq/airbyte/master/airbyte-integrations/connectors"
@@ -59,7 +59,7 @@ class CheckVersionIncrement(Check):
             return None
         return yaml.safe_load(response.text)["data"]
 
-    def _parse_version_from_metadata(self, metadata: Dict[str, Any]) -> semver.Version:
+    def _parse_version_from_metadata(self, metadata: dict[str, Any]) -> semver.Version:
         return semver.Version.parse(str(metadata["dockerImageTag"]))
 
     def _get_master_connector_version(self, connector: Connector) -> semver.Version:
@@ -95,6 +95,52 @@ class CheckVersionIncrement(Check):
             and master_version.patch == current_version.patch
         )
 
+    @staticmethod
+    def _get_registry_override_tag(
+        metadata: dict[str, Any] | None, channel: str
+    ) -> str | None:
+        """Extract the dockerImageTag from registryOverrides for a given channel.
+
+        Args:
+            metadata: The metadata dictionary (master or current).
+            channel: The channel to extract from ("cloud" or "oss").
+
+        Returns:
+            The dockerImageTag value if present, None otherwise.
+        """
+        if metadata is None:
+            return None
+        try:
+            return (
+                metadata.get("registryOverrides", {})
+                .get(channel, {})
+                .get("dockerImageTag")
+            )
+        except (TypeError, AttributeError):
+            return None
+
+    def _has_registry_override_docker_tag_change(
+        self,
+        master_metadata: dict[str, Any] | None,
+        current_metadata: dict[str, Any] | None,
+    ) -> bool:
+        """Check if registryOverrides.cloud.dockerImageTag or registryOverrides.oss.dockerImageTag has changed.
+
+        Args:
+            master_metadata: The metadata from the master branch.
+            current_metadata: The current metadata.
+
+        Returns:
+            bool: True if either cloud or oss dockerImageTag has been added, removed, or changed.
+        """
+        master_cloud = self._get_registry_override_tag(master_metadata, "cloud")
+        current_cloud = self._get_registry_override_tag(current_metadata, "cloud")
+
+        master_oss = self._get_registry_override_tag(master_metadata, "oss")
+        current_oss = self._get_registry_override_tag(current_metadata, "oss")
+
+        return (master_cloud != current_cloud) or (master_oss != current_oss)
+
     def _run(self, connector: Connector) -> CheckResult:
         """Run the version increment check."""
         if (
@@ -109,29 +155,59 @@ class CheckVersionIncrement(Check):
             )
 
         try:
-            master_version = self._get_master_connector_version(connector)
+            master_metadata = self._get_master_metadata(connector)
+            # Note: master_metadata being None is expected for new connectors that don't exist
+            # in master yet. In this case, we use 0.0.0 as the baseline, allowing any version.
+            master_version = (
+                self._parse_version_from_metadata(master_metadata)
+                if master_metadata
+                else semver.Version.parse("0.0.0")
+            )
             current_version = self._get_current_connector_version(connector)
 
             # Require a version increment
-            if current_version <= master_version:
+            if current_version < master_version:
+                return self.fail(
+                    connector,
+                    f"The dockerImageTag in {consts.METADATA_FILE_NAME} appears to be lower than the "
+                    f"version on the default branch. Master version is {master_version}, current "
+                    f"version is {current_version}. "
+                    f"Update your PR branch from the default branch to get the latest connector version. "
+                    f"Maintainers can use the `/bump-version` PR slash command. "
+                    f"AI agents can use the `bump_version_in_repo` tool from the `airbyte-ops-mcp` MCP server.",
+                )
+
+            if current_version == master_version:
+                # Allow version to stay the same if registryOverrides.cloud.dockerImageTag or
+                # registryOverrides.oss.dockerImageTag has been added, removed, or changed
+                if self._has_registry_override_docker_tag_change(
+                    master_metadata, connector.metadata
+                ):
+                    return self.skip(
+                        connector,
+                        f"The current change is modifying the registryOverrides pinned version on Cloud or OSS. "
+                        f"Skipping this check because the defined version {current_version} is allowed to be unchanged.",
+                    )
                 return self.fail(
                     connector,
                     f"The dockerImageTag in {consts.METADATA_FILE_NAME} was not incremented. "
-                    f"Master version is {master_version}, current version is {current_version}",
+                    f"Master version is {master_version}, current version is {current_version}. "
+                    f"Ignore this message if you do not intend to re-release the connector. "
+                    f"Maintainers can use the `/bump-version` PR slash command. "
+                    f"AI agents can use the `bump_version_in_repo` tool from the `airbyte-ops-mcp` MCP server.",
                 )
 
             if self._are_both_versions_release_candidates(
                 master_version, current_version
+            ) and not self._have_same_major_minor_patch(
+                master_version, current_version
             ):
-                if not self._have_same_major_minor_patch(
-                    master_version, current_version
-                ):
-                    return self.fail(
-                        connector,
-                        f"Master and current version are release candidates but they have different major, minor or patch versions. "
-                        f"Release candidates should only differ in the prerelease part. Master version is {master_version}, "
-                        f"current version is {current_version}",
-                    )
+                return self.fail(
+                    connector,
+                    f"Master and current version are release candidates but they have different major, minor or patch versions. "
+                    f"Release candidates should only differ in the prerelease part. Master version is {master_version}, "
+                    f"current version is {current_version}",
+                )
 
             return self.pass_(
                 connector,