qwak-core 0.4.284__py3-none-any.whl → 0.4.285__py3-none-any.whl

_qwak_proto/qwak/kube_deployment_captain/batch_job_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/kube_deployment_captain/batch_job.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,261 +16,9 @@ from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,qwak/kube_deployment_captain/batch_job.proto\x12\x1cqwak.kube.deployment.captain\x1a\x1egoogle/protobuf/wrappers.proto\"\xd2\x03\n\x1cListInferenceJobFilesRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x16\n\x0e\x64irectory_path\x18\x02 \x01(\t\x12\x18\n\x0ctoken_secret\x18\x03 \x01(\tB\x02\x18\x01\x12\x19\n\rsecret_secret\x18\x04 \x01(\tB\x02\x18\x01\x12\x1a\n\x12secret_service_url\x18\x05 \x01(\t\x12\x14\n\x08role_arn\x18\x06 \x01(\tB\x02\x18\x01\x12G\n\x0f\x61ws_credentials\x18\x07 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x08 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12P\n\x14grouped_task_details\x18\t \x01(\x0b\x32\x30.qwak.kube.deployment.captain.GroupedTaskDetailsH\x01\x12\x16\n\x0e\x65nvironment_id\x18\n \x01(\tB\x1a\n\x18\x63loud_client_credentialsB\x0b\n\ttask_flow\"c\n\x12GroupedTaskDetails\x12\x17\n\x0f\x63oncurrentTasks\x18\x01 \x01(\x05\x12\x18\n\x10\x64\x65stination_path\x18\x02 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x03 \x01(\t\"\xb4\x02\n)CleanStorageObjectFromCloudStorageRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x15\n\robject_prefix\x18\x02 \x03(\t\x12\x1a\n\x12secret_service_url\x18\x03 \x01(\t\x12G\n\x0f\x61ws_credentials\x18\x04 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x05 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12\x16\n\x0e\x65nvironment_id\x18\x06 \x01(\tB\x1a\n\x18\x63loud_client_credentials\"O\n\x0e\x41wsCredentials\x12\x10\n\x08role_arn\x18\x01 \x01(\t\x12\x14\n\x0ctoken_secret\x18\x02 \x01(\t\x12\x15\n\rsecret_secret\x18\x04 \x01(\t\"9\n\x0eGcpCredentials\x12\'\n\x1fservice_account_json_key_secret\x18\x01 \x01(\t\"\xa6\x01\n\x1dListInferenceJobFilesResponse\x12\x12\n\nfile_names\x18\x01 \x03(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x03 \x01(\t\x12H\n\x11list_file_details\x18\x04 \x03(\x0b\x32-.qwak.kube.deployment.captain.ListFileDetails\"W\n\x0fListFileDetails\x12\x15\n\ris_point_file\x18\x01 \x01(\x08\x12\x17\n\x0fpoint_file_path\x18\x02 \x01(\t\x12\x14\n\x0cobject_paths\x18\x03 \x03(\t\"\xf0\x01\n\"CreateInferenceTaskExecutorRequest\x12\x63\n\x1btask_executor_configuration\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.TaskExecutorConfigurationMessage\x12\x65\n\x1cinference_task_configuration\x18\x02 \x01(\x0b\x32?.qwak.kube.deployment.captain.InferenceTaskConfigurationMessage\"\x89\x01\n\x1aPrepareInferenceJobRequest\x12k\n#inference_job_configuration_message\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.InferenceJobConfigurationMessage\"F\n\x1bPrepareInferenceJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xcd\x04\n TaskExecutorConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\x12G\n\x10model_identifier\x18\x05 \x01(\x0b\x32-.qwak.kube.deployment.captain.ModelIdentifier\x12\x11\n\timage_url\x18\x06 \x01(\t\x12\x15\n\rbackoff_limit\x18\x07 \x01(\x05\x12\x0f\n\x03\x63pu\x18\x08 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\t \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x0c \x01(\t\x12\x41\n\x08job_size\x18\r \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x17\n\x0fpurchase_option\x18\x0e \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0f \x01(\t\x12%\n\x1d\x63ustom_service_account_secret\x18\x10 \x01(\t\x12#\n\x1bjfrog_token_api_secret_name\x18\x11 \x01(\t\x12\x16\n\x0ejfrog_base_url\x18\x12 \x01(\t\"q\n InferenceJobConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x03 \x01(\t\"`\n\x0fModelIdentifier\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x15\n\tbranch_id\x18\x03 \x01(\tB\x02\x18\x01\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\xa2\x01\n\x1dListFilesPreStepConfiguration\x12\x13\n\x0bsource_path\x18\x01 \x01(\t\x12\x1c\n\x14list_files_file_name\x18\x02 \x01(\t\x12\x1d\n\x15total_number_of_tasks\x18\x03 \x01(\x05\x12/\n\ntask_index\x18\x04 \x01(\x0b\x32\x1b.google.protobuf.Int32Value\"\xf4\x04\n!InferenceTaskConfigurationMessage\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x02 \x01(\t\x12\x10\n\x08\x66ilepath\x18\x03 \x01(\t\x12\x18\n\x10\x64\x65stination_path\x18\x04 \x01(\t\x12\x44\n\x0finput_file_type\x18\x05 \x01(\x0e\x32+.qwak.kube.deployment.captain.InputFileType\x12\x46\n\x10output_file_type\x18\x06 \x01(\x0e\x32,.qwak.kube.deployment.captain.OutputFileType\x12\x14\n\x0ctoken_secret\x18\x07 \x01(\t\x12\x15\n\rsecret_secret\x18\x08 \x01(\t\x12\x43\n\nparameters\x18\t \x03(\x0b\x32/.qwak.kube.deployment.captain.BatchJobParameter\x12\x43\n\x0e\x63loud_provider\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.CloudProvider\x12\'\n\x1fservice_account_json_key_secret\x18\x0b \x01(\t\x12\x1a\n\x12is_point_file_path\x18\x0c \x01(\x08\x12\x66\n!list_files_pre_step_configuration\x18\r \x01(\x0b\x32;.qwak.kube.deployment.captain.ListFilesPreStepConfiguration\"/\n\x11\x42\x61tchJobParameter\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"?\n#CleanInferenceTasksExecutorsRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\"O\n$CleanInferenceTasksExecutorsResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"X\n!CleanInferenceTaskExecutorRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\"M\n\"CleanInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"N\n#CreateInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xf7\x02\n\x1eStartInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x11\n\timage_url\x18\x04 \x01(\t\x12\x0f\n\x03\x63pu\x18\x05 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x06 \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\x07 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x11\n\texecutors\x18\x08 \x01(\x05\x12\x0f\n\x07timeout\x18\t \x01(\x05\x12\x41\n\x08job_size\x18\n \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x12\n\nmodel_uuid\x18\x0b \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0c \x01(\t\"!\n\x1fStartInferenceJobWarmupResponse\"p\n\x1f\x43\x61ncelInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\"\n CancelInferenceJobWarmupResponse\"\xd5\x01\n\x11\x42\x61tchJobResources\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\x12\x41\n\x0cmemory_units\x18\x04 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApi\x12\x41\n\rgpu_resources\x18\x05 \x01(\x0b\x32*.qwak.kube.deployment.captain.GpuResources\"[\n\x0cGpuResources\x12\x37\n\x08gpu_type\x18\x01 \x01(\x0e\x32%.qwak.kube.deployment.captain.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05*:\n\rMemoryUnitApi\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x81\x01\n\rInputFileType\x12\x1d\n\x19UNDEFINED_INPUT_FILE_TYPE\x10\x00\x12\x17\n\x13\x43SV_INPUT_FILE_TYPE\x10\x01\x12\x1b\n\x17\x46\x45\x41THER_INPUT_FILE_TYPE\x10\x02\x12\x1b\n\x17PARQUET_INPUT_FILE_TYPE\x10\x03*\x86\x01\n\x0eOutputFileType\x12\x1e\n\x1aUNDEFINED_OUTPUT_FILE_TYPE\x10\x00\x12\x18\n\x14\x43SV_OUTPUT_FILE_TYPE\x10\x01\x12\x1c\n\x18\x46\x45\x41THER_OUTPUT_FILE_TYPE\x10\x02\x12\x1c\n\x18PARQUET_OUTPUT_FILE_TYPE\x10\x03*\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\rCloudProvider\x12\x1a\n\x16UNKNOWN_CLOUD_PROVIDER\x10\x00\x12\x07\n\x03\x41WS\x10\x01\x12\x07\n\x03GCP\x10\x02\x42+\n\'com.qwak.ai.kube.deployment.captain.apiP\x01\x62\x06proto3')
 
