frogml-core 0.0.12__py3-none-any.whl → 0.0.14__py3-none-any.whl

frogml_core/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for frogml."""
 
 __author__ = "jfrog"
-__version__ = "0.0.12"
+__version__ = "0.0.14"
 
 from frogml_core.inner.di_configuration import wire_dependencies
 from frogml_core.model.model_version_tracking import (  # noqa: F401,E501

frogml_core/clients/model_version_manager/client.py

@@ -3,6 +3,8 @@ from platform import python_version
 from dependency_injector.wiring import Provide
 from grpc import RpcError
 
+from frogml_core.exceptions import FrogmlException
+from frogml_core.inner.di_configuration import FrogmlContainer
 from frogml_core.utils.model_utils import get_model_id_from_model_name
 from frogml_proto.jfml.model_version.v1.model_repository_spec_pb2 import (
     ModelRepositorySpec,
@@ -13,13 +15,13 @@ from frogml_proto.jfml.model_version.v1.model_version_framework_pb2 import (
 from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2 import (
     CreateModelVersionRequest,
     CreateModelVersionResponse,
+    GetMlBomModelVersionByIdRequest,
+    GetMlBomModelVersionByIdResponse,
 )
 from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
     ModelVersionManagerServiceStub,
 )
 from frogml_proto.jfml.model_version.v1.model_version_pb2 import ModelVersionSpec
-from frogml_core.exceptions import FrogmlException
-from frogml_core.inner.di_configuration import FrogmlContainer
 
 
 class ModelVersionManagerClient:
@@ -111,3 +113,27 @@ class ModelVersionManagerClient:
         except Exception as e:
             message = f"Failed to validate model version, details [{e}]"
             raise FrogmlException(message)
+
+    def get_mlbom_by_model_version_id(
+        self, model_version_id: str
+    ) -> GetMlBomModelVersionByIdResponse:
+        try:
+            request = GetMlBomModelVersionByIdRequest(model_version_id=model_version_id)
+            create_model_version_response: GetMlBomModelVersionByIdResponse = (
+                self.__model_version_manager_stub.GetMlBomModelVersionById(request)
+            )
+
+            return create_model_version_response
+        except RpcError as e:
+            message = (
+                f"Failed to get MLBOM for model version {model_version_id}, "
+                f"details [{e.details()}]"
+            )
+            raise FrogmlException(message)
+
+        except Exception as e:
+            message = (
+                f"Failed to get MLBOM for model version {model_version_id}, "
+                f"details [{e}]"
+            )
+            raise FrogmlException(message)

{frogml_core-0.0.12.dist-info → frogml_core-0.0.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.12
+Version: 0.0.14
 Summary: frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{frogml_core-0.0.12.dist-info → frogml_core-0.0.14.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=uaXZGrHq-qnxd5cDdqqeMFchG3ha9_v2iE0JzjYuzxI,777
+frogml_core/__init__.py,sha256=693__WVbTmV7KkdDLdXWTduAveqtsa7sTsR7ZvR6N64,777
 frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
 frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
 frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -73,7 +73,7 @@ frogml_core/clients/logging_client/client.py,sha256=A7qQJWW54Ve7O0DuJX9gmSJlUQyR
 frogml_core/clients/model_management/__init__.py,sha256=vjWVP8MjmK4_A70WOgJqa6x24AeLK-ABjGJtogGzw9w,43
 frogml_core/clients/model_management/client.py,sha256=rXnzcP3-pfXCVG0KlcCwbdeIJ1iFzV84JYtyNC8n-x8,4575
 frogml_core/clients/model_version_manager/__init__.py,sha256=4Pnfzj4Egps48__dProdbSKQl5pNip0hGJd75w67BfY,46
-frogml_core/clients/model_version_manager/client.py,sha256=6WIvzWY1IWuUINvKzHPs8klMKaRJuyIN3DDdSsAryCQ,4214
+frogml_core/clients/model_version_manager/client.py,sha256=MKw3ek_dwvm6AhjCaAWtm_s_N3LueN9-Rer4P-ItN9Y,5194
 frogml_core/clients/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/clients/project/client.py,sha256=VvnUk7mo-UkE9Gkj9jVkCPGTVx8qmuekMz7oH9Uzv2Y,2159
 frogml_core/clients/prompt_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -431,6 +431,9 @@ frogml_core/vector_store/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 frogml_core/vector_store/utils/filter_utils.py,sha256=n80IlmS7lGelCNEpZ8o3rAHzrZpHc-3EykEd8tZoeAE,854
 frogml_core/vector_store/utils/upsert_utils.py,sha256=vqnyAJOl46WqpZMUiOa4I8Zf2CDIXQxR7wM4Z8rgooc,7517
 frogml_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+frogml_proto/jfml/model_version/v1/artifact_pb2.py,sha256=HjSD0Tp2XGxwMBnDzFAYhCBftaJwiJRoYSkTcr1pJyA,1973
+frogml_proto/jfml/model_version/v1/artifact_pb2.pyi,sha256=5CGU58qXOSbe0KSu0ovjIbXWrpS9r-PIvdq4qhqPQIc,1673
+frogml_proto/jfml/model_version/v1/artifact_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml_proto/jfml/model_version/v1/build_spec_pb2.py,sha256=7FeOfvGtH2v4swm0FGl8ALvUgR4ckIcn-pSkzYd-uhQ,3779
 frogml_proto/jfml/model_version/v1/build_spec_pb2.pyi,sha256=CIBEscG5MQoOGddvsfGqxD2K4U342IS-kFpzyYFiFD8,4923
 frogml_proto/jfml/model_version/v1/build_spec_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -440,11 +443,11 @@ frogml_proto/jfml/model_version/v1/model_repository_spec_pb2_grpc.py,sha256=1obo
 frogml_proto/jfml/model_version/v1/model_version_framework_pb2.py,sha256=Vk4NO9dlhXkNVgxdLGDfh20s-mBfcoOOZEEr6hUtLag,5370
 frogml_proto/jfml/model_version/v1/model_version_framework_pb2.pyi,sha256=JgOE0Im69ir0PSgpbg6Zz4scRK_u7WAxWTtWcdzfyAw,3773
 frogml_proto/jfml/model_version/v1/model_version_framework_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py,sha256=X0KJkIj8eg41SVWD48AuJSMw1P_1R5lfD4X_L_rSajs,13816
-frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi,sha256=kLV9okhcSx5LKAwTBUknA3e0sotbkDMa_JnOwcUzrRc,12416
-frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py,sha256=2v-qLZmVDKKE2J6c9Sz1WAXPLO9Ngtw9dFnehH7Kc5U,17508
-frogml_proto/jfml/model_version/v1/model_version_pb2.py,sha256=87CsbasbiwvPWGI1b6AfGrje_wzDCZFD8Hi7meguDjU,5797
-frogml_proto/jfml/model_version/v1/model_version_pb2.pyi,sha256=NqHepmIm1HiKBYn77JDFOoBAEVwCbyFsj0Wiurhsxkc,5664
+frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py,sha256=AdtFAjC0tmEIcQD7k8m7TtTCVTa0ioexf31VSyUcPHk,14670
+frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi,sha256=ZEEgWKD0qd8zAD8nljdBAZFP3hnLN0MOAYtZA1EHfjw,13323
+frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py,sha256=5b77ei6ijZs9-z0frMZ6MngPziapHRYJAqU6CzZFMFs,17535
+frogml_proto/jfml/model_version/v1/model_version_pb2.py,sha256=S3xn4m27oPw4FCi1CIwurcOUYRroz1Vh55ctUeZQlBw,8336
+frogml_proto/jfml/model_version/v1/model_version_pb2.pyi,sha256=vLtqBIKmdB-Wb2vFPSPIu2D-WWg5UTsmqgjMJ_zWTsY,8914
 frogml_proto/jfml/model_version/v1/model_version_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml_proto/qwak/administration/account/v1/account_pb2.py,sha256=eiD7FPyN-7p5OplBBPYYavyXYVAEcu2tFp5sgZ2-fgs,8082
 frogml_proto/qwak/administration/account/v1/account_pb2.pyi,sha256=-pGJBR4NbmZDPkznUbb3sfGsTu4W6hfg72pE0jtvmpg,10934
@@ -1000,7 +1003,7 @@ frogml_services_mock/mocks/job_registry_service_api.py,sha256=Zd5lVM6h4jFfKHxnQA
 frogml_services_mock/mocks/kube_captain_service_api.py,sha256=WVCaoOHY-kFdS73bd7kuOssr1RAK1F6MUlJ-NO0eLfY,1596
 frogml_services_mock/mocks/logging_service.py,sha256=JRTCjJRIKlJQeTHr3Qt5gbkfxMLUPblo0Sp50COspxE,7385
 frogml_services_mock/mocks/model_management_service.py,sha256=YKNgZ65AcKTMFs1zkrfVfGOBjqPBpodTgD6r358iRMg,4167
-frogml_services_mock/mocks/model_version_manager_service.py,sha256=GoipKHH6-aN_ulcOzOoIotgT5eJM69SFMUmw4zpcs4M,3745
+frogml_services_mock/mocks/model_version_manager_service.py,sha256=nBSB-bL0X8AR1e_3J8R_bHuRaGQvf9P8aFskjNUvEo8,4814
 frogml_services_mock/mocks/project_manager_service.py,sha256=dxjhZwqN2UjDpHybArB8O6JOBke2JnQ9OMxy_1lLLJM,3093
 frogml_services_mock/mocks/prompt_manager_service.py,sha256=dJufY5oO4SwdO3rmbz9KaOzZKz25H5Ggx3grUQ3sIHg,11369
 frogml_services_mock/mocks/repository_service_mock.py,sha256=2KoXSZ4l1Q5kvLns_eoH_yDZQBRzgCk50yTzNFTL3Sc,1405
@@ -1016,6 +1019,6 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=_34z6rFCHFwcSni-9eyJlrH264xsL_QDNmG-EBv0zMc,20281
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_core-0.0.12.dist-info/METADATA,sha256=gyb4TAXQR4zOf51CEr6hwU0tSb9M7XLiMUUFAFV3cZQ,2099
-frogml_core-0.0.12.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-frogml_core-0.0.12.dist-info/RECORD,,
+frogml_core-0.0.14.dist-info/METADATA,sha256=Hb8qAPAxhKJKxYeXj-k3o6S_pEWONxeezG43xDNna18,2099
+frogml_core-0.0.14.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+frogml_core-0.0.14.dist-info/RECORD,,

frogml_proto/jfml/model_version/v1/artifact_pb2.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.jfml.model_version/v1/artifact.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 message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$jfml/model_version/v1/artifact.proto\x12\x15jfml.model_version.v1\"d\n\x08\x41rtifact\x12\x15\n\rartifact_path\x18\x01 \x01(\t\x12\x0c\n\x04size\x18\x02 \x01(\x04\x12\x33\n\tchecksums\x18\x03 \x01(\x0b\x32 .jfml.model_version.v1.Checksums\"\x19\n\tChecksums\x12\x0c\n\x04sha2\x18\x01 \x01(\tB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+
+
+
+_ARTIFACT = DESCRIPTOR.message_types_by_name['Artifact']
+_CHECKSUMS = DESCRIPTOR.message_types_by_name['Checksums']
+Artifact = _reflection.GeneratedProtocolMessageType('Artifact', (_message.Message,), {
+  'DESCRIPTOR' : _ARTIFACT,
+  '__module__' : 'jfml.model_version.v1.artifact_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.Artifact)
+  })
+_sym_db.RegisterMessage(Artifact)
+
+Checksums = _reflection.GeneratedProtocolMessageType('Checksums', (_message.Message,), {
+  'DESCRIPTOR' : _CHECKSUMS,
+  '__module__' : 'jfml.model_version.v1.artifact_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.Checksums)
+  })
+_sym_db.RegisterMessage(Checksums)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _ARTIFACT._serialized_start=63
+  _ARTIFACT._serialized_end=163
+  _CHECKSUMS._serialized_start=165
+  _CHECKSUMS._serialized_end=190
+# @@protoc_insertion_point(module_scope)

frogml_proto/jfml/model_version/v1/artifact_pb2.pyi

@@ -0,0 +1,55 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class Artifact(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ARTIFACT_PATH_FIELD_NUMBER: builtins.int
+    SIZE_FIELD_NUMBER: builtins.int
+    CHECKSUMS_FIELD_NUMBER: builtins.int
+    artifact_path: builtins.str
+    """Path to the artifact"""
+    size: builtins.int
+    """Size of the artifact"""
+    @property
+    def checksums(self) -> global___Checksums:
+        """Checksums of the artifact"""
+    def __init__(
+        self,
+        *,
+        artifact_path: builtins.str = ...,
+        size: builtins.int = ...,
+        checksums: global___Checksums | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["checksums", b"checksums"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["artifact_path", b"artifact_path", "checksums", b"checksums", "size", b"size"]) -> None: ...
+
+global___Artifact = Artifact
+
+class Checksums(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SHA2_FIELD_NUMBER: builtins.int
+    sha2: builtins.str
+    """SHA2 checksum"""
+    def __init__(
+        self,
+        *,
+        sha2: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["sha2", b"sha2"]) -> None: ...
+
+global___Checksums = Checksums

frogml_proto/jfml/model_version/v1/artifact_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py

@@ -17,7 +17,7 @@ from frogml_proto.jfml.model_version.v1 import model_repository_spec_pb2 as jfml
 from frogml_proto.jfml.model_version.v1 import build_spec_pb2 as jfml_dot_model__version_dot_v1_dot_build__spec__pb2
 
 
-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')
+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\"\xa3\x01\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\x12\x35\n\x0bml_bom_data\x18\x03 \x01(\x0b\x32 .jfml.model_version.v1.MlBomData\"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\x1fGetMlBomModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\"?\n GetMlBomModelVersionByIdResponse\x12\x1b\n\x13mlbom_model_version\x18\x01 \x01(\t2\xab\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\x8b\x01\n\x18GetMlBomModelVersionById\x12\x36.jfml.model_version.v1.GetMlBomModelVersionByIdRequest\x1a\x37.jfml.model_version.v1.GetMlBomModelVersionByIdResponseB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
 
 
 
@@ -35,7 +35,8 @@ _DELETEMODELVERSIONBYIDREQUEST = DESCRIPTOR.message_types_by_name['DeleteModelVe
 _DELETEMODELVERSIONBYIDRESPONSE = DESCRIPTOR.message_types_by_name['DeleteModelVersionByIdResponse']
 _PROMOTEMODELVERSIONTOBUILDREQUEST = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildRequest']
 _PROMOTEMODELVERSIONTOBUILDRESPONSE = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildResponse']
-_GETMLBOMMODELVERSIONRESPONSE = DESCRIPTOR.message_types_by_name['GetMLBOMModelVersionResponse']
+_GETMLBOMMODELVERSIONBYIDREQUEST = DESCRIPTOR.message_types_by_name['GetMlBomModelVersionByIdRequest']
+_GETMLBOMMODELVERSIONBYIDRESPONSE = DESCRIPTOR.message_types_by_name['GetMlBomModelVersionByIdResponse']
 CreateModelVersionRequest = _reflection.GeneratedProtocolMessageType('CreateModelVersionRequest', (_message.Message,), {
   'DESCRIPTOR' : _CREATEMODELVERSIONREQUEST,
   '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
@@ -134,48 +135,57 @@ PromoteModelVersionToBuildResponse = _reflection.GeneratedProtocolMessageType('P
   })
 _sym_db.RegisterMessage(PromoteModelVersionToBuildResponse)
 
-GetMLBOMModelVersionResponse = _reflection.GeneratedProtocolMessageType('GetMLBOMModelVersionResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETMLBOMMODELVERSIONRESPONSE,
+GetMlBomModelVersionByIdRequest = _reflection.GeneratedProtocolMessageType('GetMlBomModelVersionByIdRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETMLBOMMODELVERSIONBYIDREQUEST,
   '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
-  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.GetMLBOMModelVersionResponse)
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.GetMlBomModelVersionByIdRequest)
   })
-_sym_db.RegisterMessage(GetMLBOMModelVersionResponse)
+_sym_db.RegisterMessage(GetMlBomModelVersionByIdRequest)
+
+GetMlBomModelVersionByIdResponse = _reflection.GeneratedProtocolMessageType('GetMlBomModelVersionByIdResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETMLBOMMODELVERSIONBYIDRESPONSE,
+  '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.GetMlBomModelVersionByIdResponse)
+  })
+_sym_db.RegisterMessage(GetMlBomModelVersionByIdResponse)
 
 _MODELVERSIONMANAGERSERVICE = DESCRIPTOR.services_by_name['ModelVersionManagerService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
-  _CREATEMODELVERSIONREQUEST._serialized_start=218
-  _CREATEMODELVERSIONREQUEST._serialized_end=326
-  _CREATEMODELVERSIONRESPONSE._serialized_start=328
-  _CREATEMODELVERSIONRESPONSE._serialized_end=416
-  _GETMODELVERSIONBYIDREQUEST._serialized_start=418
-  _GETMODELVERSIONBYIDREQUEST._serialized_end=472
-  _GETMODELVERSIONBYIDRESPONSE._serialized_start=474
-  _GETMODELVERSIONBYIDRESPONSE._serialized_end=563
-  _GETMODELVERSIONBYNAMEREQUEST._serialized_start=565
-  _GETMODELVERSIONBYNAMEREQUEST._serialized_end=641
-  _GETMODELVERSIONBYNAMERESPONSE._serialized_start=643
-  _GETMODELVERSIONBYNAMERESPONSE._serialized_end=734
-  _INITPAGINATIONREQUEST._serialized_start=736
-  _INITPAGINATIONREQUEST._serialized_end=777
-  _INITPAGINATIONRESPONSE._serialized_start=779
-  _INITPAGINATIONRESPONSE._serialized_end=848
-  _LISTMODELVERSIONSREQUEST._serialized_start=851
-  _LISTMODELVERSIONSREQUEST._serialized_end=1009
-  _LISTMODELVERSIONSRESPONSE._serialized_start=1011
-  _LISTMODELVERSIONSRESPONSE._serialized_end=1099
-  _DELETEMODELVERSIONBYIDREQUEST._serialized_start=1101
-  _DELETEMODELVERSIONBYIDREQUEST._serialized_end=1158
-  _DELETEMODELVERSIONBYIDRESPONSE._serialized_start=1160
-  _DELETEMODELVERSIONBYIDRESPONSE._serialized_end=1192
-  _PROMOTEMODELVERSIONTOBUILDREQUEST._serialized_start=1195
-  _PROMOTEMODELVERSIONTOBUILDREQUEST._serialized_end=1411
-  _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_start=1413
-  _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_end=1467
-  _GETMLBOMMODELVERSIONRESPONSE._serialized_start=1469
-  _GETMLBOMMODELVERSIONRESPONSE._serialized_end=1528
-  _MODELVERSIONMANAGERSERVICE._serialized_start=1531
-  _MODELVERSIONMANAGERSERVICE._serialized_end=2589
+  _CREATEMODELVERSIONREQUEST._serialized_start=219
+  _CREATEMODELVERSIONREQUEST._serialized_end=382
+  _CREATEMODELVERSIONRESPONSE._serialized_start=384
+  _CREATEMODELVERSIONRESPONSE._serialized_end=472
+  _GETMODELVERSIONBYIDREQUEST._serialized_start=474
+  _GETMODELVERSIONBYIDREQUEST._serialized_end=528
+  _GETMODELVERSIONBYIDRESPONSE._serialized_start=530
+  _GETMODELVERSIONBYIDRESPONSE._serialized_end=619
+  _GETMODELVERSIONBYNAMEREQUEST._serialized_start=621
+  _GETMODELVERSIONBYNAMEREQUEST._serialized_end=697
+  _GETMODELVERSIONBYNAMERESPONSE._serialized_start=699
+  _GETMODELVERSIONBYNAMERESPONSE._serialized_end=790
+  _INITPAGINATIONREQUEST._serialized_start=792
+  _INITPAGINATIONREQUEST._serialized_end=833
+  _INITPAGINATIONRESPONSE._serialized_start=835
+  _INITPAGINATIONRESPONSE._serialized_end=904
+  _LISTMODELVERSIONSREQUEST._serialized_start=907
+  _LISTMODELVERSIONSREQUEST._serialized_end=1065
+  _LISTMODELVERSIONSRESPONSE._serialized_start=1067
+  _LISTMODELVERSIONSRESPONSE._serialized_end=1155
+  _DELETEMODELVERSIONBYIDREQUEST._serialized_start=1157
+  _DELETEMODELVERSIONBYIDREQUEST._serialized_end=1214
+  _DELETEMODELVERSIONBYIDRESPONSE._serialized_start=1216
+  _DELETEMODELVERSIONBYIDRESPONSE._serialized_end=1248
+  _PROMOTEMODELVERSIONTOBUILDREQUEST._serialized_start=1251
+  _PROMOTEMODELVERSIONTOBUILDREQUEST._serialized_end=1467
+  _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_start=1469
+  _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_end=1523
+  _GETMLBOMMODELVERSIONBYIDREQUEST._serialized_start=1525
+  _GETMLBOMMODELVERSIONBYIDREQUEST._serialized_end=1584
+  _GETMLBOMMODELVERSIONBYIDRESPONSE._serialized_start=1586
+  _GETMLBOMMODELVERSIONBYIDRESPONSE._serialized_end=1649
+  _MODELVERSIONMANAGERSERVICE._serialized_start=1652
+  _MODELVERSIONMANAGERSERVICE._serialized_end=2719
 # @@protoc_insertion_point(module_scope)

frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi

@@ -24,18 +24,23 @@ class CreateModelVersionRequest(google.protobuf.message.Message):
 
     DRY_RUN_FIELD_NUMBER: builtins.int
     MODEL_VERSION_FIELD_NUMBER: builtins.int
+    ML_BOM_DATA_FIELD_NUMBER: builtins.int
     dry_run: builtins.bool
     """When True, the request will be validated but not executed"""
     @property
     def model_version(self) -> frogml_proto.jfml.model_version.v1.model_version_pb2.ModelVersionSpec: ...
+    @property
+    def ml_bom_data(self) -> frogml_proto.jfml.model_version.v1.model_version_pb2.MlBomData:
+        """The model data for mlbom file"""
     def __init__(
         self,
         *,
         dry_run: builtins.bool = ...,
         model_version: frogml_proto.jfml.model_version.v1.model_version_pb2.ModelVersionSpec | None = ...,
+        ml_bom_data: frogml_proto.jfml.model_version.v1.model_version_pb2.MlBomData | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["model_version", b"model_version"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["dry_run", b"dry_run", "model_version", b"model_version"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["ml_bom_data", b"ml_bom_data", "model_version", b"model_version"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["dry_run", b"dry_run", "ml_bom_data", b"ml_bom_data", "model_version", b"model_version"]) -> None: ...
 
 global___CreateModelVersionRequest = CreateModelVersionRequest
 
@@ -272,7 +277,22 @@ class PromoteModelVersionToBuildResponse(google.protobuf.message.Message):
 
 global___PromoteModelVersionToBuildResponse = PromoteModelVersionToBuildResponse
 
-class GetMLBOMModelVersionResponse(google.protobuf.message.Message):
+class GetMlBomModelVersionByIdRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
+    model_version_id: builtins.str
+    """ModelVersion's id"""
+    def __init__(
+        self,
+        *,
+        model_version_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["model_version_id", b"model_version_id"]) -> None: ...
+
+global___GetMlBomModelVersionByIdRequest = GetMlBomModelVersionByIdRequest
+
+class GetMlBomModelVersionByIdResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     MLBOM_MODEL_VERSION_FIELD_NUMBER: builtins.int
@@ -285,4 +305,4 @@ class GetMLBOMModelVersionResponse(google.protobuf.message.Message):
     ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["mlbom_model_version", b"mlbom_model_version"]) -> None: ...
 
