frogml-core 0.0.9__py3-none-any.whl → 0.0.11__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.9"
4
+ __version__ = "0.0.11"
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.9
3
+ Version: 0.0.11
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=vMKd3TebJRvynO4b1Jp3qf2uXhkOyHMyml-IuKMCYDc,776
1
+ frogml_core/__init__.py,sha256=KITB_jkjEatVYfNOLDBRgtpKoSR-7CpgUjW2y8p97Lk,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,15 +434,15 @@ 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
442
442
  frogml_proto/jfml/model_version/v1/model_version_framework_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
443
- frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py,sha256=-QakaXaKJLZX68AuiYqyBGNFEy2i6EralNxddhFQy7Y,12964
444
- frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi,sha256=LE4_HRvT_zDjubYt8cRgGrYqtHcAEh1TVSmA8VzsUFc,11863
445
- frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py,sha256=qx5iX5FK7k8ICpD-T13T4wGDX7dSyVyOPe_O8U6TXSg,15434
443
+ frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py,sha256=X0KJkIj8eg41SVWD48AuJSMw1P_1R5lfD4X_L_rSajs,13816
444
+ frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi,sha256=kLV9okhcSx5LKAwTBUknA3e0sotbkDMa_JnOwcUzrRc,12416
445
+ frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py,sha256=2v-qLZmVDKKE2J6c9Sz1WAXPLO9Ngtw9dFnehH7Kc5U,17508
446
446
  frogml_proto/jfml/model_version/v1/model_version_pb2.py,sha256=87CsbasbiwvPWGI1b6AfGrje_wzDCZFD8Hi7meguDjU,5797
447
447
  frogml_proto/jfml/model_version/v1/model_version_pb2.pyi,sha256=NqHepmIm1HiKBYn77JDFOoBAEVwCbyFsj0Wiurhsxkc,5664
448
448
  frogml_proto/jfml/model_version/v1/model_version_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -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.9.dist-info/METADATA,sha256=Lx9lwhhx8TY6hOE671-Z318a3XNkDsbKTR-e6ZuwFk0,2105
1020
- frogml_core-0.0.9.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
1021
- frogml_core-0.0.9.dist-info/RECORD,,
1019
+ frogml_core-0.0.11.dist-info/METADATA,sha256=zbNFPfeKgRHN1cs_QG8dykfYYBIfBv_pMB7SZ5-9nik,2106
1020
+ frogml_core-0.0.11.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
1021
+ frogml_core-0.0.11.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.jfml.model_version.v1 import model_repository_spec_pb2 as jfml
17
17
  from frogml_proto.jfml.model_version.v1 import build_spec_pb2 as jfml_dot_model__version_dot_v1_dot_build__spec__pb2
18
18
 
19
19
 
