validio-sdk 7.2.0__tar.gz → 7.3.1__tar.gz

{validio_sdk-7.2.0 → validio_sdk-7.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: validio-sdk
-Version: 7.2.0
+Version: 7.3.1
 Summary: SDK to interact with the Validio platform
 Home-page: https://validio.io/
 License: Apache-2.0

{validio_sdk-7.2.0 → validio_sdk-7.3.1}/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 = "7.2.0"
+version = "7.3.1"
 description = "SDK to interact with the Validio platform"
 authors = ["Validio <support@validio.io>"]
 license = "Apache-2.0"

{validio_sdk-7.2.0 → validio_sdk-7.3.1}/validio_sdk/_api/api.py

@@ -553,6 +553,13 @@ async def get_sources(
           schedule
         }
       }
+      ... on TeradataSource {
+        config {
+          database
+          table
+          schedule
+        }
+      }
     }
     """
 
@@ -962,6 +969,12 @@ async def get_credentials(
           baseUrl
         }
       }
+      ... on AnthropicCredential {
+        config {
+          model
+          optBaseUrl: baseUrl
+        }
+      }
       ... on AwsCredential {
         config {
           accessKey
@@ -1166,6 +1179,21 @@ async def get_credentials(
         }
         enableCatalog
       }
+      ... on TeradataCredential {
+        config {
+          host
+          sslMode
+          httpsPort
+          tdmstPort
+          auth {
+            __typename
+            ... on TeradataCredentialUserPassword {
+              user
+            }
+          }
+        }
+        enableCatalog
+      }
     }
     """
 
@@ -1214,6 +1242,7 @@ async def get_credentials(
                 ("powerBiAuth", "auth"),
                 ("databaseRequired", "database"),
                 ("optUser", "user"),
+                ("optBaseUrl", "baseUrl"),
             ],
         )
         result[i] = credential

{validio_sdk-7.2.0 → validio_sdk-7.3.1}/validio_sdk/resource/_diffable.py

@@ -1,7 +1,7 @@
 import inspect
 from abc import ABC, abstractmethod
 from collections.abc import Iterator
-from typing import TYPE_CHECKING, Any, Optional
+from typing import TYPE_CHECKING, Any
 
 from camel_converter import to_camel
 
@@ -53,7 +53,7 @@ class Diffable(ABC):
     @abstractmethod
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         """Returns any nested objects contained within this object.
 
         Nested objects will be diff-ed recursively.

{validio_sdk-7.2.0 → validio_sdk-7.3.1}/validio_sdk/resource/_server_resources.py

@@ -38,6 +38,9 @@ from validio_sdk.resource.channels import (
     WebhookChannel,
 )
 from validio_sdk.resource.credentials import (
+    AnthropicCredential,
+    AnthropicCredentialApiKey,
+    AnthropicCredentialAuth,
     AtlanCredential,
     AwsAthenaCredential,
     AwsCredential,
@@ -74,6 +77,9 @@ from validio_sdk.resource.credentials import (
     SnowflakeCredentialUserPassword,
     TableauConnectedAppCredential,
     TableauPersonalAccessTokenCredential,
+    TeradataCredential,
+    TeradataCredentialAuth,
+    TeradataCredentialUserPassword,
 )
 from validio_sdk.resource.filters import Filter, SqlFilter
 from validio_sdk.resource.notification_rules import Conditions
@@ -205,6 +211,19 @@ async def load_credentials(
                     display_name=display_name,
                     __internal__=g,
                 )
+            case "AnthropicCredential":
+                anthropic_auth: AnthropicCredentialAuth = AnthropicCredentialApiKey(
+                    api_key="UNSET",
+                )
+
+                credential = AnthropicCredential(
+                    name=name,
+                    model=c["config"]["model"],
+                    base_url=c["config"]["baseUrl"],
+                    auth=anthropic_auth,
+                    display_name=display_name,
+                    __internal__=g,
+                )
             case "AtlanCredential":
                 credential = AtlanCredential(
                     name=name,
@@ -491,6 +510,32 @@ async def load_credentials(
                     enable_catalog=c["enableCatalog"],
                     __internal__=g,
                 )
+            case "TeradataCredential":
+                auth_type = c["config"]["auth"]["__typename"]
+
+                if auth_type == "TeradataCredentialUserPassword":
+                    teradata_auth: TeradataCredentialAuth = (
+                        TeradataCredentialUserPassword(
+                            user=c["config"]["auth"]["user"],
+                            password="UNSET",
+                        )
+                    )
+                else:
+                    raise ValidioBugError(
+                        f"Unknown Teradata auth type on {name}: '{auth_type}'"
+                    )
+
+                credential = TeradataCredential(
+                    name=name,
+                    host=c["config"]["host"],
+                    auth=teradata_auth,
+                    ssl_mode=c["config"]["sslMode"],
+                    https_port=c["config"]["httpsPort"],
+                    tdmst_port=c["config"]["tdmstPort"],
+                    display_name=display_name,
+                    enable_catalog=c["enableCatalog"],
+                    __internal__=g,
+                )
             case _:
                 raise ValidioError(
                     f"unsupported credential '{name}' of type '{type(c)}'"

{validio_sdk-7.2.0 → validio_sdk-7.3.1}/validio_sdk/resource/channels.py

@@ -1,7 +1,7 @@
 """Notification Channels."""
 
 from enum import Enum
-from typing import TYPE_CHECKING, Any, Optional, Union, cast
+from typing import TYPE_CHECKING, Any, Union, cast
 
 from camel_converter import to_camel
 
@@ -387,7 +387,7 @@ class EmailChannelAuthSmtpUserPassword(ApiSecretChangeNestedResource):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {}
 
 
@@ -456,5 +456,5 @@ class EmailChannel(Channel):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {"auth": self.auth}

{validio_sdk-7.2.0 → validio_sdk-7.3.1}/validio_sdk/resource/credentials.py

@@ -21,6 +21,7 @@ from validio_sdk.resource._serde import (
     _api_update_input_params,
     _encode_resource,
     _import_resource_params,
+    _maybe_cast,
     decode_nested_objects,
     get_children_node,
     get_config_node,
@@ -48,6 +49,15 @@ class ClickHouseProtocol(str, Enum):
     NATIVE_TLS = "NATIVE_TLS"
 
 
+class TeradataSSLMode(str, Enum):
+    """SSL encryption mode when connecting to Teradata."""
+
+    PREFER = "PREFER"
+    ALLOW = "ALLOW"
+    DISABLE = "DISABLE"
+    REQUIRE = "REQUIRE"
+
+
 class Credential(Resource):
     """
     Base class for a credential resource.