-global___GetMLBOMModelVersionResponse = GetMLBOMModelVersionResponse
+global___GetMlBomModelVersionByIdResponse = GetMlBomModelVersionByIdResponse

frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py

@@ -51,8 +51,8 @@ class ModelVersionManagerServiceStub(object):
                 )
         self.GetMlBomModelVersionById = channel.unary_unary(
                 '/jfml.model_version.v1.ModelVersionManagerService/GetMlBomModelVersionById',
-                request_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetModelVersionByIdRequest.SerializeToString,
-                response_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMLBOMModelVersionResponse.FromString,
+                request_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdRequest.SerializeToString,
+                response_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdResponse.FromString,
                 )
 
 
@@ -155,8 +155,8 @@ def add_ModelVersionManagerServiceServicer_to_server(servicer, server):
             ),
             'GetMlBomModelVersionById': grpc.unary_unary_rpc_method_handler(
                     servicer.GetMlBomModelVersionById,
-                    request_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetModelVersionByIdRequest.FromString,
-                    response_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMLBOMModelVersionResponse.SerializeToString,
+                    request_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdRequest.FromString,
+                    response_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdResponse.SerializeToString,
             ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
@@ -299,7 +299,7 @@ class ModelVersionManagerService(object):
             timeout=None,
             metadata=None):
         return grpc.experimental.unary_unary(request, target, '/jfml.model_version.v1.ModelVersionManagerService/GetMlBomModelVersionById',
-            jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetModelVersionByIdRequest.SerializeToString,
-            jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMLBOMModelVersionResponse.FromString,
+            jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdRequest.SerializeToString,
+            jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

frogml_proto/jfml/model_version/v1/model_version_pb2.py

@@ -15,14 +15,18 @@ _sym_db = _symbol_database.Default()
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 from frogml_proto.jfml.model_version.v1 import model_version_framework_pb2 as jfml_dot_model__version_dot_v1_dot_model__version__framework__pb2
 from frogml_proto.jfml.model_version.v1 import model_repository_spec_pb2 as jfml_dot_model__version_dot_v1_dot_model__repository__spec__pb2
+from frogml_proto.jfml.model_version.v1 import artifact_pb2 as jfml_dot_model__version_dot_v1_dot_artifact__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)jfml/model_version/v1/model_version.proto\x12\x15jfml.model_version.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x33jfml/model_version/v1/model_version_framework.proto\x1a\x31jfml/model_version/v1/model_repository_spec.proto\"\x8e\x02\n\x0cModelVersion\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\x12\x35\n\x04spec\x18\x02 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpec\x12\x39\n\x06status\x18\x03 \x01(\x0b\x32).jfml.model_version.v1.ModelVersionStatus\x12\x12\n\ncreated_by\x18\x04 \x01(\t\x12.\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\xbe\x01\n\x10ModelVersionSpec\x12\x43\n\x0frepository_spec\x18\x01 \x01(\x0b\x32*.jfml.model_version.v1.ModelRepositorySpec\x12\x0c\n\x04name\x18\x02 \x01(\t\x12?\n\tframework\x18\x03 \x01(\x0b\x32,.jfml.model_version.v1.ModelVersionFramework\x12\x16\n\x0epython_version\x18\x04 \x01(\t\"\xf6\x01\n\x12ModelVersionStatus\x12J\n\x0bin_progress\x18\x01 \x01(\x0b\x32\x33.jfml.model_version.v1.ModelVersionInProgressStatusH\x00\x12G\n\tcompleted\x18\x02 \x01(\x0b\x32\x32.jfml.model_version.v1.ModelVersionCompletedStatusH\x00\x12\x41\n\x06\x66\x61iled\x18\x03 \x01(\x0b\x32/.jfml.model_version.v1.ModelVersionFailedStatusH\x00\x42\x08\n\x06status\"\x1e\n\x1cModelVersionInProgressStatus\"\x1d\n\x1bModelVersionCompletedStatus\"\x1a\n\x18ModelVersionFailedStatusB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)jfml/model_version/v1/model_version.proto\x12\x15jfml.model_version.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x33jfml/model_version/v1/model_version_framework.proto\x1a\x31jfml/model_version/v1/model_repository_spec.proto\x1a$jfml/model_version/v1/artifact.proto\"\x8e\x02\n\x0cModelVersion\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\x12\x35\n\x04spec\x18\x02 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpec\x12\x39\n\x06status\x18\x03 \x01(\x0b\x32).jfml.model_version.v1.ModelVersionStatus\x12\x12\n\ncreated_by\x18\x04 \x01(\t\x12.\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\xbe\x01\n\x10ModelVersionSpec\x12\x43\n\x0frepository_spec\x18\x01 \x01(\x0b\x32*.jfml.model_version.v1.ModelRepositorySpec\x12\x0c\n\x04name\x18\x02 \x01(\t\x12?\n\tframework\x18\x03 \x01(\x0b\x32,.jfml.model_version.v1.ModelVersionFramework\x12\x16\n\x0epython_version\x18\x04 \x01(\t\"\xa5\x03\n\tMlBomData\x12\x37\n\x0emodel_artifact\x18\x01 \x03(\x0b\x32\x1f.jfml.model_version.v1.Artifact\x12\x37\n\x0e\x63ode_artifacts\x18\x02 \x03(\x0b\x32\x1f.jfml.model_version.v1.Artifact\x12=\n\x14\x64\x65pendency_artifacts\x18\x03 \x03(\x0b\x32\x1f.jfml.model_version.v1.Artifact\x12\x44\n\nparameters\x18\x04 \x03(\x0b\x32\x30.jfml.model_version.v1.MlBomData.ParametersEntry\x12>\n\x07metrics\x18\x05 \x03(\x0b\x32-.jfml.model_version.v1.MlBomData.MetricsEntry\x1a\x31\n\x0fParametersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a.\n\x0cMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xf6\x01\n\x12ModelVersionStatus\x12J\n\x0bin_progress\x18\x01 \x01(\x0b\x32\x33.jfml.model_version.v1.ModelVersionInProgressStatusH\x00\x12G\n\tcompleted\x18\x02 \x01(\x0b\x32\x32.jfml.model_version.v1.ModelVersionCompletedStatusH\x00\x12\x41\n\x06\x66\x61iled\x18\x03 \x01(\x0b\x32/.jfml.model_version.v1.ModelVersionFailedStatusH\x00\x42\x08\n\x06status\"\x1e\n\x1cModelVersionInProgressStatus\"\x1d\n\x1bModelVersionCompletedStatus\"\x1a\n\x18ModelVersionFailedStatusB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
 
 
 
 _MODELVERSION = DESCRIPTOR.message_types_by_name['ModelVersion']
 _MODELVERSIONSPEC = DESCRIPTOR.message_types_by_name['ModelVersionSpec']
