frogml-core 0.0.8__py3-none-any.whl → 0.0.10__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
frogml_core/__init__.py CHANGED
@@ -1,7 +1,7 @@
1
1
  """Top-level package for frogml."""
2
2
 
3
3
  __author__ = "jfrog"
4
- __version__ = "0.0.8"
4
+ __version__ = "0.0.10"
5
5
 
6
6
  from frogml_core.inner.di_configuration import wire_dependencies
7
7
  from frogml_core.model.model_version_tracking import ( # noqa: F401,E501
@@ -31,7 +31,7 @@ class ModelVersionManagerClient:
31
31
  self.__model_version_manager_stub = ModelVersionManagerServiceStub(grpc_channel)
32
32
 
33
33
  @staticmethod
34
- def __build_create_model_request(
34
+ def __build_create_model_version_request(
35
35
  project_key: str,
36
36
  repository_key: str,
37
37
  model_name: str,
@@ -46,6 +46,7 @@ class ModelVersionManagerClient:
46
46
  project_key=project_key,
47
47
  repository_key=repository_key,
48
48
  model_id=get_model_id_from_model_name(model_name),
49
+ model_name=model_name,
49
50
  ),
50
51
  name=model_version_name,
51
52
  framework=model_version_framework,
@@ -62,7 +63,7 @@ class ModelVersionManagerClient:
62
63
  model_version_framework: ModelVersionFramework,
63
64
  ):
64
65
  try:
65
- create_model_request = self.__build_create_model_request(
66
+ create_model_request = self.__build_create_model_version_request(
66
67
  project_key=project_key,
67
68
  repository_key=repository_key,
68
69
  model_name=model_name,
@@ -88,7 +89,7 @@ class ModelVersionManagerClient:
88
89
  model_version_framework: ModelVersionFramework,
89
90
  ) -> CreateModelVersionResponse:
90
91
  try:
91
- create_model_request = self.__build_create_model_request(
92
+ create_model_request = self.__build_create_model_version_request(
92
93
  project_key=project_key,
93
94
  repository_key=repository_key,
94
95
  model_name=model_name,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: frogml-core
3
- Version: 0.0.8
3
+ Version: 0.0.10
4
4
  Summary: frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform
5
5
  License: Apache-2.0
6
6
  Keywords: mlops,ml,deployment,serving,model
@@ -1,4 +1,4 @@
1
- frogml_core/__init__.py,sha256=JhwoSbF47ceA8YU8-lN8O9gtjsC_Eymxflr02nb2ioI,776
1
+ frogml_core/__init__.py,sha256=ZZi0L9uOJdvdTb0bbtwJR6x9GBKHoVVClTDtRVhNz2s,777
2
2
  frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
3
3
  frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
4
4
  frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -73,7 +73,7 @@ frogml_core/clients/logging_client/client.py,sha256=A7qQJWW54Ve7O0DuJX9gmSJlUQyR
73
73
  frogml_core/clients/model_management/__init__.py,sha256=vjWVP8MjmK4_A70WOgJqa6x24AeLK-ABjGJtogGzw9w,43
74
74
  frogml_core/clients/model_management/client.py,sha256=rXnzcP3-pfXCVG0KlcCwbdeIJ1iFzV84JYtyNC8n-x8,4575
75
75
  frogml_core/clients/model_version_manager/__init__.py,sha256=4Pnfzj4Egps48__dProdbSKQl5pNip0hGJd75w67BfY,46
76
- frogml_core/clients/model_version_manager/client.py,sha256=-QN6ibQ_wdkS_pUqcD_WfjSoroeGgCMFEXOcqOdndVo,4147
76
+ frogml_core/clients/model_version_manager/client.py,sha256=6WIvzWY1IWuUINvKzHPs8klMKaRJuyIN3DDdSsAryCQ,4214
77
77
  frogml_core/clients/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
78
78
  frogml_core/clients/project/client.py,sha256=VvnUk7mo-UkE9Gkj9jVkCPGTVx8qmuekMz7oH9Uzv2Y,2159
79
79
  frogml_core/clients/prompt_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -434,8 +434,8 @@ frogml_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
434
434
  frogml_proto/jfml/model_version/v1/build_spec_pb2.py,sha256=7FeOfvGtH2v4swm0FGl8ALvUgR4ckIcn-pSkzYd-uhQ,3779
435
435
  frogml_proto/jfml/model_version/v1/build_spec_pb2.pyi,sha256=CIBEscG5MQoOGddvsfGqxD2K4U342IS-kFpzyYFiFD8,4923
436
436
  frogml_proto/jfml/model_version/v1/build_spec_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
437
- frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.py,sha256=llPIiXM7XrcYZMOb68gJNnzwMiE9UQDSalcAW8ztNlU,1656
438
- frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.pyi,sha256=UXjFPw49S9jVVx0WXOdXv8aQMj74RGpBii6Kw1vrlls,1470
437
+ frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.py,sha256=qCXNf6iuagFDKi0B9AUX2pz5GdFpS7VtM_EyUQFrmcY,1701
438
+ frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.pyi,sha256=2yF-pKLxTFsxArlXdHFuPt4GzXjln2b1aYAYZQafjg4,1610
439
439
  frogml_proto/jfml/model_version/v1/model_repository_spec_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
440
440
  frogml_proto/jfml/model_version/v1/model_version_framework_pb2.py,sha256=Vk4NO9dlhXkNVgxdLGDfh20s-mBfcoOOZEEr6hUtLag,5370
441
441
  frogml_proto/jfml/model_version/v1/model_version_framework_pb2.pyi,sha256=JgOE0Im69ir0PSgpbg6Zz4scRK_u7WAxWTtWcdzfyAw,3773
@@ -569,8 +569,8 @@ frogml_proto/qwak/build_settings/build_settings_api_pb2_grpc.py,sha256=t62VlkIu6
569
569
  frogml_proto/qwak/build_settings/build_settings_pb2.py,sha256=sjTRd3bI7b9lwMZgmD0-l5RSkJfOp-LOJMBVUr_ps5Y,6029
570
570
  frogml_proto/qwak/build_settings/build_settings_pb2.pyi,sha256=MMYamilVLTgHn-WmsIqRCI8K9QJSehtxAclPeP75CWA,7049
571
571
  frogml_proto/qwak/build_settings/build_settings_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
572
- frogml_proto/qwak/builds/build_pb2.py,sha256=ftaNyfIuweuBYR9ns1fPrHIqIRoT-Z_Pj8IneqcVZTY,15143
573
- frogml_proto/qwak/builds/build_pb2.pyi,sha256=p13IwNcgtm3roLFtxsZKGs_SI6qiyyu0rsEODrk6Khw,25148
572
+ frogml_proto/qwak/builds/build_pb2.py,sha256=e1fecTsRZgEWlBtRoFP93yMPp-kbui0xozBCRPwptLE,15189
573
+ frogml_proto/qwak/builds/build_pb2.pyi,sha256=YGGVicG6kMMQpYDT67WCoy4NJN5-iB1tLPnlIiDR4aw,25381
574
574
  frogml_proto/qwak/builds/build_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
575
575
  frogml_proto/qwak/builds/build_url_pb2.py,sha256=anKHqrZRH82uplaN_LvxRm3wIIaf9DDjh9AHQ1J1u70,5565
576
576
  frogml_proto/qwak/builds/build_url_pb2.pyi,sha256=IC0L3D7zg5vf_4U9Ubc-V4lLqviPknUaBKCCrVK6w9k,7299
@@ -1016,6 +1016,6 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
1016
1016
  frogml_services_mock/services_mock.py,sha256=_34z6rFCHFwcSni-9eyJlrH264xsL_QDNmG-EBv0zMc,20281
1017
1017
  frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1018
1018
  frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
1019
- frogml_core-0.0.8.dist-info/METADATA,sha256=il6AZcJlu5LVDij3EQQSeO-Yriah3rXnwXByh6PJyIc,2105
1020
- frogml_core-0.0.8.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
1021
- frogml_core-0.0.8.dist-info/RECORD,,
1019
+ frogml_core-0.0.10.dist-info/METADATA,sha256=pwrQzUIYdzqCri8WmWPmEMf9Wf0vltBnUX8U0SGRPdI,2106
1020
+ frogml_core-0.0.10.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
1021
+ frogml_core-0.0.10.dist-info/RECORD,,
@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
14
14
 
15
15
 
16
16
 
17
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1jfml/model_version/v1/model_repository_spec.proto\x12\x15jfml.model_version.v1\"l\n\x13ModelRepositorySpec\x12\x15\n\x0bproject_key\x18\x01 \x01(\tH\x00\x12\x16\n\x0erepository_key\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\tB\x14\n\x12project_identifierB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
17
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1jfml/model_version/v1/model_repository_spec.proto\x12\x15jfml.model_version.v1\"\x80\x01\n\x13ModelRepositorySpec\x12\x15\n\x0bproject_key\x18\x01 \x01(\tH\x00\x12\x16\n\x0erepository_key\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\t\x12\x12\n\nmodel_name\x18\x04 \x01(\tB\x14\n\x12project_identifierB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
18
18
 
19
19
 
20
20
 
@@ -30,6 +30,6 @@ if _descriptor._USE_C_DESCRIPTORS == False:
30
30
 
31
31
  DESCRIPTOR._options = None
32
32
  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
33
- _MODELREPOSITORYSPEC._serialized_start=76
34
- _MODELREPOSITORYSPEC._serialized_end=184
33
+ _MODELREPOSITORYSPEC._serialized_start=77
34
+ _MODELREPOSITORYSPEC._serialized_end=205
35
35
  # @@protoc_insertion_point(module_scope)
@@ -20,19 +20,22 @@ class ModelRepositorySpec(google.protobuf.message.Message):
20
20
  PROJECT_KEY_FIELD_NUMBER: builtins.int
21
21
  REPOSITORY_KEY_FIELD_NUMBER: builtins.int
22
22
  MODEL_ID_FIELD_NUMBER: builtins.int
23
+ MODEL_NAME_FIELD_NUMBER: builtins.int
23
24
  project_key: builtins.str
24
25
  """JFrog project key"""
25
26
  repository_key: builtins.str
26
27
  model_id: builtins.str
28
+ model_name: builtins.str
27
29
  def __init__(
28
30
  self,
29
31
  *,
30
32
  project_key: builtins.str = ...,
31
33
  repository_key: builtins.str = ...,
32
34
  model_id: builtins.str = ...,
35
+ model_name: builtins.str = ...,
33
36
  ) -> None: ...
34
37
  def HasField(self, field_name: typing_extensions.Literal["project_identifier", b"project_identifier", "project_key", b"project_key"]) -> builtins.bool: ...
35
- def ClearField(self, field_name: typing_extensions.Literal["model_id", b"model_id", "project_identifier", b"project_identifier", "project_key", b"project_key", "repository_key", b"repository_key"]) -> None: ...
38
+ def ClearField(self, field_name: typing_extensions.Literal["model_id", b"model_id", "model_name", b"model_name", "project_identifier", b"project_identifier", "project_key", b"project_key", "repository_key", b"repository_key"]) -> None: ...
36
39
  def WhichOneof(self, oneof_group: typing_extensions.Literal["project_identifier", b"project_identifier"]) -> typing_extensions.Literal["project_key"] | None: ...
37
40
 
38
41
  global___ModelRepositorySpec = ModelRepositorySpec
@@ -17,7 +17,7 @@ from frogml_proto.qwak.user_application.common.v0 import resources_pb2 as qwak_d
17
17
  from frogml_proto.qwak.fitness_service import fitness_pb2 as qwak_dot_fitness__service_dot_fitness__pb2
18
18
 
19
19
 
20
- 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\"\xa5\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(\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\"|\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\"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')
20
+ 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\"|\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\"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')
21
21
 
22
22
  _MEMORYUNIT = DESCRIPTOR.enum_types_by_name['MemoryUnit']
23
23
  MemoryUnit = enum_type_wrapper.EnumTypeWrapper(_MEMORYUNIT)
@@ -190,44 +190,44 @@ if _descriptor._USE_C_DESCRIPTORS == False:
190
190
  _REMOTEBUILDSPEC.fields_by_name['cpu_resources']._serialized_options = b'\030\001'
191
191
  _REMOTEBUILDSPEC.fields_by_name['gpu_resources']._options = None
192
192
  _REMOTEBUILDSPEC.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
193
- _MEMORYUNIT._serialized_start=2694
194
- _MEMORYUNIT._serialized_end=2749
195
- _GPUTYPE._serialized_start=2752
196
- _GPUTYPE._serialized_end=3020
197
- _DATACOLUMNTYPE._serialized_start=3022
198
- _DATACOLUMNTYPE._serialized_end=3122
199
- _BASEDOCKERIMAGETYPE._serialized_start=3124
200
- _BASEDOCKERIMAGETYPE._serialized_end=3199
193
+ _MEMORYUNIT._serialized_start=2720
194
+ _MEMORYUNIT._serialized_end=2775
195
+ _GPUTYPE._serialized_start=2778
196
+ _GPUTYPE._serialized_end=3046
197
+ _DATACOLUMNTYPE._serialized_start=3048
198
+ _DATACOLUMNTYPE._serialized_end=3148
199
+ _BASEDOCKERIMAGETYPE._serialized_start=3150
200
+ _BASEDOCKERIMAGETYPE._serialized_end=3225
201
201
  _REMOTEBUILDSPEC._serialized_start=139
202
- _REMOTEBUILDSPEC._serialized_end=944
203
- _BUILDPROPERTIESV1._serialized_start=946
204
- _BUILDPROPERTIESV1._serialized_end=1049
205
- _BUILDPROPERTIES._serialized_start=1052
206
- _BUILDPROPERTIES._serialized_end=1238
207
- _MODELURISPEC._serialized_start=1241
208
- _MODELURISPEC._serialized_end=1419
209
- _BUILDENV._serialized_start=1421
210
- _BUILDENV._serialized_end=1545
211
- _DOCKERENV._serialized_start=1548
212
- _DOCKERENV._serialized_end=1686
213
- _PYTHONENV._serialized_start=1688
214
- _PYTHONENV._serialized_end=1812
215
- _CPURESOURCES._serialized_start=1814
216
- _CPURESOURCES._serialized_end=1924
217
- _GPURESOURCES._serialized_start=1926
218
- _GPURESOURCES._serialized_end=2013
219
- _DATATABLEDEFINITION._serialized_start=2016
220
- _DATATABLEDEFINITION._serialized_end=2161
221
- _DATACOLUMNDEFINITION._serialized_start=2163
222
- _DATACOLUMNDEFINITION._serialized_end=2255
223
- _DATAFORMAT._serialized_start=2257
224
- _DATAFORMAT._serialized_end=2331
225
- _CSVFORMAT._serialized_start=2333
226
- _CSVFORMAT._serialized_end=2404
227
- _BUILDINITIATOR._serialized_start=2407
228
- _BUILDINITIATOR._serialized_end=2572
229
- _USERBUILDINITIATOR._serialized_start=2574
230
- _USERBUILDINITIATOR._serialized_end=2594
231
- _AUTOMATIONBUILDINITIATOR._serialized_start=2596
232
- _AUTOMATIONBUILDINITIATOR._serialized_end=2692
202
+ _REMOTEBUILDSPEC._serialized_end=970
203
+ _BUILDPROPERTIESV1._serialized_start=972
204
+ _BUILDPROPERTIESV1._serialized_end=1075
205
+ _BUILDPROPERTIES._serialized_start=1078
206
+ _BUILDPROPERTIES._serialized_end=1264
207
+ _MODELURISPEC._serialized_start=1267
208
+ _MODELURISPEC._serialized_end=1445
209
+ _BUILDENV._serialized_start=1447
210
+ _BUILDENV._serialized_end=1571
211
+ _DOCKERENV._serialized_start=1574
212
+ _DOCKERENV._serialized_end=1712
213
+ _PYTHONENV._serialized_start=1714
214
+ _PYTHONENV._serialized_end=1838
215
+ _CPURESOURCES._serialized_start=1840
216
+ _CPURESOURCES._serialized_end=1950
217
+ _GPURESOURCES._serialized_start=1952
218
+ _GPURESOURCES._serialized_end=2039
219
+ _DATATABLEDEFINITION._serialized_start=2042
220
+ _DATATABLEDEFINITION._serialized_end=2187
221
+ _DATACOLUMNDEFINITION._serialized_start=2189
222
+ _DATACOLUMNDEFINITION._serialized_end=2281
223
+ _DATAFORMAT._serialized_start=2283
224
+ _DATAFORMAT._serialized_end=2357
225
+ _CSVFORMAT._serialized_start=2359
226
+ _CSVFORMAT._serialized_end=2430
227
+ _BUILDINITIATOR._serialized_start=2433
228
+ _BUILDINITIATOR._serialized_end=2598
229
+ _USERBUILDINITIATOR._serialized_start=2600
230
+ _USERBUILDINITIATOR._serialized_end=2620
231
+ _AUTOMATIONBUILDINITIATOR._serialized_start=2622
232
+ _AUTOMATIONBUILDINITIATOR._serialized_end=2718
233
233
  # @@protoc_insertion_point(module_scope)
@@ -152,6 +152,7 @@ class RemoteBuildSpec(google.protobuf.message.Message):
152
152
  PROVISION_INSTANCE_TIMEOUT_FIELD_NUMBER: builtins.int
153
153
  JFROG_TOKEN_ID_FIELD_NUMBER: builtins.int
154
154
  ARTIFACTORY_PROJECT_KEY_FIELD_NUMBER: builtins.int
155
+ MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
155
156
  @property
156
157
  def build_properties(self) -> global___BuildProperties: ...
157
158
  @property
@@ -192,6 +193,8 @@ class RemoteBuildSpec(google.protobuf.message.Message):
192
193
  """JFrog token id for token generation"""
193
194
  artifactory_project_key: builtins.str
194
195
  """Artifactory project key"""
196
+ model_version_id: builtins.str
197
+ """Optional - The model version's id used for the build"""
195
198
  def __init__(
196
199
  self,
197
200
  *,
@@ -213,9 +216,10 @@ class RemoteBuildSpec(google.protobuf.message.Message):
213
216
  provision_instance_timeout: builtins.int = ...,
214
217
  jfrog_token_id: builtins.str = ...,
215
218
  artifactory_project_key: builtins.str = ...,
219
+ model_version_id: builtins.str = ...,
216
220
  ) -> None: ...
217
221
  def HasField(self, field_name: typing_extensions.Literal["Resources", b"Resources", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources"]) -> builtins.bool: ...
218
- def ClearField(self, field_name: typing_extensions.Literal["Resources", b"Resources", "artifactory_project_key", b"artifactory_project_key", "build_code_path", b"build_code_path", "build_config", b"build_config", "build_destined_image", b"build_destined_image", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "build_steps", b"build_steps", "build_v1_flag", b"build_v1_flag", "client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "environment_id", b"environment_id", "gpu_resources", b"gpu_resources", "jfrog_token_id", b"jfrog_token_id", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "sdk_version", b"sdk_version", "verbose", b"verbose"]) -> None: ...
222
+ def ClearField(self, field_name: typing_extensions.Literal["Resources", b"Resources", "artifactory_project_key", b"artifactory_project_key", "build_code_path", b"build_code_path", "build_config", b"build_config", "build_destined_image", b"build_destined_image", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "build_steps", b"build_steps", "build_v1_flag", b"build_v1_flag", "client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "environment_id", b"environment_id", "gpu_resources", b"gpu_resources", "jfrog_token_id", b"jfrog_token_id", "model_version_id", b"model_version_id", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "sdk_version", b"sdk_version", "verbose", b"verbose"]) -> None: ...
219
223
  def WhichOneof(self, oneof_group: typing_extensions.Literal["Resources", b"Resources"]) -> typing_extensions.Literal["cpu_resources", "gpu_resources", "client_pod_compute_resources"] | None: ...
220
224
 
221
225
  global___RemoteBuildSpec = RemoteBuildSpec