frogml-core 0.0.9__tar.gz → 0.0.11__tar.gz

{frogml_core-0.0.9 → frogml_core-0.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.9
+Version: 0.0.11
 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.9 → frogml_core-0.0.11}/frogml_core/__init__.py

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

{frogml_core-0.0.9 → frogml_core-0.0.11}/frogml_core/clients/model_version_manager/client.py

@@ -31,7 +31,7 @@ class ModelVersionManagerClient:
         self.__model_version_manager_stub = ModelVersionManagerServiceStub(grpc_channel)
 
     @staticmethod
-    def __build_create_model_request(
+    def __build_create_model_version_request(
         project_key: str,
         repository_key: str,
         model_name: str,
@@ -46,6 +46,7 @@ class ModelVersionManagerClient:
                     project_key=project_key,
                     repository_key=repository_key,
                     model_id=get_model_id_from_model_name(model_name),
+                    model_name=model_name,
                 ),
                 name=model_version_name,
                 framework=model_version_framework,
@@ -62,7 +63,7 @@ class ModelVersionManagerClient:
         model_version_framework: ModelVersionFramework,
     ):
         try:
-            create_model_request = self.__build_create_model_request(
+            create_model_request = self.__build_create_model_version_request(
                 project_key=project_key,
                 repository_key=repository_key,
                 model_name=model_name,
@@ -88,7 +89,7 @@ class ModelVersionManagerClient:
         model_version_framework: ModelVersionFramework,
     ) -> CreateModelVersionResponse:
         try:
-            create_model_request = self.__build_create_model_request(
+            create_model_request = self.__build_create_model_version_request(
                 project_key=project_key,
                 repository_key=repository_key,
                 model_name=model_name,

{frogml_core-0.0.9 → frogml_core-0.0.11}/frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1jfml/model_version/v1/model_repository_spec.proto\x12\x15jfml.model_version.v1\"l\n\x13ModelRepositorySpec\x12\x15\n\x0bproject_key\x18\x01 \x01(\tH\x00\x12\x16\n\x0erepository_key\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\tB\x14\n\x12project_identifierB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1jfml/model_version/v1/model_repository_spec.proto\x12\x15jfml.model_version.v1\"\x80\x01\n\x13ModelRepositorySpec\x12\x15\n\x0bproject_key\x18\x01 \x01(\tH\x00\x12\x16\n\x0erepository_key\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\t\x12\x12\n\nmodel_name\x18\x04 \x01(\tB\x14\n\x12project_identifierB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
 
 
 
@@ -30,6 +30,6 @@ if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
-  _MODELREPOSITORYSPEC._serialized_start=76
-  _MODELREPOSITORYSPEC._serialized_end=184
+  _MODELREPOSITORYSPEC._serialized_start=77
+  _MODELREPOSITORYSPEC._serialized_end=205
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.9 → frogml_core-0.0.11}/frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.pyi

@@ -20,19 +20,22 @@ class ModelRepositorySpec(google.protobuf.message.Message):
     PROJECT_KEY_FIELD_NUMBER: builtins.int
     REPOSITORY_KEY_FIELD_NUMBER: builtins.int
     MODEL_ID_FIELD_NUMBER: builtins.int
+    MODEL_NAME_FIELD_NUMBER: builtins.int
     project_key: builtins.str
     """JFrog project key"""
     repository_key: builtins.str
     model_id: builtins.str
+    model_name: builtins.str
     def __init__(
         self,
         *,
         project_key: builtins.str = ...,
         repository_key: builtins.str = ...,
         model_id: builtins.str = ...,
+        model_name: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["project_identifier", b"project_identifier", "project_key", b"project_key"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["model_id", b"model_id", "project_identifier", b"project_identifier", "project_key", b"project_key", "repository_key", b"repository_key"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["model_id", b"model_id", "model_name", b"model_name", "project_identifier", b"project_identifier", "project_key", b"project_key", "repository_key", b"repository_key"]) -> None: ...
     def WhichOneof(self, oneof_group: typing_extensions.Literal["project_identifier", b"project_identifier"]) -> typing_extensions.Literal["project_key"] | None: ...
 
 global___ModelRepositorySpec = ModelRepositorySpec

{frogml_core-0.0.9 → frogml_core-0.0.11}/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(\t2\x9d\x07\n\x1aModelVersionManagerService\x12y\n\x12\x43reateModelVersion\x12\x30.jfml.model_version.v1.CreateModelVersionRequest\x1a\x31.jfml.model_version.v1.CreateModelVersionResponse\x12|\n\x13GetModelVersionById\x12\x31.jfml.model_version.v1.GetModelVersionByIdRequest\x1a\x32.jfml.model_version.v1.GetModelVersionByIdResponse\x12\x82\x01\n\x15GetModelVersionByName\x12\x33.jfml.model_version.v1.GetModelVersionByNameRequest\x1a\x34.jfml.model_version.v1.GetModelVersionByNameResponse\x12m\n\x0eInitPagination\x12,.jfml.model_version.v1.InitPaginationRequest\x1a-.jfml.model_version.v1.InitPaginationResponse\x12v\n\x11ListModelVersions\x12/.jfml.model_version.v1.ListModelVersionsRequest\x1a\x30.jfml.model_version.v1.ListModelVersionsResponse\x12\x85\x01\n\x16\x44\x65leteModelVersionById\x12\x34.jfml.model_version.v1.DeleteModelVersionByIdRequest\x1a\x35.jfml.model_version.v1.DeleteModelVersionByIdResponse\x12\x91\x01\n\x1aPromoteModelVersionToBuild\x12\x38.jfml.model_version.v1.PromoteModelVersionToBuildRequest\x1a\x39.jfml.model_version.v1.PromoteModelVersionToBuildResponseB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+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')
 
 
 
@@ -35,6 +35,7 @@ _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']
 CreateModelVersionRequest = _reflection.GeneratedProtocolMessageType('CreateModelVersionRequest', (_message.Message,), {
   'DESCRIPTOR' : _CREATEMODELVERSIONREQUEST,
   '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
@@ -133,6 +134,13 @@ PromoteModelVersionToBuildResponse = _reflection.GeneratedProtocolMessageType('P
   })
 _sym_db.RegisterMessage(PromoteModelVersionToBuildResponse)
 
+GetMLBOMModelVersionResponse = _reflection.GeneratedProtocolMessageType('GetMLBOMModelVersionResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETMLBOMMODELVERSIONRESPONSE,
+  '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.GetMLBOMModelVersionResponse)
+  })
+_sym_db.RegisterMessage(GetMLBOMModelVersionResponse)
+
 _MODELVERSIONMANAGERSERVICE = DESCRIPTOR.services_by_name['ModelVersionManagerService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
@@ -166,6 +174,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _PROMOTEMODELVERSIONTOBUILDREQUEST._serialized_end=1411
   _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_start=1413
   _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_end=1467
-  _MODELVERSIONMANAGERSERVICE._serialized_start=1470
-  _MODELVERSIONMANAGERSERVICE._serialized_end=2395
+  _GETMLBOMMODELVERSIONRESPONSE._serialized_start=1469
+  _GETMLBOMMODELVERSIONRESPONSE._serialized_end=1528
+  _MODELVERSIONMANAGERSERVICE._serialized_start=1531
+  _MODELVERSIONMANAGERSERVICE._serialized_end=2589
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.9 → frogml_core-0.0.11}/frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi

@@ -271,3 +271,18 @@ class PromoteModelVersionToBuildResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
 
 global___PromoteModelVersionToBuildResponse = PromoteModelVersionToBuildResponse
+
+class GetMLBOMModelVersionResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MLBOM_MODEL_VERSION_FIELD_NUMBER: builtins.int
+    mlbom_model_version: builtins.str
+    """The MLBOM model_version"""
+    def __init__(
+        self,
+        *,
+        mlbom_model_version: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["mlbom_model_version", b"mlbom_model_version"]) -> None: ...
+
+global___GetMLBOMModelVersionResponse = GetMLBOMModelVersionResponse

{frogml_core-0.0.9 → frogml_core-0.0.11}/frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py

@@ -49,6 +49,11 @@ class ModelVersionManagerServiceStub(object):
                 request_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildRequest.SerializeToString,
                 response_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildResponse.FromString,
                 )
+        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,
+                )
 
 
 class ModelVersionManagerServiceServicer(object):
@@ -103,6 +108,13 @@ class ModelVersionManagerServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetMlBomModelVersionById(self, request, context):
+        """Get MLBOM model_version by id
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_ModelVersionManagerServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -141,6 +153,11 @@ def add_ModelVersionManagerServiceServicer_to_server(servicer, server):
                     request_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildRequest.FromString,
                     response_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildResponse.SerializeToString,
             ),
+            '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,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'jfml.model_version.v1.ModelVersionManagerService', rpc_method_handlers)
@@ -269,3 +286,20 @@ class ModelVersionManagerService(object):
             jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetMlBomModelVersionById(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            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,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

{frogml_core-0.0.9 → frogml_core-0.0.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "frogml-core"
-version = "0.0.9"
+version = "0.0.11"
 description = "frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform"
 authors = ["Jfrog"]
 readme = "README.md"