+_MLBOMDATA = DESCRIPTOR.message_types_by_name['MlBomData']
+_MLBOMDATA_PARAMETERSENTRY = _MLBOMDATA.nested_types_by_name['ParametersEntry']
+_MLBOMDATA_METRICSENTRY = _MLBOMDATA.nested_types_by_name['MetricsEntry']
 _MODELVERSIONSTATUS = DESCRIPTOR.message_types_by_name['ModelVersionStatus']
 _MODELVERSIONINPROGRESSSTATUS = DESCRIPTOR.message_types_by_name['ModelVersionInProgressStatus']
 _MODELVERSIONCOMPLETEDSTATUS = DESCRIPTOR.message_types_by_name['ModelVersionCompletedStatus']
@@ -41,6 +45,29 @@ ModelVersionSpec = _reflection.GeneratedProtocolMessageType('ModelVersionSpec',
   })
 _sym_db.RegisterMessage(ModelVersionSpec)
 
+MlBomData = _reflection.GeneratedProtocolMessageType('MlBomData', (_message.Message,), {
+
+  'ParametersEntry' : _reflection.GeneratedProtocolMessageType('ParametersEntry', (_message.Message,), {
+    'DESCRIPTOR' : _MLBOMDATA_PARAMETERSENTRY,
+    '__module__' : 'jfml.model_version.v1.model_version_pb2'
+    # @@protoc_insertion_point(class_scope:jfml.model_version.v1.MlBomData.ParametersEntry)
+    })
+  ,
+
+  'MetricsEntry' : _reflection.GeneratedProtocolMessageType('MetricsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _MLBOMDATA_METRICSENTRY,
+    '__module__' : 'jfml.model_version.v1.model_version_pb2'
+    # @@protoc_insertion_point(class_scope:jfml.model_version.v1.MlBomData.MetricsEntry)
+    })
+  ,
+  'DESCRIPTOR' : _MLBOMDATA,
+  '__module__' : 'jfml.model_version.v1.model_version_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.MlBomData)
+  })
+_sym_db.RegisterMessage(MlBomData)
+_sym_db.RegisterMessage(MlBomData.ParametersEntry)
+_sym_db.RegisterMessage(MlBomData.MetricsEntry)
+
 ModelVersionStatus = _reflection.GeneratedProtocolMessageType('ModelVersionStatus', (_message.Message,), {
   'DESCRIPTOR' : _MODELVERSIONSTATUS,
   '__module__' : 'jfml.model_version.v1.model_version_pb2'
@@ -73,16 +100,26 @@ if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
-  _MODELVERSION._serialized_start=206
-  _MODELVERSION._serialized_end=476
-  _MODELVERSIONSPEC._serialized_start=479
-  _MODELVERSIONSPEC._serialized_end=669
-  _MODELVERSIONSTATUS._serialized_start=672
-  _MODELVERSIONSTATUS._serialized_end=918
-  _MODELVERSIONINPROGRESSSTATUS._serialized_start=920
-  _MODELVERSIONINPROGRESSSTATUS._serialized_end=950
-  _MODELVERSIONCOMPLETEDSTATUS._serialized_start=952
-  _MODELVERSIONCOMPLETEDSTATUS._serialized_end=981
-  _MODELVERSIONFAILEDSTATUS._serialized_start=983
-  _MODELVERSIONFAILEDSTATUS._serialized_end=1009
+  _MLBOMDATA_PARAMETERSENTRY._options = None
+  _MLBOMDATA_PARAMETERSENTRY._serialized_options = b'8\001'
+  _MLBOMDATA_METRICSENTRY._options = None
+  _MLBOMDATA_METRICSENTRY._serialized_options = b'8\001'
+  _MODELVERSION._serialized_start=244
+  _MODELVERSION._serialized_end=514
+  _MODELVERSIONSPEC._serialized_start=517
+  _MODELVERSIONSPEC._serialized_end=707
+  _MLBOMDATA._serialized_start=710
+  _MLBOMDATA._serialized_end=1131
+  _MLBOMDATA_PARAMETERSENTRY._serialized_start=1034
+  _MLBOMDATA_PARAMETERSENTRY._serialized_end=1083
+  _MLBOMDATA_METRICSENTRY._serialized_start=1085
+  _MLBOMDATA_METRICSENTRY._serialized_end=1131
+  _MODELVERSIONSTATUS._serialized_start=1134
+  _MODELVERSIONSTATUS._serialized_end=1380
+  _MODELVERSIONINPROGRESSSTATUS._serialized_start=1382
+  _MODELVERSIONINPROGRESSSTATUS._serialized_end=1412
+  _MODELVERSIONCOMPLETEDSTATUS._serialized_start=1414
+  _MODELVERSIONCOMPLETEDSTATUS._serialized_end=1443
+  _MODELVERSIONFAILEDSTATUS._serialized_start=1445
+  _MODELVERSIONFAILEDSTATUS._serialized_end=1471
 # @@protoc_insertion_point(module_scope)

frogml_proto/jfml/model_version/v1/model_version_pb2.pyi

@@ -3,9 +3,12 @@
 isort:skip_file
 """
 import builtins
+import collections.abc
 import google.protobuf.descriptor
+import google.protobuf.internal.containers
 import google.protobuf.message
 import google.protobuf.timestamp_pb2
+import frogml_proto.jfml.model_version.v1.artifact_pb2
 import frogml_proto.jfml.model_version.v1.model_repository_spec_pb2
 import frogml_proto.jfml.model_version.v1.model_version_framework_pb2
 import sys
@@ -80,6 +83,72 @@ class ModelVersionSpec(google.protobuf.message.Message):
 
 global___ModelVersionSpec = ModelVersionSpec
 
+class MlBomData(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class ParametersEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    class MetricsEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    MODEL_ARTIFACT_FIELD_NUMBER: builtins.int
+    CODE_ARTIFACTS_FIELD_NUMBER: builtins.int
+    DEPENDENCY_ARTIFACTS_FIELD_NUMBER: builtins.int
+    PARAMETERS_FIELD_NUMBER: builtins.int
+    METRICS_FIELD_NUMBER: builtins.int
+    @property
+    def model_artifact(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[jfml.model_version.v1.artifact_pb2.Artifact]:
+        """The model's artifact"""
+    @property
+    def code_artifacts(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[jfml.model_version.v1.artifact_pb2.Artifact]:
+        """The model's code artifact"""
+    @property
+    def dependency_artifacts(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[jfml.model_version.v1.artifact_pb2.Artifact]:
+        """The model's dependency artifacts"""
+    @property
+    def parameters(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+        """The model's parameters"""
+    @property
+    def metrics(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+        """The model's metrics"""
+    def __init__(
+        self,
+        *,
+        model_artifact: collections.abc.Iterable[jfml.model_version.v1.artifact_pb2.Artifact] | None = ...,
+        code_artifacts: collections.abc.Iterable[jfml.model_version.v1.artifact_pb2.Artifact] | None = ...,
+        dependency_artifacts: collections.abc.Iterable[jfml.model_version.v1.artifact_pb2.Artifact] | None = ...,
+        parameters: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+        metrics: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["code_artifacts", b"code_artifacts", "dependency_artifacts", b"dependency_artifacts", "metrics", b"metrics", "model_artifact", b"model_artifact", "parameters", b"parameters"]) -> None: ...
+
+global___MlBomData = MlBomData
+
 class ModelVersionStatus(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

frogml_services_mock/mocks/model_version_manager_service.py

@@ -17,6 +17,16 @@ from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2 import
     GetModelVersionByIdRequest,
     GetModelVersionByNameRequest,
     GetModelVersionByNameResponse,
+    InitPaginationRequest,
+    InitPaginationResponse,
+    ListModelVersionsRequest,
+    ListModelVersionsResponse,
+    DeleteModelVersionByIdRequest,
+    DeleteModelVersionByIdResponse,
+    PromoteModelVersionToBuildRequest,
+    PromoteModelVersionToBuildResponse,
+    GetMlBomModelVersionByIdRequest,
+    GetMlBomModelVersionByIdResponse,
 )
 from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
     ModelVersionManagerServiceServicer,
@@ -24,7 +34,8 @@ from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc i
 from frogml_proto.jfml.model_version.v1.model_version_pb2 import (
     ModelVersion,
     ModelVersionCompletedStatus,
-    ModelVersionSpec, ModelVersionStatus,
+    ModelVersionSpec,
+    ModelVersionStatus,
 )
 
 timestamp = Timestamp()
@@ -36,7 +47,9 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
         super(ModelVersionManagerServiceMock, self).__init__()
         self.model_versions: Dict[str, ModelVersion] = {}
 
-    def CreateModelVersion(self, request: CreateModelVersionRequest, context):
+    def CreateModelVersion(
+        self, request: CreateModelVersionRequest, context
+    ) -> CreateModelVersionResponse:
         return CreateModelVersionResponse(
             model_version=ModelVersion(
                 model_version_id=str(uuid.uuid4()),
@@ -48,7 +61,9 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
             )
         )
 
-    def GetModelVersionById(self, request: GetModelVersionByIdRequest, context):
+    def GetModelVersionById(
+        self, request: GetModelVersionByIdRequest, context
+    ) -> GetModelVersionByIdResponse:
         return GetModelVersionByIdResponse(
             model_version=ModelVersion(
                 model_version_id=request.model_version_id,
@@ -90,11 +105,27 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
             )
         )
 
-    def InitPagination(self, request, context):
+    def InitPagination(
+        self, request: InitPaginationRequest, context
+    ) -> InitPaginationResponse:
         return super().InitPagination(request, context)
 
-    def ListModelVersions(self, request, context):
+    def ListModelVersions(
+        self, request: ListModelVersionsRequest, context
+    ) -> ListModelVersionsResponse:
         return super().ListModelVersions(request, context)
 
-    def DeleteModelVersionById(self, request, context):
+    def DeleteModelVersionById(
+        self, request: DeleteModelVersionByIdRequest, context
+    ) -> DeleteModelVersionByIdResponse:
         return super().DeleteModelVersionById(request, context)
+
+    def PromoteModelVersionToBuild(
+        self, request: PromoteModelVersionToBuildRequest, context
+    ) -> PromoteModelVersionToBuildResponse:
+        return super().PromoteModelVersionToBuild(request, context)
+
+    def GetMlBomModelVersionById(
+        self, request: GetMlBomModelVersionByIdRequest, context
+    ) -> GetMlBomModelVersionByIdResponse:
+        return super().GetMlBomModelVersionById(request, context)