-_MEMORYUNITAPI = DESCRIPTOR.enum_types_by_name['MemoryUnitApi']
-MemoryUnitApi = enum_type_wrapper.EnumTypeWrapper(_MEMORYUNITAPI)
-_INPUTFILETYPE = DESCRIPTOR.enum_types_by_name['InputFileType']
-InputFileType = enum_type_wrapper.EnumTypeWrapper(_INPUTFILETYPE)
-_OUTPUTFILETYPE = DESCRIPTOR.enum_types_by_name['OutputFileType']
-OutputFileType = enum_type_wrapper.EnumTypeWrapper(_OUTPUTFILETYPE)
-_GPUTYPE = DESCRIPTOR.enum_types_by_name['GpuType']
-GpuType = enum_type_wrapper.EnumTypeWrapper(_GPUTYPE)
-_CLOUDPROVIDER = DESCRIPTOR.enum_types_by_name['CloudProvider']
-CloudProvider = enum_type_wrapper.EnumTypeWrapper(_CLOUDPROVIDER)
-UNKNOWN_MEMORY_UNIT = 0
-MIB = 1
-GIB = 2
-UNDEFINED_INPUT_FILE_TYPE = 0
-CSV_INPUT_FILE_TYPE = 1
-FEATHER_INPUT_FILE_TYPE = 2
-PARQUET_INPUT_FILE_TYPE = 3
-UNDEFINED_OUTPUT_FILE_TYPE = 0
-CSV_OUTPUT_FILE_TYPE = 1
-FEATHER_OUTPUT_FILE_TYPE = 2
-PARQUET_OUTPUT_FILE_TYPE = 3
-INVALID_GPU = 0
-NVIDIA_K80 = 1
-NVIDIA_V100 = 2
-NVIDIA_A100 = 3
-NVIDIA_T4 = 4
-NVIDIA_A10G = 5
-NVIDIA_L4 = 6
-NVIDIA_T4_1_4_15 = 7
-NVIDIA_T4_1_8_30 = 8
-NVIDIA_T4_1_16_60 = 9
-NVIDIA_A100_80GB_8_96_1360 = 10
-NVIDIA_V100_1_8_52 = 11
-NVIDIA_V100_4_32_208 = 12
-UNKNOWN_CLOUD_PROVIDER = 0
-AWS = 1
-GCP = 2
-
-
-_LISTINFERENCEJOBFILESREQUEST = DESCRIPTOR.message_types_by_name['ListInferenceJobFilesRequest']
-_GROUPEDTASKDETAILS = DESCRIPTOR.message_types_by_name['GroupedTaskDetails']
-_CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST = DESCRIPTOR.message_types_by_name['CleanStorageObjectFromCloudStorageRequest']
-_AWSCREDENTIALS = DESCRIPTOR.message_types_by_name['AwsCredentials']
-_GCPCREDENTIALS = DESCRIPTOR.message_types_by_name['GcpCredentials']
-_LISTINFERENCEJOBFILESRESPONSE = DESCRIPTOR.message_types_by_name['ListInferenceJobFilesResponse']
-_LISTFILEDETAILS = DESCRIPTOR.message_types_by_name['ListFileDetails']
-_CREATEINFERENCETASKEXECUTORREQUEST = DESCRIPTOR.message_types_by_name['CreateInferenceTaskExecutorRequest']
-_PREPAREINFERENCEJOBREQUEST = DESCRIPTOR.message_types_by_name['PrepareInferenceJobRequest']
-_PREPAREINFERENCEJOBRESPONSE = DESCRIPTOR.message_types_by_name['PrepareInferenceJobResponse']
-_TASKEXECUTORCONFIGURATIONMESSAGE = DESCRIPTOR.message_types_by_name['TaskExecutorConfigurationMessage']
-_INFERENCEJOBCONFIGURATIONMESSAGE = DESCRIPTOR.message_types_by_name['InferenceJobConfigurationMessage']
-_MODELIDENTIFIER = DESCRIPTOR.message_types_by_name['ModelIdentifier']
-_LISTFILESPRESTEPCONFIGURATION = DESCRIPTOR.message_types_by_name['ListFilesPreStepConfiguration']
-_INFERENCETASKCONFIGURATIONMESSAGE = DESCRIPTOR.message_types_by_name['InferenceTaskConfigurationMessage']
-_BATCHJOBPARAMETER = DESCRIPTOR.message_types_by_name['BatchJobParameter']
-_CLEANINFERENCETASKSEXECUTORSREQUEST = DESCRIPTOR.message_types_by_name['CleanInferenceTasksExecutorsRequest']
-_CLEANINFERENCETASKSEXECUTORSRESPONSE = DESCRIPTOR.message_types_by_name['CleanInferenceTasksExecutorsResponse']
-_CLEANINFERENCETASKEXECUTORREQUEST = DESCRIPTOR.message_types_by_name['CleanInferenceTaskExecutorRequest']
-_CLEANINFERENCETASKEXECUTORRESPONSE = DESCRIPTOR.message_types_by_name['CleanInferenceTaskExecutorResponse']
-_CREATEINFERENCETASKEXECUTORRESPONSE = DESCRIPTOR.message_types_by_name['CreateInferenceTaskExecutorResponse']
-_STARTINFERENCEJOBWARMUPREQUEST = DESCRIPTOR.message_types_by_name['StartInferenceJobWarmupRequest']
-_STARTINFERENCEJOBWARMUPRESPONSE = DESCRIPTOR.message_types_by_name['StartInferenceJobWarmupResponse']
-_CANCELINFERENCEJOBWARMUPREQUEST = DESCRIPTOR.message_types_by_name['CancelInferenceJobWarmupRequest']
-_CANCELINFERENCEJOBWARMUPRESPONSE = DESCRIPTOR.message_types_by_name['CancelInferenceJobWarmupResponse']
-_BATCHJOBRESOURCES = DESCRIPTOR.message_types_by_name['BatchJobResources']
-_GPURESOURCES = DESCRIPTOR.message_types_by_name['GpuResources']
-ListInferenceJobFilesRequest = _reflection.GeneratedProtocolMessageType('ListInferenceJobFilesRequest', (_message.Message,), {
-  'DESCRIPTOR' : _LISTINFERENCEJOBFILESREQUEST,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.ListInferenceJobFilesRequest)
-  })
-_sym_db.RegisterMessage(ListInferenceJobFilesRequest)
-
-GroupedTaskDetails = _reflection.GeneratedProtocolMessageType('GroupedTaskDetails', (_message.Message,), {
-  'DESCRIPTOR' : _GROUPEDTASKDETAILS,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.GroupedTaskDetails)
-  })
-_sym_db.RegisterMessage(GroupedTaskDetails)
-
-CleanStorageObjectFromCloudStorageRequest = _reflection.GeneratedProtocolMessageType('CleanStorageObjectFromCloudStorageRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.CleanStorageObjectFromCloudStorageRequest)
-  })
-_sym_db.RegisterMessage(CleanStorageObjectFromCloudStorageRequest)
-
-AwsCredentials = _reflection.GeneratedProtocolMessageType('AwsCredentials', (_message.Message,), {
-  'DESCRIPTOR' : _AWSCREDENTIALS,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.AwsCredentials)
-  })
-_sym_db.RegisterMessage(AwsCredentials)
-
-GcpCredentials = _reflection.GeneratedProtocolMessageType('GcpCredentials', (_message.Message,), {
-  'DESCRIPTOR' : _GCPCREDENTIALS,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.GcpCredentials)
-  })
-_sym_db.RegisterMessage(GcpCredentials)
-
-ListInferenceJobFilesResponse = _reflection.GeneratedProtocolMessageType('ListInferenceJobFilesResponse', (_message.Message,), {
-  'DESCRIPTOR' : _LISTINFERENCEJOBFILESRESPONSE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.ListInferenceJobFilesResponse)
-  })
-_sym_db.RegisterMessage(ListInferenceJobFilesResponse)
-
-ListFileDetails = _reflection.GeneratedProtocolMessageType('ListFileDetails', (_message.Message,), {
-  'DESCRIPTOR' : _LISTFILEDETAILS,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.ListFileDetails)
-  })
-_sym_db.RegisterMessage(ListFileDetails)
-
-CreateInferenceTaskExecutorRequest = _reflection.GeneratedProtocolMessageType('CreateInferenceTaskExecutorRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CREATEINFERENCETASKEXECUTORREQUEST,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.CreateInferenceTaskExecutorRequest)
-  })
-_sym_db.RegisterMessage(CreateInferenceTaskExecutorRequest)
-
-PrepareInferenceJobRequest = _reflection.GeneratedProtocolMessageType('PrepareInferenceJobRequest', (_message.Message,), {
-  'DESCRIPTOR' : _PREPAREINFERENCEJOBREQUEST,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.PrepareInferenceJobRequest)
-  })
-_sym_db.RegisterMessage(PrepareInferenceJobRequest)
-
-PrepareInferenceJobResponse = _reflection.GeneratedProtocolMessageType('PrepareInferenceJobResponse', (_message.Message,), {
-  'DESCRIPTOR' : _PREPAREINFERENCEJOBRESPONSE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.PrepareInferenceJobResponse)
-  })
-_sym_db.RegisterMessage(PrepareInferenceJobResponse)
-
-TaskExecutorConfigurationMessage = _reflection.GeneratedProtocolMessageType('TaskExecutorConfigurationMessage', (_message.Message,), {
-  'DESCRIPTOR' : _TASKEXECUTORCONFIGURATIONMESSAGE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.TaskExecutorConfigurationMessage)
-  })
-_sym_db.RegisterMessage(TaskExecutorConfigurationMessage)
-
-InferenceJobConfigurationMessage = _reflection.GeneratedProtocolMessageType('InferenceJobConfigurationMessage', (_message.Message,), {
-  'DESCRIPTOR' : _INFERENCEJOBCONFIGURATIONMESSAGE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.InferenceJobConfigurationMessage)
-  })
-_sym_db.RegisterMessage(InferenceJobConfigurationMessage)
-
-ModelIdentifier = _reflection.GeneratedProtocolMessageType('ModelIdentifier', (_message.Message,), {
-  'DESCRIPTOR' : _MODELIDENTIFIER,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.ModelIdentifier)
-  })
-_sym_db.RegisterMessage(ModelIdentifier)
-
-ListFilesPreStepConfiguration = _reflection.GeneratedProtocolMessageType('ListFilesPreStepConfiguration', (_message.Message,), {
-  'DESCRIPTOR' : _LISTFILESPRESTEPCONFIGURATION,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.ListFilesPreStepConfiguration)
-  })
-_sym_db.RegisterMessage(ListFilesPreStepConfiguration)
-
-InferenceTaskConfigurationMessage = _reflection.GeneratedProtocolMessageType('InferenceTaskConfigurationMessage', (_message.Message,), {
-  'DESCRIPTOR' : _INFERENCETASKCONFIGURATIONMESSAGE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.InferenceTaskConfigurationMessage)
-  })
-_sym_db.RegisterMessage(InferenceTaskConfigurationMessage)
-
-BatchJobParameter = _reflection.GeneratedProtocolMessageType('BatchJobParameter', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHJOBPARAMETER,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.BatchJobParameter)
-  })
-_sym_db.RegisterMessage(BatchJobParameter)
-
-CleanInferenceTasksExecutorsRequest = _reflection.GeneratedProtocolMessageType('CleanInferenceTasksExecutorsRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CLEANINFERENCETASKSEXECUTORSREQUEST,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.CleanInferenceTasksExecutorsRequest)
-  })
-_sym_db.RegisterMessage(CleanInferenceTasksExecutorsRequest)
-
-CleanInferenceTasksExecutorsResponse = _reflection.GeneratedProtocolMessageType('CleanInferenceTasksExecutorsResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CLEANINFERENCETASKSEXECUTORSRESPONSE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.CleanInferenceTasksExecutorsResponse)
-  })
-_sym_db.RegisterMessage(CleanInferenceTasksExecutorsResponse)
-
-CleanInferenceTaskExecutorRequest = _reflection.GeneratedProtocolMessageType('CleanInferenceTaskExecutorRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CLEANINFERENCETASKEXECUTORREQUEST,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.CleanInferenceTaskExecutorRequest)
-  })
-_sym_db.RegisterMessage(CleanInferenceTaskExecutorRequest)
-
-CleanInferenceTaskExecutorResponse = _reflection.GeneratedProtocolMessageType('CleanInferenceTaskExecutorResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CLEANINFERENCETASKEXECUTORRESPONSE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.CleanInferenceTaskExecutorResponse)
-  })
-_sym_db.RegisterMessage(CleanInferenceTaskExecutorResponse)
-
-CreateInferenceTaskExecutorResponse = _reflection.GeneratedProtocolMessageType('CreateInferenceTaskExecutorResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CREATEINFERENCETASKEXECUTORRESPONSE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.CreateInferenceTaskExecutorResponse)
-  })
-_sym_db.RegisterMessage(CreateInferenceTaskExecutorResponse)
-
-StartInferenceJobWarmupRequest = _reflection.GeneratedProtocolMessageType('StartInferenceJobWarmupRequest', (_message.Message,), {
-  'DESCRIPTOR' : _STARTINFERENCEJOBWARMUPREQUEST,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.StartInferenceJobWarmupRequest)
-  })
-_sym_db.RegisterMessage(StartInferenceJobWarmupRequest)
-
-StartInferenceJobWarmupResponse = _reflection.GeneratedProtocolMessageType('StartInferenceJobWarmupResponse', (_message.Message,), {
-  'DESCRIPTOR' : _STARTINFERENCEJOBWARMUPRESPONSE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.StartInferenceJobWarmupResponse)
-  })
-_sym_db.RegisterMessage(StartInferenceJobWarmupResponse)
-
-CancelInferenceJobWarmupRequest = _reflection.GeneratedProtocolMessageType('CancelInferenceJobWarmupRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CANCELINFERENCEJOBWARMUPREQUEST,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.CancelInferenceJobWarmupRequest)
-  })
-_sym_db.RegisterMessage(CancelInferenceJobWarmupRequest)
-
-CancelInferenceJobWarmupResponse = _reflection.GeneratedProtocolMessageType('CancelInferenceJobWarmupResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CANCELINFERENCEJOBWARMUPRESPONSE,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.CancelInferenceJobWarmupResponse)
-  })
-_sym_db.RegisterMessage(CancelInferenceJobWarmupResponse)
-
-BatchJobResources = _reflection.GeneratedProtocolMessageType('BatchJobResources', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHJOBRESOURCES,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.BatchJobResources)
-  })
-_sym_db.RegisterMessage(BatchJobResources)
-
-GpuResources = _reflection.GeneratedProtocolMessageType('GpuResources', (_message.Message,), {
-  'DESCRIPTOR' : _GPURESOURCES,
-  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.GpuResources)
-  })
-_sym_db.RegisterMessage(GpuResources)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.kube_deployment_captain.batch_job_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -301,68 +47,68 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _STARTINFERENCEJOBWARMUPREQUEST.fields_by_name['memory_units']._serialized_options = b'\030\001'
   _CANCELINFERENCEJOBWARMUPREQUEST.fields_by_name['branch_id']._options = None
   _CANCELINFERENCEJOBWARMUPREQUEST.fields_by_name['branch_id']._serialized_options = b'\030\001'
-  _MEMORYUNITAPI._serialized_start=4761
-  _MEMORYUNITAPI._serialized_end=4819
-  _INPUTFILETYPE._serialized_start=4822
-  _INPUTFILETYPE._serialized_end=4951
-  _OUTPUTFILETYPE._serialized_start=4954
-  _OUTPUTFILETYPE._serialized_end=5088
-  _GPUTYPE._serialized_start=5091
-  _GPUTYPE._serialized_end=5363
-  _CLOUDPROVIDER._serialized_start=5365
-  _CLOUDPROVIDER._serialized_end=5426
-  _LISTINFERENCEJOBFILESREQUEST._serialized_start=111
-  _LISTINFERENCEJOBFILESREQUEST._serialized_end=577
-  _GROUPEDTASKDETAILS._serialized_start=579
-  _GROUPEDTASKDETAILS._serialized_end=678
-  _CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST._serialized_start=681
-  _CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST._serialized_end=989
-  _AWSCREDENTIALS._serialized_start=991
-  _AWSCREDENTIALS._serialized_end=1070
-  _GCPCREDENTIALS._serialized_start=1072
-  _GCPCREDENTIALS._serialized_end=1129
-  _LISTINFERENCEJOBFILESRESPONSE._serialized_start=1132
-  _LISTINFERENCEJOBFILESRESPONSE._serialized_end=1298
-  _LISTFILEDETAILS._serialized_start=1300
-  _LISTFILEDETAILS._serialized_end=1387
-  _CREATEINFERENCETASKEXECUTORREQUEST._serialized_start=1390
-  _CREATEINFERENCETASKEXECUTORREQUEST._serialized_end=1630
-  _PREPAREINFERENCEJOBREQUEST._serialized_start=1633
-  _PREPAREINFERENCEJOBREQUEST._serialized_end=1770
-  _PREPAREINFERENCEJOBRESPONSE._serialized_start=1772
-  _PREPAREINFERENCEJOBRESPONSE._serialized_end=1842
-  _TASKEXECUTORCONFIGURATIONMESSAGE._serialized_start=1845
-  _TASKEXECUTORCONFIGURATIONMESSAGE._serialized_end=2434
-  _INFERENCEJOBCONFIGURATIONMESSAGE._serialized_start=2436
-  _INFERENCEJOBCONFIGURATIONMESSAGE._serialized_end=2549
-  _MODELIDENTIFIER._serialized_start=2551
-  _MODELIDENTIFIER._serialized_end=2647
-  _LISTFILESPRESTEPCONFIGURATION._serialized_start=2650
-  _LISTFILESPRESTEPCONFIGURATION._serialized_end=2812
-  _INFERENCETASKCONFIGURATIONMESSAGE._serialized_start=2815
-  _INFERENCETASKCONFIGURATIONMESSAGE._serialized_end=3443
-  _BATCHJOBPARAMETER._serialized_start=3445
-  _BATCHJOBPARAMETER._serialized_end=3492
-  _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_start=3494
-  _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_end=3557
-  _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_start=3559
-  _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_end=3638
-  _CLEANINFERENCETASKEXECUTORREQUEST._serialized_start=3640
-  _CLEANINFERENCETASKEXECUTORREQUEST._serialized_end=3728
-  _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_start=3730
-  _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_end=3807
-  _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_start=3809
-  _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_end=3887
-  _STARTINFERENCEJOBWARMUPREQUEST._serialized_start=3890
-  _STARTINFERENCEJOBWARMUPREQUEST._serialized_end=4265
-  _STARTINFERENCEJOBWARMUPRESPONSE._serialized_start=4267
-  _STARTINFERENCEJOBWARMUPRESPONSE._serialized_end=4300
-  _CANCELINFERENCEJOBWARMUPREQUEST._serialized_start=4302
-  _CANCELINFERENCEJOBWARMUPREQUEST._serialized_end=4414
-  _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_start=4416
-  _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_end=4450
-  _BATCHJOBRESOURCES._serialized_start=4453
-  _BATCHJOBRESOURCES._serialized_end=4666
-  _GPURESOURCES._serialized_start=4668
-  _GPURESOURCES._serialized_end=4759
+  _globals['_MEMORYUNITAPI']._serialized_start=4761
+  _globals['_MEMORYUNITAPI']._serialized_end=4819
+  _globals['_INPUTFILETYPE']._serialized_start=4822
+  _globals['_INPUTFILETYPE']._serialized_end=4951
+  _globals['_OUTPUTFILETYPE']._serialized_start=4954
+  _globals['_OUTPUTFILETYPE']._serialized_end=5088
+  _globals['_GPUTYPE']._serialized_start=5091
+  _globals['_GPUTYPE']._serialized_end=5363
+  _globals['_CLOUDPROVIDER']._serialized_start=5365
+  _globals['_CLOUDPROVIDER']._serialized_end=5426
+  _globals['_LISTINFERENCEJOBFILESREQUEST']._serialized_start=111
+  _globals['_LISTINFERENCEJOBFILESREQUEST']._serialized_end=577
+  _globals['_GROUPEDTASKDETAILS']._serialized_start=579
+  _globals['_GROUPEDTASKDETAILS']._serialized_end=678
+  _globals['_CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST']._serialized_start=681
+  _globals['_CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST']._serialized_end=989
+  _globals['_AWSCREDENTIALS']._serialized_start=991
+  _globals['_AWSCREDENTIALS']._serialized_end=1070
+  _globals['_GCPCREDENTIALS']._serialized_start=1072
+  _globals['_GCPCREDENTIALS']._serialized_end=1129
+  _globals['_LISTINFERENCEJOBFILESRESPONSE']._serialized_start=1132
+  _globals['_LISTINFERENCEJOBFILESRESPONSE']._serialized_end=1298
+  _globals['_LISTFILEDETAILS']._serialized_start=1300
+  _globals['_LISTFILEDETAILS']._serialized_end=1387
+  _globals['_CREATEINFERENCETASKEXECUTORREQUEST']._serialized_start=1390
+  _globals['_CREATEINFERENCETASKEXECUTORREQUEST']._serialized_end=1630
+  _globals['_PREPAREINFERENCEJOBREQUEST']._serialized_start=1633
+  _globals['_PREPAREINFERENCEJOBREQUEST']._serialized_end=1770
+  _globals['_PREPAREINFERENCEJOBRESPONSE']._serialized_start=1772
+  _globals['_PREPAREINFERENCEJOBRESPONSE']._serialized_end=1842
+  _globals['_TASKEXECUTORCONFIGURATIONMESSAGE']._serialized_start=1845
+  _globals['_TASKEXECUTORCONFIGURATIONMESSAGE']._serialized_end=2434
+  _globals['_INFERENCEJOBCONFIGURATIONMESSAGE']._serialized_start=2436
+  _globals['_INFERENCEJOBCONFIGURATIONMESSAGE']._serialized_end=2549
+  _globals['_MODELIDENTIFIER']._serialized_start=2551
+  _globals['_MODELIDENTIFIER']._serialized_end=2647
+  _globals['_LISTFILESPRESTEPCONFIGURATION']._serialized_start=2650
+  _globals['_LISTFILESPRESTEPCONFIGURATION']._serialized_end=2812
+  _globals['_INFERENCETASKCONFIGURATIONMESSAGE']._serialized_start=2815
+  _globals['_INFERENCETASKCONFIGURATIONMESSAGE']._serialized_end=3443
+  _globals['_BATCHJOBPARAMETER']._serialized_start=3445
+  _globals['_BATCHJOBPARAMETER']._serialized_end=3492
+  _globals['_CLEANINFERENCETASKSEXECUTORSREQUEST']._serialized_start=3494
+  _globals['_CLEANINFERENCETASKSEXECUTORSREQUEST']._serialized_end=3557
+  _globals['_CLEANINFERENCETASKSEXECUTORSRESPONSE']._serialized_start=3559
+  _globals['_CLEANINFERENCETASKSEXECUTORSRESPONSE']._serialized_end=3638
+  _globals['_CLEANINFERENCETASKEXECUTORREQUEST']._serialized_start=3640
+  _globals['_CLEANINFERENCETASKEXECUTORREQUEST']._serialized_end=3728
+  _globals['_CLEANINFERENCETASKEXECUTORRESPONSE']._serialized_start=3730
+  _globals['_CLEANINFERENCETASKEXECUTORRESPONSE']._serialized_end=3807
+  _globals['_CREATEINFERENCETASKEXECUTORRESPONSE']._serialized_start=3809
+  _globals['_CREATEINFERENCETASKEXECUTORRESPONSE']._serialized_end=3887
+  _globals['_STARTINFERENCEJOBWARMUPREQUEST']._serialized_start=3890
+  _globals['_STARTINFERENCEJOBWARMUPREQUEST']._serialized_end=4265
+  _globals['_STARTINFERENCEJOBWARMUPRESPONSE']._serialized_start=4267
+  _globals['_STARTINFERENCEJOBWARMUPRESPONSE']._serialized_end=4300
+  _globals['_CANCELINFERENCEJOBWARMUPREQUEST']._serialized_start=4302
+  _globals['_CANCELINFERENCEJOBWARMUPREQUEST']._serialized_end=4414
+  _globals['_CANCELINFERENCEJOBWARMUPRESPONSE']._serialized_start=4416
+  _globals['_CANCELINFERENCEJOBWARMUPRESPONSE']._serialized_end=4450
+  _globals['_BATCHJOBRESOURCES']._serialized_start=4453
+  _globals['_BATCHJOBRESOURCES']._serialized_end=4666
+  _globals['_GPURESOURCES']._serialized_start=4668
+  _globals['_GPURESOURCES']._serialized_end=4759
 # @@protoc_insertion_point(module_scope)