qwak-core 0.1.37__py3-none-any.whl → 0.5.19__py3-none-any.whl

_qwak_proto/qwak/kube_deployment_captain/deployment_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/kube_deployment_captain/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
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -16,167 +14,53 @@ _sym_db = _symbol_database.Default()
 from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/kube_deployment_captain/deployment.proto\x12\x1cqwak.kube.deployment.captain\x1a\x1egoogle/protobuf/wrappers.proto\"\x84\x02\n\x1dKubeAdvancedDeploymentOptions\x12\x19\n\x11number_of_workers\x18\x01 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x02 \x01(\x05\x12/\n\x0b\x64\x61\x65mon_mode\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.BoolValue\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x04 \x01(\t\x12\x16\n\x0emax_batch_size\x18\x05 \x01(\x05\x12(\n deployment_process_timeout_limit\x18\x06 \x01(\x05\x12\x17\n\x0fpurchase_option\x18\x07 \x01(\t\"\xaf\x01\n\x13KubeServingStrategy\x12G\n\x0frealtime_config\x18\x01 \x01(\x0b\x32,.qwak.kube.deployment.captain.RealTimeConfigH\x00\x12\x43\n\rstream_config\x18\x02 \x01(\x0b\x32*.qwak.kube.deployment.captain.StreamConfigH\x00\x42\n\n\x08Strategy\"\xb5\x01\n\x0f\x41utoScalingSpec\x12\x19\n\x11min_replica_count\x18\x01 \x01(\x05\x12\x19\n\x11max_replica_count\x18\x02 \x01(\x05\x12\x18\n\x10polling_interval\x18\x03 \x01(\x05\x12\x18\n\x10\x63ool_down_period\x18\x04 \x01(\x05\x12\x38\n\x08triggers\x18\x05 \x03(\x0b\x32&.qwak.kube.deployment.captain.Triggers\"i\n\x08Triggers\x12M\n\x12prometheus_trigger\x18\x01 \x01(\x0b\x32/.qwak.kube.deployment.captain.PrometheusTriggerH\x00\x42\x0e\n\x0ctrigger_type\"J\n\x11PrometheusTrigger\x12\x13\n\x0bmetric_name\x18\x01 \x01(\t\x12\x11\n\tthreshold\x18\x02 \x01(\x05\x12\r\n\x05query\x18\x03 \x01(\t\"W\n\x0eRealTimeConfig\x12\'\n\x1fselected_variation_mapping_name\x18\x01 \x01(\t\x12\x1c\n\x14\x61ssumed_iam_role_arn\x18\x02 \x01(\t\"[\n\x0cStreamConfig\x12:\n\x05kafka\x18\x01 \x01(\x0b\x32).qwak.kube.deployment.captain.KafkaConfigH\x00\x42\x0f\n\rconfiguration\"\xb7\x05\n\x0bKafkaConfig\x12\x44\n\x08\x63onsumer\x18\x01 \x01(\x0b\x32\x32.qwak.kube.deployment.captain.KafkaConfig.Consumer\x12\x44\n\x08producer\x18\x02 \x01(\x0b\x32\x32.qwak.kube.deployment.captain.KafkaConfig.Producer\x12\x0f\n\x07workers\x18\x03 \x01(\x05\x1a\x9b\x02\n\x08\x43onsumer\x12\x18\n\x10\x62ootstrap_server\x18\x01 \x03(\t\x12\r\n\x05topic\x18\x02 \x01(\t\x12\r\n\x05group\x18\x03 \x01(\t\x12\x0f\n\x07timeout\x18\x04 \x01(\x05\x12[\n\x10\x61uto_offset_type\x18\x05 \x01(\x0e\x32\x41.qwak.kube.deployment.captain.KafkaConfig.Consumer.AutoOffSetType\x12\x16\n\x0emax_batch_size\x18\x06 \x01(\x05\x12\x18\n\x10max_poll_latency\x18\x07 \x01(\x01\"7\n\x0e\x41utoOffSetType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0c\n\x08\x45\x41RLIEST\x10\x01\x12\n\n\x06LATEST\x10\x02\x1a\xec\x01\n\x08Producer\x12\x18\n\x10\x62ootstrap_server\x18\x01 \x03(\t\x12\r\n\x05topic\x18\x02 \x01(\t\x12\\\n\x10\x63ompression_type\x18\x03 \x01(\x0e\x32\x42.qwak.kube.deployment.captain.KafkaConfig.Producer.CompressionType\"Y\n\x0f\x43ompressionType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x10\n\x0cUNCOMPRESSED\x10\x01\x12\x08\n\x04GZIP\x10\x02\x12\n\n\x06SNAPPY\x10\x03\x12\x07\n\x03LZ4\x10\x04\x12\x08\n\x04ZSTD\x10\x05\"\x84\x04\n\x12KubeDeploymentSize\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0b\n\x03\x63pu\x18\x02 \x01(\x02\x12\x15\n\rmemory_amount\x18\x03 \x01(\x05\x12Q\n\x0cmemory_units\x18\x04 \x01(\x0e\x32;.qwak.kube.deployment.captain.KubeDeploymentSize.MemoryUnit\x12T\n\rgpu_resources\x18\x05 \x01(\x0b\x32=.qwak.kube.deployment.captain.KubeDeploymentSize.GpuResources\x1an\n\x0cGpuResources\x12J\n\x08gpu_type\x18\x01 \x01(\x0e\x32\x38.qwak.kube.deployment.captain.KubeDeploymentSize.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"l\n\x07GpuType\x12\x0f\n\x0bINVALID_GPU\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\"+\n\nMemoryUnit\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\"\xc3\x01\n\x1dKubeRuntimeDeploymentSettings\x12_\n\x11root_logger_level\x18\x01 \x01(\x0e\x32\x44.qwak.kube.deployment.captain.KubeRuntimeDeploymentSettings.LogLevel\"A\n\x08LogLevel\x12\x0b\n\x07NOT_SET\x10\x00\x12\t\n\x05\x45RROR\x10\x01\x12\x08\n\x04WARN\x10\x02\x12\x08\n\x04INFO\x10\x03\x12\t\n\x05\x44\x45\x42UG\x10\x04*;\n\x12KubeDeploymentType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0c\n\x08REALTIME\x10\x01\x12\n\n\x06STREAM\x10\x02\x42+\n\'com.qwak.ai.kube.deployment.captain.apiP\x01\x62\x06proto3')
