validio-sdk 9.0.6__tar.gz → 9.2.0__tar.gz

{validio_sdk-9.0.6 → validio_sdk-9.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: validio-sdk
-Version: 9.0.6
+Version: 9.2.0
 Summary: SDK to interact with the Validio platform
 License: Apache-2.0
 License-File: LICENSE

{validio_sdk-9.0.6 → validio_sdk-9.2.0}/pyproject.toml

@@ -3,7 +3,7 @@ name = "validio-sdk"
 # This version does not represent the released version or any tag. For each
 # release we automatically bump this before building and publishing so this
 # should be kept at 0.0.1dev1
-version = "9.0.6"
+version = "9.2.0"
 description = "SDK to interact with the Validio platform"
 authors = ["Validio <support@validio.io>"]
 license = "Apache-2.0"

{validio_sdk-9.0.6 → validio_sdk-9.2.0}/validio_sdk/_api/api.py

@@ -401,7 +401,33 @@ async def get_sources(
     source_id: str | None = None,
     namespace_id: str | None = None,
 ) -> list[dict[str, Any]] | dict[str, Any]:
-    source_details = """
+    sap_iq_fragment = ""
+    if is_enabled(Feature.SAP_IQ):
+        sap_iq_fragment = """
+      ... on SapIqSource {
+        config {
+          database
+          schema
+          table
+          schedule
+        }
+      }
+"""
+
+    open_edge_fragment = ""
+    if is_enabled(Feature.OPEN_EDGE):
+        open_edge_fragment = """
+      ... on OpenEdgeSource {
+        config {
+          schema
+          table
+          schedule
+        }
+      }
+"""
+
+    source_details = (
+        """
     fragment SourceDetails on Source {
       __typename
       id
@@ -513,6 +539,9 @@ async def get_sources(
           schedule
         }
       }
+"""
+        + sap_iq_fragment
+        + """
       ... on MsSqlServerSource {
         config {
           database
@@ -576,8 +605,12 @@ async def get_sources(
           schedule
         }
       }
+"""
+        + open_edge_fragment
+        + """
     }
     """
+    )
 
     if source_id and source_id.startswith("SRC_"):
         query = f"""
@@ -984,6 +1017,50 @@ async def get_credentials(
       }
 """
 
+    sap_iq_fragment = ""
+    if is_enabled(Feature.SAP_IQ):
+        sap_iq_fragment = """
+      ... on SapIqCredential {
+        config {
+          host
+          port
+          server
+          sapIqSslMode: sslMode
+          tlsCaCertificate
+          certificateName
+          auth {
+            __typename
+            ... on SapIqCredentialUserPassword {
+              user
+            }
+          }
+        }
+        enableCatalog
+      }
+"""
+
+    open_edge_fragment = ""
+    if is_enabled(Feature.OPEN_EDGE):
+        open_edge_fragment = """
+      ... on OpenEdgeCredential {
+        config {
+          host
+          port
+          databaseRequired: database
+          openEdgeSslMode: sslMode
+          tlsCaCertificate
+          certificateName
+          auth {
+            __typename
+            ... on OpenEdgeCredentialUserPassword {
+              user
+            }
+          }
+        }
+        enableCatalog
+      }
+"""
+
     credential_details = (
         """
     fragment CredentialBase on Credential {
@@ -1035,6 +1112,8 @@ async def get_credentials(
       }
 """
         + gcp_gemini_enterprise_fragment
+        + sap_iq_fragment
+        + open_edge_fragment
         + """
       ... on AwsCredential {
         config {
@@ -1133,6 +1212,7 @@ async def get_credentials(
           }
           tlsEnabledRequired: tlsEnabled
           tlsCaCertificates
+          packageCollection
         }
         enableCatalog
       }
@@ -1336,6 +1416,8 @@ async def get_credentials(
                 ("optUser", "user"),
                 ("optBaseUrl", "baseUrl"),
                 ("tlsEnabledRequired", "tlsEnabled"),
+                ("sapIqSslMode", "sslMode"),
+                ("openEdgeSslMode", "sslMode"),
             ],
         )
         result[i] = credential

{validio_sdk-9.0.6 → validio_sdk-9.2.0}/validio_sdk/features.py

@@ -31,6 +31,8 @@ class Feature(str, Enum):
 
     GCP_GEMINI_ENTERPRISE_CREDENTIAL = "gemini"
     JIRA = "jira"
+    OPEN_EDGE = "open-edge"
+    SAP_IQ = "sap-iq"
 
 
 def is_enabled(feature: Feature) -> bool:

{validio_sdk-9.0.6 → validio_sdk-9.2.0}/validio_sdk/resource/_server_resources.py

@@ -94,10 +94,16 @@ from validio_sdk.resource.credentials import (
     OpenAiCredential,
     OpenAiCredentialApiKey,
     OpenAiCredentialAuth,
+    OpenEdgeCredential,
+    OpenEdgeCredentialAuth,
+    OpenEdgeCredentialUserPassword,
     OracleCredential,
     OracleCredentialAuth,
     OracleCredentialUserPassword,
     PostgreSqlCredential,
+    SapIqCredential,
+    SapIqCredentialAuth,
+    SapIqCredentialUserPassword,
     SigmaCredential,
     SnowflakeCredential,
     SnowflakeCredentialAuth,
@@ -366,6 +372,35 @@ async def load_credentials(
                     auth=db2_auth,
                     tls_enabled=c["config"]["tlsEnabled"],
                     tls_ca_certificates=c["config"]["tlsCaCertificates"],
+                    package_collection=c["config"]["packageCollection"],
+                    display_name=display_name,
+                    enable_catalog=c["enableCatalog"],
+                    __internal__=g,
+                )
+            case "OpenEdgeCredential" if is_enabled(Feature.OPEN_EDGE):
+                auth_type = c["config"]["auth"]["__typename"]
+
+                if auth_type == "OpenEdgeCredentialUserPassword":
+                    open_edge_auth: OpenEdgeCredentialAuth = (
+                        OpenEdgeCredentialUserPassword(
+                            user=c["config"]["auth"]["user"],
+                            password="UNSET",
+                        )
+                    )
+                else:
+                    raise ValidioBugError(
+                        f"Unknown OpenEdge auth type on {name}: '{auth_type}'"
+                    )
+
+                credential = OpenEdgeCredential(
+                    name=name,
+                    host=c["config"]["host"],
+                    port=c["config"]["port"],
+                    database=c["config"]["database"],
+                    auth=open_edge_auth,
+                    ssl_mode=c["config"]["sslMode"],
+                    tls_ca_certificate=c["config"]["tlsCaCertificate"],
+                    certificate_name=c["config"]["certificateName"],
                     display_name=display_name,
                     enable_catalog=c["enableCatalog"],
                     __internal__=g,
@@ -659,6 +694,32 @@ async def load_credentials(
                     enable_catalog=c["enableCatalog"],
                     __internal__=g,
                 )
+            case "SapIqCredential" if is_enabled(Feature.SAP_IQ):
+                auth_type = c["config"]["auth"]["__typename"]
+
+                if auth_type == "SapIqCredentialUserPassword":
+                    sap_iq_auth: SapIqCredentialAuth = SapIqCredentialUserPassword(
+                        user=c["config"]["auth"]["user"],
+                        password="UNSET",
+                    )
+                else:
+                    raise ValidioBugError(
+                        f"Unknown SAP IQ auth type on {name}: '{auth_type}'"
+                    )
+
+                credential = SapIqCredential(
+                    name=name,
+                    host=c["config"]["host"],
+                    port=c["config"]["port"],
+                    server=c["config"]["server"],
+                    auth=sap_iq_auth,
+                    ssl_mode=c["config"]["sslMode"],
+                    tls_ca_certificate=c["config"]["tlsCaCertificate"],
+                    certificate_name=c["config"]["certificateName"],
+                    display_name=display_name,
+                    enable_catalog=c["enableCatalog"],
+                    __internal__=g,
+                )
             case "TeradataCredential":
                 auth_type = c["config"]["auth"]["__typename"]
 
@@ -858,6 +919,13 @@ async def load_sources(
         owner = s.get("owner") or {}
 
         cls = eval(f"validio_sdk.resource.sources.{s['__typename']}")
+        if cls._feature is not None and not is_enabled(cls._feature):
+            raise ValidioError(
+                f"unsupported source '{name}' of type '{s['__typename']}'. "
+                "If this source type requires a feature flag, "
+                "ensure it is enabled."
+            )
+
         source = cls(
             **{
                 **{to_snake(k): v for k, v in source_config.items()},

{validio_sdk-9.0.6 → validio_sdk-9.2.0}/validio_sdk/resource/channels.py

@@ -78,7 +78,13 @@ class Channel(Resource):
         self._resource_graph._add_root(self)
 
     def _immutable_fields(self) -> set[str]:
-        return {"integration_id"}
+        return set()
+
+    def _mutable_fields(self) -> set[str]:
+        return {
+            *super()._mutable_fields(),
+            *{"integration_id"},
+        }
 
     @abstractmethod
     def _deprecated_fields(self) -> set[str]:
@@ -109,15 +115,19 @@ class Channel(Resource):
     def _encode(self) -> dict[str, object]:
         return _encode_resource(self)
 
+    def _api_top_level_fields(
+        self,
+        namespace: str,  # noqa: ARG002
+        ctx: "DiffContext",  # noqa: ARG002
+    ) -> dict[str, Any]:
+        return {"integrationId": self.integration_id}
+
     def _api_top_level_create_fields(
         self,
         namespace: str,
         ctx: "DiffContext",  # noqa: ARG002
     ) -> dict[str, Any]:
-        return {"namespaceId": namespace, "integrationId": self.integration_id}
-
-    def _api_config_excludes(self) -> set[str]:
-        return {"integration_id"}
+        return {"namespaceId": namespace}
 
     @staticmethod
     def _decode(

{validio_sdk-9.0.6 → validio_sdk-9.2.0}/validio_sdk/resource/credentials.py

@@ -53,6 +53,20 @@ class IbmDb2BackendType(str, Enum):
     ZOS = "ZOS"
 
 
+class OpenEdgeSSLMode(str, Enum):
+    """SSL encryption mode when connecting to OpenEdge."""
+
+    DISABLE = "DISABLE"
+    REQUIRE = "REQUIRE"
+
+
+class SapIqSslMode(str, Enum):
+    """SSL mode when connecting to SAP IQ."""
+
+    DISABLE = "DISABLE"
+    REQUIRE = "REQUIRE"
+
+
 class TeradataSSLMode(str, Enum):
     """SSL encryption mode when connecting to Teradata."""
 
@@ -1162,6 +1176,7 @@ class IbmDb2Credential(Credential):
         auth: IbmDb2CredentialAuth,
         tls_enabled: bool,
         tls_ca_certificates: str | None = None,
+        package_collection: str | None = None,
         enable_catalog: bool = False,
         display_name: str | None = None,
         ignore_changes: bool = False,
@@ -1181,6 +1196,8 @@ class IbmDb2Credential(Credential):
             established using the TLS protocol.
         :param tls_ca_certificates: PEM certificate bundle used to
             verify the server TLS certificate.
+        :param package_collection: The DB2 package collection to
+            run queries under.
         :param enable_catalog: If set to true, this credential will
             be used to fetch catalog information.
         :param display_name: Human-readable name for the credential. This name is
@@ -1205,6 +1222,7 @@ class IbmDb2Credential(Credential):
         self.enable_catalog = enable_catalog
         self.tls_enabled = tls_enabled
         self.tls_ca_certificates = tls_ca_certificates
+        self.package_collection = package_collection
 
     def _immutable_fields(self) -> set[str]:
         return {"database", "backend_type"}
@@ -1217,6 +1235,7 @@ class IbmDb2Credential(Credential):
                 "port",
                 "tls_enabled",
                 "tls_ca_certificates",
+                "package_collection",
                 "enable_catalog",
             },
         }
@@ -2791,6 +2810,257 @@ class SigmaCredential(Credential):
         return {"client_secret"}
 
 
+class OpenEdgeCredentialUserPassword(ApiSecretChangeNestedResource):
+    """OpenEdge credential via user-password auth."""
+
+    def __init__(
+        self,
+        user: str,
+        password: str,
+    ):
+        """
+        Constructor.
+
+        :param user: Username.
+        :param password: Password.
+        """
+        self.user = user
+        self.password = password
+
+    def _api_variant_name(self) -> str:
+        return "userPassword"
+
+    def _immutable_fields(self) -> set[str]:
+        return set()
+
+    def _mutable_fields(self) -> set[str]:
+        return {"user"}
+
+    def _secret_fields(self) -> set[str]:
+        return {"password"}
+
+    def _mutable_nested_objects(
+        self,
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
+        return {}
+
+
+OpenEdgeCredentialAuth = Union[OpenEdgeCredentialUserPassword]
+
+
+class OpenEdgeCredential(Credential):
+    """Credential to connect to Progress OpenEdge."""
+
+    _feature = Feature.OPEN_EDGE
+
+    def __init__(
+        self,
+        *,
+        name: str,
+        host: str,
+        port: int,
+        database: str,
+        auth: OpenEdgeCredentialAuth,
+        ssl_mode: OpenEdgeSSLMode,
+        tls_ca_certificate: str | None = None,
+        certificate_name: str | None = None,
+        enable_catalog: bool = False,
+        display_name: str | None = None,
+        ignore_changes: bool = False,
+        recreate_on_change: bool = False,
+        __internal__: ResourceGraph | None = None,
+    ):
+        """
+        Constructor.
+
+        :param name: Unique resource name assigned to the credential
+        :param host: Host address of the database server.
+        :param port: Port number to use to access the OpenEdge SQL Engine.
+        :param database: Name of the OpenEdge database to connect to.
+            This field is immutable; changing it requires recreating the
+            credential.
+        :param auth: Credentials to use for authentication.
+        :param ssl_mode: Specifies which SSL mode the client operates in.
+        :param tls_ca_certificate: Optional CA certificate (PEM format)
+            used to verify the server certificate.
+        :param certificate_name: Optional certificate name (CN/SAN) to
+            expect in the server certificate. Set when it doesn't match
+            the dialed host.
+        :param enable_catalog: If set to true, this credential will
+            be used to fetch catalog information.
+        :param display_name: Human-readable name for the credential. This name is
+          visible in the UI and does not need to be unique.
+        :param ignore_changes: If set to true, changes to the resource will be ignored.
+        :param recreate_on_change: If set to true, any change to the resource will
+          cause it to be deleted and recreated instead of updated.
+        """
+        super().__init__(
+            name=name,
+            display_name=display_name,
+            ignore_changes=ignore_changes,
+            recreate_on_change=recreate_on_change,
+            __internal__=__internal__,
+        )
+
+        self.host = host
+        self.port = port
+        self.database = database
+        self.auth = auth
+        self.ssl_mode = _maybe_cast(ssl_mode, OpenEdgeSSLMode)
+        self.tls_ca_certificate = tls_ca_certificate
+        self.certificate_name = certificate_name
+        self.enable_catalog = enable_catalog
+
+    def _immutable_fields(self) -> set[str]:
+        return {"database"}
+
+    def _mutable_fields(self) -> set[str]:
+        return {
+            *super()._mutable_fields(),
+            *{
+                "host",
+                "port",
+                "ssl_mode",
+                "tls_ca_certificate",
+                "certificate_name",
+                "enable_catalog",
+            },
+        }
+
+    def _secret_fields(self) -> set[str]:
+        return set()
+
+    def _mutable_nested_objects(
+        self,
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
+        return {"auth": self.auth}
+
+
+class SapIqCredentialUserPassword(ApiSecretChangeNestedResource):
+    """SAP IQ credential via user-password auth."""
+
+    def __init__(
+        self,
+        user: str,
+        password: str,
+    ):
+        """
+        Constructor.
+
+        :param user: Username.
+        :param password: Password.
+        """
+        self.user = user
+        self.password = password
+
+    def _api_variant_name(self) -> str:
+        return "userPassword"
+
+    def _immutable_fields(self) -> set[str]:
+        return set({})
+
+    def _mutable_fields(self) -> set[str]:
+        return {"user"}
+
+    def _secret_fields(self) -> set[str]:
+        return {"password"}
+
+    def _mutable_nested_objects(
+        self,
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
+        return {}
+
+
+SapIqCredentialAuth = Union[SapIqCredentialUserPassword]
+
+
+class SapIqCredential(Credential):
+    """Credential to connect to SAP IQ."""
+
+    _feature = Feature.SAP_IQ
+
+    def __init__(
+        self,
+        *,
+        name: str,
+        host: str,
+        port: int,
+        server: str,
+        auth: SapIqCredentialAuth,
+        ssl_mode: SapIqSslMode,
+        tls_ca_certificate: str | None = None,
+        certificate_name: str | None = None,
+        enable_catalog: bool = False,
+        display_name: str | None = None,
+        ignore_changes: bool = False,
+        recreate_on_change: bool = False,
+        __internal__: ResourceGraph | None = None,
+    ):
+        """
+        Constructor.
+
+        :param name: Unique resource name assigned to the credential
+        :param host: Host address of the server.
+        :param port: Port number of the server.
+        :param server: Server (engine) name to connect to.
+        :param auth: Credentials to use for authentication.
+        :param ssl_mode: Specifies which SSL mode the client operates in.
+        :param tls_ca_certificate: Optional CA certificate (PEM format)
+            used to verify the server certificate.
+        :param certificate_name: Optional certificate name (CN/SAN) to
+            expect in the server certificate. Set when it doesn't match
+            the dialed host.
+        :param enable_catalog: If set to true, this credential will
+            be used to fetch catalog information.
+        :param display_name: Human-readable name for the credential. This name is
+          visible in the UI and does not need to be unique.
+        :param ignore_changes: If set to true, changes to the resource will be ignored.
+        :param recreate_on_change: If set to true, any change to the resource will
+          cause it to be deleted and recreated instead of updated.
+        """
+        super().__init__(
+            name=name,
+            display_name=display_name,
+            ignore_changes=ignore_changes,
+            recreate_on_change=recreate_on_change,
+            __internal__=__internal__,
+        )
+
+        self.host = host
+        self.port = port
+        self.server = server
+        self.auth = auth
+        self.ssl_mode = _maybe_cast(ssl_mode, SapIqSslMode)
+        self.tls_ca_certificate = tls_ca_certificate
+        self.certificate_name = certificate_name
+        self.enable_catalog = enable_catalog
+
+    def _immutable_fields(self) -> set[str]:
+        return set({})
+
+    def _mutable_fields(self) -> set[str]:
+        return {
+            *super()._mutable_fields(),
+            *{
+                "host",
+                "port",
+                "server",
+                "ssl_mode",
+                "tls_ca_certificate",
+                "certificate_name",
+                "enable_catalog",
+            },
+        }
+
+    def _secret_fields(self) -> set[str]:
+        return set()
+
+    def _mutable_nested_objects(
+        self,
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
+        return {"auth": self.auth}
+
+
 class TeradataCredentialUserPassword(ApiSecretChangeNestedResource):
     """Teradata credential via user-password auth."""
 
@@ -2915,6 +3185,8 @@ WarehouseCredential = Union[
     IbmDb2Credential,
     PostgreSqlCredential,
     OracleCredential,
+    OpenEdgeCredential,
+    SapIqCredential,
     SnowflakeCredential,
     TeradataCredential,
 ]

{validio_sdk-9.0.6 → validio_sdk-9.2.0}/validio_sdk/resource/sources.py

@@ -11,6 +11,7 @@ import validio_sdk
 from validio_sdk._api.api import infer_schema, sql_source_preview
 from validio_sdk.client import Session
 from validio_sdk.exception import ValidioResourceError
+from validio_sdk.features import Feature
 from validio_sdk.resource._diffable import Diffable
 from validio_sdk.resource._errors import ManifestConfigurationError
 from validio_sdk.resource._resource import Resource
@@ -36,8 +37,10 @@ from validio_sdk.resource.credentials import (
     IbmDb2Credential,
     KafkaCredential,
     MsSqlServerCredential,
+    OpenEdgeCredential,
     OracleCredential,
     PostgreSqlCredential,
+    SapIqCredential,
     SnowflakeCredential,
     TeradataCredential,
     WarehouseCredential,
@@ -1513,6 +1516,167 @@ class OracleSource(Source):
         }
 
 
+class OpenEdgeSource(Source):
+    """A Progress OpenEdge source configuration."""
+
+    _feature = Feature.OPEN_EDGE
+
+    def __init__(
+        self,
+        *,
+        name: str,
+        credential: OpenEdgeCredential,
+        schema: str,
+        table: str,
+        schedule: str | None,
+        display_name: str | None = None,
+        owner: str | None = None,
+        ignore_changes: bool = False,
+        recreate_on_change: bool = False,
+        jtd_schema: JsonTypeDefinition | None = None,
+        description: str | None = None,
+        tags: list[Tag] | None = None,
+        priority: IncidentGroupPriority | None = None,
+    ):
+        """
+        Constructor.
+
+        :param schema: Name of the schema in the database.
+        :param table: Name of table to monitor.
+        :param schedule: A 5-digit cron expression specifying how when the source
+            polls for new data. Example: '0 0 * * *' to poll daily at midnight.
+        :param display_name: Human-readable name for the source. This name is
+          visible in the UI and does not need to be unique.
+        :param owner: User email address of the source owner.
+        :param ignore_changes: If set to true, changes to the resource will be ignored.
+        :param recreate_on_change: If set to true, any change to the resource will
+          cause it to be deleted and recreated instead of updated.
+        :param description: Description of the resource.
+        :param tags: Tags to add to the resource
+        :param priority: Priority to assign to incidents created by this source.
+        """
+        super().__init__(
+            name=name,
+            credential=credential,
+            display_name=display_name,
+            owner=owner,
+            ignore_changes=ignore_changes,
+            recreate_on_change=recreate_on_change,
+            jtd_schema=jtd_schema,
+            description=description,
+            tags=tags,
+            priority=priority,
+        )
+
+        self.schema = schema
+        self.table = table
+        self.schedule = schedule
+
+    def _immutable_fields(self) -> set[str]:
+        return {
+            *super()._immutable_fields(),
+            *{
+                "schema",
+                "table",
+            },
+        }
+
+    def _mutable_fields(self) -> set[str]:
+        return {
+            *super()._mutable_fields(),
+            *{"schedule"},
+        }
+
+    def _api_infer_schema_input(self) -> dict[str, object] | None:
+        return {
+            "schema": self.schema,
+            "table": self.table,
+        }
+
+
+class SapIqSource(Source):
+    """A SAP IQ source configuration."""
+
+    _feature = Feature.SAP_IQ
+
+    def __init__(
+        self,
+        *,
+        name: str,
+        credential: SapIqCredential,
+        database: str,
+        schema: str,
+        table: str,
+        schedule: str | None,
+        display_name: str | None = None,
+        owner: str | None = None,
+        ignore_changes: bool = False,
+        recreate_on_change: bool = False,
+        jtd_schema: JsonTypeDefinition | None = None,
+        description: str | None = None,
+        tags: list[Tag] | None = None,
+        priority: IncidentGroupPriority | None = None,
+    ):
+        """
+        Constructor.
+
+        :param database: Name of the database to connect to.
+        :param schema: Name of the schema in the database.
+        :param table: Name of table to monitor.
+        :param schedule: A 5-digit cron expression specifying how when the source
+            polls for new data. Example: '0 0 * * *' to poll daily at midnight.
+        :param display_name: Human-readable name for the source. This name is
+          visible in the UI and does not need to be unique.
+        :param owner: User email address of the source owner.
+        :param ignore_changes: If set to true, changes to the resource will be ignored.
+        :param recreate_on_change: If set to true, any change to the resource will
+          cause it to be deleted and recreated instead of updated.
+        :param description: Description of the resource.
+        :param tags: Tags to add to the resource
+        :param priority: Priority to assign to incidents created by this source.
+        """
+        super().__init__(
+            name=name,
+            credential=credential,
+            display_name=display_name,
+            owner=owner,
+            ignore_changes=ignore_changes,
+            recreate_on_change=recreate_on_change,
+            jtd_schema=jtd_schema,
+            description=description,
+            tags=tags,
+            priority=priority,
+        )
+
+        self.database = database
+        self.schema = schema
+        self.table = table
+        self.schedule = schedule
+
+    def _immutable_fields(self) -> set[str]:
+        return {
+            *super()._immutable_fields(),
+            *{
+                "database",
+                "schema",
+                "table",
+            },
+        }
+
+    def _mutable_fields(self) -> set[str]:
+        return {
+            *super()._mutable_fields(),
+            *{"schedule"},
+        }
+
+    def _api_infer_schema_input(self) -> dict[str, object] | None:
+        return {
+            "database": self.database,
+            "schema": self.schema,
+            "table": self.table,
+        }
+
+
 class TeradataSource(Source):
     """A Teradata source configuration."""