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

_qwak_proto/qwak/automation/v1/action_pb2.py

@@ -2,153 +2,27 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/automation/v1/action.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()
 
 
+from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
 from _qwak_proto.qwak.automation.v1 import common_pb2 as qwak_dot_automation_dot_v1_dot_common__pb2
 from _qwak_proto.qwak.automation.v1 import auto_scaling_pb2 as qwak_dot_automation_dot_v1_dot_auto__scaling__pb2
 from _qwak_proto.qwak.batch_job.v1 import batch_job_service_pb2 as qwak_dot_batch__job_dot_v1_dot_batch__job__service__pb2
 from _qwak_proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fqwak/automation/v1/action.proto\x12\x12qwak.automation.v1\x1a\x1fqwak/automation/v1/common.proto\x1a%qwak/automation/v1/auto_scaling.proto\x1a)qwak/batch_job/v1/batch_job_service.proto\x1a/qwak/user_application/common/v0/resources.proto\"\x99\x01\n\x06\x41\x63tion\x12@\n\x0c\x62uild_deploy\x18\x01 \x01(\x0b\x32(.qwak.automation.v1.BuildAndDeployActionH\x00\x12\x43\n\x0f\x62\x61tch_execution\x18\x02 \x01(\x0b\x32(.qwak.automation.v1.BatchExecutionActionH\x00\x42\x08\n\x06\x61\x63tion\"Q\n\x14\x42\x61tchExecutionAction\x12\x39\n\x11\x65xecute_batch_job\x18\x01 \x01(\x0b\x32\x1e.qwak.batchjob.BatchJobRequest\"\xcd\x01\n\x14\x42uildAndDeployAction\x12\x31\n\nbuild_spec\x18\x01 \x01(\x0b\x32\x1d.qwak.automation.v1.BuildSpec\x12\x45\n\x14\x64\x65ployment_condition\x18\x02 \x01(\x0b\x32\'.qwak.automation.v1.DeploymentCondition\x12;\n\x0f\x64\x65ployment_spec\x18\x03 \x01(\x0b\x32\".qwak.automation.v1.DeploymentSpec\"d\n\x13\x44\x65ploymentCondition\x12@\n\x0c\x62uild_metric\x18\x01 \x01(\x0b\x32(.qwak.automation.v1.BuildMetricConditionH\x00\x42\x0b\n\tcondition\"\x89\x01\n\x14\x42uildMetricCondition\x12\x13\n\x0bmetric_name\x18\x01 \x01(\t\x12I\n\x13threshold_direction\x18\x02 \x01(\x0e\x32,.qwak.automation.v1.MetricThresholdDirection\x12\x11\n\tthreshold\x18\x03 \x01(\t\"\xdf\x01\n\x0e\x44\x65ploymentSpec\x12;\n\x0f\x64\x65ployment_size\x18\x01 \x01(\x0b\x32\".qwak.automation.v1.DeploymentSize\x12G\n\x10\x61\x64vanced_options\x18\x02 \x01(\x0b\x32-.qwak.automation.v1.AdvancedDeploymentOptions\x12\x1f\n\x17selected_variation_name\x18\x03 \x01(\t\x12\x14\n\x0c\x65nvironments\x18\x04 \x03(\t\x12\x10\n\x08\x65nv_vars\x18\x05 \x03(\t\"\x9b\x02\n\x19\x41\x64vancedDeploymentOptions\x12%\n\x1dnumber_of_http_server_workers\x18\x01 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x02 \x01(\x05\x12\x13\n\x0b\x64\x61\x65mon_mode\x18\x03 \x01(\x08\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x04 \x01(\t\x12\x42\n\x13\x61uto_scaling_config\x18\x05 \x01(\x0b\x32%.qwak.automation.v1.AutoScalingConfig\x12\x16\n\x0emax_batch_size\x18\x06 \x01(\x05\x12(\n deployment_process_timeout_limit\x18\x07 \x01(\x05\"\xad\x02\n\x0e\x44\x65ploymentSize\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0f\n\x03\x63pu\x18\x02 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x03 \x01(\x05\x42\x02\x18\x01\x12\x38\n\x0cmemory_units\x18\x04 \x01(\x0e\x32\x1e.qwak.automation.v1.MemoryUnitB\x02\x18\x01\x12;\n\rgpu_resources\x18\x05 \x01(\x0b\x32 .qwak.automation.v1.GpuResourcesB\x02\x18\x01\x12`\n\x1c\x63lient_pod_compute_resources\x18\x06 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\"\xee\x02\n\tBuildSpec\x12<\n\x10git_model_source\x18\x01 \x01(\x0b\x32\".qwak.automation.v1.GitModelSource\x12\x41\n\nparameters\x18\x02 \x03(\x0b\x32-.qwak.automation.v1.BuildSpec.ParametersEntry\x12\x0c\n\x04tags\x18\x03 \x03(\t\x12\x12\n\nbase_image\x18\x04 \x01(\t\x12\x18\n\x10\x61ssumed_iam_role\x18\x05 \x01(\t\x12/\n\x08resource\x18\x06 \x01(\x0b\x32\x1d.qwak.automation.v1.Resources\x12\x10\n\x08main_dir\x18\x07 \x01(\t\x12\x1c\n\x14\x64\x65pendency_file_path\x18\x08 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\t \x03(\t\x1a\x31\n\x0fParametersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xf9\x01\n\tResources\x12=\n\rcpu_resources\x18\x01 \x01(\x0b\x32 .qwak.automation.v1.CpuResourcesB\x02\x18\x01H\x00\x12=\n\rgpu_resources\x18\x02 \x01(\x0b\x32 .qwak.automation.v1.GpuResourcesB\x02\x18\x01H\x00\x12\x62\n\x1c\x63lient_pod_compute_resources\x18\x03 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResourcesH\x00\x42\n\n\x08resource\"h\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12\x34\n\x0cmemory_units\x18\x03 \x01(\x0e\x32\x1e.qwak.automation.v1.MemoryUnit\"Q\n\x0cGpuResources\x12-\n\x08gpu_type\x18\x01 \x01(\x0e\x32\x1b.qwak.automation.v1.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"Z\n\x0eGitModelSource\x12\x0f\n\x07git_uri\x18\x01 \x01(\t\x12#\n\x1bgit_credentials_secret_name\x18\x02 \x01(\t\x12\x12\n\ngit_branch\x18\x03 \x01(\t*h\n\x07GpuType\x12\x0b\n\x07INVALID\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\x42!\n\x1d\x63om.qwak.ai.automation.api.v1P\x01\x62\x06proto3')
-
-_GPUTYPE = DESCRIPTOR.enum_types_by_name['GpuType']
-GpuType = enum_type_wrapper.EnumTypeWrapper(_GPUTYPE)
-_MEMORYUNIT = DESCRIPTOR.enum_types_by_name['MemoryUnit']
-MemoryUnit = enum_type_wrapper.EnumTypeWrapper(_MEMORYUNIT)
-INVALID = 0
-NVIDIA_K80 = 1
-NVIDIA_V100 = 2
-NVIDIA_A100 = 3
-NVIDIA_T4 = 4
-NVIDIA_A10G = 5
-UNKNOWN = 0
-MIB = 1
-GIB = 2
-
-
-_ACTION = DESCRIPTOR.message_types_by_name['Action']
-_BATCHEXECUTIONACTION = DESCRIPTOR.message_types_by_name['BatchExecutionAction']
-_BUILDANDDEPLOYACTION = DESCRIPTOR.message_types_by_name['BuildAndDeployAction']
-_DEPLOYMENTCONDITION = DESCRIPTOR.message_types_by_name['DeploymentCondition']
-_BUILDMETRICCONDITION = DESCRIPTOR.message_types_by_name['BuildMetricCondition']
-_DEPLOYMENTSPEC = DESCRIPTOR.message_types_by_name['DeploymentSpec']
-_ADVANCEDDEPLOYMENTOPTIONS = DESCRIPTOR.message_types_by_name['AdvancedDeploymentOptions']
-_DEPLOYMENTSIZE = DESCRIPTOR.message_types_by_name['DeploymentSize']
-_BUILDSPEC = DESCRIPTOR.message_types_by_name['BuildSpec']
-_BUILDSPEC_PARAMETERSENTRY = _BUILDSPEC.nested_types_by_name['ParametersEntry']
-_RESOURCES = DESCRIPTOR.message_types_by_name['Resources']
-_CPURESOURCES = DESCRIPTOR.message_types_by_name['CpuResources']
-_GPURESOURCES = DESCRIPTOR.message_types_by_name['GpuResources']
-_GITMODELSOURCE = DESCRIPTOR.message_types_by_name['GitModelSource']
-Action = _reflection.GeneratedProtocolMessageType('Action', (_message.Message,), {
-  'DESCRIPTOR' : _ACTION,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.Action)
-  })
-_sym_db.RegisterMessage(Action)
-
-BatchExecutionAction = _reflection.GeneratedProtocolMessageType('BatchExecutionAction', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHEXECUTIONACTION,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.BatchExecutionAction)
-  })
-_sym_db.RegisterMessage(BatchExecutionAction)
-
-BuildAndDeployAction = _reflection.GeneratedProtocolMessageType('BuildAndDeployAction', (_message.Message,), {
-  'DESCRIPTOR' : _BUILDANDDEPLOYACTION,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.BuildAndDeployAction)
-  })
-_sym_db.RegisterMessage(BuildAndDeployAction)
-
-DeploymentCondition = _reflection.GeneratedProtocolMessageType('DeploymentCondition', (_message.Message,), {
-  'DESCRIPTOR' : _DEPLOYMENTCONDITION,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.DeploymentCondition)
-  })
-_sym_db.RegisterMessage(DeploymentCondition)
-
-BuildMetricCondition = _reflection.GeneratedProtocolMessageType('BuildMetricCondition', (_message.Message,), {
-  'DESCRIPTOR' : _BUILDMETRICCONDITION,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.BuildMetricCondition)
-  })
-_sym_db.RegisterMessage(BuildMetricCondition)
-
-DeploymentSpec = _reflection.GeneratedProtocolMessageType('DeploymentSpec', (_message.Message,), {
-  'DESCRIPTOR' : _DEPLOYMENTSPEC,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.DeploymentSpec)
-  })
-_sym_db.RegisterMessage(DeploymentSpec)
-
-AdvancedDeploymentOptions = _reflection.GeneratedProtocolMessageType('AdvancedDeploymentOptions', (_message.Message,), {
-  'DESCRIPTOR' : _ADVANCEDDEPLOYMENTOPTIONS,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.AdvancedDeploymentOptions)
-  })
-_sym_db.RegisterMessage(AdvancedDeploymentOptions)
-
-DeploymentSize = _reflection.GeneratedProtocolMessageType('DeploymentSize', (_message.Message,), {
-  'DESCRIPTOR' : _DEPLOYMENTSIZE,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.DeploymentSize)
-  })
-_sym_db.RegisterMessage(DeploymentSize)
-
-BuildSpec = _reflection.GeneratedProtocolMessageType('BuildSpec', (_message.Message,), {
-
-  'ParametersEntry' : _reflection.GeneratedProtocolMessageType('ParametersEntry', (_message.Message,), {
-    'DESCRIPTOR' : _BUILDSPEC_PARAMETERSENTRY,
-    '__module__' : 'qwak.automation.v1.action_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.automation.v1.BuildSpec.ParametersEntry)
-    })
-  ,
-  'DESCRIPTOR' : _BUILDSPEC,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.BuildSpec)
-  })
-_sym_db.RegisterMessage(BuildSpec)
-_sym_db.RegisterMessage(BuildSpec.ParametersEntry)
-
-Resources = _reflection.GeneratedProtocolMessageType('Resources', (_message.Message,), {
-  'DESCRIPTOR' : _RESOURCES,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.Resources)
-  })
-_sym_db.RegisterMessage(Resources)
-
-CpuResources = _reflection.GeneratedProtocolMessageType('CpuResources', (_message.Message,), {
-  'DESCRIPTOR' : _CPURESOURCES,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.CpuResources)
-  })
-_sym_db.RegisterMessage(CpuResources)
-
-GpuResources = _reflection.GeneratedProtocolMessageType('GpuResources', (_message.Message,), {
-  'DESCRIPTOR' : _GPURESOURCES,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.GpuResources)
-  })
-_sym_db.RegisterMessage(GpuResources)
-
-GitModelSource = _reflection.GeneratedProtocolMessageType('GitModelSource', (_message.Message,), {
-  'DESCRIPTOR' : _GITMODELSOURCE,
-  '__module__' : 'qwak.automation.v1.action_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.GitModelSource)
-  })
-_sym_db.RegisterMessage(GitModelSource)
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fqwak/automation/v1/action.proto\x12\x12qwak.automation.v1\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x1fqwak/automation/v1/common.proto\x1a%qwak/automation/v1/auto_scaling.proto\x1a)qwak/batch_job/v1/batch_job_service.proto\x1a/qwak/user_application/common/v0/resources.proto\"\x99\x01\n\x06\x41\x63tion\x12@\n\x0c\x62uild_deploy\x18\x01 \x01(\x0b\x32(.qwak.automation.v1.BuildAndDeployActionH\x00\x12\x43\n\x0f\x62\x61tch_execution\x18\x02 \x01(\x0b\x32(.qwak.automation.v1.BatchExecutionActionH\x00\x42\x08\n\x06\x61\x63tion\"Q\n\x14\x42\x61tchExecutionAction\x12\x39\n\x11\x65xecute_batch_job\x18\x01 \x01(\x0b\x32\x1e.qwak.batchjob.BatchJobRequest\"\xcd\x01\n\x14\x42uildAndDeployAction\x12\x31\n\nbuild_spec\x18\x01 \x01(\x0b\x32\x1d.qwak.automation.v1.BuildSpec\x12\x45\n\x14\x64\x65ployment_condition\x18\x02 \x01(\x0b\x32\'.qwak.automation.v1.DeploymentCondition\x12;\n\x0f\x64\x65ployment_spec\x18\x03 \x01(\x0b\x32\".qwak.automation.v1.DeploymentSpec\"\xb7\x01\n\x13\x44\x65ploymentCondition\x12@\n\x0c\x62uild_metric\x18\x01 \x01(\x0b\x32(.qwak.automation.v1.BuildMetricConditionH\x00\x12Q\n\x15\x64\x65ployed_build_metric\x18\x02 \x01(\x0b\x32\x30.qwak.automation.v1.DeployedBuildMetricConditionH\x00\x42\x0b\n\tcondition\"\x89\x01\n\x14\x42uildMetricCondition\x12\x13\n\x0bmetric_name\x18\x01 \x01(\t\x12I\n\x13threshold_direction\x18\x02 \x01(\x0e\x32,.qwak.automation.v1.MetricThresholdDirection\x12\x11\n\tthreshold\x18\x03 \x01(\t\"\x91\x01\n\x1c\x44\x65ployedBuildMetricCondition\x12\x13\n\x0bmetric_name\x18\x01 \x01(\t\x12I\n\x13threshold_direction\x18\x02 \x01(\x0e\x32,.qwak.automation.v1.MetricThresholdDirection\x12\x11\n\tvariation\x18\x03 \x01(\t\"\xdf\x01\n\x0e\x44\x65ploymentSpec\x12;\n\x0f\x64\x65ployment_size\x18\x01 \x01(\x0b\x32\".qwak.automation.v1.DeploymentSize\x12G\n\x10\x61\x64vanced_options\x18\x02 \x01(\x0b\x32-.qwak.automation.v1.AdvancedDeploymentOptions\x12\x1f\n\x17selected_variation_name\x18\x03 \x01(\t\x12\x14\n\x0c\x65nvironments\x18\x04 \x03(\t\x12\x10\n\x08\x65nv_vars\x18\x05 \x03(\t\"\xc4\x02\n\x19\x41\x64vancedDeploymentOptions\x12%\n\x1dnumber_of_http_server_workers\x18\x01 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x02 \x01(\x05\x12\x13\n\x0b\x64\x61\x65mon_mode\x18\x03 \x01(\x08\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x04 \x01(\t\x12\x42\n\x13\x61uto_scaling_config\x18\x05 \x01(\x0b\x32%.qwak.automation.v1.AutoScalingConfig\x12\x16\n\x0emax_batch_size\x18\x06 \x01(\x05\x12(\n deployment_process_timeout_limit\x18\x07 \x01(\x05\x12\'\n\x1fservice_account_key_secret_name\x18\x08 \x01(\t\"\xad\x02\n\x0e\x44\x65ploymentSize\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0f\n\x03\x63pu\x18\x02 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x03 \x01(\x05\x42\x02\x18\x01\x12\x38\n\x0cmemory_units\x18\x04 \x01(\x0e\x32\x1e.qwak.automation.v1.MemoryUnitB\x02\x18\x01\x12;\n\rgpu_resources\x18\x05 \x01(\x0b\x32 .qwak.automation.v1.GpuResourcesB\x02\x18\x01\x12`\n\x1c\x63lient_pod_compute_resources\x18\x06 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\"\xc0\x04\n\tBuildSpec\x12<\n\x10git_model_source\x18\x01 \x01(\x0b\x32\".qwak.automation.v1.GitModelSource\x12\x41\n\nparameters\x18\x02 \x03(\x0b\x32-.qwak.automation.v1.BuildSpec.ParametersEntry\x12\x0c\n\x04tags\x18\x03 \x03(\t\x12\x12\n\nbase_image\x18\x04 \x01(\t\x12\x18\n\x10\x61ssumed_iam_role\x18\x05 \x01(\t\x12/\n\x08resource\x18\x06 \x01(\x0b\x32\x1d.qwak.automation.v1.Resources\x12\x10\n\x08main_dir\x18\x07 \x01(\t\x12\x1c\n\x14\x64\x65pendency_file_path\x18\x08 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\t \x03(\t\x12\x16\n\x0egpu_compatible\x18\n \x01(\x08\x12.\n\npush_image\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.BoolValue\x12;\n\x0fpurchase_option\x18\x0c \x01(\x0e\x32\".qwak.automation.v1.PurchaseOption\x12\'\n\x1fservice_account_key_secret_name\x18\r \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x0e \x01(\x05\x1a\x31\n\x0fParametersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xf9\x01\n\tResources\x12=\n\rcpu_resources\x18\x01 \x01(\x0b\x32 .qwak.automation.v1.CpuResourcesB\x02\x18\x01H\x00\x12=\n\rgpu_resources\x18\x02 \x01(\x0b\x32 .qwak.automation.v1.GpuResourcesB\x02\x18\x01H\x00\x12\x62\n\x1c\x63lient_pod_compute_resources\x18\x03 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResourcesH\x00\x42\n\n\x08resource\"h\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12\x34\n\x0cmemory_units\x18\x03 \x01(\x0e\x32\x1e.qwak.automation.v1.MemoryUnit\"Q\n\x0cGpuResources\x12-\n\x08gpu_type\x18\x01 \x01(\x0e\x32\x1b.qwak.automation.v1.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"w\n\x0eGitModelSource\x12\x0f\n\x07git_uri\x18\x01 \x01(\t\x12#\n\x1bgit_credentials_secret_name\x18\x02 \x01(\t\x12\x12\n\ngit_branch\x18\x03 \x01(\t\x12\x1b\n\x13git_ssh_secret_name\x18\x04 \x01(\t*w\n\x07GpuType\x12\x0b\n\x07INVALID\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*+\n\nMemoryUnit\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*)\n\x0ePurchaseOption\x12\r\n\tON_DEMAND\x10\x00\x12\x08\n\x04SPOT\x10\x01\x42!\n\x1d\x63om.qwak.ai.automation.api.v1P\x01\x62\x06proto3')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.automation.v1.action_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -167,36 +41,40 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _RESOURCES.fields_by_name['cpu_resources']._serialized_options = b'\030\001'
   _RESOURCES.fields_by_name['gpu_resources']._options = None
   _RESOURCES.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
-  _GPUTYPE._serialized_start=2626
-  _GPUTYPE._serialized_end=2730
-  _MEMORYUNIT._serialized_start=2732
-  _MEMORYUNIT._serialized_end=2775
-  _ACTION._serialized_start=220
-  _ACTION._serialized_end=373
-  _BATCHEXECUTIONACTION._serialized_start=375
-  _BATCHEXECUTIONACTION._serialized_end=456
-  _BUILDANDDEPLOYACTION._serialized_start=459
-  _BUILDANDDEPLOYACTION._serialized_end=664
-  _DEPLOYMENTCONDITION._serialized_start=666
-  _DEPLOYMENTCONDITION._serialized_end=766
-  _BUILDMETRICCONDITION._serialized_start=769
-  _BUILDMETRICCONDITION._serialized_end=906
-  _DEPLOYMENTSPEC._serialized_start=909
-  _DEPLOYMENTSPEC._serialized_end=1132
-  _ADVANCEDDEPLOYMENTOPTIONS._serialized_start=1135
-  _ADVANCEDDEPLOYMENTOPTIONS._serialized_end=1418
-  _DEPLOYMENTSIZE._serialized_start=1421
-  _DEPLOYMENTSIZE._serialized_end=1722
-  _BUILDSPEC._serialized_start=1725
-  _BUILDSPEC._serialized_end=2091
-  _BUILDSPEC_PARAMETERSENTRY._serialized_start=2042
-  _BUILDSPEC_PARAMETERSENTRY._serialized_end=2091
-  _RESOURCES._serialized_start=2094
-  _RESOURCES._serialized_end=2343
-  _CPURESOURCES._serialized_start=2345
-  _CPURESOURCES._serialized_end=2449
-  _GPURESOURCES._serialized_start=2451
-  _GPURESOURCES._serialized_end=2532
-  _GITMODELSOURCE._serialized_start=2534
-  _GITMODELSOURCE._serialized_end=2624
+  _globals['_GPUTYPE']._serialized_start=3170
+  _globals['_GPUTYPE']._serialized_end=3289
+  _globals['_MEMORYUNIT']._serialized_start=3291
+  _globals['_MEMORYUNIT']._serialized_end=3334
+  _globals['_PURCHASEOPTION']._serialized_start=3336
+  _globals['_PURCHASEOPTION']._serialized_end=3377
+  _globals['_ACTION']._serialized_start=252
+  _globals['_ACTION']._serialized_end=405
+  _globals['_BATCHEXECUTIONACTION']._serialized_start=407
+  _globals['_BATCHEXECUTIONACTION']._serialized_end=488
+  _globals['_BUILDANDDEPLOYACTION']._serialized_start=491
+  _globals['_BUILDANDDEPLOYACTION']._serialized_end=696
+  _globals['_DEPLOYMENTCONDITION']._serialized_start=699
+  _globals['_DEPLOYMENTCONDITION']._serialized_end=882
+  _globals['_BUILDMETRICCONDITION']._serialized_start=885
+  _globals['_BUILDMETRICCONDITION']._serialized_end=1022
+  _globals['_DEPLOYEDBUILDMETRICCONDITION']._serialized_start=1025
+  _globals['_DEPLOYEDBUILDMETRICCONDITION']._serialized_end=1170
+  _globals['_DEPLOYMENTSPEC']._serialized_start=1173
+  _globals['_DEPLOYMENTSPEC']._serialized_end=1396
+  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_start=1399
+  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_end=1723
+  _globals['_DEPLOYMENTSIZE']._serialized_start=1726
+  _globals['_DEPLOYMENTSIZE']._serialized_end=2027
+  _globals['_BUILDSPEC']._serialized_start=2030
+  _globals['_BUILDSPEC']._serialized_end=2606
+  _globals['_BUILDSPEC_PARAMETERSENTRY']._serialized_start=2557
+  _globals['_BUILDSPEC_PARAMETERSENTRY']._serialized_end=2606
+  _globals['_RESOURCES']._serialized_start=2609
+  _globals['_RESOURCES']._serialized_end=2858
+  _globals['_CPURESOURCES']._serialized_start=2860
+  _globals['_CPURESOURCES']._serialized_end=2964
+  _globals['_GPURESOURCES']._serialized_start=2966
+  _globals['_GPURESOURCES']._serialized_end=3047
+  _globals['_GITMODELSOURCE']._serialized_start=3049
+  _globals['_GITMODELSOURCE']._serialized_end=3168
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/automation/v1/action_pb2.pyi

@@ -8,6 +8,7 @@ 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 qwak.automation.v1.auto_scaling_pb2
 import qwak.automation.v1.common_pb2
 import qwak.batch_job.v1.batch_job_service_pb2
@@ -34,6 +35,7 @@ class _GpuTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTy
     NVIDIA_A100: _GpuType.ValueType  # 3
     NVIDIA_T4: _GpuType.ValueType  # 4
     NVIDIA_A10G: _GpuType.ValueType  # 5
+    NVIDIA_L4: _GpuType.ValueType  # 6
 
 class GpuType(_GpuType, metaclass=_GpuTypeEnumTypeWrapper): ...
 
@@ -43,6 +45,7 @@ NVIDIA_V100: GpuType.ValueType  # 2
 NVIDIA_A100: GpuType.ValueType  # 3
 NVIDIA_T4: GpuType.ValueType  # 4
 NVIDIA_A10G: GpuType.ValueType  # 5
+NVIDIA_L4: GpuType.ValueType  # 6
 global___GpuType = GpuType
 
 class _MemoryUnit:
@@ -63,6 +66,21 @@ MIB: MemoryUnit.ValueType  # 1
 GIB: MemoryUnit.ValueType  # 2
 global___MemoryUnit = MemoryUnit
 
+class _PurchaseOption:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _PurchaseOptionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PurchaseOption.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    ON_DEMAND: _PurchaseOption.ValueType  # 0
+    SPOT: _PurchaseOption.ValueType  # 1
+
+class PurchaseOption(_PurchaseOption, metaclass=_PurchaseOptionEnumTypeWrapper): ...
+
+ON_DEMAND: PurchaseOption.ValueType  # 0
+SPOT: PurchaseOption.ValueType  # 1
+global___PurchaseOption = PurchaseOption
+
 class Action(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -128,20 +146,26 @@ class DeploymentCondition(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     BUILD_METRIC_FIELD_NUMBER: builtins.int
+    DEPLOYED_BUILD_METRIC_FIELD_NUMBER: builtins.int
     @property
     def build_metric(self) -> global___BuildMetricCondition: ...
+    @property
+    def deployed_build_metric(self) -> global___DeployedBuildMetricCondition: ...
     def __init__(
         self,
         *,
         build_metric: global___BuildMetricCondition | None = ...,
+        deployed_build_metric: global___DeployedBuildMetricCondition | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["build_metric", b"build_metric", "condition", b"condition"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_metric", b"build_metric", "condition", b"condition"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["condition", b"condition"]) -> typing_extensions.Literal["build_metric"] | None: ...
+    def HasField(self, field_name: typing_extensions.Literal["build_metric", b"build_metric", "condition", b"condition", "deployed_build_metric", b"deployed_build_metric"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_metric", b"build_metric", "condition", b"condition", "deployed_build_metric", b"deployed_build_metric"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["condition", b"condition"]) -> typing_extensions.Literal["build_metric", "deployed_build_metric"] | None: ...
 
 global___DeploymentCondition = DeploymentCondition
 
 class BuildMetricCondition(google.protobuf.message.Message):
+    """checks the metric of the current build the automation is executing"""
+
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     METRIC_NAME_FIELD_NUMBER: builtins.int
@@ -164,6 +188,31 @@ class BuildMetricCondition(google.protobuf.message.Message):
 
 global___BuildMetricCondition = BuildMetricCondition
 
+class DeployedBuildMetricCondition(google.protobuf.message.Message):
+    """checks the metric of the deployed build (if there is one)"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    METRIC_NAME_FIELD_NUMBER: builtins.int
+    THRESHOLD_DIRECTION_FIELD_NUMBER: builtins.int
+    VARIATION_FIELD_NUMBER: builtins.int
+    metric_name: builtins.str
+    """The metric name to check"""
+    threshold_direction: qwak.automation.v1.common_pb2.MetricThresholdDirection.ValueType
+    """Metric direction to check"""
+    variation: builtins.str
+    """The deployed variation name to check"""
+    def __init__(
+        self,
+        *,
+        metric_name: builtins.str = ...,
+        threshold_direction: qwak.automation.v1.common_pb2.MetricThresholdDirection.ValueType = ...,
+        variation: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["metric_name", b"metric_name", "threshold_direction", b"threshold_direction", "variation", b"variation"]) -> None: ...
+
+global___DeployedBuildMetricCondition = DeployedBuildMetricCondition
+
 class DeploymentSpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -209,6 +258,7 @@ class AdvancedDeploymentOptions(google.protobuf.message.Message):
     AUTO_SCALING_CONFIG_FIELD_NUMBER: builtins.int
     MAX_BATCH_SIZE_FIELD_NUMBER: builtins.int
     DEPLOYMENT_PROCESS_TIMEOUT_LIMIT_FIELD_NUMBER: builtins.int
+    SERVICE_ACCOUNT_KEY_SECRET_NAME_FIELD_NUMBER: builtins.int
     number_of_http_server_workers: builtins.int
     """Amount of http server workers"""
     http_request_timeout_ms: builtins.int
@@ -224,6 +274,8 @@ class AdvancedDeploymentOptions(google.protobuf.message.Message):
     """Max batch size of records"""
     deployment_process_timeout_limit: builtins.int
     """The timeout for the deployment (in seconds)"""
+    service_account_key_secret_name: builtins.str
+    """Custom service account for GCP"""
     def __init__(
         self,
         *,
@@ -234,9 +286,10 @@ class AdvancedDeploymentOptions(google.protobuf.message.Message):
         auto_scaling_config: qwak.automation.v1.auto_scaling_pb2.AutoScalingConfig | None = ...,
         max_batch_size: builtins.int = ...,
         deployment_process_timeout_limit: builtins.int = ...,
+        service_account_key_secret_name: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["auto_scaling_config", b"auto_scaling_config"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["auto_scaling_config", b"auto_scaling_config", "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_http_server_workers", b"number_of_http_server_workers"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["auto_scaling_config", b"auto_scaling_config", "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_http_server_workers", b"number_of_http_server_workers", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
 
 global___AdvancedDeploymentOptions = AdvancedDeploymentOptions
 
@@ -307,6 +360,11 @@ class BuildSpec(google.protobuf.message.Message):
     MAIN_DIR_FIELD_NUMBER: builtins.int
     DEPENDENCY_FILE_PATH_FIELD_NUMBER: builtins.int
     ENV_VARS_FIELD_NUMBER: builtins.int
+    GPU_COMPATIBLE_FIELD_NUMBER: builtins.int
+    PUSH_IMAGE_FIELD_NUMBER: builtins.int
+    PURCHASE_OPTION_FIELD_NUMBER: builtins.int
+    SERVICE_ACCOUNT_KEY_SECRET_NAME_FIELD_NUMBER: builtins.int
+    PROVISION_INSTANCE_TIMEOUT_FIELD_NUMBER: builtins.int
     @property
     def git_model_source(self) -> global___GitModelSource:
         """Git Details to get model from."""
@@ -327,6 +385,14 @@ class BuildSpec(google.protobuf.message.Message):
     @property
     def env_vars(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         """Environment variables for the model code"""
+    gpu_compatible: builtins.bool
+    """If the model is GPU compatible"""
+    @property
+    def push_image(self) -> google.protobuf.wrappers_pb2.BoolValue: ...
+    purchase_option: global___PurchaseOption.ValueType
+    service_account_key_secret_name: builtins.str
+    """Custom service account for GCP"""
+    provision_instance_timeout: builtins.int
     def __init__(
         self,
         *,
@@ -339,9 +405,14 @@ class BuildSpec(google.protobuf.message.Message):
         main_dir: builtins.str = ...,
         dependency_file_path: builtins.str = ...,
         env_vars: collections.abc.Iterable[builtins.str] | None = ...,
+        gpu_compatible: builtins.bool = ...,
+        push_image: google.protobuf.wrappers_pb2.BoolValue | None = ...,
+        purchase_option: global___PurchaseOption.ValueType = ...,
+        service_account_key_secret_name: builtins.str = ...,
+        provision_instance_timeout: builtins.int = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["git_model_source", b"git_model_source", "resource", b"resource"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["assumed_iam_role", b"assumed_iam_role", "base_image", b"base_image", "dependency_file_path", b"dependency_file_path", "env_vars", b"env_vars", "git_model_source", b"git_model_source", "main_dir", b"main_dir", "parameters", b"parameters", "resource", b"resource", "tags", b"tags"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["git_model_source", b"git_model_source", "push_image", b"push_image", "resource", b"resource"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["assumed_iam_role", b"assumed_iam_role", "base_image", b"base_image", "dependency_file_path", b"dependency_file_path", "env_vars", b"env_vars", "git_model_source", b"git_model_source", "gpu_compatible", b"gpu_compatible", "main_dir", b"main_dir", "parameters", b"parameters", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "push_image", b"push_image", "resource", b"resource", "service_account_key_secret_name", b"service_account_key_secret_name", "tags", b"tags"]) -> None: ...
 
 global___BuildSpec = BuildSpec
 
@@ -421,18 +492,22 @@ class GitModelSource(google.protobuf.message.Message):
     GIT_URI_FIELD_NUMBER: builtins.int
     GIT_CREDENTIALS_SECRET_NAME_FIELD_NUMBER: builtins.int
     GIT_BRANCH_FIELD_NUMBER: builtins.int
+    GIT_SSH_SECRET_NAME_FIELD_NUMBER: builtins.int
     git_uri: builtins.str
     """git uri to fetch models code from"""
     git_credentials_secret_name: builtins.str
     """Secret to fetch git credentials, overrides git_credentials if set"""
     git_branch: builtins.str
+    git_ssh_secret_name: builtins.str
+    """Ssh keu to fetch git credentials, overrides git_secret_ssh"""
     def __init__(
         self,
         *,
         git_uri: builtins.str = ...,
         git_credentials_secret_name: builtins.str = ...,
         git_branch: builtins.str = ...,
+        git_ssh_secret_name: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["git_branch", b"git_branch", "git_credentials_secret_name", b"git_credentials_secret_name", "git_uri", b"git_uri"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["git_branch", b"git_branch", "git_credentials_secret_name", b"git_credentials_secret_name", "git_ssh_secret_name", b"git_ssh_secret_name", "git_uri", b"git_uri"]) -> None: ...
 
 global___GitModelSource = GitModelSource

_qwak_proto/qwak/automation/v1/auto_scaling_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/automation/v1/auto_scaling.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()
@@ -15,83 +13,27 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%qwak/automation/v1/auto_scaling.proto\x12\x12qwak.automation.v1\"\xad\x01\n\x11\x41utoScalingConfig\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.\n\x08triggers\x18\x05 \x01(\x0b\x32\x1c.qwak.automation.v1.Triggers\">\n\x08Triggers\x12\x32\n\x08triggers\x18\x01 \x03(\x0b\x32 .qwak.automation.v1.ScaleTrigger\"n\n\x0cScaleTrigger\x12N\n\x12prometheus_trigger\x18\x01 \x01(\x0b\x32\x30.qwak.automation.v1.AutoScalingPrometheusTriggerH\x00\x42\x0e\n\x0ctrigger_type\"d\n\x1c\x41utoScalingPrometheusTrigger\x12\x31\n\nquery_spec\x18\x01 \x01(\x0b\x32\x1d.qwak.automation.v1.QuerySpec\x12\x11\n\tthreshold\x18\x02 \x01(\x05\"\xaa\x01\n\tQuerySpec\x12>\n\x0bmetric_type\x18\x01 \x01(\x0e\x32).qwak.automation.v1.AutoScalingMetricType\x12H\n\x10\x61ggregation_type\x18\x02 \x01(\x0e\x32..qwak.automation.v1.AutoScalingAggregationType\x12\x13\n\x0btime_period\x18\x03 \x01(\x05*x\n\x15\x41utoScalingMetricType\x12\x19\n\x15METRIC_TYPE_NOT_VALID\x10\x00\x12\x13\n\x0fMETRIC_TYPE_CPU\x10\x01\x12\x16\n\x12METRIC_TYPE_MEMORY\x10\x02\x12\x17\n\x13METRIC_TYPE_LATENCY\x10\x03*\x90\x02\n\x1a\x41utoScalingAggregationType\x12\x1e\n\x1a\x41GGREGATION_TYPE_NOT_VALID\x10\x00\x12\x18\n\x14\x41GGREGATION_TYPE_MIN\x10\x01\x12\x18\n\x14\x41GGREGATION_TYPE_MAX\x10\x02\x12\x1c\n\x18\x41GGREGATION_TYPE_AVERAGE\x10\x03\x12\x18\n\x14\x41GGREGATION_TYPE_SUM\x10\x04\x12\x18\n\x14\x41GGREGATION_TYPE_P50\x10\x05\x12\x18\n\x14\x41GGREGATION_TYPE_P90\x10\x06\x12\x18\n\x14\x41GGREGATION_TYPE_P95\x10\x07\x12\x18\n\x14\x41GGREGATION_TYPE_P99\x10\x08\x42!\n\x1d\x63om.qwak.ai.automation.api.v1P\x01\x62\x06proto3')
-
-_AUTOSCALINGMETRICTYPE = DESCRIPTOR.enum_types_by_name['AutoScalingMetricType']
-AutoScalingMetricType = enum_type_wrapper.EnumTypeWrapper(_AUTOSCALINGMETRICTYPE)
-_AUTOSCALINGAGGREGATIONTYPE = DESCRIPTOR.enum_types_by_name['AutoScalingAggregationType']
-AutoScalingAggregationType = enum_type_wrapper.EnumTypeWrapper(_AUTOSCALINGAGGREGATIONTYPE)
-METRIC_TYPE_NOT_VALID = 0
-METRIC_TYPE_CPU = 1
-METRIC_TYPE_MEMORY = 2
-METRIC_TYPE_LATENCY = 3
-AGGREGATION_TYPE_NOT_VALID = 0
-AGGREGATION_TYPE_MIN = 1
-AGGREGATION_TYPE_MAX = 2
-AGGREGATION_TYPE_AVERAGE = 3
-AGGREGATION_TYPE_SUM = 4
-AGGREGATION_TYPE_P50 = 5
-AGGREGATION_TYPE_P90 = 6
-AGGREGATION_TYPE_P95 = 7
-AGGREGATION_TYPE_P99 = 8
-
-
-_AUTOSCALINGCONFIG = DESCRIPTOR.message_types_by_name['AutoScalingConfig']
-_TRIGGERS = DESCRIPTOR.message_types_by_name['Triggers']
-_SCALETRIGGER = DESCRIPTOR.message_types_by_name['ScaleTrigger']
-_AUTOSCALINGPROMETHEUSTRIGGER = DESCRIPTOR.message_types_by_name['AutoScalingPrometheusTrigger']
-_QUERYSPEC = DESCRIPTOR.message_types_by_name['QuerySpec']
-AutoScalingConfig = _reflection.GeneratedProtocolMessageType('AutoScalingConfig', (_message.Message,), {
-  'DESCRIPTOR' : _AUTOSCALINGCONFIG,
-  '__module__' : 'qwak.automation.v1.auto_scaling_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.AutoScalingConfig)
-  })
-_sym_db.RegisterMessage(AutoScalingConfig)
-
-Triggers = _reflection.GeneratedProtocolMessageType('Triggers', (_message.Message,), {
-  'DESCRIPTOR' : _TRIGGERS,
-  '__module__' : 'qwak.automation.v1.auto_scaling_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.Triggers)
-  })
-_sym_db.RegisterMessage(Triggers)
-
-ScaleTrigger = _reflection.GeneratedProtocolMessageType('ScaleTrigger', (_message.Message,), {
-  'DESCRIPTOR' : _SCALETRIGGER,
-  '__module__' : 'qwak.automation.v1.auto_scaling_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.ScaleTrigger)
-  })
-_sym_db.RegisterMessage(ScaleTrigger)
-
-AutoScalingPrometheusTrigger = _reflection.GeneratedProtocolMessageType('AutoScalingPrometheusTrigger', (_message.Message,), {
-  'DESCRIPTOR' : _AUTOSCALINGPROMETHEUSTRIGGER,
-  '__module__' : 'qwak.automation.v1.auto_scaling_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.AutoScalingPrometheusTrigger)
-  })
-_sym_db.RegisterMessage(AutoScalingPrometheusTrigger)
-
-QuerySpec = _reflection.GeneratedProtocolMessageType('QuerySpec', (_message.Message,), {
-  'DESCRIPTOR' : _QUERYSPEC,
-  '__module__' : 'qwak.automation.v1.auto_scaling_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.QuerySpec)
-  })
-_sym_db.RegisterMessage(QuerySpec)
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%qwak/automation/v1/auto_scaling.proto\x12\x12qwak.automation.v1\"\xad\x01\n\x11\x41utoScalingConfig\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.\n\x08triggers\x18\x05 \x01(\x0b\x32\x1c.qwak.automation.v1.Triggers\">\n\x08Triggers\x12\x32\n\x08triggers\x18\x01 \x03(\x0b\x32 .qwak.automation.v1.ScaleTrigger\"n\n\x0cScaleTrigger\x12N\n\x12prometheus_trigger\x18\x01 \x01(\x0b\x32\x30.qwak.automation.v1.AutoScalingPrometheusTriggerH\x00\x42\x0e\n\x0ctrigger_type\"d\n\x1c\x41utoScalingPrometheusTrigger\x12\x31\n\nquery_spec\x18\x01 \x01(\x0b\x32\x1d.qwak.automation.v1.QuerySpec\x12\x11\n\tthreshold\x18\x02 \x01(\x05\"\xaa\x01\n\tQuerySpec\x12>\n\x0bmetric_type\x18\x01 \x01(\x0e\x32).qwak.automation.v1.AutoScalingMetricType\x12H\n\x10\x61ggregation_type\x18\x02 \x01(\x0e\x32..qwak.automation.v1.AutoScalingAggregationType\x12\x13\n\x0btime_period\x18\x03 \x01(\x05*\xc5\x01\n\x15\x41utoScalingMetricType\x12\x19\n\x15METRIC_TYPE_NOT_VALID\x10\x00\x12\x13\n\x0fMETRIC_TYPE_CPU\x10\x01\x12\x16\n\x12METRIC_TYPE_MEMORY\x10\x02\x12\x17\n\x13METRIC_TYPE_LATENCY\x10\x03\x12\x13\n\x0fMETRIC_TYPE_GPU\x10\x04\x12\x1a\n\x16METRIC_TYPE_ERROR_RATE\x10\x05\x12\x1a\n\x16METRIC_TYPE_THROUGHPUT\x10\x06*\x90\x02\n\x1a\x41utoScalingAggregationType\x12\x1e\n\x1a\x41GGREGATION_TYPE_NOT_VALID\x10\x00\x12\x18\n\x14\x41GGREGATION_TYPE_MIN\x10\x01\x12\x18\n\x14\x41GGREGATION_TYPE_MAX\x10\x02\x12\x1c\n\x18\x41GGREGATION_TYPE_AVERAGE\x10\x03\x12\x18\n\x14\x41GGREGATION_TYPE_SUM\x10\x04\x12\x18\n\x14\x41GGREGATION_TYPE_P50\x10\x05\x12\x18\n\x14\x41GGREGATION_TYPE_P90\x10\x06\x12\x18\n\x14\x41GGREGATION_TYPE_P95\x10\x07\x12\x18\n\x14\x41GGREGATION_TYPE_P99\x10\x08\x42!\n\x1d\x63om.qwak.ai.automation.api.v1P\x01\x62\x06proto3')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.automation.v1.auto_scaling_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n\035com.qwak.ai.automation.api.v1P\001'
-  _AUTOSCALINGMETRICTYPE._serialized_start=688
-  _AUTOSCALINGMETRICTYPE._serialized_end=808
-  _AUTOSCALINGAGGREGATIONTYPE._serialized_start=811
-  _AUTOSCALINGAGGREGATIONTYPE._serialized_end=1083
-  _AUTOSCALINGCONFIG._serialized_start=62
-  _AUTOSCALINGCONFIG._serialized_end=235
-  _TRIGGERS._serialized_start=237
-  _TRIGGERS._serialized_end=299
-  _SCALETRIGGER._serialized_start=301
-  _SCALETRIGGER._serialized_end=411
-  _AUTOSCALINGPROMETHEUSTRIGGER._serialized_start=413
-  _AUTOSCALINGPROMETHEUSTRIGGER._serialized_end=513
-  _QUERYSPEC._serialized_start=516
-  _QUERYSPEC._serialized_end=686
+  _globals['_AUTOSCALINGMETRICTYPE']._serialized_start=689
+  _globals['_AUTOSCALINGMETRICTYPE']._serialized_end=886
+  _globals['_AUTOSCALINGAGGREGATIONTYPE']._serialized_start=889
+  _globals['_AUTOSCALINGAGGREGATIONTYPE']._serialized_end=1161
+  _globals['_AUTOSCALINGCONFIG']._serialized_start=62
+  _globals['_AUTOSCALINGCONFIG']._serialized_end=235
+  _globals['_TRIGGERS']._serialized_start=237
+  _globals['_TRIGGERS']._serialized_end=299
+  _globals['_SCALETRIGGER']._serialized_start=301
+  _globals['_SCALETRIGGER']._serialized_end=411
+  _globals['_AUTOSCALINGPROMETHEUSTRIGGER']._serialized_start=413
+  _globals['_AUTOSCALINGPROMETHEUSTRIGGER']._serialized_end=513
+  _globals['_QUERYSPEC']._serialized_start=516
+  _globals['_QUERYSPEC']._serialized_end=686
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/automation/v1/auto_scaling_pb2.pyi

@@ -28,6 +28,9 @@ class _AutoScalingMetricTypeEnumTypeWrapper(google.protobuf.internal.enum_type_w
     METRIC_TYPE_CPU: _AutoScalingMetricType.ValueType  # 1
     METRIC_TYPE_MEMORY: _AutoScalingMetricType.ValueType  # 2
     METRIC_TYPE_LATENCY: _AutoScalingMetricType.ValueType  # 3
+    METRIC_TYPE_GPU: _AutoScalingMetricType.ValueType  # 4
+    METRIC_TYPE_ERROR_RATE: _AutoScalingMetricType.ValueType  # 5
+    METRIC_TYPE_THROUGHPUT: _AutoScalingMetricType.ValueType  # 6
 
 class AutoScalingMetricType(_AutoScalingMetricType, metaclass=_AutoScalingMetricTypeEnumTypeWrapper): ...
 
@@ -35,6 +38,9 @@ METRIC_TYPE_NOT_VALID: AutoScalingMetricType.ValueType  # 0
 METRIC_TYPE_CPU: AutoScalingMetricType.ValueType  # 1
 METRIC_TYPE_MEMORY: AutoScalingMetricType.ValueType  # 2
 METRIC_TYPE_LATENCY: AutoScalingMetricType.ValueType  # 3
+METRIC_TYPE_GPU: AutoScalingMetricType.ValueType  # 4
+METRIC_TYPE_ERROR_RATE: AutoScalingMetricType.ValueType  # 5
+METRIC_TYPE_THROUGHPUT: AutoScalingMetricType.ValueType  # 6
 global___AutoScalingMetricType = AutoScalingMetricType
 
 class _AutoScalingAggregationType: