qwak-core 0.4.362__py3-none-any.whl → 0.5.4__py3-none-any.whl

_qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.py

@@ -17,7 +17,7 @@ from _qwak_proto.qwak.batch_job.v1 import batch_job_events_pb2 as qwak_dot_batch
 from _qwak_proto.qwak.administration.authenticated_user.v1 import credentials_pb2 as qwak_dot_administration_dot_authenticated__user_dot_v1_dot_credentials__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)qwak/batch_job/v1/batch_job_service.proto\x12\rqwak.batchjob\x1a\x1fgoogle/protobuf/timestamp.proto\x1a/qwak/user_application/common/v0/resources.proto\x1a(qwak/batch_job/v1/batch_job_events.proto\x1a;qwak/administration/authenticated_user/v1/credentials.proto\"\xb4\x01\n\x15StartWarmupJobRequest\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\x05 \x01(\t\x12\x16\n\x0ewarmup_timeout\x18\x03 \x01(\x05\x12H\n\x19\x62\x61tch_job_deployment_size\x18\x04 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\"B\n\x16StartWarmupJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\"S\n\x16\x43\x61ncelWarmupJobRequest\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\"C\n\x17\x43\x61ncelWarmupJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\"\x89\x05\n\x14StartBatchJobRequest\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\x0f \x01(\t\x12\x15\n\rsource_bucket\x18\x03 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x04 \x01(\t\x12\x15\n\rsource_folder\x18\x05 \x01(\t\x12\x1a\n\x12\x64\x65stination_folder\x18\x06 \x01(\t\x12H\n\x19\x62\x61tch_job_deployment_size\x18\x07 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12\x13\n\x0bjob_timeout\x18\x08 \x01(\x05\x12\x14\n\x0ctask_timeout\x18\t \x01(\x05\x12\x35\n\x0finput_file_type\x18\n \x01(\x0e\x32\x1c.qwak.batchjob.InputFileType\x12\x37\n\x10output_file_type\x18\x0b \x01(\x0e\x32\x1d.qwak.batchjob.OutputFileType\x12\x14\n\x0ctoken_secret\x18\x0c \x01(\t\x12\x15\n\rsecret_secret\x18\r \x01(\t\x12M\n\x1b\x61\x64vanced_deployment_options\x18\x0e \x01(\x0b\x32(.qwak.batchjob.AdvancedDeploymentOptions\x12\x34\n\nparameters\x18\x10 \x03(\x0b\x32 .qwak.batchjob.BatchJobParameter\x12\x39\n\x11\x62\x61tch_job_request\x18\x11 \x01(\x0b\x32\x1e.qwak.batchjob.BatchJobRequest\"\xcb\x01\n\x0f\x42\x61tchJobRequest\x12:\n\rmodel_details\x18\x01 \x01(\x0b\x32#.qwak.batchjob.BatchJobModelDetails\x12\x38\n\x0c\x64\x61ta_details\x18\x02 \x01(\x0b\x32\".qwak.batchjob.BatchJobDataDetails\x12\x42\n\x11\x65xecution_details\x18\x04 \x01(\x0b\x32\'.qwak.batchjob.BatchJobExecutionDetails\":\n\x14\x42\x61tchJobModelDetails\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x0f \x01(\t\"\xe5\x01\n\x13\x42\x61tchJobDataDetails\x12\x36\n\x0bsource_path\x18\x01 \x01(\x0b\x32!.qwak.batchjob.BatchJobSourcePath\x12@\n\x10\x64\x65stination_path\x18\x02 \x01(\x0b\x32&.qwak.batchjob.BatchJobDestinationPath\x12\x14\n\x0ctoken_secret\x18\x0c \x01(\t\x12\x15\n\rsecret_secret\x18\r \x01(\t\x12\'\n\x1fservice_account_json_key_secret\x18\x05 \x01(\t\"y\n\x12\x42\x61tchJobSourcePath\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x15\n\rsource_folder\x18\x02 \x01(\t\x12\x35\n\x0finput_file_type\x18\x03 \x01(\x0e\x32\x1c.qwak.batchjob.InputFileType\"\x8a\x01\n\x17\x42\x61tchJobDestinationPath\x12\x1a\n\x12\x64\x65stination_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_folder\x18\x02 \x01(\t\x12\x37\n\x10output_file_type\x18\x03 \x01(\x0e\x32\x1d.qwak.batchjob.OutputFileType\"\x94\x02\n\x18\x42\x61tchJobExecutionDetails\x12\x13\n\x0bjob_timeout\x18\x01 \x01(\x05\x12\x14\n\x0ctask_timeout\x18\x02 \x01(\x05\x12H\n\x19\x62\x61tch_job_deployment_size\x18\x03 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12M\n\x1b\x61\x64vanced_deployment_options\x18\x04 \x01(\x0b\x32(.qwak.batchjob.AdvancedDeploymentOptions\x12\x34\n\nparameters\x18\x05 \x03(\x0b\x32 .qwak.batchjob.BatchJobParameter\"/\n\x11\x42\x61tchJobParameter\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"S\n\x15StartBatchJobResponse\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\")\n\x15\x43\x61ncelBatchJobRequest\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\"B\n\x16\x43\x61ncelBatchJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\",\n\x18GetBatchJobStatusRequest\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\"\xac\x01\n\x19GetBatchJobStatusResponse\x12\x38\n\njob_status\x18\x01 \x01(\x0e\x32$.qwak.batchjob.BatchJobStatusMessage\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\x12\x16\n\x0e\x66inished_files\x18\x04 \x01(\x05\x12\x13\n\x0btotal_files\x18\x05 \x01(\x05\",\n\x18GetBatchJobReportRequest\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\"a\n\x19GetBatchJobReportResponse\x12\x12\n\nsuccessful\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\x12\x17\n\x0freport_messages\x18\x03 \x03(\t\"`\n\x1dUpdateBatchTasksStatusRequest\x12?\n\x11\x62\x61tch_task_events\x18\x01 \x03(\x0b\x32$.qwak.batchjob.BatchTaskEventMessage\"I\n\x1dUpdateBatchTaskStatusResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\"z\n\x19\x41\x64vancedDeploymentOptions\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x01 \x01(\t\x12\x17\n\x0fpurchase_option\x18\x02 \x01(\t\x12\'\n\x1fservice_account_key_secret_name\x18\x03 \x01(\t\"\x85\x03\n\x16\x42\x61tchJobDeploymentSize\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\x12J\n\x0cmemory_units\x18\x04 \x01(\x0e\x32\x30.qwak.batchjob.BatchJobDeploymentSize.MemoryUnitB\x02\x18\x01\x12H\n\rgpu_resources\x18\x05 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResourcesB\x02\x18\x01\x12`\n\x1c\x63lient_pod_compute_resources\x18\x06 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\"/\n\nMemoryUnit\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x1a\x02\x18\x01\"\x99\x01\n\x15\x42\x61tchJobExecutionSize\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12S\n\x15pod_compute_resources\x18\x02 \x01(\x0b\x32\x34.qwak.user_application.common.v0.PodComputeResources\x12\x13\n\x0btemplate_id\x18\x03 \x01(\t\"Q\n\x14ListBatchJobsRequest\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\"K\n\x15ListBatchJobsResponse\x12\x32\n\nbatch_jobs\x18\x01 \x03(\x0b\x32\x1e.qwak.batchjob.BatchJobDetails\"\x81\x03\n\x0f\x42\x61tchJobDetails\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x0e\n\x06job_id\x18\x02 \x01(\t\x12\x38\n\njob_status\x18\x03 \x01(\x0e\x32$.qwak.batchjob.BatchJobStatusMessage\x12.\n\nstart_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x17\n\x0f\x66\x61ilure_message\x18\x06 \x01(\t\x12;\n\x08job_size\x18\x07 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSizeB\x02\x18\x01\x12\x46\n\x18\x62\x61tch_job_execution_size\x18\x08 \x01(\x0b\x32$.qwak.batchjob.BatchJobExecutionSize\x12\x16\n\x0e\x65nvironment_id\x18\t \x01(\t\"+\n\x19GetBatchJobDetailsRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"y\n\x1aGetBatchJobDetailsResponse\x12\x31\n\tbatch_job\x18\x01 \x01(\x0b\x32\x1e.qwak.batchjob.BatchJobMessage\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\"\x84\x04\n\x0f\x42\x61tchJobMessage\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12\x15\n\tbranch_id\x18\x03 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x04 \x01(\t\x12.\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x38\n\njob_status\x18\x07 \x01(\x0e\x32$.qwak.batchjob.BatchJobStatusMessage\x12\x15\n\rtotal_records\x18\x08 \x01(\x05\x12\x37\n\x08job_size\x18\t \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12;\n\x0freport_messages\x18\n \x03(\x0b\x32\".qwak.batchjob.ExecutionReportLine\x12\x43\n\x16\x65xecution_file_details\x18\x0b \x01(\x0b\x32#.qwak.batchjob.ExecutionFileDetails\x12<\n\x0ftask_executions\x18\x0c \x03(\x0b\x32#.qwak.batchjob.TaskExecutionDetails\"M\n\x13\x45xecutionReportLine\x12(\n\x04time\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0c\n\x04line\x18\x02 \x01(\t\"\x95\x02\n\x14\x45xecutionFileDetails\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x02 \x01(\t\x12\x13\n\x0bsource_path\x18\x03 \x01(\t\x12\x18\n\x10\x64\x65stination_path\x18\x04 \x01(\t\x12\x35\n\x0finput_file_type\x18\x05 \x01(\x0e\x32\x1c.qwak.batchjob.InputFileType\x12\x37\n\x10output_file_type\x18\x06 \x01(\x0e\x32\x1d.qwak.batchjob.OutputFileType\x12\x16\n\x0e\x66inished_files\x18\x07 \x01(\x05\x12\x13\n\x0btotal_files\x18\x08 \x01(\x05\"\xfd\x01\n\x14TaskExecutionDetails\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x35\n\x06status\x18\x02 \x01(\x0e\x32%.qwak.batchjob.BatchTaskStatusMessage\x12.\n\nstart_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x16\n\x0enum_of_records\x18\x05 \x01(\x05\x12\x14\n\x08\x66ilename\x18\x06 \x01(\tB\x02\x18\x01\x12\x11\n\tfilenames\x18\x07 \x03(\t\"d\n$GetBatchJobPreSignedUploadUrlRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x17\n\x0fnumber_of_files\x18\x02 \x01(\x05\x12\x11\n\tfile_type\x18\x03 \x01(\t\"\x98\x01\n%GetBatchJobPreSignedUploadUrlResponse\x12\x12\n\ninput_path\x18\x01 \x01(\t\x12\x13\n\x0boutput_path\x18\x02 \x01(\t\x12\x0e\n\x06\x62ucket\x18\x03 \x01(\t\x12\x0c\n\x04urls\x18\x04 \x03(\t\x12\x0f\n\x07success\x18\x05 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x06 \x01(\t\"8\n&GetBatchJobPreSignedDownloadUrlRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"a\n\'GetBatchJobPreSignedDownloadUrlResponse\x12\x0c\n\x04urls\x18\x01 \x03(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\"3\n\x1fGetBatchJobUploadDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\xb4\x01\n GetBatchJobUploadDetailsResponse\x12\x12\n\ninput_path\x18\x01 \x01(\t\x12\x13\n\x0boutput_path\x18\x02 \x01(\t\x12\x0e\n\x06\x62ucket\x18\x03 \x01(\t\x12W\n\x0b\x63redentials\x18\x04 \x01(\x0b\x32\x42.qwak.administration.authenticated_user.v1.AwsTemporaryCredentials\"3\n!GetBatchJobDownloadDetailsRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"\x9b\x01\n\"GetBatchJobDownloadDetailsResponse\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x0c\n\x04keys\x18\x02 \x03(\t\x12W\n\x0b\x63redentials\x18\x03 \x01(\x0b\x32\x42.qwak.administration.authenticated_user.v1.AwsTemporaryCredentials\"\xd7\x01\n\x1aUpdateDefaultParamsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x46\n\x17\x64\x65\x66\x61ult_deployment_size\x18\x03 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12M\n\x1b\x61\x64vanced_deployment_options\x18\x04 \x01(\x0b\x32(.qwak.batchjob.AdvancedDeploymentOptions\"\x1d\n\x1bUpdateDefaultParamsResponse\".\n\x1a\x44\x65leteDefaultParamsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\x1d\n\x1b\x44\x65leteDefaultParamsResponse\"S\n&UpdateTaskIngestedRecordsAmountRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x18\n\x10ingested_records\x18\x02 \x01(\x03\")\n\'UpdateTaskIngestedRecordsAmountResponse\"S\n\x19UpdateTasksDetailsRequest\x12\x36\n\rtasks_details\x18\x01 \x03(\x0b\x32\x1f.qwak.batchjob.BatchTaskDetails\"\x1c\n\x1aUpdateTasksDetailsResponse\"a\n\x10\x42\x61tchTaskDetails\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12<\n\x13input_files_details\x18\x02 \x03(\x0b\x32\x1f.qwak.batchjob.InputFileDetails\" \n\x10InputFileDetails\x12\x0c\n\x04path\x18\x01 \x01(\t*\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*\x8d\x02\n\x15\x42\x61tchJobStatusMessage\x12\x1e\n\x1aUNDEFINED_BATCH_JOB_STATUS\x10\x00\x12\x1e\n\x1a\x42\x41TCH_JOB_COMMITTED_STATUS\x10\x01\x12\x1c\n\x18\x42\x41TCH_JOB_PENDING_STATUS\x10\x02\x12\x1c\n\x18\x42\x41TCH_JOB_RUNNING_STATUS\x10\x03\x12\x1d\n\x19\x42\x41TCH_JOB_FINISHED_STATUS\x10\x04\x12\x1b\n\x17\x42\x41TCH_JOB_FAILED_STATUS\x10\x05\x12\x1e\n\x1a\x42\x41TCH_JOB_CANCELLED_STATUS\x10\x06\x12\x1c\n\x18\x42\x41TCH_JOB_TIMEOUT_STATUS\x10\x07*\x9c\x02\n\x16\x42\x61tchTaskStatusMessage\x12\x1f\n\x1bUNDEFINED_BATCH_TASK_STATUS\x10\x00\x12\x1f\n\x1b\x42\x41TCH_TASK_COMMITTED_STATUS\x10\x01\x12#\n\x1f\x42\x41TCH_TASK_PENDING_START_STATUS\x10\x02\x12\x1d\n\x19\x42\x41TCH_TASK_RUNNING_STATUS\x10\x03\x12\x1e\n\x1a\x42\x41TCH_TASK_FINISHED_STATUS\x10\x04\x12\x1c\n\x18\x42\x41TCH_TASK_FAILED_STATUS\x10\x05\x12\x1f\n\x1b\x42\x41TCH_TASK_CANCELLED_STATUS\x10\x06\x12\x1d\n\x19\x42\x41TCH_TASK_TIMEOUT_STATUS\x10\x07\x32\x9f\x0f\n\x19\x42\x61tchJobManagementService\x12Z\n\rStartBatchJob\x12#.qwak.batchjob.StartBatchJobRequest\x1a$.qwak.batchjob.StartBatchJobResponse\x12]\n\x0e\x43\x61ncelBatchJob\x12$.qwak.batchjob.CancelBatchJobRequest\x1a%.qwak.batchjob.CancelBatchJobResponse\x12]\n\x0eStartWarmupJob\x12$.qwak.batchjob.StartWarmupJobRequest\x1a%.qwak.batchjob.StartWarmupJobResponse\x12`\n\x0f\x43\x61ncelWarmupJob\x12%.qwak.batchjob.CancelWarmupJobRequest\x1a&.qwak.batchjob.CancelWarmupJobResponse\x12\x66\n\x11GetBatchJobStatus\x12\'.qwak.batchjob.GetBatchJobStatusRequest\x1a(.qwak.batchjob.GetBatchJobStatusResponse\x12\x66\n\x11GetBatchJobReport\x12\'.qwak.batchjob.GetBatchJobReportRequest\x1a(.qwak.batchjob.GetBatchJobReportResponse\x12t\n\x16UpdateBatchTasksStatus\x12,.qwak.batchjob.UpdateBatchTasksStatusRequest\x1a,.qwak.batchjob.UpdateBatchTaskStatusResponse\x12Z\n\rListBatchJobs\x12#.qwak.batchjob.ListBatchJobsRequest\x1a$.qwak.batchjob.ListBatchJobsResponse\x12i\n\x12GetBatchJobDetails\x12(.qwak.batchjob.GetBatchJobDetailsRequest\x1a).qwak.batchjob.GetBatchJobDetailsResponse\x12\x8a\x01\n\x1dGetBatchJobPreSignedUploadUrl\x12\x33.qwak.batchjob.GetBatchJobPreSignedUploadUrlRequest\x1a\x34.qwak.batchjob.GetBatchJobPreSignedUploadUrlResponse\x12\x90\x01\n\x1fGetBatchJobPreSignedDownloadUrl\x12\x35.qwak.batchjob.GetBatchJobPreSignedDownloadUrlRequest\x1a\x36.qwak.batchjob.GetBatchJobPreSignedDownloadUrlResponse\x12{\n\x18GetBatchJobUploadDetails\x12..qwak.batchjob.GetBatchJobUploadDetailsRequest\x1a/.qwak.batchjob.GetBatchJobUploadDetailsResponse\x12\x81\x01\n\x1aGetBatchJobDownloadDetails\x12\x30.qwak.batchjob.GetBatchJobDownloadDetailsRequest\x1a\x31.qwak.batchjob.GetBatchJobDownloadDetailsResponse\x12l\n\x13UpdateDefaultParams\x12).qwak.batchjob.UpdateDefaultParamsRequest\x1a*.qwak.batchjob.UpdateDefaultParamsResponse\x12l\n\x13\x44\x65leteDefaultParams\x12).qwak.batchjob.DeleteDefaultParamsRequest\x1a*.qwak.batchjob.DeleteDefaultParamsResponse\x12\x90\x01\n\x1fUpdateTaskIngestedRecordsAmount\x12\x35.qwak.batchjob.UpdateTaskIngestedRecordsAmountRequest\x1a\x36.qwak.batchjob.UpdateTaskIngestedRecordsAmountResponse\x12i\n\x12UpdateTasksDetails\x12(.qwak.batchjob.UpdateTasksDetailsRequest\x1a).qwak.batchjob.UpdateTasksDetailsResponseB.\n\x19\x63om.qwak.ai.batch.job.apiP\x01Z\x0f./;inferencejobb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)qwak/batch_job/v1/batch_job_service.proto\x12\rqwak.batchjob\x1a\x1fgoogle/protobuf/timestamp.proto\x1a/qwak/user_application/common/v0/resources.proto\x1a(qwak/batch_job/v1/batch_job_events.proto\x1a;qwak/administration/authenticated_user/v1/credentials.proto\"\xb4\x01\n\x15StartWarmupJobRequest\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\x05 \x01(\t\x12\x16\n\x0ewarmup_timeout\x18\x03 \x01(\x05\x12H\n\x19\x62\x61tch_job_deployment_size\x18\x04 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\"B\n\x16StartWarmupJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\"S\n\x16\x43\x61ncelWarmupJobRequest\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\"C\n\x17\x43\x61ncelWarmupJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\"\x89\x05\n\x14StartBatchJobRequest\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\x0f \x01(\t\x12\x15\n\rsource_bucket\x18\x03 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x04 \x01(\t\x12\x15\n\rsource_folder\x18\x05 \x01(\t\x12\x1a\n\x12\x64\x65stination_folder\x18\x06 \x01(\t\x12H\n\x19\x62\x61tch_job_deployment_size\x18\x07 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12\x13\n\x0bjob_timeout\x18\x08 \x01(\x05\x12\x14\n\x0ctask_timeout\x18\t \x01(\x05\x12\x35\n\x0finput_file_type\x18\n \x01(\x0e\x32\x1c.qwak.batchjob.InputFileType\x12\x37\n\x10output_file_type\x18\x0b \x01(\x0e\x32\x1d.qwak.batchjob.OutputFileType\x12\x14\n\x0ctoken_secret\x18\x0c \x01(\t\x12\x15\n\rsecret_secret\x18\r \x01(\t\x12M\n\x1b\x61\x64vanced_deployment_options\x18\x0e \x01(\x0b\x32(.qwak.batchjob.AdvancedDeploymentOptions\x12\x34\n\nparameters\x18\x10 \x03(\x0b\x32 .qwak.batchjob.BatchJobParameter\x12\x39\n\x11\x62\x61tch_job_request\x18\x11 \x01(\x0b\x32\x1e.qwak.batchjob.BatchJobRequest\"\xcb\x01\n\x0f\x42\x61tchJobRequest\x12:\n\rmodel_details\x18\x01 \x01(\x0b\x32#.qwak.batchjob.BatchJobModelDetails\x12\x38\n\x0c\x64\x61ta_details\x18\x02 \x01(\x0b\x32\".qwak.batchjob.BatchJobDataDetails\x12\x42\n\x11\x65xecution_details\x18\x04 \x01(\x0b\x32\'.qwak.batchjob.BatchJobExecutionDetails\":\n\x14\x42\x61tchJobModelDetails\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x0f \x01(\t\"\xfc\x01\n\x13\x42\x61tchJobDataDetails\x12\x36\n\x0bsource_path\x18\x01 \x01(\x0b\x32!.qwak.batchjob.BatchJobSourcePath\x12@\n\x10\x64\x65stination_path\x18\x02 \x01(\x0b\x32&.qwak.batchjob.BatchJobDestinationPath\x12\x14\n\x0ctoken_secret\x18\x0c \x01(\t\x12\x15\n\rsecret_secret\x18\r \x01(\t\x12\x15\n\rsession_token\x18\x03 \x01(\t\x12\'\n\x1fservice_account_json_key_secret\x18\x05 \x01(\t\"y\n\x12\x42\x61tchJobSourcePath\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x15\n\rsource_folder\x18\x02 \x01(\t\x12\x35\n\x0finput_file_type\x18\x03 \x01(\x0e\x32\x1c.qwak.batchjob.InputFileType\"\x8a\x01\n\x17\x42\x61tchJobDestinationPath\x12\x1a\n\x12\x64\x65stination_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_folder\x18\x02 \x01(\t\x12\x37\n\x10output_file_type\x18\x03 \x01(\x0e\x32\x1d.qwak.batchjob.OutputFileType\"\x94\x02\n\x18\x42\x61tchJobExecutionDetails\x12\x13\n\x0bjob_timeout\x18\x01 \x01(\x05\x12\x14\n\x0ctask_timeout\x18\x02 \x01(\x05\x12H\n\x19\x62\x61tch_job_deployment_size\x18\x03 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12M\n\x1b\x61\x64vanced_deployment_options\x18\x04 \x01(\x0b\x32(.qwak.batchjob.AdvancedDeploymentOptions\x12\x34\n\nparameters\x18\x05 \x03(\x0b\x32 .qwak.batchjob.BatchJobParameter\"/\n\x11\x42\x61tchJobParameter\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"S\n\x15StartBatchJobResponse\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\")\n\x15\x43\x61ncelBatchJobRequest\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\"B\n\x16\x43\x61ncelBatchJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\",\n\x18GetBatchJobStatusRequest\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\"\xac\x01\n\x19GetBatchJobStatusResponse\x12\x38\n\njob_status\x18\x01 \x01(\x0e\x32$.qwak.batchjob.BatchJobStatusMessage\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\x12\x16\n\x0e\x66inished_files\x18\x04 \x01(\x05\x12\x13\n\x0btotal_files\x18\x05 \x01(\x05\",\n\x18GetBatchJobReportRequest\x12\x10\n\x08\x62\x61tch_id\x18\x01 \x01(\t\"a\n\x19GetBatchJobReportResponse\x12\x12\n\nsuccessful\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\x12\x17\n\x0freport_messages\x18\x03 \x03(\t\"`\n\x1dUpdateBatchTasksStatusRequest\x12?\n\x11\x62\x61tch_task_events\x18\x01 \x03(\x0b\x32$.qwak.batchjob.BatchTaskEventMessage\"I\n\x1dUpdateBatchTaskStatusResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x02 \x01(\t\"z\n\x19\x41\x64vancedDeploymentOptions\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x01 \x01(\t\x12\x17\n\x0fpurchase_option\x18\x02 \x01(\t\x12\'\n\x1fservice_account_key_secret_name\x18\x03 \x01(\t\"\x85\x03\n\x16\x42\x61tchJobDeploymentSize\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\x12J\n\x0cmemory_units\x18\x04 \x01(\x0e\x32\x30.qwak.batchjob.BatchJobDeploymentSize.MemoryUnitB\x02\x18\x01\x12H\n\rgpu_resources\x18\x05 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResourcesB\x02\x18\x01\x12`\n\x1c\x63lient_pod_compute_resources\x18\x06 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\"/\n\nMemoryUnit\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x1a\x02\x18\x01\"\x99\x01\n\x15\x42\x61tchJobExecutionSize\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12S\n\x15pod_compute_resources\x18\x02 \x01(\x0b\x32\x34.qwak.user_application.common.v0.PodComputeResources\x12\x13\n\x0btemplate_id\x18\x03 \x01(\t\"Q\n\x14ListBatchJobsRequest\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\"K\n\x15ListBatchJobsResponse\x12\x32\n\nbatch_jobs\x18\x01 \x03(\x0b\x32\x1e.qwak.batchjob.BatchJobDetails\"\x81\x03\n\x0f\x42\x61tchJobDetails\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x0e\n\x06job_id\x18\x02 \x01(\t\x12\x38\n\njob_status\x18\x03 \x01(\x0e\x32$.qwak.batchjob.BatchJobStatusMessage\x12.\n\nstart_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x17\n\x0f\x66\x61ilure_message\x18\x06 \x01(\t\x12;\n\x08job_size\x18\x07 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSizeB\x02\x18\x01\x12\x46\n\x18\x62\x61tch_job_execution_size\x18\x08 \x01(\x0b\x32$.qwak.batchjob.BatchJobExecutionSize\x12\x16\n\x0e\x65nvironment_id\x18\t \x01(\t\"+\n\x19GetBatchJobDetailsRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"y\n\x1aGetBatchJobDetailsResponse\x12\x31\n\tbatch_job\x18\x01 \x01(\x0b\x32\x1e.qwak.batchjob.BatchJobMessage\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\"\x84\x04\n\x0f\x42\x61tchJobMessage\x12\x0e\n\x06job_id\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12\x15\n\tbranch_id\x18\x03 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x04 \x01(\t\x12.\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x38\n\njob_status\x18\x07 \x01(\x0e\x32$.qwak.batchjob.BatchJobStatusMessage\x12\x15\n\rtotal_records\x18\x08 \x01(\x05\x12\x37\n\x08job_size\x18\t \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12;\n\x0freport_messages\x18\n \x03(\x0b\x32\".qwak.batchjob.ExecutionReportLine\x12\x43\n\x16\x65xecution_file_details\x18\x0b \x01(\x0b\x32#.qwak.batchjob.ExecutionFileDetails\x12<\n\x0ftask_executions\x18\x0c \x03(\x0b\x32#.qwak.batchjob.TaskExecutionDetails\"M\n\x13\x45xecutionReportLine\x12(\n\x04time\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0c\n\x04line\x18\x02 \x01(\t\"\x95\x02\n\x14\x45xecutionFileDetails\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x02 \x01(\t\x12\x13\n\x0bsource_path\x18\x03 \x01(\t\x12\x18\n\x10\x64\x65stination_path\x18\x04 \x01(\t\x12\x35\n\x0finput_file_type\x18\x05 \x01(\x0e\x32\x1c.qwak.batchjob.InputFileType\x12\x37\n\x10output_file_type\x18\x06 \x01(\x0e\x32\x1d.qwak.batchjob.OutputFileType\x12\x16\n\x0e\x66inished_files\x18\x07 \x01(\x05\x12\x13\n\x0btotal_files\x18\x08 \x01(\x05\"\xfd\x01\n\x14TaskExecutionDetails\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x35\n\x06status\x18\x02 \x01(\x0e\x32%.qwak.batchjob.BatchTaskStatusMessage\x12.\n\nstart_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12,\n\x08\x65nd_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x16\n\x0enum_of_records\x18\x05 \x01(\x05\x12\x14\n\x08\x66ilename\x18\x06 \x01(\tB\x02\x18\x01\x12\x11\n\tfilenames\x18\x07 \x03(\t\"d\n$GetBatchJobPreSignedUploadUrlRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x17\n\x0fnumber_of_files\x18\x02 \x01(\x05\x12\x11\n\tfile_type\x18\x03 \x01(\t\"\x98\x01\n%GetBatchJobPreSignedUploadUrlResponse\x12\x12\n\ninput_path\x18\x01 \x01(\t\x12\x13\n\x0boutput_path\x18\x02 \x01(\t\x12\x0e\n\x06\x62ucket\x18\x03 \x01(\t\x12\x0c\n\x04urls\x18\x04 \x03(\t\x12\x0f\n\x07success\x18\x05 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x06 \x01(\t\"8\n&GetBatchJobPreSignedDownloadUrlRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"a\n\'GetBatchJobPreSignedDownloadUrlResponse\x12\x0c\n\x04urls\x18\x01 \x03(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x17\n\x0f\x66\x61ilure_message\x18\x03 \x01(\t\"3\n\x1fGetBatchJobUploadDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\xb4\x01\n GetBatchJobUploadDetailsResponse\x12\x12\n\ninput_path\x18\x01 \x01(\t\x12\x13\n\x0boutput_path\x18\x02 \x01(\t\x12\x0e\n\x06\x62ucket\x18\x03 \x01(\t\x12W\n\x0b\x63redentials\x18\x04 \x01(\x0b\x32\x42.qwak.administration.authenticated_user.v1.AwsTemporaryCredentials\"3\n!GetBatchJobDownloadDetailsRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\t\"\x9b\x01\n\"GetBatchJobDownloadDetailsResponse\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x0c\n\x04keys\x18\x02 \x03(\t\x12W\n\x0b\x63redentials\x18\x03 \x01(\x0b\x32\x42.qwak.administration.authenticated_user.v1.AwsTemporaryCredentials\"\xd7\x01\n\x1aUpdateDefaultParamsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x46\n\x17\x64\x65\x66\x61ult_deployment_size\x18\x03 \x01(\x0b\x32%.qwak.batchjob.BatchJobDeploymentSize\x12M\n\x1b\x61\x64vanced_deployment_options\x18\x04 \x01(\x0b\x32(.qwak.batchjob.AdvancedDeploymentOptions\"\x1d\n\x1bUpdateDefaultParamsResponse\".\n\x1a\x44\x65leteDefaultParamsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\x1d\n\x1b\x44\x65leteDefaultParamsResponse\"S\n&UpdateTaskIngestedRecordsAmountRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x18\n\x10ingested_records\x18\x02 \x01(\x03\")\n\'UpdateTaskIngestedRecordsAmountResponse\"S\n\x19UpdateTasksDetailsRequest\x12\x36\n\rtasks_details\x18\x01 \x03(\x0b\x32\x1f.qwak.batchjob.BatchTaskDetails\"\x1c\n\x1aUpdateTasksDetailsResponse\"a\n\x10\x42\x61tchTaskDetails\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12<\n\x13input_files_details\x18\x02 \x03(\x0b\x32\x1f.qwak.batchjob.InputFileDetails\" \n\x10InputFileDetails\x12\x0c\n\x04path\x18\x01 \x01(\t*\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*\x8d\x02\n\x15\x42\x61tchJobStatusMessage\x12\x1e\n\x1aUNDEFINED_BATCH_JOB_STATUS\x10\x00\x12\x1e\n\x1a\x42\x41TCH_JOB_COMMITTED_STATUS\x10\x01\x12\x1c\n\x18\x42\x41TCH_JOB_PENDING_STATUS\x10\x02\x12\x1c\n\x18\x42\x41TCH_JOB_RUNNING_STATUS\x10\x03\x12\x1d\n\x19\x42\x41TCH_JOB_FINISHED_STATUS\x10\x04\x12\x1b\n\x17\x42\x41TCH_JOB_FAILED_STATUS\x10\x05\x12\x1e\n\x1a\x42\x41TCH_JOB_CANCELLED_STATUS\x10\x06\x12\x1c\n\x18\x42\x41TCH_JOB_TIMEOUT_STATUS\x10\x07*\x9c\x02\n\x16\x42\x61tchTaskStatusMessage\x12\x1f\n\x1bUNDEFINED_BATCH_TASK_STATUS\x10\x00\x12\x1f\n\x1b\x42\x41TCH_TASK_COMMITTED_STATUS\x10\x01\x12#\n\x1f\x42\x41TCH_TASK_PENDING_START_STATUS\x10\x02\x12\x1d\n\x19\x42\x41TCH_TASK_RUNNING_STATUS\x10\x03\x12\x1e\n\x1a\x42\x41TCH_TASK_FINISHED_STATUS\x10\x04\x12\x1c\n\x18\x42\x41TCH_TASK_FAILED_STATUS\x10\x05\x12\x1f\n\x1b\x42\x41TCH_TASK_CANCELLED_STATUS\x10\x06\x12\x1d\n\x19\x42\x41TCH_TASK_TIMEOUT_STATUS\x10\x07\x32\x9f\x0f\n\x19\x42\x61tchJobManagementService\x12Z\n\rStartBatchJob\x12#.qwak.batchjob.StartBatchJobRequest\x1a$.qwak.batchjob.StartBatchJobResponse\x12]\n\x0e\x43\x61ncelBatchJob\x12$.qwak.batchjob.CancelBatchJobRequest\x1a%.qwak.batchjob.CancelBatchJobResponse\x12]\n\x0eStartWarmupJob\x12$.qwak.batchjob.StartWarmupJobRequest\x1a%.qwak.batchjob.StartWarmupJobResponse\x12`\n\x0f\x43\x61ncelWarmupJob\x12%.qwak.batchjob.CancelWarmupJobRequest\x1a&.qwak.batchjob.CancelWarmupJobResponse\x12\x66\n\x11GetBatchJobStatus\x12\'.qwak.batchjob.GetBatchJobStatusRequest\x1a(.qwak.batchjob.GetBatchJobStatusResponse\x12\x66\n\x11GetBatchJobReport\x12\'.qwak.batchjob.GetBatchJobReportRequest\x1a(.qwak.batchjob.GetBatchJobReportResponse\x12t\n\x16UpdateBatchTasksStatus\x12,.qwak.batchjob.UpdateBatchTasksStatusRequest\x1a,.qwak.batchjob.UpdateBatchTaskStatusResponse\x12Z\n\rListBatchJobs\x12#.qwak.batchjob.ListBatchJobsRequest\x1a$.qwak.batchjob.ListBatchJobsResponse\x12i\n\x12GetBatchJobDetails\x12(.qwak.batchjob.GetBatchJobDetailsRequest\x1a).qwak.batchjob.GetBatchJobDetailsResponse\x12\x8a\x01\n\x1dGetBatchJobPreSignedUploadUrl\x12\x33.qwak.batchjob.GetBatchJobPreSignedUploadUrlRequest\x1a\x34.qwak.batchjob.GetBatchJobPreSignedUploadUrlResponse\x12\x90\x01\n\x1fGetBatchJobPreSignedDownloadUrl\x12\x35.qwak.batchjob.GetBatchJobPreSignedDownloadUrlRequest\x1a\x36.qwak.batchjob.GetBatchJobPreSignedDownloadUrlResponse\x12{\n\x18GetBatchJobUploadDetails\x12..qwak.batchjob.GetBatchJobUploadDetailsRequest\x1a/.qwak.batchjob.GetBatchJobUploadDetailsResponse\x12\x81\x01\n\x1aGetBatchJobDownloadDetails\x12\x30.qwak.batchjob.GetBatchJobDownloadDetailsRequest\x1a\x31.qwak.batchjob.GetBatchJobDownloadDetailsResponse\x12l\n\x13UpdateDefaultParams\x12).qwak.batchjob.UpdateDefaultParamsRequest\x1a*.qwak.batchjob.UpdateDefaultParamsResponse\x12l\n\x13\x44\x65leteDefaultParams\x12).qwak.batchjob.DeleteDefaultParamsRequest\x1a*.qwak.batchjob.DeleteDefaultParamsResponse\x12\x90\x01\n\x1fUpdateTaskIngestedRecordsAmount\x12\x35.qwak.batchjob.UpdateTaskIngestedRecordsAmountRequest\x1a\x36.qwak.batchjob.UpdateTaskIngestedRecordsAmountResponse\x12i\n\x12UpdateTasksDetails\x12(.qwak.batchjob.UpdateTasksDetailsRequest\x1a).qwak.batchjob.UpdateTasksDetailsResponseB.\n\x19\x63om.qwak.ai.batch.job.apiP\x01Z\x0f./;inferencejobb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -50,14 +50,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _BATCHJOBMESSAGE.fields_by_name['branch_id']._serialized_options = b'\030\001'
   _TASKEXECUTIONDETAILS.fields_by_name['filename']._options = None
   _TASKEXECUTIONDETAILS.fields_by_name['filename']._serialized_options = b'\030\001'
-  _globals['_INPUTFILETYPE']._serialized_start=7215
-  _globals['_INPUTFILETYPE']._serialized_end=7344
-  _globals['_OUTPUTFILETYPE']._serialized_start=7347
-  _globals['_OUTPUTFILETYPE']._serialized_end=7481
-  _globals['_BATCHJOBSTATUSMESSAGE']._serialized_start=7484
-  _globals['_BATCHJOBSTATUSMESSAGE']._serialized_end=7753
-  _globals['_BATCHTASKSTATUSMESSAGE']._serialized_start=7756
-  _globals['_BATCHTASKSTATUSMESSAGE']._serialized_end=8040
+  _globals['_INPUTFILETYPE']._serialized_start=7238
+  _globals['_INPUTFILETYPE']._serialized_end=7367
+  _globals['_OUTPUTFILETYPE']._serialized_start=7370
+  _globals['_OUTPUTFILETYPE']._serialized_end=7504
+  _globals['_BATCHJOBSTATUSMESSAGE']._serialized_start=7507
+  _globals['_BATCHJOBSTATUSMESSAGE']._serialized_end=7776
+  _globals['_BATCHTASKSTATUSMESSAGE']._serialized_start=7779
+  _globals['_BATCHTASKSTATUSMESSAGE']._serialized_end=8063
   _globals['_STARTWARMUPJOBREQUEST']._serialized_start=246
   _globals['_STARTWARMUPJOBREQUEST']._serialized_end=426
   _globals['_STARTWARMUPJOBRESPONSE']._serialized_start=428
@@ -73,95 +73,95 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_BATCHJOBMODELDETAILS']._serialized_start=1508
   _globals['_BATCHJOBMODELDETAILS']._serialized_end=1566
   _globals['_BATCHJOBDATADETAILS']._serialized_start=1569
-  _globals['_BATCHJOBDATADETAILS']._serialized_end=1798
-  _globals['_BATCHJOBSOURCEPATH']._serialized_start=1800
-  _globals['_BATCHJOBSOURCEPATH']._serialized_end=1921
-  _globals['_BATCHJOBDESTINATIONPATH']._serialized_start=1924
-  _globals['_BATCHJOBDESTINATIONPATH']._serialized_end=2062
-  _globals['_BATCHJOBEXECUTIONDETAILS']._serialized_start=2065
-  _globals['_BATCHJOBEXECUTIONDETAILS']._serialized_end=2341
-  _globals['_BATCHJOBPARAMETER']._serialized_start=2343
-  _globals['_BATCHJOBPARAMETER']._serialized_end=2390
-  _globals['_STARTBATCHJOBRESPONSE']._serialized_start=2392
-  _globals['_STARTBATCHJOBRESPONSE']._serialized_end=2475
-  _globals['_CANCELBATCHJOBREQUEST']._serialized_start=2477
-  _globals['_CANCELBATCHJOBREQUEST']._serialized_end=2518
-  _globals['_CANCELBATCHJOBRESPONSE']._serialized_start=2520
-  _globals['_CANCELBATCHJOBRESPONSE']._serialized_end=2586
-  _globals['_GETBATCHJOBSTATUSREQUEST']._serialized_start=2588
-  _globals['_GETBATCHJOBSTATUSREQUEST']._serialized_end=2632
-  _globals['_GETBATCHJOBSTATUSRESPONSE']._serialized_start=2635
-  _globals['_GETBATCHJOBSTATUSRESPONSE']._serialized_end=2807
-  _globals['_GETBATCHJOBREPORTREQUEST']._serialized_start=2809
-  _globals['_GETBATCHJOBREPORTREQUEST']._serialized_end=2853
-  _globals['_GETBATCHJOBREPORTRESPONSE']._serialized_start=2855
-  _globals['_GETBATCHJOBREPORTRESPONSE']._serialized_end=2952
-  _globals['_UPDATEBATCHTASKSSTATUSREQUEST']._serialized_start=2954
-  _globals['_UPDATEBATCHTASKSSTATUSREQUEST']._serialized_end=3050
-  _globals['_UPDATEBATCHTASKSTATUSRESPONSE']._serialized_start=3052
-  _globals['_UPDATEBATCHTASKSTATUSRESPONSE']._serialized_end=3125
-  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_start=3127
-  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_end=3249
-  _globals['_BATCHJOBDEPLOYMENTSIZE']._serialized_start=3252
-  _globals['_BATCHJOBDEPLOYMENTSIZE']._serialized_end=3641
-  _globals['_BATCHJOBDEPLOYMENTSIZE_MEMORYUNIT']._serialized_start=3594
-  _globals['_BATCHJOBDEPLOYMENTSIZE_MEMORYUNIT']._serialized_end=3641
-  _globals['_BATCHJOBEXECUTIONSIZE']._serialized_start=3644
-  _globals['_BATCHJOBEXECUTIONSIZE']._serialized_end=3797
-  _globals['_LISTBATCHJOBSREQUEST']._serialized_start=3799
-  _globals['_LISTBATCHJOBSREQUEST']._serialized_end=3880
-  _globals['_LISTBATCHJOBSRESPONSE']._serialized_start=3882
-  _globals['_LISTBATCHJOBSRESPONSE']._serialized_end=3957
-  _globals['_BATCHJOBDETAILS']._serialized_start=3960
-  _globals['_BATCHJOBDETAILS']._serialized_end=4345
-  _globals['_GETBATCHJOBDETAILSREQUEST']._serialized_start=4347
-  _globals['_GETBATCHJOBDETAILSREQUEST']._serialized_end=4390
-  _globals['_GETBATCHJOBDETAILSRESPONSE']._serialized_start=4392
-  _globals['_GETBATCHJOBDETAILSRESPONSE']._serialized_end=4513
-  _globals['_BATCHJOBMESSAGE']._serialized_start=4516
-  _globals['_BATCHJOBMESSAGE']._serialized_end=5032
-  _globals['_EXECUTIONREPORTLINE']._serialized_start=5034
-  _globals['_EXECUTIONREPORTLINE']._serialized_end=5111
-  _globals['_EXECUTIONFILEDETAILS']._serialized_start=5114
-  _globals['_EXECUTIONFILEDETAILS']._serialized_end=5391
-  _globals['_TASKEXECUTIONDETAILS']._serialized_start=5394
-  _globals['_TASKEXECUTIONDETAILS']._serialized_end=5647
-  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLREQUEST']._serialized_start=5649
-  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLREQUEST']._serialized_end=5749
-  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLRESPONSE']._serialized_start=5752
-  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLRESPONSE']._serialized_end=5904
-  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLREQUEST']._serialized_start=5906
-  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLREQUEST']._serialized_end=5962
-  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLRESPONSE']._serialized_start=5964
-  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLRESPONSE']._serialized_end=6061
-  _globals['_GETBATCHJOBUPLOADDETAILSREQUEST']._serialized_start=6063
-  _globals['_GETBATCHJOBUPLOADDETAILSREQUEST']._serialized_end=6114
-  _globals['_GETBATCHJOBUPLOADDETAILSRESPONSE']._serialized_start=6117
-  _globals['_GETBATCHJOBUPLOADDETAILSRESPONSE']._serialized_end=6297
-  _globals['_GETBATCHJOBDOWNLOADDETAILSREQUEST']._serialized_start=6299
-  _globals['_GETBATCHJOBDOWNLOADDETAILSREQUEST']._serialized_end=6350
-  _globals['_GETBATCHJOBDOWNLOADDETAILSRESPONSE']._serialized_start=6353
-  _globals['_GETBATCHJOBDOWNLOADDETAILSRESPONSE']._serialized_end=6508
-  _globals['_UPDATEDEFAULTPARAMSREQUEST']._serialized_start=6511
-  _globals['_UPDATEDEFAULTPARAMSREQUEST']._serialized_end=6726
-  _globals['_UPDATEDEFAULTPARAMSRESPONSE']._serialized_start=6728
-  _globals['_UPDATEDEFAULTPARAMSRESPONSE']._serialized_end=6757
-  _globals['_DELETEDEFAULTPARAMSREQUEST']._serialized_start=6759
-  _globals['_DELETEDEFAULTPARAMSREQUEST']._serialized_end=6805
-  _globals['_DELETEDEFAULTPARAMSRESPONSE']._serialized_start=6807
-  _globals['_DELETEDEFAULTPARAMSRESPONSE']._serialized_end=6836
-  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTREQUEST']._serialized_start=6838
-  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTREQUEST']._serialized_end=6921
-  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTRESPONSE']._serialized_start=6923
-  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTRESPONSE']._serialized_end=6964
-  _globals['_UPDATETASKSDETAILSREQUEST']._serialized_start=6966
-  _globals['_UPDATETASKSDETAILSREQUEST']._serialized_end=7049
-  _globals['_UPDATETASKSDETAILSRESPONSE']._serialized_start=7051
-  _globals['_UPDATETASKSDETAILSRESPONSE']._serialized_end=7079
-  _globals['_BATCHTASKDETAILS']._serialized_start=7081
-  _globals['_BATCHTASKDETAILS']._serialized_end=7178
-  _globals['_INPUTFILEDETAILS']._serialized_start=7180
-  _globals['_INPUTFILEDETAILS']._serialized_end=7212
-  _globals['_BATCHJOBMANAGEMENTSERVICE']._serialized_start=8043
-  _globals['_BATCHJOBMANAGEMENTSERVICE']._serialized_end=9994
+  _globals['_BATCHJOBDATADETAILS']._serialized_end=1821
+  _globals['_BATCHJOBSOURCEPATH']._serialized_start=1823
+  _globals['_BATCHJOBSOURCEPATH']._serialized_end=1944
+  _globals['_BATCHJOBDESTINATIONPATH']._serialized_start=1947
+  _globals['_BATCHJOBDESTINATIONPATH']._serialized_end=2085
+  _globals['_BATCHJOBEXECUTIONDETAILS']._serialized_start=2088
+  _globals['_BATCHJOBEXECUTIONDETAILS']._serialized_end=2364
+  _globals['_BATCHJOBPARAMETER']._serialized_start=2366
+  _globals['_BATCHJOBPARAMETER']._serialized_end=2413
+  _globals['_STARTBATCHJOBRESPONSE']._serialized_start=2415
+  _globals['_STARTBATCHJOBRESPONSE']._serialized_end=2498
+  _globals['_CANCELBATCHJOBREQUEST']._serialized_start=2500
+  _globals['_CANCELBATCHJOBREQUEST']._serialized_end=2541
+  _globals['_CANCELBATCHJOBRESPONSE']._serialized_start=2543
+  _globals['_CANCELBATCHJOBRESPONSE']._serialized_end=2609
+  _globals['_GETBATCHJOBSTATUSREQUEST']._serialized_start=2611
+  _globals['_GETBATCHJOBSTATUSREQUEST']._serialized_end=2655
+  _globals['_GETBATCHJOBSTATUSRESPONSE']._serialized_start=2658
+  _globals['_GETBATCHJOBSTATUSRESPONSE']._serialized_end=2830
+  _globals['_GETBATCHJOBREPORTREQUEST']._serialized_start=2832
+  _globals['_GETBATCHJOBREPORTREQUEST']._serialized_end=2876
+  _globals['_GETBATCHJOBREPORTRESPONSE']._serialized_start=2878
+  _globals['_GETBATCHJOBREPORTRESPONSE']._serialized_end=2975
+  _globals['_UPDATEBATCHTASKSSTATUSREQUEST']._serialized_start=2977
+  _globals['_UPDATEBATCHTASKSSTATUSREQUEST']._serialized_end=3073
+  _globals['_UPDATEBATCHTASKSTATUSRESPONSE']._serialized_start=3075
+  _globals['_UPDATEBATCHTASKSTATUSRESPONSE']._serialized_end=3148
+  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_start=3150
+  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_end=3272
+  _globals['_BATCHJOBDEPLOYMENTSIZE']._serialized_start=3275
+  _globals['_BATCHJOBDEPLOYMENTSIZE']._serialized_end=3664
+  _globals['_BATCHJOBDEPLOYMENTSIZE_MEMORYUNIT']._serialized_start=3617
+  _globals['_BATCHJOBDEPLOYMENTSIZE_MEMORYUNIT']._serialized_end=3664
+  _globals['_BATCHJOBEXECUTIONSIZE']._serialized_start=3667
+  _globals['_BATCHJOBEXECUTIONSIZE']._serialized_end=3820
+  _globals['_LISTBATCHJOBSREQUEST']._serialized_start=3822
+  _globals['_LISTBATCHJOBSREQUEST']._serialized_end=3903
+  _globals['_LISTBATCHJOBSRESPONSE']._serialized_start=3905
+  _globals['_LISTBATCHJOBSRESPONSE']._serialized_end=3980
+  _globals['_BATCHJOBDETAILS']._serialized_start=3983
+  _globals['_BATCHJOBDETAILS']._serialized_end=4368
+  _globals['_GETBATCHJOBDETAILSREQUEST']._serialized_start=4370
+  _globals['_GETBATCHJOBDETAILSREQUEST']._serialized_end=4413
+  _globals['_GETBATCHJOBDETAILSRESPONSE']._serialized_start=4415
+  _globals['_GETBATCHJOBDETAILSRESPONSE']._serialized_end=4536
+  _globals['_BATCHJOBMESSAGE']._serialized_start=4539
+  _globals['_BATCHJOBMESSAGE']._serialized_end=5055
+  _globals['_EXECUTIONREPORTLINE']._serialized_start=5057
+  _globals['_EXECUTIONREPORTLINE']._serialized_end=5134
+  _globals['_EXECUTIONFILEDETAILS']._serialized_start=5137
+  _globals['_EXECUTIONFILEDETAILS']._serialized_end=5414
+  _globals['_TASKEXECUTIONDETAILS']._serialized_start=5417
+  _globals['_TASKEXECUTIONDETAILS']._serialized_end=5670
+  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLREQUEST']._serialized_start=5672
+  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLREQUEST']._serialized_end=5772
+  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLRESPONSE']._serialized_start=5775
+  _globals['_GETBATCHJOBPRESIGNEDUPLOADURLRESPONSE']._serialized_end=5927
+  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLREQUEST']._serialized_start=5929
+  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLREQUEST']._serialized_end=5985
+  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLRESPONSE']._serialized_start=5987
+  _globals['_GETBATCHJOBPRESIGNEDDOWNLOADURLRESPONSE']._serialized_end=6084
+  _globals['_GETBATCHJOBUPLOADDETAILSREQUEST']._serialized_start=6086
+  _globals['_GETBATCHJOBUPLOADDETAILSREQUEST']._serialized_end=6137
+  _globals['_GETBATCHJOBUPLOADDETAILSRESPONSE']._serialized_start=6140
+  _globals['_GETBATCHJOBUPLOADDETAILSRESPONSE']._serialized_end=6320
+  _globals['_GETBATCHJOBDOWNLOADDETAILSREQUEST']._serialized_start=6322
+  _globals['_GETBATCHJOBDOWNLOADDETAILSREQUEST']._serialized_end=6373
+  _globals['_GETBATCHJOBDOWNLOADDETAILSRESPONSE']._serialized_start=6376
+  _globals['_GETBATCHJOBDOWNLOADDETAILSRESPONSE']._serialized_end=6531
+  _globals['_UPDATEDEFAULTPARAMSREQUEST']._serialized_start=6534
+  _globals['_UPDATEDEFAULTPARAMSREQUEST']._serialized_end=6749
+  _globals['_UPDATEDEFAULTPARAMSRESPONSE']._serialized_start=6751
+  _globals['_UPDATEDEFAULTPARAMSRESPONSE']._serialized_end=6780
+  _globals['_DELETEDEFAULTPARAMSREQUEST']._serialized_start=6782
+  _globals['_DELETEDEFAULTPARAMSREQUEST']._serialized_end=6828
+  _globals['_DELETEDEFAULTPARAMSRESPONSE']._serialized_start=6830
+  _globals['_DELETEDEFAULTPARAMSRESPONSE']._serialized_end=6859
+  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTREQUEST']._serialized_start=6861
+  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTREQUEST']._serialized_end=6944
+  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTRESPONSE']._serialized_start=6946
+  _globals['_UPDATETASKINGESTEDRECORDSAMOUNTRESPONSE']._serialized_end=6987
+  _globals['_UPDATETASKSDETAILSREQUEST']._serialized_start=6989
+  _globals['_UPDATETASKSDETAILSREQUEST']._serialized_end=7072
+  _globals['_UPDATETASKSDETAILSRESPONSE']._serialized_start=7074
+  _globals['_UPDATETASKSDETAILSRESPONSE']._serialized_end=7102
+  _globals['_BATCHTASKDETAILS']._serialized_start=7104
+  _globals['_BATCHTASKDETAILS']._serialized_end=7201
+  _globals['_INPUTFILEDETAILS']._serialized_start=7203
+  _globals['_INPUTFILEDETAILS']._serialized_end=7235
+  _globals['_BATCHJOBMANAGEMENTSERVICE']._serialized_start=8066
+  _globals['_BATCHJOBMANAGEMENTSERVICE']._serialized_end=10017
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/batch_job/v1/batch_job_service_pb2.pyi

@@ -352,6 +352,7 @@ class BatchJobDataDetails(google.protobuf.message.Message):
     DESTINATION_PATH_FIELD_NUMBER: builtins.int
     TOKEN_SECRET_FIELD_NUMBER: builtins.int
     SECRET_SECRET_FIELD_NUMBER: builtins.int
+    SESSION_TOKEN_FIELD_NUMBER: builtins.int
     SERVICE_ACCOUNT_JSON_KEY_SECRET_FIELD_NUMBER: builtins.int
     @property
     def source_path(self) -> global___BatchJobSourcePath:
@@ -363,6 +364,8 @@ class BatchJobDataDetails(google.protobuf.message.Message):
     """The access token's secret name in secret service"""
     secret_secret: builtins.str
     """The access secret's secret name in secret service"""
+    session_token: builtins.str
+    """The session token required to use the bucket for batch inference."""
     service_account_json_key_secret: builtins.str
     """The service account json key secret name"""
     def __init__(
@@ -372,10 +375,11 @@ class BatchJobDataDetails(google.protobuf.message.Message):
         destination_path: global___BatchJobDestinationPath | None = ...,
         token_secret: builtins.str = ...,
         secret_secret: builtins.str = ...,
+        session_token: builtins.str = ...,
         service_account_json_key_secret: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["destination_path", b"destination_path", "source_path", b"source_path"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["destination_path", b"destination_path", "secret_secret", b"secret_secret", "service_account_json_key_secret", b"service_account_json_key_secret", "source_path", b"source_path", "token_secret", b"token_secret"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["destination_path", b"destination_path", "secret_secret", b"secret_secret", "service_account_json_key_secret", b"service_account_json_key_secret", "session_token", b"session_token", "source_path", b"source_path", "token_secret", b"token_secret"]) -> None: ...
 
 global___BatchJobDataDetails = BatchJobDataDetails
 

_qwak_proto/qwak/builds/build_pb2.py

@@ -13,9 +13,10 @@ _sym_db = _symbol_database.Default()
 
 from _qwak_proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
 from _qwak_proto.qwak.fitness_service import fitness_pb2 as qwak_dot_fitness__service_dot_fitness__pb2
+from _qwak_proto.qwak.builds import build_values_pb2 as qwak_dot_builds_dot_build__values__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17qwak/builds/build.proto\x12\x18qwak.builds.orchestrator\x1a/qwak/user_application/common/v0/resources.proto\x1a\"qwak/fitness_service/fitness.proto\"\xbf\x06\n\x0fRemoteBuildSpec\x12\x43\n\x10\x62uild_properties\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.BuildProperties\x12\x35\n\tbuild_env\x18\x02 \x01(\x0b\x32\".qwak.builds.orchestrator.BuildEnv\x12\x43\n\rcpu_resources\x18\x03 \x01(\x0b\x32&.qwak.builds.orchestrator.CpuResourcesB\x02\x18\x01H\x00\x12\x43\n\rgpu_resources\x18\x07 \x01(\x0b\x32&.qwak.builds.orchestrator.GpuResourcesB\x02\x18\x01H\x00\x12\x62\n\x1c\x63lient_pod_compute_resources\x18\r \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResourcesH\x00\x12\x0f\n\x07verbose\x18\x04 \x01(\x05\x12\x17\n\x0f\x62uild_code_path\x18\x05 \x01(\t\x12\x14\n\x0c\x62uild_config\x18\x06 \x01(\t\x12\x15\n\rbuild_v1_flag\x18\x08 \x01(\x08\x12H\n\x13\x62uild_properties_v1\x18\t \x01(\x0b\x32+.qwak.builds.orchestrator.BuildPropertiesV1\x12\x13\n\x0b\x62uild_steps\x18\n \x03(\t\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x13\n\x0bsdk_version\x18\x0c \x01(\t\x12=\n\x0fpurchase_option\x18\x0e \x01(\x0e\x32$.qwak.fitness.service.PurchaseOption\x12\x1c\n\x14\x62uild_destined_image\x18\x0f \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x10 \x01(\x05\x12\x16\n\x0ejfrog_token_id\x18\x11 \x01(\t\x12\x1f\n\x17\x61rtifactory_project_key\x18\x12 \x01(\t\x12\x18\n\x10model_version_id\x18\x13 \x01(\tB\x0b\n\tResources\"g\n\x11\x42uildPropertiesV1\x12\x18\n\x10\x62uild_config_url\x18\x01 \x01(\t\x12\x1a\n\x12qwak_sdk_wheel_url\x18\x02 \x01(\t\x12\x1c\n\x14qwak_sdk_version_url\x18\x03 \x01(\t\"\xba\x01\n\x0f\x42uildProperties\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x0e\n\x06\x62ranch\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12\x39\n\tmodel_uri\x18\x05 \x01(\x0b\x32&.qwak.builds.orchestrator.ModelUriSpec\x12\x16\n\x0egpu_compatible\x18\x06 \x01(\x08\x12\x12\n\nbuild_name\x18\x07 \x01(\t\"\xb2\x01\n\x0cModelUriSpec\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x12\n\ngit_branch\x18\x02 \x01(\t\x12\x17\n\x0fgit_credentials\x18\x03 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x04 \x01(\t\x12\x11\n\tcommit_id\x18\x05 \x01(\t\x12\x10\n\x08main_dir\x18\x06 \x01(\t\x12#\n\x1b\x64\x65pendency_required_folders\x18\x07 \x03(\t\"|\n\x08\x42uildEnv\x12\x37\n\ndocker_env\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.DockerEnv\x12\x37\n\npython_env\x18\x02 \x01(\x0b\x32#.qwak.builds.orchestrator.PythonEnv\"\x8a\x01\n\tDockerEnv\x12\x12\n\nbase_image\x18\x01 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\x02 \x03(\t\x12\x10\n\x08no_cache\x18\x03 \x01(\x08\x12\x1c\n\x14\x61ssumed_iam_role_arn\x18\x04 \x01(\t\x12\'\n\x1fservice_account_key_secret_name\x18\x05 \x01(\t\"\xb8\x01\n\tPythonEnv\x12\x17\n\x0fgit_credentials\x18\x01 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x02 \x01(\t\x12\x18\n\x10qwak_sdk_version\x18\x03 \x01(\t\x12\x1c\n\x14qwak_sdk_extra_index\x18\x04 \x01(\t\x12\x1a\n\x12\x66rogml_cli_version\x18\x05 \x01(\t\x12\x1e\n\x16\x66rogml_cli_extra_index\x18\x06 \x01(\t\"n\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12:\n\x0cmemory_units\x18\x03 \x01(\x0e\x32$.qwak.builds.orchestrator.MemoryUnit\"W\n\x0cGpuResources\x12\x33\n\x08gpu_type\x18\x01 \x01(\x0e\x32!.qwak.builds.orchestrator.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"\x91\x01\n\x13\x44\x61taTableDefinition\x12?\n\x07\x63olumns\x18\x01 \x03(\x0b\x32..qwak.builds.orchestrator.DataColumnDefinition\x12\x39\n\x0b\x64\x61ta_format\x18\x02 \x01(\x0b\x32$.qwak.builds.orchestrator.DataFormat\"\\\n\x14\x44\x61taColumnDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x36\n\x04type\x18\x02 \x01(\x0e\x32(.qwak.builds.orchestrator.DataColumnType\"J\n\nDataFormat\x12\x32\n\x03\x63sv\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.CsvFormatH\x00\x42\x08\n\x06\x66ormat\"G\n\tCsvFormat\x12\x11\n\tdelimiter\x18\x01 \x01(\t\x12\x12\n\nquote_char\x18\x02 \x01(\t\x12\x13\n\x0b\x65scape_char\x18\x03 \x01(\t\"\xa5\x01\n\x0e\x42uildInitiator\x12<\n\x04user\x18\x01 \x01(\x0b\x32,.qwak.builds.orchestrator.UserBuildInitiatorH\x00\x12H\n\nautomation\x18\x02 \x01(\x0b\x32\x32.qwak.builds.orchestrator.AutomationBuildInitiatorH\x00\x42\x0b\n\tInitiator\"\x14\n\x12UserBuildInitiator\"`\n\x18\x41utomationBuildInitiator\x12\x15\n\rautomation_id\x18\x01 \x01(\t\x12\x14\n\x0c\x65xecution_id\x18\x02 \x01(\t\x12\x17\n\x0f\x61utomation_name\x18\x03 \x01(\t*7\n\nMemoryUnit\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x8c\x02\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\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*d\n\x0e\x44\x61taColumnType\x12\x17\n\x13INVALID_COLUMN_TYPE\x10\x00\x12\n\n\x06OBJECT\x10\x01\x12\x07\n\x03INT\x10\x02\x12\t\n\x05\x46LOAT\x10\x03\x12\x0b\n\x07\x42OOLEAN\x10\x04\x12\x0c\n\x08\x44\x41TETIME\x10\x05*K\n\x13\x42\x61seDockerImageType\x12\"\n\x1eUNKNOWN_BASE_DOCKER_IMAGE_TYPE\x10\x00\x12\x07\n\x03\x43PU\x10\x01\x12\x07\n\x03GPU\x10\x02\x42\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17qwak/builds/build.proto\x12\x18qwak.builds.orchestrator\x1a/qwak/user_application/common/v0/resources.proto\x1a\"qwak/fitness_service/fitness.proto\x1a\x1eqwak/builds/build_values.proto\"\xbf\x06\n\x0fRemoteBuildSpec\x12\x43\n\x10\x62uild_properties\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.BuildProperties\x12\x35\n\tbuild_env\x18\x02 \x01(\x0b\x32\".qwak.builds.orchestrator.BuildEnv\x12\x43\n\rcpu_resources\x18\x03 \x01(\x0b\x32&.qwak.builds.orchestrator.CpuResourcesB\x02\x18\x01H\x00\x12\x43\n\rgpu_resources\x18\x07 \x01(\x0b\x32&.qwak.builds.orchestrator.GpuResourcesB\x02\x18\x01H\x00\x12\x62\n\x1c\x63lient_pod_compute_resources\x18\r \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResourcesH\x00\x12\x0f\n\x07verbose\x18\x04 \x01(\x05\x12\x17\n\x0f\x62uild_code_path\x18\x05 \x01(\t\x12\x14\n\x0c\x62uild_config\x18\x06 \x01(\t\x12\x15\n\rbuild_v1_flag\x18\x08 \x01(\x08\x12H\n\x13\x62uild_properties_v1\x18\t \x01(\x0b\x32+.qwak.builds.orchestrator.BuildPropertiesV1\x12\x13\n\x0b\x62uild_steps\x18\n \x03(\t\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x13\n\x0bsdk_version\x18\x0c \x01(\t\x12=\n\x0fpurchase_option\x18\x0e \x01(\x0e\x32$.qwak.fitness.service.PurchaseOption\x12\x1c\n\x14\x62uild_destined_image\x18\x0f \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x10 \x01(\x05\x12\x16\n\x0ejfrog_token_id\x18\x11 \x01(\t\x12\x1f\n\x17\x61rtifactory_project_key\x18\x12 \x01(\t\x12\x18\n\x10model_version_id\x18\x13 \x01(\tB\x0b\n\tResources\"g\n\x11\x42uildPropertiesV1\x12\x18\n\x10\x62uild_config_url\x18\x01 \x01(\t\x12\x1a\n\x12qwak_sdk_wheel_url\x18\x02 \x01(\t\x12\x1c\n\x14qwak_sdk_version_url\x18\x03 \x01(\t\"\xba\x01\n\x0f\x42uildProperties\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x0e\n\x06\x62ranch\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12\x39\n\tmodel_uri\x18\x05 \x01(\x0b\x32&.qwak.builds.orchestrator.ModelUriSpec\x12\x16\n\x0egpu_compatible\x18\x06 \x01(\x08\x12\x12\n\nbuild_name\x18\x07 \x01(\t\"\xb2\x01\n\x0cModelUriSpec\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x12\n\ngit_branch\x18\x02 \x01(\t\x12\x17\n\x0fgit_credentials\x18\x03 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x04 \x01(\t\x12\x11\n\tcommit_id\x18\x05 \x01(\t\x12\x10\n\x08main_dir\x18\x06 \x01(\t\x12#\n\x1b\x64\x65pendency_required_folders\x18\x07 \x03(\t\"|\n\x08\x42uildEnv\x12\x37\n\ndocker_env\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.DockerEnv\x12\x37\n\npython_env\x18\x02 \x01(\x0b\x32#.qwak.builds.orchestrator.PythonEnv\"\xcc\x01\n\tDockerEnv\x12\x12\n\nbase_image\x18\x01 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\x02 \x03(\t\x12\x10\n\x08no_cache\x18\x03 \x01(\x08\x12\x1c\n\x14\x61ssumed_iam_role_arn\x18\x04 \x01(\t\x12\'\n\x1fservice_account_key_secret_name\x18\x05 \x01(\t\x12@\n\x10\x63ontainer_engine\x18\x06 \x01(\x0b\x32&.qwak.build.management.ContainerEngine\"\xb8\x01\n\tPythonEnv\x12\x17\n\x0fgit_credentials\x18\x01 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x02 \x01(\t\x12\x18\n\x10qwak_sdk_version\x18\x03 \x01(\t\x12\x1c\n\x14qwak_sdk_extra_index\x18\x04 \x01(\t\x12\x1a\n\x12\x66rogml_cli_version\x18\x05 \x01(\t\x12\x1e\n\x16\x66rogml_cli_extra_index\x18\x06 \x01(\t\"n\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12:\n\x0cmemory_units\x18\x03 \x01(\x0e\x32$.qwak.builds.orchestrator.MemoryUnit\"W\n\x0cGpuResources\x12\x33\n\x08gpu_type\x18\x01 \x01(\x0e\x32!.qwak.builds.orchestrator.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"\x91\x01\n\x13\x44\x61taTableDefinition\x12?\n\x07\x63olumns\x18\x01 \x03(\x0b\x32..qwak.builds.orchestrator.DataColumnDefinition\x12\x39\n\x0b\x64\x61ta_format\x18\x02 \x01(\x0b\x32$.qwak.builds.orchestrator.DataFormat\"\\\n\x14\x44\x61taColumnDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x36\n\x04type\x18\x02 \x01(\x0e\x32(.qwak.builds.orchestrator.DataColumnType\"J\n\nDataFormat\x12\x32\n\x03\x63sv\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.CsvFormatH\x00\x42\x08\n\x06\x66ormat\"G\n\tCsvFormat\x12\x11\n\tdelimiter\x18\x01 \x01(\t\x12\x12\n\nquote_char\x18\x02 \x01(\t\x12\x13\n\x0b\x65scape_char\x18\x03 \x01(\t\"\xa5\x01\n\x0e\x42uildInitiator\x12<\n\x04user\x18\x01 \x01(\x0b\x32,.qwak.builds.orchestrator.UserBuildInitiatorH\x00\x12H\n\nautomation\x18\x02 \x01(\x0b\x32\x32.qwak.builds.orchestrator.AutomationBuildInitiatorH\x00\x42\x0b\n\tInitiator\"\x14\n\x12UserBuildInitiator\"`\n\x18\x41utomationBuildInitiator\x12\x15\n\rautomation_id\x18\x01 \x01(\t\x12\x14\n\x0c\x65xecution_id\x18\x02 \x01(\t\x12\x17\n\x0f\x61utomation_name\x18\x03 \x01(\t*7\n\nMemoryUnit\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x82\x03\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\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\x12\x13\n\x0fNVIDIA_H100_NVL\x10\r\x12\x0e\n\nNVIDIA_A10\x10\x0e\x12\x13\n\x0fNVIDIA_H100_SXM\x10\x0f\x12\x14\n\x10NVIDIA_H100_40GB\x10\x10\x12\x14\n\x10NVIDIA_H100_80GB\x10\x11\x12\x0e\n\nNVIDIA_M60\x10\x12*d\n\x0e\x44\x61taColumnType\x12\x17\n\x13INVALID_COLUMN_TYPE\x10\x00\x12\n\n\x06OBJECT\x10\x01\x12\x07\n\x03INT\x10\x02\x12\t\n\x05\x46LOAT\x10\x03\x12\x0b\n\x07\x42OOLEAN\x10\x04\x12\x0c\n\x08\x44\x41TETIME\x10\x05*K\n\x13\x42\x61seDockerImageType\x12\"\n\x1eUNKNOWN_BASE_DOCKER_IMAGE_TYPE\x10\x00\x12\x07\n\x03\x43PU\x10\x01\x12\x07\n\x03GPU\x10\x02\x42\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -28,44 +29,44 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _REMOTEBUILDSPEC.fields_by_name['cpu_resources']._serialized_options = b'\030\001'
   _REMOTEBUILDSPEC.fields_by_name['gpu_resources']._options = None
   _REMOTEBUILDSPEC.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
-  _globals['_MEMORYUNIT']._serialized_start=2781
-  _globals['_MEMORYUNIT']._serialized_end=2836
-  _globals['_GPUTYPE']._serialized_start=2839
-  _globals['_GPUTYPE']._serialized_end=3107
-  _globals['_DATACOLUMNTYPE']._serialized_start=3109
-  _globals['_DATACOLUMNTYPE']._serialized_end=3209
-  _globals['_BASEDOCKERIMAGETYPE']._serialized_start=3211
-  _globals['_BASEDOCKERIMAGETYPE']._serialized_end=3286
-  _globals['_REMOTEBUILDSPEC']._serialized_start=139
-  _globals['_REMOTEBUILDSPEC']._serialized_end=970
-  _globals['_BUILDPROPERTIESV1']._serialized_start=972
-  _globals['_BUILDPROPERTIESV1']._serialized_end=1075
-  _globals['_BUILDPROPERTIES']._serialized_start=1078
-  _globals['_BUILDPROPERTIES']._serialized_end=1264
-  _globals['_MODELURISPEC']._serialized_start=1267
-  _globals['_MODELURISPEC']._serialized_end=1445
-  _globals['_BUILDENV']._serialized_start=1447
-  _globals['_BUILDENV']._serialized_end=1571
-  _globals['_DOCKERENV']._serialized_start=1574
-  _globals['_DOCKERENV']._serialized_end=1712
-  _globals['_PYTHONENV']._serialized_start=1715
-  _globals['_PYTHONENV']._serialized_end=1899
-  _globals['_CPURESOURCES']._serialized_start=1901
-  _globals['_CPURESOURCES']._serialized_end=2011
-  _globals['_GPURESOURCES']._serialized_start=2013
-  _globals['_GPURESOURCES']._serialized_end=2100
-  _globals['_DATATABLEDEFINITION']._serialized_start=2103
-  _globals['_DATATABLEDEFINITION']._serialized_end=2248
-  _globals['_DATACOLUMNDEFINITION']._serialized_start=2250
-  _globals['_DATACOLUMNDEFINITION']._serialized_end=2342
-  _globals['_DATAFORMAT']._serialized_start=2344
-  _globals['_DATAFORMAT']._serialized_end=2418
-  _globals['_CSVFORMAT']._serialized_start=2420
-  _globals['_CSVFORMAT']._serialized_end=2491
-  _globals['_BUILDINITIATOR']._serialized_start=2494
-  _globals['_BUILDINITIATOR']._serialized_end=2659
-  _globals['_USERBUILDINITIATOR']._serialized_start=2661
-  _globals['_USERBUILDINITIATOR']._serialized_end=2681
-  _globals['_AUTOMATIONBUILDINITIATOR']._serialized_start=2683
-  _globals['_AUTOMATIONBUILDINITIATOR']._serialized_end=2779
+  _globals['_MEMORYUNIT']._serialized_start=2879
+  _globals['_MEMORYUNIT']._serialized_end=2934
+  _globals['_GPUTYPE']._serialized_start=2937
+  _globals['_GPUTYPE']._serialized_end=3323
+  _globals['_DATACOLUMNTYPE']._serialized_start=3325
+  _globals['_DATACOLUMNTYPE']._serialized_end=3425
+  _globals['_BASEDOCKERIMAGETYPE']._serialized_start=3427
+  _globals['_BASEDOCKERIMAGETYPE']._serialized_end=3502
+  _globals['_REMOTEBUILDSPEC']._serialized_start=171
+  _globals['_REMOTEBUILDSPEC']._serialized_end=1002
+  _globals['_BUILDPROPERTIESV1']._serialized_start=1004
+  _globals['_BUILDPROPERTIESV1']._serialized_end=1107
+  _globals['_BUILDPROPERTIES']._serialized_start=1110
+  _globals['_BUILDPROPERTIES']._serialized_end=1296
+  _globals['_MODELURISPEC']._serialized_start=1299
+  _globals['_MODELURISPEC']._serialized_end=1477
+  _globals['_BUILDENV']._serialized_start=1479
+  _globals['_BUILDENV']._serialized_end=1603
+  _globals['_DOCKERENV']._serialized_start=1606
+  _globals['_DOCKERENV']._serialized_end=1810
+  _globals['_PYTHONENV']._serialized_start=1813
+  _globals['_PYTHONENV']._serialized_end=1997
+  _globals['_CPURESOURCES']._serialized_start=1999
+  _globals['_CPURESOURCES']._serialized_end=2109
+  _globals['_GPURESOURCES']._serialized_start=2111
+  _globals['_GPURESOURCES']._serialized_end=2198
+  _globals['_DATATABLEDEFINITION']._serialized_start=2201
+  _globals['_DATATABLEDEFINITION']._serialized_end=2346
+  _globals['_DATACOLUMNDEFINITION']._serialized_start=2348
+  _globals['_DATACOLUMNDEFINITION']._serialized_end=2440
+  _globals['_DATAFORMAT']._serialized_start=2442
+  _globals['_DATAFORMAT']._serialized_end=2516
+  _globals['_CSVFORMAT']._serialized_start=2518
+  _globals['_CSVFORMAT']._serialized_end=2589
+  _globals['_BUILDINITIATOR']._serialized_start=2592
+  _globals['_BUILDINITIATOR']._serialized_end=2757
+  _globals['_USERBUILDINITIATOR']._serialized_start=2759
+  _globals['_USERBUILDINITIATOR']._serialized_end=2779
+  _globals['_AUTOMATIONBUILDINITIATOR']._serialized_start=2781
+  _globals['_AUTOMATIONBUILDINITIATOR']._serialized_end=2877
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/builds/build_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 qwak.builds.build_values_pb2
 import qwak.fitness_service.fitness_pb2
 import qwak.user_application.common.v0.resources_pb2
 import sys
@@ -64,6 +65,18 @@ class _GpuTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTy
     """a100.xl"""
     NVIDIA_V100_4_32_208: _GpuType.ValueType  # 12
     """v100.4xl"""
+    NVIDIA_H100_NVL: _GpuType.ValueType  # 13
+    """H100 NVL XLarge"""
+    NVIDIA_A10: _GpuType.ValueType  # 14
+    """A10 Small"""
+    NVIDIA_H100_SXM: _GpuType.ValueType  # 15
+    """H100 SXM 8XLarge"""
+    NVIDIA_H100_40GB: _GpuType.ValueType  # 16
+    """H100 40GB 8XLarge"""
+    NVIDIA_H100_80GB: _GpuType.ValueType  # 17
+    """H100 80GB 8XLarge"""
+    NVIDIA_M60: _GpuType.ValueType  # 18
+    """M60 XLarge"""
 
 class GpuType(_GpuType, metaclass=_GpuTypeEnumTypeWrapper): ...
 
@@ -88,6 +101,18 @@ NVIDIA_V100_1_8_52: GpuType.ValueType  # 11
 """a100.xl"""
 NVIDIA_V100_4_32_208: GpuType.ValueType  # 12
 """v100.4xl"""
+NVIDIA_H100_NVL: GpuType.ValueType  # 13
+"""H100 NVL XLarge"""
+NVIDIA_A10: GpuType.ValueType  # 14
+"""A10 Small"""
+NVIDIA_H100_SXM: GpuType.ValueType  # 15
+"""H100 SXM 8XLarge"""
+NVIDIA_H100_40GB: GpuType.ValueType  # 16
+"""H100 40GB 8XLarge"""
+NVIDIA_H100_80GB: GpuType.ValueType  # 17
+"""H100 80GB 8XLarge"""
+NVIDIA_M60: GpuType.ValueType  # 18
+"""M60 XLarge"""
 global___GpuType = GpuType
 
 class _DataColumnType:
@@ -356,6 +381,7 @@ class DockerEnv(google.protobuf.message.Message):
     NO_CACHE_FIELD_NUMBER: builtins.int
     ASSUMED_IAM_ROLE_ARN_FIELD_NUMBER: builtins.int
     SERVICE_ACCOUNT_KEY_SECRET_NAME_FIELD_NUMBER: builtins.int
+    CONTAINER_ENGINE_FIELD_NUMBER: builtins.int
     base_image: builtins.str
     """Base image to use for serving"""
     @property
@@ -369,6 +395,9 @@ class DockerEnv(google.protobuf.message.Message):
     """Custom IAM role to be assumed by model"""
     service_account_key_secret_name: builtins.str
     """Service account key secret name for gcp"""
+    @property
+    def container_engine(self) -> qwak.builds.build_values_pb2.ContainerEngine:
+        """Container engine to use for building and pushing the image (e.g., Docker, Podman)"""
     def __init__(
         self,
         *,
@@ -377,8 +406,10 @@ class DockerEnv(google.protobuf.message.Message):
         no_cache: builtins.bool = ...,
         assumed_iam_role_arn: builtins.str = ...,
         service_account_key_secret_name: builtins.str = ...,
+        container_engine: qwak.builds.build_values_pb2.ContainerEngine | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["assumed_iam_role_arn", b"assumed_iam_role_arn", "base_image", b"base_image", "env_vars", b"env_vars", "no_cache", b"no_cache", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["container_engine", b"container_engine"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["assumed_iam_role_arn", b"assumed_iam_role_arn", "base_image", b"base_image", "container_engine", b"container_engine", "env_vars", b"env_vars", "no_cache", b"no_cache", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
 
 global___DockerEnv = DockerEnv
 

_qwak_proto/qwak/builds/build_values_pb2.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/builds/build_values.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+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 _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\x1eqwak/builds/build_values.proto\x12\x15qwak.build.management\x1a/qwak/user_application/common/v0/resources.proto\"\xc3\x03\n\x10ModelBuildValues\x12\x34\n\nbuild_info\x18\x01 \x01(\x0b\x32 .qwak.build.management.BuildInfo\x12<\n\x0emodel_metadata\x18\x02 \x01(\x0b\x32$.qwak.build.management.ModelMetadata\x12\x45\n\x13model_code_location\x18\x03 \x01(\x0b\x32(.qwak.build.management.ModelCodeLocation\x12G\n\x14\x62uild_execution_spec\x18\x04 \x01(\x0b\x32).qwak.build.management.BuildExecutionSpec\x12\x30\n\x08sdk_spec\x18\x05 \x01(\x0b\x32\x1e.qwak.build.management.SdkSpec\x12G\n\x14jfrog_ecosystem_spec\x18\x06 \x01(\x0b\x32).qwak.build.management.JfrogEcosystemSpec\x12\x30\n\x08job_spec\x18\x07 \x01(\x0b\x32\x1e.qwak.build.management.JobSpec\"\xa3\x01\n\tBuildInfo\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x12\n\nbuild_tags\x18\x03 \x03(\t\x12\x14\n\x0c\x62uild_config\x18\x04 \x01(\t\x12\x45\n\x13\x62uild_properties_v1\x18\x05 \x01(\x0b\x32(.qwak.build.management.BuildPropertiesV1\"]\n\x11\x42uildPropertiesV1\x12\x18\n\x10\x62uild_config_url\x18\x01 \x01(\t\x12\x15\n\rsdk_wheel_url\x18\x02 \x01(\t\x12\x17\n\x0fsdk_version_url\x18\x03 \x01(\t\"\x8e\x01\n\rModelMetadata\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\naccount_id\x18\x02 \x01(\t\x12\x0f\n\x07user_id\x18\x03 \x01(\t\x12\x46\n\x13\x61rtifactory_details\x18\x04 \x01(\x0b\x32).qwak.build.management.ArtifactoryDetails\")\n\x12\x41rtifactoryDetails\x12\x13\n\x0bproject_key\x18\x01 \x01(\t\"\xa0\x01\n\x11ModelCodeLocation\x12\x1c\n\x14main_model_directory\x18\x01 \x01(\t\x12\x34\n\tlocal_uri\x18\x02 \x01(\x0b\x32\x1f.qwak.build.management.LocalUriH\x00\x12\x30\n\x07git_uri\x18\x03 \x01(\x0b\x32\x1d.qwak.build.management.GitUriH\x00\x42\x05\n\x03uri\"\x18\n\x08LocalUri\x12\x0c\n\x04path\x18\x01 \x01(\t\"M\n\x06GitUri\x12\x0b\n\x03url\x18\x01 \x01(\t\x12\x36\n\x0bgit_details\x18\x02 \x01(\x0b\x32!.qwak.build.management.GitDetails\"|\n\nGitDetails\x12\x0e\n\x06\x62ranch\x18\x01 \x01(\t\x12\x11\n\tcommit_id\x18\x02 \x01(\t\x12\x15\n\x0b\x63redentials\x18\x03 \x01(\tH\x00\x12!\n\x17\x63redentials_secret_name\x18\x04 \x01(\tH\x00\x42\x11\n\x0fgit_credentials\"\xd0\x02\n\x12\x42uildExecutionSpec\x12\x1f\n\x17\x62uild_image_destination\x18\x01 \x01(\t\x12\"\n\x1ashould_build_without_cache\x18\x02 \x01(\x08\x12G\n\x14logs_verbosity_level\x18\x03 \x01(\x0e\x32).qwak.build.management.LogsVerbosityLevel\x12\x12\n\nbase_image\x18\x04 \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x05 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\x06 \x03(\t\x12\'\n\x1fremote_builder_auth_secret_name\x18\x07 \x01(\t\x12@\n\x10\x63ontainer_engine\x18\x08 \x01(\x0b\x32&.qwak.build.management.ContainerEngine\"a\n\x07SdkSpec\x12\x13\n\x0bsdk_version\x18\x01 \x01(\t\x12\x1b\n\x13sdk_extra_index_url\x18\x02 \x01(\t\x12$\n\x1csdk_runtime_version_override\x18\x03 \x01(\t\"k\n\x12JfrogEcosystemSpec\x12\x0f\n\x07\x62\x61seUrl\x18\x01 \x01(\t\x12\x1d\n\x15\x61\x63\x63\x65ssTokenSecretName\x18\x02 \x01(\t\x12%\n\x1d\x61rtifactorySettingsSecretName\x18\x03 \x01(\t\"\xb5\x04\n\x07JobSpec\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x02 \x01(\x05\x12>\n\x0fpurchase_option\x18\x03 \x01(\x0e\x32%.qwak.build.management.PurchaseOption\x12$\n\x1cis_node_optimization_enabled\x18\x04 \x01(\x08\x12\"\n\x1ashould_use_build_isolation\x18\x05 \x01(\x08\x12#\n\x1bis_multi_tenant_environment\x18\x06 \x01(\x08\x12\x1a\n\x12is_self_hosted_jpd\x18\n \x01(\x08\x12 \n\x18\x62uild_executor_image_url\x18\x07 \x01(\t\x12<\n\x0e\x63loud_provider\x18\x08 \x01(\x0e\x32$.qwak.build.management.CloudProvider\x12\x30\n\x08job_size\x18\t \x01(\x0b\x32\x1e.qwak.build.management.JobSize\x12Y\n\x17\x62uild_executor_env_vars\x18\x0b \x03(\x0b\x32\x38.qwak.build.management.JobSpec.BuildExecutorEnvVarsEntry\x1a;\n\x19\x42uildExecutorEnvVarsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x9e\x02\n\x07JobSize\x12\x0f\n\x03\x63pu\x18\x01 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x02 \x01(\x05\x42\x02\x18\x01\x12;\n\x0cmemory_units\x18\x03 \x01(\x0e\x32!.qwak.build.management.MemoryUnitB\x02\x18\x01\x12H\n\rgpu_resources\x18\x04 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResourcesB\x02\x18\x01\x12`\n\x1c\x63lient_pod_compute_resources\x18\x05 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\"\x9d\x01\n\x0f\x43ontainerEngine\x12>\n\x06\x64ocker\x18\x01 \x01(\x0b\x32,.qwak.build.management.DockerContainerEngineH\x00\x12\x42\n\x06podman\x18\x02 \x01(\x0b\x32,.qwak.build.management.PodmanContainerEngineB\x02\x18\x01H\x00\x42\x06\n\x04type\"\x17\n\x15\x44ockerContainerEngine\"\x17\n\x15PodmanContainerEngine*e\n\x0ePurchaseOption\x12\x1b\n\x17INVALID_PURCHASE_OPTION\x10\x00\x12\x18\n\x14SPOT_PURCHASE_OPTION\x10\x01\x12\x1c\n\x18ONDEMAND_PURCHASE_OPTION\x10\x02*H\n\rCloudProvider\x12\x1a\n\x16INVALID_CLOUD_PROVIDER\x10\x00\x12\x07\n\x03\x41WS\x10\x01\x12\x07\n\x03GCP\x10\x02\x12\t\n\x05\x41ZURE\x10\x03*7\n\nMemoryUnit\x12\x17\n\x13INVALID_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*S\n\x12LogsVerbosityLevel\x12\x1a\n\x16INVALID_LOGS_VERBOSITY\x10\x00\x12\t\n\x05\x44\x45\x42UG\x10\n\x12\x08\n\x04INFO\x10\x14\x12\x0c\n\x08\x43RITICAL\x10\x32\x42\x36\n com.qwak.ai.management.build.apiP\x01Z\x10qwak/build;buildb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.builds.build_values_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n com.qwak.ai.management.build.apiP\001Z\020qwak/build;build'
+  _JOBSPEC_BUILDEXECUTORENVVARSENTRY._options = None
+  _JOBSPEC_BUILDEXECUTORENVVARSENTRY._serialized_options = b'8\001'
+  _JOBSIZE.fields_by_name['cpu']._options = None
+  _JOBSIZE.fields_by_name['cpu']._serialized_options = b'\030\001'
+  _JOBSIZE.fields_by_name['memory_amount']._options = None
+  _JOBSIZE.fields_by_name['memory_amount']._serialized_options = b'\030\001'
+  _JOBSIZE.fields_by_name['memory_units']._options = None
+  _JOBSIZE.fields_by_name['memory_units']._serialized_options = b'\030\001'
+  _JOBSIZE.fields_by_name['gpu_resources']._options = None
+  _JOBSIZE.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
+  _CONTAINERENGINE.fields_by_name['podman']._options = None
+  _CONTAINERENGINE.fields_by_name['podman']._serialized_options = b'\030\001'
+  _globals['_PURCHASEOPTION']._serialized_start=3017
+  _globals['_PURCHASEOPTION']._serialized_end=3118
+  _globals['_CLOUDPROVIDER']._serialized_start=3120
+  _globals['_CLOUDPROVIDER']._serialized_end=3192
+  _globals['_MEMORYUNIT']._serialized_start=3194
+  _globals['_MEMORYUNIT']._serialized_end=3249
+  _globals['_LOGSVERBOSITYLEVEL']._serialized_start=3251
+  _globals['_LOGSVERBOSITYLEVEL']._serialized_end=3334
+  _globals['_MODELBUILDVALUES']._serialized_start=107
+  _globals['_MODELBUILDVALUES']._serialized_end=558
+  _globals['_BUILDINFO']._serialized_start=561
+  _globals['_BUILDINFO']._serialized_end=724
+  _globals['_BUILDPROPERTIESV1']._serialized_start=726
+  _globals['_BUILDPROPERTIESV1']._serialized_end=819
+  _globals['_MODELMETADATA']._serialized_start=822
+  _globals['_MODELMETADATA']._serialized_end=964
+  _globals['_ARTIFACTORYDETAILS']._serialized_start=966
+  _globals['_ARTIFACTORYDETAILS']._serialized_end=1007
+  _globals['_MODELCODELOCATION']._serialized_start=1010
+  _globals['_MODELCODELOCATION']._serialized_end=1170
+  _globals['_LOCALURI']._serialized_start=1172
+  _globals['_LOCALURI']._serialized_end=1196
+  _globals['_GITURI']._serialized_start=1198
+  _globals['_GITURI']._serialized_end=1275
+  _globals['_GITDETAILS']._serialized_start=1277
+  _globals['_GITDETAILS']._serialized_end=1401
+  _globals['_BUILDEXECUTIONSPEC']._serialized_start=1404
+  _globals['_BUILDEXECUTIONSPEC']._serialized_end=1740
+  _globals['_SDKSPEC']._serialized_start=1742
+  _globals['_SDKSPEC']._serialized_end=1839
+  _globals['_JFROGECOSYSTEMSPEC']._serialized_start=1841
+  _globals['_JFROGECOSYSTEMSPEC']._serialized_end=1948
+  _globals['_JOBSPEC']._serialized_start=1951
+  _globals['_JOBSPEC']._serialized_end=2516
+  _globals['_JOBSPEC_BUILDEXECUTORENVVARSENTRY']._serialized_start=2457
+  _globals['_JOBSPEC_BUILDEXECUTORENVVARSENTRY']._serialized_end=2516
+  _globals['_JOBSIZE']._serialized_start=2519
+  _globals['_JOBSIZE']._serialized_end=2805
+  _globals['_CONTAINERENGINE']._serialized_start=2808
+  _globals['_CONTAINERENGINE']._serialized_end=2965
+  _globals['_DOCKERCONTAINERENGINE']._serialized_start=2967
+  _globals['_DOCKERCONTAINERENGINE']._serialized_end=2990
+  _globals['_PODMANCONTAINERENGINE']._serialized_start=2992
+  _globals['_PODMANCONTAINERENGINE']._serialized_end=3015
+# @@protoc_insertion_point(module_scope)