-
-_KUBEDEPLOYMENTTYPE = DESCRIPTOR.enum_types_by_name['KubeDeploymentType']
-KubeDeploymentType = enum_type_wrapper.EnumTypeWrapper(_KUBEDEPLOYMENTTYPE)
-UNKNOWN = 0
-REALTIME = 1
-STREAM = 2
-
-
-_KUBEADVANCEDDEPLOYMENTOPTIONS = DESCRIPTOR.message_types_by_name['KubeAdvancedDeploymentOptions']
-_KUBESERVINGSTRATEGY = DESCRIPTOR.message_types_by_name['KubeServingStrategy']
-_AUTOSCALINGSPEC = DESCRIPTOR.message_types_by_name['AutoScalingSpec']
-_TRIGGERS = DESCRIPTOR.message_types_by_name['Triggers']
-_PROMETHEUSTRIGGER = DESCRIPTOR.message_types_by_name['PrometheusTrigger']
-_REALTIMECONFIG = DESCRIPTOR.message_types_by_name['RealTimeConfig']
-_STREAMCONFIG = DESCRIPTOR.message_types_by_name['StreamConfig']
-_KAFKACONFIG = DESCRIPTOR.message_types_by_name['KafkaConfig']
-_KAFKACONFIG_CONSUMER = _KAFKACONFIG.nested_types_by_name['Consumer']
-_KAFKACONFIG_PRODUCER = _KAFKACONFIG.nested_types_by_name['Producer']
-_KUBEDEPLOYMENTSIZE = DESCRIPTOR.message_types_by_name['KubeDeploymentSize']
-_KUBEDEPLOYMENTSIZE_GPURESOURCES = _KUBEDEPLOYMENTSIZE.nested_types_by_name['GpuResources']
-_KUBERUNTIMEDEPLOYMENTSETTINGS = DESCRIPTOR.message_types_by_name['KubeRuntimeDeploymentSettings']
-_KAFKACONFIG_CONSUMER_AUTOOFFSETTYPE = _KAFKACONFIG_CONSUMER.enum_types_by_name['AutoOffSetType']
-_KAFKACONFIG_PRODUCER_COMPRESSIONTYPE = _KAFKACONFIG_PRODUCER.enum_types_by_name['CompressionType']
-_KUBEDEPLOYMENTSIZE_GPUTYPE = _KUBEDEPLOYMENTSIZE.enum_types_by_name['GpuType']
-_KUBEDEPLOYMENTSIZE_MEMORYUNIT = _KUBEDEPLOYMENTSIZE.enum_types_by_name['MemoryUnit']
-_KUBERUNTIMEDEPLOYMENTSETTINGS_LOGLEVEL = _KUBERUNTIMEDEPLOYMENTSETTINGS.enum_types_by_name['LogLevel']
-KubeAdvancedDeploymentOptions = _reflection.GeneratedProtocolMessageType('KubeAdvancedDeploymentOptions', (_message.Message,), {
-  'DESCRIPTOR' : _KUBEADVANCEDDEPLOYMENTOPTIONS,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.KubeAdvancedDeploymentOptions)
-  })
-_sym_db.RegisterMessage(KubeAdvancedDeploymentOptions)
-
-KubeServingStrategy = _reflection.GeneratedProtocolMessageType('KubeServingStrategy', (_message.Message,), {
-  'DESCRIPTOR' : _KUBESERVINGSTRATEGY,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.KubeServingStrategy)
-  })
-_sym_db.RegisterMessage(KubeServingStrategy)
-
-AutoScalingSpec = _reflection.GeneratedProtocolMessageType('AutoScalingSpec', (_message.Message,), {
-  'DESCRIPTOR' : _AUTOSCALINGSPEC,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.AutoScalingSpec)
-  })
-_sym_db.RegisterMessage(AutoScalingSpec)
-
-Triggers = _reflection.GeneratedProtocolMessageType('Triggers', (_message.Message,), {
-  'DESCRIPTOR' : _TRIGGERS,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.Triggers)
-  })
-_sym_db.RegisterMessage(Triggers)
-
-PrometheusTrigger = _reflection.GeneratedProtocolMessageType('PrometheusTrigger', (_message.Message,), {
-  'DESCRIPTOR' : _PROMETHEUSTRIGGER,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.PrometheusTrigger)
-  })
-_sym_db.RegisterMessage(PrometheusTrigger)
-
-RealTimeConfig = _reflection.GeneratedProtocolMessageType('RealTimeConfig', (_message.Message,), {
-  'DESCRIPTOR' : _REALTIMECONFIG,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.RealTimeConfig)
-  })
-_sym_db.RegisterMessage(RealTimeConfig)
-
-StreamConfig = _reflection.GeneratedProtocolMessageType('StreamConfig', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMCONFIG,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.StreamConfig)
-  })
-_sym_db.RegisterMessage(StreamConfig)
-
-KafkaConfig = _reflection.GeneratedProtocolMessageType('KafkaConfig', (_message.Message,), {
-
-  'Consumer' : _reflection.GeneratedProtocolMessageType('Consumer', (_message.Message,), {
-    'DESCRIPTOR' : _KAFKACONFIG_CONSUMER,
-    '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.KafkaConfig.Consumer)
-    })
-  ,
-
-  'Producer' : _reflection.GeneratedProtocolMessageType('Producer', (_message.Message,), {
-    'DESCRIPTOR' : _KAFKACONFIG_PRODUCER,
-    '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.KafkaConfig.Producer)
-    })
-  ,
-  'DESCRIPTOR' : _KAFKACONFIG,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.KafkaConfig)
-  })
-_sym_db.RegisterMessage(KafkaConfig)
-_sym_db.RegisterMessage(KafkaConfig.Consumer)
-_sym_db.RegisterMessage(KafkaConfig.Producer)
-
-KubeDeploymentSize = _reflection.GeneratedProtocolMessageType('KubeDeploymentSize', (_message.Message,), {
-
-  'GpuResources' : _reflection.GeneratedProtocolMessageType('GpuResources', (_message.Message,), {
-    'DESCRIPTOR' : _KUBEDEPLOYMENTSIZE_GPURESOURCES,
-    '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.KubeDeploymentSize.GpuResources)
-    })
-  ,
-  'DESCRIPTOR' : _KUBEDEPLOYMENTSIZE,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.KubeDeploymentSize)
-  })
-_sym_db.RegisterMessage(KubeDeploymentSize)
-_sym_db.RegisterMessage(KubeDeploymentSize.GpuResources)
-
-KubeRuntimeDeploymentSettings = _reflection.GeneratedProtocolMessageType('KubeRuntimeDeploymentSettings', (_message.Message,), {
-  'DESCRIPTOR' : _KUBERUNTIMEDEPLOYMENTSETTINGS,
-  '__module__' : 'qwak.kube_deployment_captain.deployment_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.KubeRuntimeDeploymentSettings)
-  })
-_sym_db.RegisterMessage(KubeRuntimeDeploymentSettings)
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/kube_deployment_captain/deployment.proto\x12\x1cqwak.kube.deployment.captain\x1a\x1egoogle/protobuf/wrappers.proto\"\x84\x02\n\x1dKubeAdvancedDeploymentOptions\x12\x19\n\x11number_of_workers\x18\x01 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x02 \x01(\x05\x12/\n\x0b\x64\x61\x65mon_mode\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.BoolValue\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x04 \x01(\t\x12\x16\n\x0emax_batch_size\x18\x05 \x01(\x05\x12(\n deployment_process_timeout_limit\x18\x06 \x01(\x05\x12\x17\n\x0fpurchase_option\x18\x07 \x01(\t\"\xaf\x01\n\x13KubeServingStrategy\x12G\n\x0frealtime_config\x18\x01 \x01(\x0b\x32,.qwak.kube.deployment.captain.RealTimeConfigH\x00\x12\x43\n\rstream_config\x18\x02 \x01(\x0b\x32*.qwak.kube.deployment.captain.StreamConfigH\x00\x42\n\n\x08Strategy\"\xb5\x01\n\x0f\x41utoScalingSpec\x12\x19\n\x11min_replica_count\x18\x01 \x01(\x05\x12\x19\n\x11max_replica_count\x18\x02 \x01(\x05\x12\x18\n\x10polling_interval\x18\x03 \x01(\x05\x12\x18\n\x10\x63ool_down_period\x18\x04 \x01(\x05\x12\x38\n\x08triggers\x18\x05 \x03(\x0b\x32&.qwak.kube.deployment.captain.Triggers\"i\n\x08Triggers\x12M\n\x12prometheus_trigger\x18\x01 \x01(\x0b\x32/.qwak.kube.deployment.captain.PrometheusTriggerH\x00\x42\x0e\n\x0ctrigger_type\"\x90\x01\n\x11PrometheusTrigger\x12\x13\n\x0bmetric_name\x18\x01 \x01(\t\x12\x11\n\tthreshold\x18\x02 \x01(\x05\x12\r\n\x05query\x18\x03 \x01(\t\x12\x44\n\x0bmetric_type\x18\x04 \x01(\x0e\x32/.qwak.kube.deployment.captain.TriggerMetricType\"W\n\x0eRealTimeConfig\x12\'\n\x1fselected_variation_mapping_name\x18\x01 \x01(\t\x12\x1c\n\x14\x61ssumed_iam_role_arn\x18\x02 \x01(\t\"[\n\x0cStreamConfig\x12:\n\x05kafka\x18\x01 \x01(\x0b\x32).qwak.kube.deployment.captain.KafkaConfigH\x00\x42\x0f\n\rconfiguration\"\xb7\x05\n\x0bKafkaConfig\x12\x44\n\x08\x63onsumer\x18\x01 \x01(\x0b\x32\x32.qwak.kube.deployment.captain.KafkaConfig.Consumer\x12\x44\n\x08producer\x18\x02 \x01(\x0b\x32\x32.qwak.kube.deployment.captain.KafkaConfig.Producer\x12\x0f\n\x07workers\x18\x03 \x01(\x05\x1a\x9b\x02\n\x08\x43onsumer\x12\x18\n\x10\x62ootstrap_server\x18\x01 \x03(\t\x12\r\n\x05topic\x18\x02 \x01(\t\x12\r\n\x05group\x18\x03 \x01(\t\x12\x0f\n\x07timeout\x18\x04 \x01(\x05\x12[\n\x10\x61uto_offset_type\x18\x05 \x01(\x0e\x32\x41.qwak.kube.deployment.captain.KafkaConfig.Consumer.AutoOffSetType\x12\x16\n\x0emax_batch_size\x18\x06 \x01(\x05\x12\x18\n\x10max_poll_latency\x18\x07 \x01(\x01\"7\n\x0e\x41utoOffSetType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0c\n\x08\x45\x41RLIEST\x10\x01\x12\n\n\x06LATEST\x10\x02\x1a\xec\x01\n\x08Producer\x12\x18\n\x10\x62ootstrap_server\x18\x01 \x03(\t\x12\r\n\x05topic\x18\x02 \x01(\t\x12\\\n\x10\x63ompression_type\x18\x03 \x01(\x0e\x32\x42.qwak.kube.deployment.captain.KafkaConfig.Producer.CompressionType\"Y\n\x0f\x43ompressionType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x10\n\x0cUNCOMPRESSED\x10\x01\x12\x08\n\x04GZIP\x10\x02\x12\n\n\x06SNAPPY\x10\x03\x12\x07\n\x03LZ4\x10\x04\x12\x08\n\x04ZSTD\x10\x05\"\xa9\x05\n\x12KubeDeploymentSize\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0b\n\x03\x63pu\x18\x02 \x01(\x02\x12\x15\n\rmemory_amount\x18\x03 \x01(\x05\x12Q\n\x0cmemory_units\x18\x04 \x01(\x0e\x32;.qwak.kube.deployment.captain.KubeDeploymentSize.MemoryUnit\x12T\n\rgpu_resources\x18\x05 \x01(\x0b\x32=.qwak.kube.deployment.captain.KubeDeploymentSize.GpuResources\x1an\n\x0cGpuResources\x12J\n\x08gpu_type\x18\x01 \x01(\x0e\x32\x38.qwak.kube.deployment.captain.KubeDeploymentSize.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"\x90\x02\n\x07GpuType\x12\x0f\n\x0bINVALID_GPU\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06\x12\x14\n\x10NVIDIA_T4_1_4_15\x10\x07\x12\x14\n\x10NVIDIA_T4_1_8_30\x10\x08\x12\x15\n\x11NVIDIA_T4_1_16_60\x10\t\x12\x1e\n\x1aNVIDIA_A100_80GB_8_96_1360\x10\n\x12\x16\n\x12NVIDIA_V100_1_8_52\x10\x0b\x12\x18\n\x14NVIDIA_V100_4_32_208\x10\x0c\"+\n\nMemoryUnit\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\"\xc3\x01\n\x1dKubeRuntimeDeploymentSettings\x12_\n\x11root_logger_level\x18\x01 \x01(\x0e\x32\x44.qwak.kube.deployment.captain.KubeRuntimeDeploymentSettings.LogLevel\"A\n\x08LogLevel\x12\x0b\n\x07NOT_SET\x10\x00\x12\t\n\x05\x45RROR\x10\x01\x12\x08\n\x04WARN\x10\x02\x12\x08\n\x04INFO\x10\x03\x12\t\n\x05\x44\x45\x42UG\x10\x04*|\n\x11TriggerMetricType\x12!\n\x1dTRIGGER_METRIC_TYPE_NOT_VALID\x10\x00\x12\x1d\n\x19TRIGGER_METRIC_TYPE_VALUE\x10\x01\x12%\n!TRIGGER_METRIC_TYPE_AVERAGE_VALUE\x10\x02*;\n\x12KubeDeploymentType\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0c\n\x08REALTIME\x10\x01\x12\n\n\x06STREAM\x10\x02\x42+\n\'com.qwak.ai.kube.deployment.captain.apiP\x01\x62\x06proto3')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.kube_deployment_captain.deployment_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n\'com.qwak.ai.kube.deployment.captain.apiP\001'
-  _KUBEDEPLOYMENTTYPE._serialized_start=2516
-  _KUBEDEPLOYMENTTYPE._serialized_end=2575
-  _KUBEADVANCEDDEPLOYMENTOPTIONS._serialized_start=112
-  _KUBEADVANCEDDEPLOYMENTOPTIONS._serialized_end=372
-  _KUBESERVINGSTRATEGY._serialized_start=375
-  _KUBESERVINGSTRATEGY._serialized_end=550
-  _AUTOSCALINGSPEC._serialized_start=553
-  _AUTOSCALINGSPEC._serialized_end=734
-  _TRIGGERS._serialized_start=736
-  _TRIGGERS._serialized_end=841
-  _PROMETHEUSTRIGGER._serialized_start=843
-  _PROMETHEUSTRIGGER._serialized_end=917
-  _REALTIMECONFIG._serialized_start=919
-  _REALTIMECONFIG._serialized_end=1006
-  _STREAMCONFIG._serialized_start=1008
-  _STREAMCONFIG._serialized_end=1099
-  _KAFKACONFIG._serialized_start=1102
-  _KAFKACONFIG._serialized_end=1797
-  _KAFKACONFIG_CONSUMER._serialized_start=1275
-  _KAFKACONFIG_CONSUMER._serialized_end=1558
-  _KAFKACONFIG_CONSUMER_AUTOOFFSETTYPE._serialized_start=1503
-  _KAFKACONFIG_CONSUMER_AUTOOFFSETTYPE._serialized_end=1558
-  _KAFKACONFIG_PRODUCER._serialized_start=1561
-  _KAFKACONFIG_PRODUCER._serialized_end=1797
-  _KAFKACONFIG_PRODUCER_COMPRESSIONTYPE._serialized_start=1708
-  _KAFKACONFIG_PRODUCER_COMPRESSIONTYPE._serialized_end=1797
-  _KUBEDEPLOYMENTSIZE._serialized_start=1800
-  _KUBEDEPLOYMENTSIZE._serialized_end=2316
-  _KUBEDEPLOYMENTSIZE_GPURESOURCES._serialized_start=2051
-  _KUBEDEPLOYMENTSIZE_GPURESOURCES._serialized_end=2161
-  _KUBEDEPLOYMENTSIZE_GPUTYPE._serialized_start=2163
-  _KUBEDEPLOYMENTSIZE_GPUTYPE._serialized_end=2271
-  _KUBEDEPLOYMENTSIZE_MEMORYUNIT._serialized_start=2273
-  _KUBEDEPLOYMENTSIZE_MEMORYUNIT._serialized_end=2316
-  _KUBERUNTIMEDEPLOYMENTSETTINGS._serialized_start=2319
-  _KUBERUNTIMEDEPLOYMENTSETTINGS._serialized_end=2514
-  _KUBERUNTIMEDEPLOYMENTSETTINGS_LOGLEVEL._serialized_start=2449
-  _KUBERUNTIMEDEPLOYMENTSETTINGS_LOGLEVEL._serialized_end=2514
+  _globals['_TRIGGERMETRICTYPE']._serialized_start=2752
+  _globals['_TRIGGERMETRICTYPE']._serialized_end=2876
+  _globals['_KUBEDEPLOYMENTTYPE']._serialized_start=2878
+  _globals['_KUBEDEPLOYMENTTYPE']._serialized_end=2937
+  _globals['_KUBEADVANCEDDEPLOYMENTOPTIONS']._serialized_start=112
+  _globals['_KUBEADVANCEDDEPLOYMENTOPTIONS']._serialized_end=372
+  _globals['_KUBESERVINGSTRATEGY']._serialized_start=375
+  _globals['_KUBESERVINGSTRATEGY']._serialized_end=550
+  _globals['_AUTOSCALINGSPEC']._serialized_start=553
+  _globals['_AUTOSCALINGSPEC']._serialized_end=734
+  _globals['_TRIGGERS']._serialized_start=736
+  _globals['_TRIGGERS']._serialized_end=841
+  _globals['_PROMETHEUSTRIGGER']._serialized_start=844
+  _globals['_PROMETHEUSTRIGGER']._serialized_end=988
+  _globals['_REALTIMECONFIG']._serialized_start=990
+  _globals['_REALTIMECONFIG']._serialized_end=1077
+  _globals['_STREAMCONFIG']._serialized_start=1079
+  _globals['_STREAMCONFIG']._serialized_end=1170
+  _globals['_KAFKACONFIG']._serialized_start=1173
+  _globals['_KAFKACONFIG']._serialized_end=1868
+  _globals['_KAFKACONFIG_CONSUMER']._serialized_start=1346
+  _globals['_KAFKACONFIG_CONSUMER']._serialized_end=1629
+  _globals['_KAFKACONFIG_CONSUMER_AUTOOFFSETTYPE']._serialized_start=1574
+  _globals['_KAFKACONFIG_CONSUMER_AUTOOFFSETTYPE']._serialized_end=1629
+  _globals['_KAFKACONFIG_PRODUCER']._serialized_start=1632
+  _globals['_KAFKACONFIG_PRODUCER']._serialized_end=1868
+  _globals['_KAFKACONFIG_PRODUCER_COMPRESSIONTYPE']._serialized_start=1779
+  _globals['_KAFKACONFIG_PRODUCER_COMPRESSIONTYPE']._serialized_end=1868
+  _globals['_KUBEDEPLOYMENTSIZE']._serialized_start=1871
+  _globals['_KUBEDEPLOYMENTSIZE']._serialized_end=2552
+  _globals['_KUBEDEPLOYMENTSIZE_GPURESOURCES']._serialized_start=2122
+  _globals['_KUBEDEPLOYMENTSIZE_GPURESOURCES']._serialized_end=2232
+  _globals['_KUBEDEPLOYMENTSIZE_GPUTYPE']._serialized_start=2235
+  _globals['_KUBEDEPLOYMENTSIZE_GPUTYPE']._serialized_end=2507
+  _globals['_KUBEDEPLOYMENTSIZE_MEMORYUNIT']._serialized_start=2509
+  _globals['_KUBEDEPLOYMENTSIZE_MEMORYUNIT']._serialized_end=2552
+  _globals['_KUBERUNTIMEDEPLOYMENTSETTINGS']._serialized_start=2555
+  _globals['_KUBERUNTIMEDEPLOYMENTSETTINGS']._serialized_end=2750
+  _globals['_KUBERUNTIMEDEPLOYMENTSETTINGS_LOGLEVEL']._serialized_start=2685
+  _globals['_KUBERUNTIMEDEPLOYMENTSETTINGS_LOGLEVEL']._serialized_end=2750
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/kube_deployment_captain/deployment_pb2.pyi

@@ -19,6 +19,23 @@ else:
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+class _TriggerMetricType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _TriggerMetricTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TriggerMetricType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    TRIGGER_METRIC_TYPE_NOT_VALID: _TriggerMetricType.ValueType  # 0
+    TRIGGER_METRIC_TYPE_VALUE: _TriggerMetricType.ValueType  # 1
+    TRIGGER_METRIC_TYPE_AVERAGE_VALUE: _TriggerMetricType.ValueType  # 2
+
+class TriggerMetricType(_TriggerMetricType, metaclass=_TriggerMetricTypeEnumTypeWrapper): ...
+
+TRIGGER_METRIC_TYPE_NOT_VALID: TriggerMetricType.ValueType  # 0
+TRIGGER_METRIC_TYPE_VALUE: TriggerMetricType.ValueType  # 1
+TRIGGER_METRIC_TYPE_AVERAGE_VALUE: TriggerMetricType.ValueType  # 2
+global___TriggerMetricType = TriggerMetricType
+
 class _KubeDeploymentType:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
@@ -155,20 +172,24 @@ class PrometheusTrigger(google.protobuf.message.Message):
     METRIC_NAME_FIELD_NUMBER: builtins.int
     THRESHOLD_FIELD_NUMBER: builtins.int
     QUERY_FIELD_NUMBER: builtins.int
