frogml-core 0.0.1__py3-none-any.whl

_frogml_proto/qwak/kube_deployment_captain/deployment_pb2.pyi

@@ -0,0 +1,482 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import google.protobuf.wrappers_pb2
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _KubeDeploymentType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _KubeDeploymentTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_KubeDeploymentType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    UNKNOWN: _KubeDeploymentType.ValueType  # 0
+    REALTIME: _KubeDeploymentType.ValueType  # 1
+    STREAM: _KubeDeploymentType.ValueType  # 2
+
+class KubeDeploymentType(_KubeDeploymentType, metaclass=_KubeDeploymentTypeEnumTypeWrapper): ...
+
+UNKNOWN: KubeDeploymentType.ValueType  # 0
+REALTIME: KubeDeploymentType.ValueType  # 1
+STREAM: KubeDeploymentType.ValueType  # 2
+global___KubeDeploymentType = KubeDeploymentType
+
+class KubeAdvancedDeploymentOptions(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NUMBER_OF_WORKERS_FIELD_NUMBER: builtins.int
+    HTTP_REQUEST_TIMEOUT_MS_FIELD_NUMBER: builtins.int
+    DAEMON_MODE_FIELD_NUMBER: builtins.int
+    CUSTOM_IAM_ROLE_ARN_FIELD_NUMBER: builtins.int
+    MAX_BATCH_SIZE_FIELD_NUMBER: builtins.int
+    DEPLOYMENT_PROCESS_TIMEOUT_LIMIT_FIELD_NUMBER: builtins.int
+    PURCHASE_OPTION_FIELD_NUMBER: builtins.int
+    number_of_workers: builtins.int
+    """Number of workers http server"""
+    http_request_timeout_ms: builtins.int
+    """Timeout for request timeout"""
+    @property
+    def daemon_mode(self) -> google.protobuf.wrappers_pb2.BoolValue:
+        """Serve model container in daemon mode"""
+    custom_iam_role_arn: builtins.str
+    """Custom IAM Role ARN"""
+    max_batch_size: builtins.int
+    """Max batch size of records"""
+    deployment_process_timeout_limit: builtins.int
+    """The timeout for the deployment (in seconds)"""
+    purchase_option: builtins.str
+    """Whether it is spot/ondemand (default - spot)"""
+    def __init__(
+        self,
+        *,
+        number_of_workers: builtins.int = ...,
+        http_request_timeout_ms: builtins.int = ...,
+        daemon_mode: google.protobuf.wrappers_pb2.BoolValue | None = ...,
+        custom_iam_role_arn: builtins.str = ...,
+        max_batch_size: builtins.int = ...,
+        deployment_process_timeout_limit: builtins.int = ...,
+        purchase_option: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["daemon_mode", b"daemon_mode"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["custom_iam_role_arn", b"custom_iam_role_arn", "daemon_mode", b"daemon_mode", "deployment_process_timeout_limit", b"deployment_process_timeout_limit", "http_request_timeout_ms", b"http_request_timeout_ms", "max_batch_size", b"max_batch_size", "number_of_workers", b"number_of_workers", "purchase_option", b"purchase_option"]) -> None: ...
+
+global___KubeAdvancedDeploymentOptions = KubeAdvancedDeploymentOptions
+
+class KubeServingStrategy(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    REALTIME_CONFIG_FIELD_NUMBER: builtins.int
+    STREAM_CONFIG_FIELD_NUMBER: builtins.int
+    @property
+    def realtime_config(self) -> global___RealTimeConfig:
+        """RealTime will replace advanced - a placeholder for now"""
+    @property
+    def stream_config(self) -> global___StreamConfig:
+        """Deploy using streaming solution (kafka)"""
+    def __init__(
+        self,
+        *,
+        realtime_config: global___RealTimeConfig | None = ...,
+        stream_config: global___StreamConfig | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["Strategy", b"Strategy", "realtime_config", b"realtime_config", "stream_config", b"stream_config"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["Strategy", b"Strategy", "realtime_config", b"realtime_config", "stream_config", b"stream_config"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["Strategy", b"Strategy"]) -> typing_extensions.Literal["realtime_config", "stream_config"] | None: ...
+
+global___KubeServingStrategy = KubeServingStrategy
+
+class AutoScalingSpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MIN_REPLICA_COUNT_FIELD_NUMBER: builtins.int
+    MAX_REPLICA_COUNT_FIELD_NUMBER: builtins.int
+    POLLING_INTERVAL_FIELD_NUMBER: builtins.int
+    COOL_DOWN_PERIOD_FIELD_NUMBER: builtins.int
+    TRIGGERS_FIELD_NUMBER: builtins.int
+    min_replica_count: builtins.int
+    """The minimum number of replicas will scale the resource down to"""
+    max_replica_count: builtins.int
+    """The maximum number of replicas of the target resource"""
+    polling_interval: builtins.int
+    """This is the interval to check each trigger on. By default it's every 30 seconds."""
+    cool_down_period: builtins.int
+    """The period to wait after the last trigger reported active before scaling the resource back to 0. By default it’s 5 minutes (300 seconds)."""
+    @property
+    def triggers(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Triggers]:
+        """List of triggers to activate scaling(up/down) of the target resource based on"""
+    def __init__(
+        self,
+        *,
+        min_replica_count: builtins.int = ...,
+        max_replica_count: builtins.int = ...,
+        polling_interval: builtins.int = ...,
+        cool_down_period: builtins.int = ...,
+        triggers: collections.abc.Iterable[global___Triggers] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cool_down_period", b"cool_down_period", "max_replica_count", b"max_replica_count", "min_replica_count", b"min_replica_count", "polling_interval", b"polling_interval", "triggers", b"triggers"]) -> None: ...
+
+global___AutoScalingSpec = AutoScalingSpec
+
+class Triggers(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PROMETHEUS_TRIGGER_FIELD_NUMBER: builtins.int
+    @property
+    def prometheus_trigger(self) -> global___PrometheusTrigger: ...
+    def __init__(
+        self,
+        *,
+        prometheus_trigger: global___PrometheusTrigger | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["trigger_type", b"trigger_type"]) -> typing_extensions.Literal["prometheus_trigger"] | None: ...
+
+global___Triggers = Triggers
+
+class PrometheusTrigger(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    METRIC_NAME_FIELD_NUMBER: builtins.int
+    THRESHOLD_FIELD_NUMBER: builtins.int
+    QUERY_FIELD_NUMBER: builtins.int
+    metric_name: builtins.str
+    """Name to identify the Metric in the external.metrics.k8s.io API"""
+    threshold: builtins.int
+    """Value to initiate scaling on"""
+    query: builtins.str
+    """Promql syntax query"""
+    def __init__(
+        self,
+        *,
+        metric_name: builtins.str = ...,
+        threshold: builtins.int = ...,
+        query: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["metric_name", b"metric_name", "query", b"query", "threshold", b"threshold"]) -> None: ...
+
+global___PrometheusTrigger = PrometheusTrigger
+
+class RealTimeConfig(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SELECTED_VARIATION_MAPPING_NAME_FIELD_NUMBER: builtins.int
+    ASSUMED_IAM_ROLE_ARN_FIELD_NUMBER: builtins.int
+    selected_variation_mapping_name: builtins.str
+    assumed_iam_role_arn: builtins.str
+    def __init__(
+        self,
+        *,
+        selected_variation_mapping_name: builtins.str = ...,
+        assumed_iam_role_arn: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["assumed_iam_role_arn", b"assumed_iam_role_arn", "selected_variation_mapping_name", b"selected_variation_mapping_name"]) -> None: ...
+
+global___RealTimeConfig = RealTimeConfig
+
+class StreamConfig(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    KAFKA_FIELD_NUMBER: builtins.int
+    @property
+    def kafka(self) -> global___KafkaConfig: ...
+    def __init__(
+        self,
+        *,
+        kafka: global___KafkaConfig | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["configuration", b"configuration", "kafka", b"kafka"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["configuration", b"configuration", "kafka", b"kafka"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["configuration", b"configuration"]) -> typing_extensions.Literal["kafka"] | None: ...
+
+global___StreamConfig = StreamConfig
+
+class KafkaConfig(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class Consumer(google.protobuf.message.Message):
+        """The consumer config"""
+
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        class _AutoOffSetType:
+            ValueType = typing.NewType("ValueType", builtins.int)
+            V: typing_extensions.TypeAlias = ValueType
+
+        class _AutoOffSetTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[KafkaConfig.Consumer._AutoOffSetType.ValueType], builtins.type):  # noqa: F821
+            DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+            UNKNOWN: KafkaConfig.Consumer._AutoOffSetType.ValueType  # 0
+            EARLIEST: KafkaConfig.Consumer._AutoOffSetType.ValueType  # 1
+            LATEST: KafkaConfig.Consumer._AutoOffSetType.ValueType  # 2
+
+        class AutoOffSetType(_AutoOffSetType, metaclass=_AutoOffSetTypeEnumTypeWrapper):
+            """The offset types"""
+
+        UNKNOWN: KafkaConfig.Consumer.AutoOffSetType.ValueType  # 0
+        EARLIEST: KafkaConfig.Consumer.AutoOffSetType.ValueType  # 1
+        LATEST: KafkaConfig.Consumer.AutoOffSetType.ValueType  # 2
+
+        BOOTSTRAP_SERVER_FIELD_NUMBER: builtins.int
+        TOPIC_FIELD_NUMBER: builtins.int
+        GROUP_FIELD_NUMBER: builtins.int
+        TIMEOUT_FIELD_NUMBER: builtins.int
+        AUTO_OFFSET_TYPE_FIELD_NUMBER: builtins.int
+        MAX_BATCH_SIZE_FIELD_NUMBER: builtins.int
+        MAX_POLL_LATENCY_FIELD_NUMBER: builtins.int
+        @property
+        def bootstrap_server(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+            """Bootstrap servers"""
+        topic: builtins.str
+        """Consumer Topic"""
+        group: builtins.str
+        """Consumer Group"""
+        timeout: builtins.int
+        """Consumer timeout"""
+        auto_offset_type: global___KafkaConfig.Consumer.AutoOffSetType.ValueType
+        """The offset for the consumer"""
+        max_batch_size: builtins.int
+        max_poll_latency: builtins.float
+        def __init__(
+            self,
+            *,
+            bootstrap_server: collections.abc.Iterable[builtins.str] | None = ...,
+            topic: builtins.str = ...,
+            group: builtins.str = ...,
+            timeout: builtins.int = ...,
+            auto_offset_type: global___KafkaConfig.Consumer.AutoOffSetType.ValueType = ...,
+            max_batch_size: builtins.int = ...,
+            max_poll_latency: builtins.float = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["auto_offset_type", b"auto_offset_type", "bootstrap_server", b"bootstrap_server", "group", b"group", "max_batch_size", b"max_batch_size", "max_poll_latency", b"max_poll_latency", "timeout", b"timeout", "topic", b"topic"]) -> None: ...
+
+    class Producer(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        class _CompressionType:
+            ValueType = typing.NewType("ValueType", builtins.int)
+            V: typing_extensions.TypeAlias = ValueType
+
+        class _CompressionTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[KafkaConfig.Producer._CompressionType.ValueType], builtins.type):  # noqa: F821
+            DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+            UNKNOWN: KafkaConfig.Producer._CompressionType.ValueType  # 0
+            UNCOMPRESSED: KafkaConfig.Producer._CompressionType.ValueType  # 1
+            GZIP: KafkaConfig.Producer._CompressionType.ValueType  # 2
+            SNAPPY: KafkaConfig.Producer._CompressionType.ValueType  # 3
+            LZ4: KafkaConfig.Producer._CompressionType.ValueType  # 4
+            ZSTD: KafkaConfig.Producer._CompressionType.ValueType  # 5
+
+        class CompressionType(_CompressionType, metaclass=_CompressionTypeEnumTypeWrapper):
+            """The compression types"""
+
+        UNKNOWN: KafkaConfig.Producer.CompressionType.ValueType  # 0
+        UNCOMPRESSED: KafkaConfig.Producer.CompressionType.ValueType  # 1
+        GZIP: KafkaConfig.Producer.CompressionType.ValueType  # 2
+        SNAPPY: KafkaConfig.Producer.CompressionType.ValueType  # 3
+        LZ4: KafkaConfig.Producer.CompressionType.ValueType  # 4
+        ZSTD: KafkaConfig.Producer.CompressionType.ValueType  # 5
+
+        BOOTSTRAP_SERVER_FIELD_NUMBER: builtins.int
+        TOPIC_FIELD_NUMBER: builtins.int
+        COMPRESSION_TYPE_FIELD_NUMBER: builtins.int
+        @property
+        def bootstrap_server(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+            """The producer bootstrap servers"""
+        topic: builtins.str
+        """The producer topic"""
+        compression_type: global___KafkaConfig.Producer.CompressionType.ValueType
+        """The producer compression type"""
+        def __init__(
+            self,
+            *,
+            bootstrap_server: collections.abc.Iterable[builtins.str] | None = ...,
+            topic: builtins.str = ...,
+            compression_type: global___KafkaConfig.Producer.CompressionType.ValueType = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["bootstrap_server", b"bootstrap_server", "compression_type", b"compression_type", "topic", b"topic"]) -> None: ...
+
+    CONSUMER_FIELD_NUMBER: builtins.int
+    PRODUCER_FIELD_NUMBER: builtins.int
+    WORKERS_FIELD_NUMBER: builtins.int
+    @property
+    def consumer(self) -> global___KafkaConfig.Consumer:
+        """Consumer config"""
+    @property
+    def producer(self) -> global___KafkaConfig.Producer:
+        """Producer config"""
+    workers: builtins.int
+    """Number of prediction workers"""
+    def __init__(
+        self,
+        *,
+        consumer: global___KafkaConfig.Consumer | None = ...,
+        producer: global___KafkaConfig.Producer | None = ...,
+        workers: builtins.int = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["consumer", b"consumer", "producer", b"producer"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["consumer", b"consumer", "producer", b"producer", "workers", b"workers"]) -> None: ...
+
+global___KafkaConfig = KafkaConfig
+
+class KubeDeploymentSize(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _GpuType:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _GpuTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[KubeDeploymentSize._GpuType.ValueType], builtins.type):  # noqa: F821
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        INVALID_GPU: KubeDeploymentSize._GpuType.ValueType  # 0
+        NVIDIA_K80: KubeDeploymentSize._GpuType.ValueType  # 1
+        NVIDIA_V100: KubeDeploymentSize._GpuType.ValueType  # 2
+        NVIDIA_A100: KubeDeploymentSize._GpuType.ValueType  # 3
+        NVIDIA_T4: KubeDeploymentSize._GpuType.ValueType  # 4
+        NVIDIA_A10G: KubeDeploymentSize._GpuType.ValueType  # 5
+        NVIDIA_L4: KubeDeploymentSize._GpuType.ValueType  # 6
+        NVIDIA_T4_1_4_15: KubeDeploymentSize._GpuType.ValueType  # 7
+        """More specific node types
+        t4.xl
+        """
+        NVIDIA_T4_1_8_30: KubeDeploymentSize._GpuType.ValueType  # 8
+        """t4.2xl"""
+        NVIDIA_T4_1_16_60: KubeDeploymentSize._GpuType.ValueType  # 9
+        """t4.4xl"""
+        NVIDIA_A100_80GB_8_96_1360: KubeDeploymentSize._GpuType.ValueType  # 10
+        """a100.8xl"""
+        NVIDIA_V100_1_8_52: KubeDeploymentSize._GpuType.ValueType  # 11
+        """a100.xl"""
+        NVIDIA_V100_4_32_208: KubeDeploymentSize._GpuType.ValueType  # 12
+        """v100.4xl"""
+
+    class GpuType(_GpuType, metaclass=_GpuTypeEnumTypeWrapper): ...
+    INVALID_GPU: KubeDeploymentSize.GpuType.ValueType  # 0
+    NVIDIA_K80: KubeDeploymentSize.GpuType.ValueType  # 1
+    NVIDIA_V100: KubeDeploymentSize.GpuType.ValueType  # 2
+    NVIDIA_A100: KubeDeploymentSize.GpuType.ValueType  # 3
+    NVIDIA_T4: KubeDeploymentSize.GpuType.ValueType  # 4
+    NVIDIA_A10G: KubeDeploymentSize.GpuType.ValueType  # 5
+    NVIDIA_L4: KubeDeploymentSize.GpuType.ValueType  # 6
+    NVIDIA_T4_1_4_15: KubeDeploymentSize.GpuType.ValueType  # 7
+    """More specific node types
+    t4.xl
+    """
+    NVIDIA_T4_1_8_30: KubeDeploymentSize.GpuType.ValueType  # 8
+    """t4.2xl"""
+    NVIDIA_T4_1_16_60: KubeDeploymentSize.GpuType.ValueType  # 9
+    """t4.4xl"""
+    NVIDIA_A100_80GB_8_96_1360: KubeDeploymentSize.GpuType.ValueType  # 10
+    """a100.8xl"""
+    NVIDIA_V100_1_8_52: KubeDeploymentSize.GpuType.ValueType  # 11
+    """a100.xl"""
+    NVIDIA_V100_4_32_208: KubeDeploymentSize.GpuType.ValueType  # 12
+    """v100.4xl"""
+
+    class _MemoryUnit:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _MemoryUnitEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[KubeDeploymentSize._MemoryUnit.ValueType], builtins.type):  # noqa: F821
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        UNKNOWN: KubeDeploymentSize._MemoryUnit.ValueType  # 0
+        MIB: KubeDeploymentSize._MemoryUnit.ValueType  # 1
+        GIB: KubeDeploymentSize._MemoryUnit.ValueType  # 2
+
+    class MemoryUnit(_MemoryUnit, metaclass=_MemoryUnitEnumTypeWrapper): ...
+    UNKNOWN: KubeDeploymentSize.MemoryUnit.ValueType  # 0
+    MIB: KubeDeploymentSize.MemoryUnit.ValueType  # 1
+    GIB: KubeDeploymentSize.MemoryUnit.ValueType  # 2
+
+    class GpuResources(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        GPU_TYPE_FIELD_NUMBER: builtins.int
+        GPU_AMOUNT_FIELD_NUMBER: builtins.int
+        gpu_type: global___KubeDeploymentSize.GpuType.ValueType
+        """The type of the GPU"""
+        gpu_amount: builtins.int
+        """Amount of GPUs"""
+        def __init__(
+            self,
+            *,
+            gpu_type: global___KubeDeploymentSize.GpuType.ValueType = ...,
+            gpu_amount: builtins.int = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["gpu_amount", b"gpu_amount", "gpu_type", b"gpu_type"]) -> None: ...
+
+    NUMBER_OF_PODS_FIELD_NUMBER: builtins.int
+    CPU_FIELD_NUMBER: builtins.int
+    MEMORY_AMOUNT_FIELD_NUMBER: builtins.int
+    MEMORY_UNITS_FIELD_NUMBER: builtins.int
+    GPU_RESOURCES_FIELD_NUMBER: builtins.int
+    number_of_pods: builtins.int
+    """Number of pods to deploy"""
+    cpu: builtins.float
+    """Cpu"""
+    memory_amount: builtins.int
+    """Amount of memory"""
+    memory_units: global___KubeDeploymentSize.MemoryUnit.ValueType
+    """Units type of memory"""
+    @property
+    def gpu_resources(self) -> global___KubeDeploymentSize.GpuResources:
+        """GPU Resource for deployed model"""
+    def __init__(
+        self,
+        *,
+        number_of_pods: builtins.int = ...,
+        cpu: builtins.float = ...,
+        memory_amount: builtins.int = ...,
+        memory_units: global___KubeDeploymentSize.MemoryUnit.ValueType = ...,
+        gpu_resources: global___KubeDeploymentSize.GpuResources | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["gpu_resources", b"gpu_resources"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cpu", b"cpu", "gpu_resources", b"gpu_resources", "memory_amount", b"memory_amount", "memory_units", b"memory_units", "number_of_pods", b"number_of_pods"]) -> None: ...
+
+global___KubeDeploymentSize = KubeDeploymentSize
+
+class KubeRuntimeDeploymentSettings(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _LogLevel:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _LogLevelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[KubeRuntimeDeploymentSettings._LogLevel.ValueType], builtins.type):  # noqa: F821
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        NOT_SET: KubeRuntimeDeploymentSettings._LogLevel.ValueType  # 0
+        ERROR: KubeRuntimeDeploymentSettings._LogLevel.ValueType  # 1
+        WARN: KubeRuntimeDeploymentSettings._LogLevel.ValueType  # 2
+        INFO: KubeRuntimeDeploymentSettings._LogLevel.ValueType  # 3
+        DEBUG: KubeRuntimeDeploymentSettings._LogLevel.ValueType  # 4
+
+    class LogLevel(_LogLevel, metaclass=_LogLevelEnumTypeWrapper): ...
+    NOT_SET: KubeRuntimeDeploymentSettings.LogLevel.ValueType  # 0
+    ERROR: KubeRuntimeDeploymentSettings.LogLevel.ValueType  # 1
+    WARN: KubeRuntimeDeploymentSettings.LogLevel.ValueType  # 2
+    INFO: KubeRuntimeDeploymentSettings.LogLevel.ValueType  # 3
+    DEBUG: KubeRuntimeDeploymentSettings.LogLevel.ValueType  # 4
+
+    ROOT_LOGGER_LEVEL_FIELD_NUMBER: builtins.int
+    root_logger_level: global___KubeRuntimeDeploymentSettings.LogLevel.ValueType
+    def __init__(
+        self,
+        *,
+        root_logger_level: global___KubeRuntimeDeploymentSettings.LogLevel.ValueType = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["root_logger_level", b"root_logger_level"]) -> None: ...
+
+global___KubeRuntimeDeploymentSettings = KubeRuntimeDeploymentSettings

_frogml_proto/qwak/kube_deployment_captain/deployment_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

_frogml_proto/qwak/kube_deployment_captain/feature_set_deployment_pb2.py

@@ -0,0 +1,172 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.kube_deployment_captain/feature_set_deployment.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from _frogml_proto.qwak.feature_store.features import feature_set_types_pb2 as qwak_dot_feature__store_dot_features_dot_feature__set__types__pb2
+from _frogml_proto.qwak.feature_store.entities import entity_pb2 as qwak_dot_feature__store_dot_entities_dot_entity__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n9qwak/kube_deployment_captain/feature_set_deployment.proto\x12\x1cqwak.kube.deployment.captain\x1a\x33qwak/feature_store/features/feature_set_types.proto\x1a(qwak/feature_store/entities/entity.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x81\x07\n\x14\x46\x65\x61tureSetDeployment\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\x12S\n\x15streaming_feature_set\x18\x02 \x01(\x0b\x32\x32.qwak.feature.store.features.StreamingFeatureSetV1H\x00\x12K\n\x11\x62\x61tch_feature_set\x18\x03 \x01(\x0b\x32..qwak.feature.store.features.BatchFeatureSetV1H\x00\x12h\n!streaming_aggregation_feature_set\x18\x04 \x01(\x0b\x32;.qwak.feature.store.features.StreamingAggregationFeatureSetH\x00\x12|\n\x1e\x65xtra_deployment_configuration\x18\x05 \x03(\x0b\x32T.qwak.kube.deployment.captain.FeatureSetDeployment.ExtraDeploymentConfigurationEntry\x12n\n\x17\x65xtra_env_configuration\x18\x06 \x03(\x0b\x32M.qwak.kube.deployment.captain.FeatureSetDeployment.ExtraEnvConfigurationEntry\x12=\n\x06\x65ntity\x18\x07 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\x12\x15\n\rfeatureset_id\x18\x08 \x01(\t\x12\x1a\n\x12secret_service_url\x18\t \x01(\t\x12\x0e\n\x06run_id\x18\n \x01(\x03\x12&\n\x1eqwak_internal_protocol_version\x18\x0b \x01(\x05\x12\x1a\n\x12\x66\x65\x61tureset_version\x18\x0c \x01(\x05\x1a\x43\n!ExtraDeploymentConfigurationEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a<\n\x1a\x45xtraEnvConfigurationEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\r\n\x0b\x66\x65\x61ture_set\"\x94\x01\n\x0c\x42\x61tchDetails\x12\x38\n\x14\x62\x61tch_execution_date\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ntry_number\x18\x02 \x01(\x05\x12\x36\n\x12\x62\x61tch_trigger_date\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"+\n\x13\x45xecutionIdentifier\x12\x14\n\x0c\x65xecution_id\x18\x01 \x01(\t\"\x84\x02\n\x18\x46\x65\x61tureSetDeploymentSpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\n\n\x02id\x18\x02 \x01(\t\x12\x46\n\x08metadata\x18\x03 \x01(\x0b\x32\x34.qwak.kube.deployment.captain.FeatureSetUserMetadata\x12>\n\x06\x65ntity\x18\x04 \x01(\x0b\x32..qwak.kube.deployment.captain.EntityDefinition\x12\x46\n\x10\x66\x65\x61ture_set_type\x18\x05 \x01(\x0b\x32,.qwak.kube.deployment.captain.FeatureSetType\"R\n\x16\x46\x65\x61tureSetUserMetadata\x12\r\n\x05owner\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x03 \x01(\t\"d\n\x10\x45ntityDefinition\x12\x11\n\tentity_id\x18\x01 \x01(\t\x12=\n\x0b\x65ntity_spec\x18\x02 \x01(\x0b\x32(.qwak.kube.deployment.captain.EntitySpec\"p\n\x0e\x46\x65\x61tureSetType\x12R\n\x15streaming_feature_set\x18\x01 \x01(\x0b\x32\x31.qwak.kube.deployment.captain.StreamingFeatureSetH\x00\x42\n\n\x08set_type\"]\n\x13StreamingFeatureSet\x12\x46\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32\x31.qwak.kube.deployment.captain.StreamingDataSource\"\xe6\x01\n\x0bKafkaSource\x12\x16\n\x0e\x63onsumer_topic\x18\x01 \x01(\t\x12!\n\x19\x63onsumer_bootstrap_server\x18\x02 \x01(\t\x12\x18\n\x10\x63onsumer_timeout\x18\x03 \x01(\t\x12\x16\n\x0e\x63onsumer_group\x18\x04 \x01(\t\x12\x18\n\x10\x61uto_offset_type\x18\x05 \x01(\t\x12\x17\n\x0fsecret_username\x18\x06 \x01(\t\x12\x17\n\x0fsecret_password\x18\x07 \x01(\t\x12\x1e\n\x16\x63onsumer_configuration\x18\x08 \x01(\t\"`\n\x13StreamingDataSource\x12\x41\n\x0ckafka_source\x18\x01 \x01(\x0b\x32).qwak.kube.deployment.captain.KafkaSourceH\x00\x42\x06\n\x04type\"z\n\nEntitySpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04keys\x18\x02 \x03(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12;\n\nvalue_type\x18\x04 \x01(\x0e\x32\'.qwak.kube.deployment.captain.ValueType*-\n\tValueType\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06STRING\x10\x01\x12\x07\n\x03INT\x10\x02\x42+\n\'com.qwak.ai.kube.deployment.captain.apiP\x01\x62\x06proto3')
+
+_VALUETYPE = DESCRIPTOR.enum_types_by_name['ValueType']
+ValueType = enum_type_wrapper.EnumTypeWrapper(_VALUETYPE)
+INVALID = 0
+STRING = 1
+INT = 2
+
+
+_FEATURESETDEPLOYMENT = DESCRIPTOR.message_types_by_name['FeatureSetDeployment']
+_FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY = _FEATURESETDEPLOYMENT.nested_types_by_name['ExtraDeploymentConfigurationEntry']
+_FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY = _FEATURESETDEPLOYMENT.nested_types_by_name['ExtraEnvConfigurationEntry']
+_BATCHDETAILS = DESCRIPTOR.message_types_by_name['BatchDetails']
+_EXECUTIONIDENTIFIER = DESCRIPTOR.message_types_by_name['ExecutionIdentifier']
+_FEATURESETDEPLOYMENTSPEC = DESCRIPTOR.message_types_by_name['FeatureSetDeploymentSpec']
+_FEATURESETUSERMETADATA = DESCRIPTOR.message_types_by_name['FeatureSetUserMetadata']
+_ENTITYDEFINITION = DESCRIPTOR.message_types_by_name['EntityDefinition']
+_FEATURESETTYPE = DESCRIPTOR.message_types_by_name['FeatureSetType']
+_STREAMINGFEATURESET = DESCRIPTOR.message_types_by_name['StreamingFeatureSet']
+_KAFKASOURCE = DESCRIPTOR.message_types_by_name['KafkaSource']
+_STREAMINGDATASOURCE = DESCRIPTOR.message_types_by_name['StreamingDataSource']
+_ENTITYSPEC = DESCRIPTOR.message_types_by_name['EntitySpec']
+FeatureSetDeployment = _reflection.GeneratedProtocolMessageType('FeatureSetDeployment', (_message.Message,), {
+
+  'ExtraDeploymentConfigurationEntry' : _reflection.GeneratedProtocolMessageType('ExtraDeploymentConfigurationEntry', (_message.Message,), {
+    'DESCRIPTOR' : _FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY,
+    '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+    # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.FeatureSetDeployment.ExtraDeploymentConfigurationEntry)
+    })
+  ,
+
+  'ExtraEnvConfigurationEntry' : _reflection.GeneratedProtocolMessageType('ExtraEnvConfigurationEntry', (_message.Message,), {
+    'DESCRIPTOR' : _FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY,
+    '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+    # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.FeatureSetDeployment.ExtraEnvConfigurationEntry)
+    })
+  ,
+  'DESCRIPTOR' : _FEATURESETDEPLOYMENT,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.FeatureSetDeployment)
+  })
+_sym_db.RegisterMessage(FeatureSetDeployment)
+_sym_db.RegisterMessage(FeatureSetDeployment.ExtraDeploymentConfigurationEntry)
+_sym_db.RegisterMessage(FeatureSetDeployment.ExtraEnvConfigurationEntry)
+
+BatchDetails = _reflection.GeneratedProtocolMessageType('BatchDetails', (_message.Message,), {
+  'DESCRIPTOR' : _BATCHDETAILS,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.BatchDetails)
+  })
+_sym_db.RegisterMessage(BatchDetails)
+
+ExecutionIdentifier = _reflection.GeneratedProtocolMessageType('ExecutionIdentifier', (_message.Message,), {
+  'DESCRIPTOR' : _EXECUTIONIDENTIFIER,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.ExecutionIdentifier)
+  })
+_sym_db.RegisterMessage(ExecutionIdentifier)
+
+FeatureSetDeploymentSpec = _reflection.GeneratedProtocolMessageType('FeatureSetDeploymentSpec', (_message.Message,), {
+  'DESCRIPTOR' : _FEATURESETDEPLOYMENTSPEC,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.FeatureSetDeploymentSpec)
+  })
+_sym_db.RegisterMessage(FeatureSetDeploymentSpec)
+
+FeatureSetUserMetadata = _reflection.GeneratedProtocolMessageType('FeatureSetUserMetadata', (_message.Message,), {
+  'DESCRIPTOR' : _FEATURESETUSERMETADATA,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.FeatureSetUserMetadata)
+  })
+_sym_db.RegisterMessage(FeatureSetUserMetadata)
+
+EntityDefinition = _reflection.GeneratedProtocolMessageType('EntityDefinition', (_message.Message,), {
+  'DESCRIPTOR' : _ENTITYDEFINITION,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.EntityDefinition)
+  })
+_sym_db.RegisterMessage(EntityDefinition)
+
+FeatureSetType = _reflection.GeneratedProtocolMessageType('FeatureSetType', (_message.Message,), {
+  'DESCRIPTOR' : _FEATURESETTYPE,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.FeatureSetType)
+  })
+_sym_db.RegisterMessage(FeatureSetType)
+
+StreamingFeatureSet = _reflection.GeneratedProtocolMessageType('StreamingFeatureSet', (_message.Message,), {
+  'DESCRIPTOR' : _STREAMINGFEATURESET,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.StreamingFeatureSet)
+  })
+_sym_db.RegisterMessage(StreamingFeatureSet)
+
+KafkaSource = _reflection.GeneratedProtocolMessageType('KafkaSource', (_message.Message,), {
+  'DESCRIPTOR' : _KAFKASOURCE,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.KafkaSource)
+  })
+_sym_db.RegisterMessage(KafkaSource)
+
+StreamingDataSource = _reflection.GeneratedProtocolMessageType('StreamingDataSource', (_message.Message,), {
+  'DESCRIPTOR' : _STREAMINGDATASOURCE,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.StreamingDataSource)
+  })
+_sym_db.RegisterMessage(StreamingDataSource)
+
+EntitySpec = _reflection.GeneratedProtocolMessageType('EntitySpec', (_message.Message,), {
+  'DESCRIPTOR' : _ENTITYSPEC,
+  '__module__' : 'qwak.kube_deployment_captain.feature_set_deployment_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.EntitySpec)
+  })
+_sym_db.RegisterMessage(EntitySpec)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\'com.qwak.ai.kube.deployment.captain.apiP\001'
+  _FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY._options = None
+  _FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY._serialized_options = b'8\001'
+  _FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY._options = None
+  _FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY._serialized_options = b'8\001'
+  _VALUETYPE._serialized_start=2428
+  _VALUETYPE._serialized_end=2473
+  _FEATURESETDEPLOYMENT._serialized_start=220
+  _FEATURESETDEPLOYMENT._serialized_end=1117
+  _FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY._serialized_start=973
+  _FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY._serialized_end=1040
+  _FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY._serialized_start=1042
+  _FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY._serialized_end=1102
+  _BATCHDETAILS._serialized_start=1120
+  _BATCHDETAILS._serialized_end=1268
+  _EXECUTIONIDENTIFIER._serialized_start=1270
+  _EXECUTIONIDENTIFIER._serialized_end=1313
+  _FEATURESETDEPLOYMENTSPEC._serialized_start=1316
+  _FEATURESETDEPLOYMENTSPEC._serialized_end=1576
+  _FEATURESETUSERMETADATA._serialized_start=1578
+  _FEATURESETUSERMETADATA._serialized_end=1660
+  _ENTITYDEFINITION._serialized_start=1662
+  _ENTITYDEFINITION._serialized_end=1762
+  _FEATURESETTYPE._serialized_start=1764
+  _FEATURESETTYPE._serialized_end=1876
+  _STREAMINGFEATURESET._serialized_start=1878
+  _STREAMINGFEATURESET._serialized_end=1971
+  _KAFKASOURCE._serialized_start=1974
+  _KAFKASOURCE._serialized_end=2204
+  _STREAMINGDATASOURCE._serialized_start=2206
+  _STREAMINGDATASOURCE._serialized_end=2302
+  _ENTITYSPEC._serialized_start=2304
+  _ENTITYSPEC._serialized_end=2426
+# @@protoc_insertion_point(module_scope)