@@ -236,6 +246,100 @@ class Credential(Resource):
         return test_input
 
 
+class AnthropicCredentialApiKey(ApiSecretChangeNestedResource):
+    """Anthropic credential using API key for authentication."""
+
+    def __init__(
+        self,
+        *,
+        api_key: str,
+    ):
+        """
+        Constructor.
+
+        :param api_key: API key to use for connecting to Anthropic.
+        """
+        self.api_key = api_key
+
+    def _api_variant_name(self) -> str:
+        return "apiKey"
+
+    def _immutable_fields(self) -> set[str]:
+        return set({})
+
+    def _mutable_fields(self) -> set[str]:
+        return set({})
+
+    def _secret_fields(self) -> set[str]:
+        return {"api_key"}
+
+    def _nested_objects(
+        self,
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
+        return {}
+
+
+AnthropicCredentialAuth = Union[AnthropicCredentialApiKey]
+
+
+class AnthropicCredential(Credential):
+    """Credential to connect to Anthropic."""
+
+    def __init__(
+        self,
+        *,
+        name: str,
+        model: str,
+        auth: AnthropicCredentialAuth,
+        base_url: str | None = None,
+        display_name: str | None = None,
+        ignore_changes: bool = False,
+        __internal__: ResourceGraph | None = None,
+    ):
+        """
+        Constructor.
+
+        :param name: Unique resource name assigned to the credential
+        :param model: Name of the model to use. E.g. claude-sonnet-4-5
+        :param auth: Auth method to use for connecting to Anthropic.
+        :param base_url: Base URL of Anthropic API. E.g. https://api.anthropic.com/v1.
+            If not provided, defaults to the standard Anthropic API endpoint.
+        :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.
+        """
+        super().__init__(
+            name=name,
+            display_name=display_name,
+            ignore_changes=ignore_changes,
+            __internal__=__internal__,
+        )
+
+        self.model = model
+        self.auth = auth
+        self.base_url = base_url
+
+    def _immutable_fields(self) -> set[str]:
+        return set({})
+
+    def _mutable_fields(self) -> set[str]:
+        return {
+            *super()._mutable_fields(),
+            *{
+                "model",
+                "base_url",
+            },
+        }
+
+    def _secret_fields(self) -> set[str]:
+        return set()
+
+    def _nested_objects(
+        self,
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
+        return {"auth": self.auth}
+
+
 class AtlanCredential(Credential):
     """(BETA) An Atlan credential resource."""
 
@@ -647,7 +751,7 @@ class SnowflakeCredentialKeyPair(ApiSecretChangeNestedResource):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {}
 
 
@@ -683,7 +787,7 @@ class SnowflakeCredentialUserPassword(ApiSecretChangeNestedResource):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {}
 
 
@@ -748,7 +852,7 @@ class SnowflakeCredential(Credential):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {"auth": self.auth}
 
 
@@ -1337,7 +1441,7 @@ class MsSqlServerCredentialUserPassword(ApiSecretChangeNestedResource):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {}
 
 
@@ -1372,7 +1476,7 @@ class MsSqlServerCredentialEntraId(ApiSecretChangeNestedResource):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {}
 
 
@@ -1442,7 +1546,7 @@ class MsSqlServerCredential(Credential):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {"auth": self.auth}
 
 
@@ -1475,7 +1579,7 @@ class OmniCredentialApiKey(ApiSecretChangeNestedResource):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {}
 
 
@@ -1536,7 +1640,7 @@ class OmniCredential(Credential):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {"auth": self.auth}
 
 
@@ -1571,7 +1675,7 @@ class OracleCredentialUserPassword(ApiSecretChangeNestedResource):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {}
 
 
@@ -1639,7 +1743,7 @@ class OracleCredential(Credential):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {"auth": self.auth}
 
 
@@ -1871,7 +1975,7 @@ class MsPowerBiCredentialAuthEntraId(ApiSecretChangeNestedResource):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {}
 
 
@@ -1922,7 +2026,7 @@ class MsPowerBiCredential(Credential):
 
     def _nested_objects(
         self,
-    ) -> dict[str, Optional["Diffable | list[Diffable]"] | None]:
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
         return {"auth": self.auth}
 
 
@@ -1978,6 +2082,115 @@ class SigmaCredential(Credential):
         return {"client_secret"}
 
 
+class TeradataCredentialUserPassword(ApiSecretChangeNestedResource):
+    """Teradata 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 _nested_objects(
+        self,
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
+        return {}
+
+
+TeradataCredentialAuth = Union[TeradataCredentialUserPassword]
+
+
+class TeradataCredential(Credential):
+    """Credential to connect to a Teradata warehouse."""
+
+    def __init__(
+        self,
+        name: str,
+        host: str,
+        auth: TeradataCredentialAuth,
+        ssl_mode: TeradataSSLMode,
+        https_port: int | None = None,
+        tdmst_port: int | None = None,
+        enable_catalog: bool = False,
+        display_name: str | None = None,
+        ignore_changes: bool = False,
+        __internal__: ResourceGraph | None = None,
+    ):
+        """
+        Constructor.
+
+        :param name: Unique resource name assigned to the credential
+        :param host: Host address of the server.
+        :param auth: Credentials to use for authentication.
+        :param ssl_mode: SSL encryption mode when connecting to Teradata.
+        :param https_port: Port number to use to access the Teradata server
+            via SSL connection (default 443).
+        :param tdmst_port: Port number to use to access the Teradata server
+            via non-SSL connection (default 1025).
+        :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.
+        """
+        super().__init__(
+            name=name,
+            display_name=display_name,
+            ignore_changes=ignore_changes,
+            __internal__=__internal__,
+        )
+
+        self.host = host
+        self.auth = auth
+        self.ssl_mode = _maybe_cast(ssl_mode, TeradataSSLMode)
+        self.https_port = https_port
+        self.tdmst_port = tdmst_port
+        self.enable_catalog = enable_catalog
+
+    def _immutable_fields(self) -> set[str]:
+        return set({})
+
+    def _mutable_fields(self) -> set[str]:
+        return {
+            *super()._mutable_fields(),
+            *{
+                "host",
+                "enable_catalog",
+                "ssl_mode",
+                "https_port",
+                "tdmst_port",
+            },
+        }
+
+    def _secret_fields(self) -> set[str]:
+        return set()
+
+    def _nested_objects(
+        self,
+    ) -> dict[str, "Diffable | list[Diffable] | None"]:
+        return {"auth": self.auth}
+
+
 WarehouseCredential = Union[
     AzureSynapseCredential,
     AwsAthenaCredential,
@@ -1986,5 +2199,7 @@ WarehouseCredential = Union[
     DatabricksCredential,
     GcpCredential,
     PostgreSqlCredential,
+    OracleCredential,
     SnowflakeCredential,
+    TeradataCredential,
 ]

{validio_sdk-7.2.0 → validio_sdk-7.3.1}/validio_sdk/resource/sources.py

@@ -39,6 +39,7 @@ from validio_sdk.resource.credentials import (
     OracleCredential,
     PostgreSqlCredential,
     SnowflakeCredential,
+    TeradataCredential,
     WarehouseCredential,
 )
 from validio_sdk.resource.enums import IncidentGroupPriority
@@ -1444,6 +1445,77 @@ class OracleSource(Source):
         }
 
 
+class TeradataSource(Source):
+    """A Teradata source configuration."""
+
+    def __init__(
+        self,
+        name: str,
+        credential: TeradataCredential,
+        database: str,
+        table: str,
+        schedule: str | None,
+        display_name: str | None = None,
+        owner: str | None = None,
+        ignore_changes: 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 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 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,
+            jtd_schema=jtd_schema,
+            description=description,
+            tags=tags,
+            priority=priority,
+        )
+
+        self.database = database
+        self.table = table
+        self.schedule = schedule
+
+    def _immutable_fields(self) -> set[str]:
+        return {
+            *super()._immutable_fields(),
+            *{
+                "database",
+                "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,
+            "table": self.table,
+        }
+
+
 # Streaming