+    METRIC_TYPE_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"""
+    metric_type: global___TriggerMetricType.ValueType
+    """Type of the metric"""
     def __init__(
         self,
         *,
         metric_name: builtins.str = ...,
         threshold: builtins.int = ...,
         query: builtins.str = ...,
+        metric_type: global___TriggerMetricType.ValueType = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["metric_name", b"metric_name", "query", b"query", "threshold", b"threshold"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["metric_name", b"metric_name", "metric_type", b"metric_type", "query", b"query", "threshold", b"threshold"]) -> None: ...
 
 global___PrometheusTrigger = PrometheusTrigger
 
@@ -347,6 +368,21 @@ class KubeDeploymentSize(google.protobuf.message.Message):
         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
@@ -355,6 +391,21 @@ class KubeDeploymentSize(google.protobuf.message.Message):
     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)

_qwak_proto/qwak/kube_deployment_captain/feature_set_deployment_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: 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
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -18,121 +16,11 @@ from _qwak_proto.qwak.feature_store.entities import entity_pb2 as qwak_dot_featu
 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\"\xe5\x06\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\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)
+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')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.kube_deployment_captain.feature_set_deployment_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -141,32 +29,32 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY._serialized_options = b'8\001'
   _FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY._options = None
   _FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY._serialized_options = b'8\001'
-  _VALUETYPE._serialized_start=2400
-  _VALUETYPE._serialized_end=2445
-  _FEATURESETDEPLOYMENT._serialized_start=220
-  _FEATURESETDEPLOYMENT._serialized_end=1089
-  _FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY._serialized_start=945
-  _FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY._serialized_end=1012
-  _FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY._serialized_start=1014
-  _FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY._serialized_end=1074
-  _BATCHDETAILS._serialized_start=1092
-  _BATCHDETAILS._serialized_end=1240
-  _EXECUTIONIDENTIFIER._serialized_start=1242
-  _EXECUTIONIDENTIFIER._serialized_end=1285
-  _FEATURESETDEPLOYMENTSPEC._serialized_start=1288
-  _FEATURESETDEPLOYMENTSPEC._serialized_end=1548
-  _FEATURESETUSERMETADATA._serialized_start=1550
-  _FEATURESETUSERMETADATA._serialized_end=1632
-  _ENTITYDEFINITION._serialized_start=1634
-  _ENTITYDEFINITION._serialized_end=1734
-  _FEATURESETTYPE._serialized_start=1736
-  _FEATURESETTYPE._serialized_end=1848
-  _STREAMINGFEATURESET._serialized_start=1850
-  _STREAMINGFEATURESET._serialized_end=1943
-  _KAFKASOURCE._serialized_start=1946
-  _KAFKASOURCE._serialized_end=2176
-  _STREAMINGDATASOURCE._serialized_start=2178
-  _STREAMINGDATASOURCE._serialized_end=2274
-  _ENTITYSPEC._serialized_start=2276
-  _ENTITYSPEC._serialized_end=2398
+  _globals['_VALUETYPE']._serialized_start=2428
+  _globals['_VALUETYPE']._serialized_end=2473
+  _globals['_FEATURESETDEPLOYMENT']._serialized_start=220
+  _globals['_FEATURESETDEPLOYMENT']._serialized_end=1117
+  _globals['_FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY']._serialized_start=973
+  _globals['_FEATURESETDEPLOYMENT_EXTRADEPLOYMENTCONFIGURATIONENTRY']._serialized_end=1040
+  _globals['_FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY']._serialized_start=1042
+  _globals['_FEATURESETDEPLOYMENT_EXTRAENVCONFIGURATIONENTRY']._serialized_end=1102
+  _globals['_BATCHDETAILS']._serialized_start=1120
+  _globals['_BATCHDETAILS']._serialized_end=1268
+  _globals['_EXECUTIONIDENTIFIER']._serialized_start=1270
+  _globals['_EXECUTIONIDENTIFIER']._serialized_end=1313
+  _globals['_FEATURESETDEPLOYMENTSPEC']._serialized_start=1316
+  _globals['_FEATURESETDEPLOYMENTSPEC']._serialized_end=1576
+  _globals['_FEATURESETUSERMETADATA']._serialized_start=1578
+  _globals['_FEATURESETUSERMETADATA']._serialized_end=1660
+  _globals['_ENTITYDEFINITION']._serialized_start=1662
+  _globals['_ENTITYDEFINITION']._serialized_end=1762
+  _globals['_FEATURESETTYPE']._serialized_start=1764
+  _globals['_FEATURESETTYPE']._serialized_end=1876
+  _globals['_STREAMINGFEATURESET']._serialized_start=1878
+  _globals['_STREAMINGFEATURESET']._serialized_end=1971
+  _globals['_KAFKASOURCE']._serialized_start=1974
+  _globals['_KAFKASOURCE']._serialized_end=2204
+  _globals['_STREAMINGDATASOURCE']._serialized_start=2206
+  _globals['_STREAMINGDATASOURCE']._serialized_end=2302
+  _globals['_ENTITYSPEC']._serialized_start=2304
+  _globals['_ENTITYSPEC']._serialized_end=2426
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/kube_deployment_captain/feature_set_deployment_pb2.pyi

@@ -88,6 +88,7 @@ class FeatureSetDeployment(google.protobuf.message.Message):
     SECRET_SERVICE_URL_FIELD_NUMBER: builtins.int
     RUN_ID_FIELD_NUMBER: builtins.int
     QWAK_INTERNAL_PROTOCOL_VERSION_FIELD_NUMBER: builtins.int
+    FEATURESET_VERSION_FIELD_NUMBER: builtins.int
     featureset_name: builtins.str
     """FeatureSet name"""
     @property
@@ -112,7 +113,9 @@ class FeatureSetDeployment(google.protobuf.message.Message):
     run_id: builtins.int
     """Run id which represents the job id"""
     qwak_internal_protocol_version: builtins.int
-    """Qwak featureset version"""
+    """Qwak featureset protocol version"""
+    featureset_version: builtins.int
+    """Featureset version"""
     def __init__(
         self,
         *,
@@ -127,9 +130,10 @@ class FeatureSetDeployment(google.protobuf.message.Message):
         secret_service_url: builtins.str = ...,
         run_id: builtins.int = ...,
         qwak_internal_protocol_version: builtins.int = ...,
+        featureset_version: builtins.int = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["batch_feature_set", b"batch_feature_set", "entity", b"entity", "feature_set", b"feature_set", "streaming_aggregation_feature_set", b"streaming_aggregation_feature_set", "streaming_feature_set", b"streaming_feature_set"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["batch_feature_set", b"batch_feature_set", "entity", b"entity", "extra_deployment_configuration", b"extra_deployment_configuration", "extra_env_configuration", b"extra_env_configuration", "feature_set", b"feature_set", "featureset_id", b"featureset_id", "featureset_name", b"featureset_name", "qwak_internal_protocol_version", b"qwak_internal_protocol_version", "run_id", b"run_id", "secret_service_url", b"secret_service_url", "streaming_aggregation_feature_set", b"streaming_aggregation_feature_set", "streaming_feature_set", b"streaming_feature_set"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["batch_feature_set", b"batch_feature_set", "entity", b"entity", "extra_deployment_configuration", b"extra_deployment_configuration", "extra_env_configuration", b"extra_env_configuration", "feature_set", b"feature_set", "featureset_id", b"featureset_id", "featureset_name", b"featureset_name", "featureset_version", b"featureset_version", "qwak_internal_protocol_version", b"qwak_internal_protocol_version", "run_id", b"run_id", "secret_service_url", b"secret_service_url", "streaming_aggregation_feature_set", b"streaming_aggregation_feature_set", "streaming_feature_set", b"streaming_feature_set"]) -> None: ...
     def WhichOneof(self, oneof_group: typing_extensions.Literal["feature_set", b"feature_set"]) -> typing_extensions.Literal["streaming_feature_set", "batch_feature_set", "streaming_aggregation_feature_set"] | None: ...
 
 global___FeatureSetDeployment = FeatureSetDeployment