20
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n9jfml/model_version/v1/model_version_manager_service.proto\x12\x15jfml.model_version.v1\x1a)jfml/model_version/v1/model_version.proto\x1a\x31jfml/model_version/v1/model_repository_spec.proto\x1a&jfml/model_version/v1/build_spec.proto\"l\n\x19\x43reateModelVersionRequest\x12\x0f\n\x07\x64ry_run\x18\x01 \x01(\x08\x12>\n\rmodel_version\x18\x02 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpec\"X\n\x1a\x43reateModelVersionResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\"6\n\x1aGetModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\"Y\n\x1bGetModelVersionByIdResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\"L\n\x1cGetModelVersionByNameRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x1a\n\x12model_version_name\x18\x02 \x01(\t\"[\n\x1dGetModelVersionByNameResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\")\n\x15InitPaginationRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"E\n\x16InitPaginationResponse\x12\x14\n\x0crecord_count\x18\x01 \x01(\x03\x12\x15\n\rmax_record_id\x18\x02 \x01(\x03\"\x9e\x01\n\x18ListModelVersionsRequest\x12\x43\n\x0frepository_spec\x18\x01 \x01(\x0b\x32*.jfml.model_version.v1.ModelRepositorySpec\x12\x11\n\tpage_size\x18\x02 \x01(\x05\x12\x13\n\x0bpage_number\x18\x03 \x01(\x05\x12\x15\n\rmax_record_id\x18\x04 \x01(\x03\"X\n\x19ListModelVersionsResponse\x12;\n\x0emodel_versions\x18\x01 \x03(\x0b\x32#.jfml.model_version.v1.ModelVersion\"9\n\x1d\x44\x65leteModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\" \n\x1e\x44\x65leteModelVersionByIdResponse\"\xd8\x01\n!PromoteModelVersionToBuildRequest\x12\x45\n\x12model_version_spec\x18\x01 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpecH\x00\x12\x1a\n\x10model_version_id\x18\x02 \x01(\tH\x00\x12\x34\n\nbuild_spec\x18\x03 \x01(\x0b\x32 .jfml.model_version.v1.BuildSpecB\x1a\n\x18model_version_identifier\"6\n\"PromoteModelVersionToBuildResponse\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t2\x9d\x07\n\x1aModelVersionManagerService\x12y\n\x12\x43reateModelVersion\x12\x30.jfml.model_version.v1.CreateModelVersionRequest\x1a\x31.jfml.model_version.v1.CreateModelVersionResponse\x12|\n\x13GetModelVersionById\x12\x31.jfml.model_version.v1.GetModelVersionByIdRequest\x1a\x32.jfml.model_version.v1.GetModelVersionByIdResponse\x12\x82\x01\n\x15GetModelVersionByName\x12\x33.jfml.model_version.v1.GetModelVersionByNameRequest\x1a\x34.jfml.model_version.v1.GetModelVersionByNameResponse\x12m\n\x0eInitPagination\x12,.jfml.model_version.v1.InitPaginationRequest\x1a-.jfml.model_version.v1.InitPaginationResponse\x12v\n\x11ListModelVersions\x12/.jfml.model_version.v1.ListModelVersionsRequest\x1a\x30.jfml.model_version.v1.ListModelVersionsResponse\x12\x85\x01\n\x16\x44\x65leteModelVersionById\x12\x34.jfml.model_version.v1.DeleteModelVersionByIdRequest\x1a\x35.jfml.model_version.v1.DeleteModelVersionByIdResponse\x12\x91\x01\n\x1aPromoteModelVersionToBuild\x12\x38.jfml.model_version.v1.PromoteModelVersionToBuildRequest\x1a\x39.jfml.model_version.v1.PromoteModelVersionToBuildResponseB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
20
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n9jfml/model_version/v1/model_version_manager_service.proto\x12\x15jfml.model_version.v1\x1a)jfml/model_version/v1/model_version.proto\x1a\x31jfml/model_version/v1/model_repository_spec.proto\x1a&jfml/model_version/v1/build_spec.proto\"l\n\x19\x43reateModelVersionRequest\x12\x0f\n\x07\x64ry_run\x18\x01 \x01(\x08\x12>\n\rmodel_version\x18\x02 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpec\"X\n\x1a\x43reateModelVersionResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\"6\n\x1aGetModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\"Y\n\x1bGetModelVersionByIdResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\"L\n\x1cGetModelVersionByNameRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x1a\n\x12model_version_name\x18\x02 \x01(\t\"[\n\x1dGetModelVersionByNameResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\")\n\x15InitPaginationRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"E\n\x16InitPaginationResponse\x12\x14\n\x0crecord_count\x18\x01 \x01(\x03\x12\x15\n\rmax_record_id\x18\x02 \x01(\x03\"\x9e\x01\n\x18ListModelVersionsRequest\x12\x43\n\x0frepository_spec\x18\x01 \x01(\x0b\x32*.jfml.model_version.v1.ModelRepositorySpec\x12\x11\n\tpage_size\x18\x02 \x01(\x05\x12\x13\n\x0bpage_number\x18\x03 \x01(\x05\x12\x15\n\rmax_record_id\x18\x04 \x01(\x03\"X\n\x19ListModelVersionsResponse\x12;\n\x0emodel_versions\x18\x01 \x03(\x0b\x32#.jfml.model_version.v1.ModelVersion\"9\n\x1d\x44\x65leteModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\" \n\x1e\x44\x65leteModelVersionByIdResponse\"\xd8\x01\n!PromoteModelVersionToBuildRequest\x12\x45\n\x12model_version_spec\x18\x01 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpecH\x00\x12\x1a\n\x10model_version_id\x18\x02 \x01(\tH\x00\x12\x34\n\nbuild_spec\x18\x03 \x01(\x0b\x32 .jfml.model_version.v1.BuildSpecB\x1a\n\x18model_version_identifier\"6\n\"PromoteModelVersionToBuildResponse\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\";\n\x1cGetMLBOMModelVersionResponse\x12\x1b\n\x13mlbom_model_version\x18\x01 \x01(\t2\xa2\x08\n\x1aModelVersionManagerService\x12y\n\x12\x43reateModelVersion\x12\x30.jfml.model_version.v1.CreateModelVersionRequest\x1a\x31.jfml.model_version.v1.CreateModelVersionResponse\x12|\n\x13GetModelVersionById\x12\x31.jfml.model_version.v1.GetModelVersionByIdRequest\x1a\x32.jfml.model_version.v1.GetModelVersionByIdResponse\x12\x82\x01\n\x15GetModelVersionByName\x12\x33.jfml.model_version.v1.GetModelVersionByNameRequest\x1a\x34.jfml.model_version.v1.GetModelVersionByNameResponse\x12m\n\x0eInitPagination\x12,.jfml.model_version.v1.InitPaginationRequest\x1a-.jfml.model_version.v1.InitPaginationResponse\x12v\n\x11ListModelVersions\x12/.jfml.model_version.v1.ListModelVersionsRequest\x1a\x30.jfml.model_version.v1.ListModelVersionsResponse\x12\x85\x01\n\x16\x44\x65leteModelVersionById\x12\x34.jfml.model_version.v1.DeleteModelVersionByIdRequest\x1a\x35.jfml.model_version.v1.DeleteModelVersionByIdResponse\x12\x91\x01\n\x1aPromoteModelVersionToBuild\x12\x38.jfml.model_version.v1.PromoteModelVersionToBuildRequest\x1a\x39.jfml.model_version.v1.PromoteModelVersionToBuildResponse\x12\x82\x01\n\x18GetMlBomModelVersionById\x12\x31.jfml.model_version.v1.GetModelVersionByIdRequest\x1a\x33.jfml.model_version.v1.GetMLBOMModelVersionResponseB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
21
21
 
22
22
 
23
23
 
@@ -35,6 +35,7 @@ _DELETEMODELVERSIONBYIDREQUEST = DESCRIPTOR.message_types_by_name['DeleteModelVe
35
35
  _DELETEMODELVERSIONBYIDRESPONSE = DESCRIPTOR.message_types_by_name['DeleteModelVersionByIdResponse']
36
36
  _PROMOTEMODELVERSIONTOBUILDREQUEST = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildRequest']
37
37
  _PROMOTEMODELVERSIONTOBUILDRESPONSE = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildResponse']
38
+ _GETMLBOMMODELVERSIONRESPONSE = DESCRIPTOR.message_types_by_name['GetMLBOMModelVersionResponse']
38
39
  CreateModelVersionRequest = _reflection.GeneratedProtocolMessageType('CreateModelVersionRequest', (_message.Message,), {
39
40
  'DESCRIPTOR' : _CREATEMODELVERSIONREQUEST,
40
41
  '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
@@ -133,6 +134,13 @@ PromoteModelVersionToBuildResponse = _reflection.GeneratedProtocolMessageType('P
133
134
  })
134
135
  _sym_db.RegisterMessage(PromoteModelVersionToBuildResponse)
135
136
 
137
+ GetMLBOMModelVersionResponse = _reflection.GeneratedProtocolMessageType('GetMLBOMModelVersionResponse', (_message.Message,), {
138
+ 'DESCRIPTOR' : _GETMLBOMMODELVERSIONRESPONSE,
139
+ '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
140
+ # @@protoc_insertion_point(class_scope:jfml.model_version.v1.GetMLBOMModelVersionResponse)
141
+ })
142
+ _sym_db.RegisterMessage(GetMLBOMModelVersionResponse)
143
+
136
144
  _MODELVERSIONMANAGERSERVICE = DESCRIPTOR.services_by_name['ModelVersionManagerService']
137
145
  if _descriptor._USE_C_DESCRIPTORS == False:
138
146
 
@@ -166,6 +174,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
166
174
  _PROMOTEMODELVERSIONTOBUILDREQUEST._serialized_end=1411
167
175
  _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_start=1413
168
176
  _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_end=1467
169
- _MODELVERSIONMANAGERSERVICE._serialized_start=1470
170
- _MODELVERSIONMANAGERSERVICE._serialized_end=2395
177
+ _GETMLBOMMODELVERSIONRESPONSE._serialized_start=1469
178
+ _GETMLBOMMODELVERSIONRESPONSE._serialized_end=1528
179
+ _MODELVERSIONMANAGERSERVICE._serialized_start=1531
180
+ _MODELVERSIONMANAGERSERVICE._serialized_end=2589
171
181
  # @@protoc_insertion_point(module_scope)
@@ -271,3 +271,18 @@ class PromoteModelVersionToBuildResponse(google.protobuf.message.Message):
271
271
  def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
272
272
 
273
273
  global___PromoteModelVersionToBuildResponse = PromoteModelVersionToBuildResponse
274
+
275
+ class GetMLBOMModelVersionResponse(google.protobuf.message.Message):
276
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
277
+
278
+ MLBOM_MODEL_VERSION_FIELD_NUMBER: builtins.int
279
+ mlbom_model_version: builtins.str
280
+ """The MLBOM model_version"""
281
+ def __init__(
282
+ self,
283
+ *,
284
+ mlbom_model_version: builtins.str = ...,
285
+ ) -> None: ...
286
+ def ClearField(self, field_name: typing_extensions.Literal["mlbom_model_version", b"mlbom_model_version"]) -> None: ...
287
+
288
+ global___GetMLBOMModelVersionResponse = GetMLBOMModelVersionResponse
@@ -49,6 +49,11 @@ class ModelVersionManagerServiceStub(object):
49
49
  request_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildRequest.SerializeToString,
50
50
  response_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildResponse.FromString,
51
51
  )
52
+ self.GetMlBomModelVersionById = channel.unary_unary(
53
+ '/jfml.model_version.v1.ModelVersionManagerService/GetMlBomModelVersionById',
54
+ request_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetModelVersionByIdRequest.SerializeToString,
55
+ response_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMLBOMModelVersionResponse.FromString,
56
+ )
52
57
 
53
58
 
54
59
  class ModelVersionManagerServiceServicer(object):
@@ -103,6 +108,13 @@ class ModelVersionManagerServiceServicer(object):
103
108
  context.set_details('Method not implemented!')
104
109
  raise NotImplementedError('Method not implemented!')
105
110
 
111
+ def GetMlBomModelVersionById(self, request, context):
112
+ """Get MLBOM model_version by id
113
+ """
114
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
115
+ context.set_details('Method not implemented!')
116
+ raise NotImplementedError('Method not implemented!')
117
+
106
118
 
107
119
  def add_ModelVersionManagerServiceServicer_to_server(servicer, server):
108
120
  rpc_method_handlers = {
@@ -141,6 +153,11 @@ def add_ModelVersionManagerServiceServicer_to_server(servicer, server):
141
153
  request_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildRequest.FromString,
142
154
  response_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildResponse.SerializeToString,
143
155
  ),
156
+ 'GetMlBomModelVersionById': grpc.unary_unary_rpc_method_handler(
157
+ servicer.GetMlBomModelVersionById,
158
+ request_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetModelVersionByIdRequest.FromString,
159
+ response_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMLBOMModelVersionResponse.SerializeToString,
160
+ ),
144
161
  }
145
162
  generic_handler = grpc.method_handlers_generic_handler(
146
163
  'jfml.model_version.v1.ModelVersionManagerService', rpc_method_handlers)
@@ -269,3 +286,20 @@ class ModelVersionManagerService(object):
269
286
  jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildResponse.FromString,
270
287
  options, channel_credentials,
271
288
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
289
+
290
+ @staticmethod
291
+ def GetMlBomModelVersionById(request,
292
+ target,
293
+ options=(),
294
+ channel_credentials=None,
295
+ call_credentials=None,
296
+ insecure=False,
297
+ compression=None,
298
+ wait_for_ready=None,
299
+ timeout=None,
300
+ metadata=None):
301
+ return grpc.experimental.unary_unary(request, target, '/jfml.model_version.v1.ModelVersionManagerService/GetMlBomModelVersionById',
302
+ jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetModelVersionByIdRequest.SerializeToString,
303
+ jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMLBOMModelVersionResponse.FromString,
304
+ options, channel_credentials,
305
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)