PyPI - frogml-core - Versions diffs - 0.0.12__py3-none-any.whl → 0.0.14__py3-none-any.whl - Mend

frogml-core 0.0.12py3-none-any.whl → 0.0.14py3-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.

Files changed (14) hide show

frogml_core/__init__.py CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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 RENAMED Viewed

@@ -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 RENAMED Viewed

@@ -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 ADDED Viewed

@@ -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 ADDED Viewed

@@ -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 ADDED Viewed

@@ -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 CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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 CHANGED Viewed

@@ -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)

{frogml_core-0.0.12.dist-info → frogml_core-0.0.14.dist-info}/WHEEL RENAMED Viewed

File without changes

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

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