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

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

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

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

{frogml_core-0.0.7 → frogml_core-0.0.9}/frogml_core/utils/proto_utils.py

@@ -19,23 +19,23 @@ SCIKIT_LEARN_FRAMEWORK_FORMAT = "joblib"
 class ProtoUtils:
     @staticmethod
     def model_framework_from_file_format(
-        serialization_format: str,
+        serialization_format: str, framework_version: str = ""
     ) -> ModelVersionFramework:
-        framework_to_define: dict
+        framework_to_define: dict = {"version": framework_version}
         stripped_and_lowered_format = serialization_format.strip().lower()
 
         if stripped_and_lowered_format == CATBOOST_SERIALIZED_TYPE:
-            framework_to_define = {"catboost": CatboostFramework()}
+            framework_to_define["catboost"] = CatboostFramework()
         elif stripped_and_lowered_format == HUGGINGFACE_FRAMEWORK_FORMAT:
-            framework_to_define = {"hugging_face": HuggingFaceFramework()}
+            framework_to_define["hugging_face"] = HuggingFaceFramework()
         elif stripped_and_lowered_format == ONNX_FRAMEWORK_FORMAT:
-            framework_to_define = {"onnx": OnnxFramework()}
+            framework_to_define["onnx"] = OnnxFramework()
         elif stripped_and_lowered_format == PYTHON_FRAMEWORK_FORMAT:
-            framework_to_define = {"python_pickle": PythonPickleFramework()}
+            framework_to_define["python_pickle"] = PythonPickleFramework()
         elif stripped_and_lowered_format == PYTORCH_FRAMEWORK_FORMAT:
-            framework_to_define = {"pytorch": PytorchFramework()}
+            framework_to_define["pytorch"] = PytorchFramework()
         elif stripped_and_lowered_format == SCIKIT_LEARN_FRAMEWORK_FORMAT:
-            framework_to_define = {"scikit_learn": ScikitLearnFramework()}
+            framework_to_define["scikit_learn"] = ScikitLearnFramework()
         else:
             raise ValueError(f"Format {serialization_format} is not supported yet")
 

frogml_core-0.0.9/frogml_proto/jfml/model_version/v1/build_spec_pb2.py

@@ -0,0 +1,67 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.jfml.model_version/v1/build_spec.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()
+
+
+from frogml_proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
+from frogml_proto.qwak.fitness_service import fitness_pb2 as qwak_dot_fitness__service_dot_fitness__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&jfml/model_version/v1/build_spec.proto\x12\x15jfml.model_version.v1\x1a/qwak/user_application/common/v0/resources.proto\x1a\"qwak/fitness_service/fitness.proto\"\xbc\x02\n\tBuildSpec\x12@\n\x10\x62uild_properties\x18\x01 \x01(\x0b\x32&.jfml.model_version.v1.BuildProperties\x12\x32\n\tbuild_env\x18\x02 \x01(\x0b\x32\x1f.jfml.model_version.v1.BuildEnv\x12Q\n\rpod_resources\x18\x03 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\x12\x0f\n\x07verbose\x18\x04 \x01(\r\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\x12=\n\x0fpurchase_option\x18\x06 \x01(\x0e\x32$.qwak.fitness.service.PurchaseOption\"K\n\x0f\x42uildProperties\x12\x12\n\nbuild_name\x18\x01 \x01(\t\x12\x0c\n\x04tags\x18\x02 \x03(\t\x12\x16\n\x0egpu_compatible\x18\x03 \x01(\x08\"@\n\x08\x42uildEnv\x12\x34\n\ndocker_env\x18\x01 \x01(\x0b\x32 .jfml.model_version.v1.DockerEnv\"1\n\tDockerEnv\x12\x12\n\nbase_image\x18\x01 \x01(\t\x12\x10\n\x08no_cache\x18\x02 \x01(\x08\x42%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+
+
+
+_BUILDSPEC = DESCRIPTOR.message_types_by_name['BuildSpec']
+_BUILDPROPERTIES = DESCRIPTOR.message_types_by_name['BuildProperties']
+_BUILDENV = DESCRIPTOR.message_types_by_name['BuildEnv']
+_DOCKERENV = DESCRIPTOR.message_types_by_name['DockerEnv']
+BuildSpec = _reflection.GeneratedProtocolMessageType('BuildSpec', (_message.Message,), {
+  'DESCRIPTOR' : _BUILDSPEC,
+  '__module__' : 'jfml.model_version.v1.build_spec_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.BuildSpec)
+  })
+_sym_db.RegisterMessage(BuildSpec)
+
+BuildProperties = _reflection.GeneratedProtocolMessageType('BuildProperties', (_message.Message,), {
+  'DESCRIPTOR' : _BUILDPROPERTIES,
+  '__module__' : 'jfml.model_version.v1.build_spec_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.BuildProperties)
+  })
+_sym_db.RegisterMessage(BuildProperties)
+
+BuildEnv = _reflection.GeneratedProtocolMessageType('BuildEnv', (_message.Message,), {
+  'DESCRIPTOR' : _BUILDENV,
+  '__module__' : 'jfml.model_version.v1.build_spec_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.BuildEnv)
+  })
+_sym_db.RegisterMessage(BuildEnv)
+
+DockerEnv = _reflection.GeneratedProtocolMessageType('DockerEnv', (_message.Message,), {
+  'DESCRIPTOR' : _DOCKERENV,
+  '__module__' : 'jfml.model_version.v1.build_spec_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.DockerEnv)
+  })
+_sym_db.RegisterMessage(DockerEnv)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _BUILDSPEC._serialized_start=151
+  _BUILDSPEC._serialized_end=467
+  _BUILDPROPERTIES._serialized_start=469
+  _BUILDPROPERTIES._serialized_end=544
+  _BUILDENV._serialized_start=546
+  _BUILDENV._serialized_end=610
+  _DOCKERENV._serialized_start=612
+  _DOCKERENV._serialized_end=661
+# @@protoc_insertion_point(module_scope)

frogml_core-0.0.9/frogml_proto/jfml/model_version/v1/build_spec_pb2.pyi

@@ -0,0 +1,120 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import frogml_proto.qwak.fitness_service.fitness_pb2
+import frogml_proto.qwak.user_application.common.v0.resources_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class BuildSpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_PROPERTIES_FIELD_NUMBER: builtins.int
+    BUILD_ENV_FIELD_NUMBER: builtins.int
+    POD_RESOURCES_FIELD_NUMBER: builtins.int
+    VERBOSE_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    PURCHASE_OPTION_FIELD_NUMBER: builtins.int
+    @property
+    def build_properties(self) -> global___BuildProperties:
+        """The build properties"""
+    @property
+    def build_env(self) -> global___BuildEnv:
+        """The build environment"""
+    @property
+    def pod_resources(self) -> frogml_proto.qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources: ...
+    verbose: builtins.int
+    """The build verbosity level (0: warning, 1: info, 2: debug)"""
+    environment_id: builtins.str
+    """The environment to build on"""
+    purchase_option: frogml_proto.qwak.fitness_service.fitness_pb2.PurchaseOption.ValueType
+    """Purchase option for the build"""
+    def __init__(
+        self,
+        *,
+        build_properties: global___BuildProperties | None = ...,
+        build_env: global___BuildEnv | None = ...,
+        pod_resources: frogml_proto.qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources | None = ...,
+        verbose: builtins.int = ...,
+        environment_id: builtins.str = ...,
+        purchase_option: frogml_proto.qwak.fitness_service.fitness_pb2.PurchaseOption.ValueType = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["build_env", b"build_env", "build_properties", b"build_properties", "pod_resources", b"pod_resources"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_env", b"build_env", "build_properties", b"build_properties", "environment_id", b"environment_id", "pod_resources", b"pod_resources", "purchase_option", b"purchase_option", "verbose", b"verbose"]) -> None: ...
+
+global___BuildSpec = BuildSpec
+
+class BuildProperties(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_NAME_FIELD_NUMBER: builtins.int
+    TAGS_FIELD_NUMBER: builtins.int
+    GPU_COMPATIBLE_FIELD_NUMBER: builtins.int
+    build_name: builtins.str
+    """Optional - The name of the build we want"""
+    @property
+    def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Optional - The Build's tags"""
+    gpu_compatible: builtins.bool
+    """Optional - Should the build be GPU compatible"""
+    def __init__(
+        self,
+        *,
+        build_name: builtins.str = ...,
+        tags: collections.abc.Iterable[builtins.str] | None = ...,
+        gpu_compatible: builtins.bool = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_name", b"build_name", "gpu_compatible", b"gpu_compatible", "tags", b"tags"]) -> None: ...
+
+global___BuildProperties = BuildProperties
+
+class BuildEnv(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DOCKER_ENV_FIELD_NUMBER: builtins.int
+    @property
+    def docker_env(self) -> global___DockerEnv:
+        """Optional - The docker specs to use for the build"""
+    def __init__(
+        self,
+        *,
+        docker_env: global___DockerEnv | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["docker_env", b"docker_env"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["docker_env", b"docker_env"]) -> None: ...
+
+global___BuildEnv = BuildEnv
+
+class DockerEnv(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BASE_IMAGE_FIELD_NUMBER: builtins.int
+    NO_CACHE_FIELD_NUMBER: builtins.int
+    base_image: builtins.str
+    """Optional - Base image to use for serving"""
+    no_cache: builtins.bool
+    """Optional -
+    Don't use remote cache to build the images, defaults to use cache,
+    if the model dependencies changed significantly it could be faster and better to run with no cache
+    """
+    def __init__(
+        self,
+        *,
+        base_image: builtins.str = ...,
+        no_cache: builtins.bool = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["base_image", b"base_image", "no_cache", b"no_cache"]) -> None: ...
+
+global___DockerEnv = DockerEnv

{frogml_core-0.0.7 → frogml_core-0.0.9}/frogml_proto/jfml/model_version/v1/model_version_framework_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3jfml/model_version/v1/model_version_framework.proto\x12\x15jfml.model_version.v1\"\xa0\x03\n\x15ModelVersionFramework\x12<\n\x08\x63\x61tboost\x18\x01 \x01(\x0b\x32(.jfml.model_version.v1.CatboostFrameworkH\x00\x12\x43\n\x0chugging_face\x18\x02 \x01(\x0b\x32+.jfml.model_version.v1.HuggingFaceFrameworkH\x00\x12\x34\n\x04onnx\x18\x03 \x01(\x0b\x32$.jfml.model_version.v1.OnnxFrameworkH\x00\x12\x45\n\rpython_pickle\x18\x04 \x01(\x0b\x32,.jfml.model_version.v1.PythonPickleFrameworkH\x00\x12:\n\x07pytorch\x18\x05 \x01(\x0b\x32\'.jfml.model_version.v1.PytorchFrameworkH\x00\x12\x43\n\x0cscikit_learn\x18\x06 \x01(\x0b\x32+.jfml.model_version.v1.ScikitLearnFrameworkH\x00\x42\x06\n\x04name\"\x13\n\x11\x43\x61tboostFramework\"\x16\n\x14HuggingFaceFramework\"\x0f\n\rOnnxFramework\"\x17\n\x15PythonPickleFramework\"\x12\n\x10PytorchFramework\"\x16\n\x14ScikitLearnFrameworkB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3jfml/model_version/v1/model_version_framework.proto\x12\x15jfml.model_version.v1\"\xb1\x03\n\x15ModelVersionFramework\x12<\n\x08\x63\x61tboost\x18\x01 \x01(\x0b\x32(.jfml.model_version.v1.CatboostFrameworkH\x00\x12\x43\n\x0chugging_face\x18\x02 \x01(\x0b\x32+.jfml.model_version.v1.HuggingFaceFrameworkH\x00\x12\x34\n\x04onnx\x18\x03 \x01(\x0b\x32$.jfml.model_version.v1.OnnxFrameworkH\x00\x12\x45\n\rpython_pickle\x18\x04 \x01(\x0b\x32,.jfml.model_version.v1.PythonPickleFrameworkH\x00\x12:\n\x07pytorch\x18\x05 \x01(\x0b\x32\'.jfml.model_version.v1.PytorchFrameworkH\x00\x12\x43\n\x0cscikit_learn\x18\x06 \x01(\x0b\x32+.jfml.model_version.v1.ScikitLearnFrameworkH\x00\x12\x0f\n\x07version\x18\x07 \x01(\tB\x06\n\x04name\"\x13\n\x11\x43\x61tboostFramework\"\x16\n\x14HuggingFaceFramework\"\x0f\n\rOnnxFramework\"\x17\n\x15PythonPickleFramework\"\x12\n\x10PytorchFramework\"\x16\n\x14ScikitLearnFrameworkB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
 
 
 
@@ -79,17 +79,17 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
   _MODELVERSIONFRAMEWORK._serialized_start=79
-  _MODELVERSIONFRAMEWORK._serialized_end=495
-  _CATBOOSTFRAMEWORK._serialized_start=497
-  _CATBOOSTFRAMEWORK._serialized_end=516
-  _HUGGINGFACEFRAMEWORK._serialized_start=518
-  _HUGGINGFACEFRAMEWORK._serialized_end=540
-  _ONNXFRAMEWORK._serialized_start=542
-  _ONNXFRAMEWORK._serialized_end=557
-  _PYTHONPICKLEFRAMEWORK._serialized_start=559
-  _PYTHONPICKLEFRAMEWORK._serialized_end=582
-  _PYTORCHFRAMEWORK._serialized_start=584
-  _PYTORCHFRAMEWORK._serialized_end=602
-  _SCIKITLEARNFRAMEWORK._serialized_start=604
-  _SCIKITLEARNFRAMEWORK._serialized_end=626
+  _MODELVERSIONFRAMEWORK._serialized_end=512
+  _CATBOOSTFRAMEWORK._serialized_start=514
+  _CATBOOSTFRAMEWORK._serialized_end=533
+  _HUGGINGFACEFRAMEWORK._serialized_start=535
+  _HUGGINGFACEFRAMEWORK._serialized_end=557
+  _ONNXFRAMEWORK._serialized_start=559
+  _ONNXFRAMEWORK._serialized_end=574
+  _PYTHONPICKLEFRAMEWORK._serialized_start=576
+  _PYTHONPICKLEFRAMEWORK._serialized_end=599
+  _PYTORCHFRAMEWORK._serialized_start=601
+  _PYTORCHFRAMEWORK._serialized_end=619
+  _SCIKITLEARNFRAMEWORK._serialized_start=621
+  _SCIKITLEARNFRAMEWORK._serialized_end=643
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.7 → frogml_core-0.0.9}/frogml_proto/jfml/model_version/v1/model_version_framework_pb2.pyi

@@ -23,6 +23,7 @@ class ModelVersionFramework(google.protobuf.message.Message):
     PYTHON_PICKLE_FIELD_NUMBER: builtins.int
     PYTORCH_FIELD_NUMBER: builtins.int
     SCIKIT_LEARN_FIELD_NUMBER: builtins.int
+    VERSION_FIELD_NUMBER: builtins.int
     @property
     def catboost(self) -> global___CatboostFramework: ...
     @property
@@ -35,6 +36,7 @@ class ModelVersionFramework(google.protobuf.message.Message):
     def pytorch(self) -> global___PytorchFramework: ...
     @property
     def scikit_learn(self) -> global___ScikitLearnFramework: ...
+    version: builtins.str
     def __init__(
         self,
         *,
@@ -44,9 +46,10 @@ class ModelVersionFramework(google.protobuf.message.Message):
         python_pickle: global___PythonPickleFramework | None = ...,
         pytorch: global___PytorchFramework | None = ...,
         scikit_learn: global___ScikitLearnFramework | None = ...,
+        version: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["catboost", b"catboost", "hugging_face", b"hugging_face", "name", b"name", "onnx", b"onnx", "python_pickle", b"python_pickle", "pytorch", b"pytorch", "scikit_learn", b"scikit_learn"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["catboost", b"catboost", "hugging_face", b"hugging_face", "name", b"name", "onnx", b"onnx", "python_pickle", b"python_pickle", "pytorch", b"pytorch", "scikit_learn", b"scikit_learn"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["catboost", b"catboost", "hugging_face", b"hugging_face", "name", b"name", "onnx", b"onnx", "python_pickle", b"python_pickle", "pytorch", b"pytorch", "scikit_learn", b"scikit_learn", "version", b"version"]) -> None: ...
     def WhichOneof(self, oneof_group: typing_extensions.Literal["name", b"name"]) -> typing_extensions.Literal["catboost", "hugging_face", "onnx", "python_pickle", "pytorch", "scikit_learn"] | None: ...
 
 global___ModelVersionFramework = ModelVersionFramework

{frogml_core-0.0.7 → frogml_core-0.0.9}/frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py

@@ -14,9 +14,10 @@ _sym_db = _symbol_database.Default()
 
 from frogml_proto.jfml.model_version.v1 import model_version_pb2 as jfml_dot_model__version_dot_v1_dot_model__version__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 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\"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\x65leteModelVersionByIdResponse2\x89\x06\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.DeleteModelVersionByIdResponseB%\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(\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')
 
 
 
@@ -32,6 +33,8 @@ _LISTMODELVERSIONSREQUEST = DESCRIPTOR.message_types_by_name['ListModelVersionsR
 _LISTMODELVERSIONSRESPONSE = DESCRIPTOR.message_types_by_name['ListModelVersionsResponse']
 _DELETEMODELVERSIONBYIDREQUEST = DESCRIPTOR.message_types_by_name['DeleteModelVersionByIdRequest']
 _DELETEMODELVERSIONBYIDRESPONSE = DESCRIPTOR.message_types_by_name['DeleteModelVersionByIdResponse']
+_PROMOTEMODELVERSIONTOBUILDREQUEST = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildRequest']
+_PROMOTEMODELVERSIONTOBUILDRESPONSE = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildResponse']
 CreateModelVersionRequest = _reflection.GeneratedProtocolMessageType('CreateModelVersionRequest', (_message.Message,), {
   'DESCRIPTOR' : _CREATEMODELVERSIONREQUEST,
   '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
@@ -116,35 +119,53 @@ DeleteModelVersionByIdResponse = _reflection.GeneratedProtocolMessageType('Delet
   })
 _sym_db.RegisterMessage(DeleteModelVersionByIdResponse)
 
+PromoteModelVersionToBuildRequest = _reflection.GeneratedProtocolMessageType('PromoteModelVersionToBuildRequest', (_message.Message,), {
+  'DESCRIPTOR' : _PROMOTEMODELVERSIONTOBUILDREQUEST,
+  '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.PromoteModelVersionToBuildRequest)
+  })
+_sym_db.RegisterMessage(PromoteModelVersionToBuildRequest)
+
+PromoteModelVersionToBuildResponse = _reflection.GeneratedProtocolMessageType('PromoteModelVersionToBuildResponse', (_message.Message,), {
+  'DESCRIPTOR' : _PROMOTEMODELVERSIONTOBUILDRESPONSE,
+  '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.PromoteModelVersionToBuildResponse)
+  })
+_sym_db.RegisterMessage(PromoteModelVersionToBuildResponse)
+
 _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=178
-  _CREATEMODELVERSIONREQUEST._serialized_end=286
-  _CREATEMODELVERSIONRESPONSE._serialized_start=288
-  _CREATEMODELVERSIONRESPONSE._serialized_end=376
-  _GETMODELVERSIONBYIDREQUEST._serialized_start=378
-  _GETMODELVERSIONBYIDREQUEST._serialized_end=432
-  _GETMODELVERSIONBYIDRESPONSE._serialized_start=434
-  _GETMODELVERSIONBYIDRESPONSE._serialized_end=523
-  _GETMODELVERSIONBYNAMEREQUEST._serialized_start=525
-  _GETMODELVERSIONBYNAMEREQUEST._serialized_end=601
-  _GETMODELVERSIONBYNAMERESPONSE._serialized_start=603
-  _GETMODELVERSIONBYNAMERESPONSE._serialized_end=694
-  _INITPAGINATIONREQUEST._serialized_start=696
-  _INITPAGINATIONREQUEST._serialized_end=737
-  _INITPAGINATIONRESPONSE._serialized_start=739
-  _INITPAGINATIONRESPONSE._serialized_end=808
-  _LISTMODELVERSIONSREQUEST._serialized_start=811
-  _LISTMODELVERSIONSREQUEST._serialized_end=969
-  _LISTMODELVERSIONSRESPONSE._serialized_start=971
-  _LISTMODELVERSIONSRESPONSE._serialized_end=1059
-  _DELETEMODELVERSIONBYIDREQUEST._serialized_start=1061
-  _DELETEMODELVERSIONBYIDREQUEST._serialized_end=1118
-  _DELETEMODELVERSIONBYIDRESPONSE._serialized_start=1120
-  _DELETEMODELVERSIONBYIDRESPONSE._serialized_end=1152
-  _MODELVERSIONMANAGERSERVICE._serialized_start=1155
-  _MODELVERSIONMANAGERSERVICE._serialized_end=1932
+  _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
+  _MODELVERSIONMANAGERSERVICE._serialized_start=1470
+  _MODELVERSIONMANAGERSERVICE._serialized_end=2395
 # @@protoc_insertion_point(module_scope)

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

@@ -7,6 +7,7 @@ import collections.abc
 import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.message
+import frogml_proto.jfml.model_version.v1.build_spec_pb2
 import frogml_proto.jfml.model_version.v1.model_repository_spec_pb2
 import frogml_proto.jfml.model_version.v1.model_version_pb2
 import sys
@@ -228,3 +229,45 @@ class DeleteModelVersionByIdResponse(google.protobuf.message.Message):
     ) -> None: ...
 
 global___DeleteModelVersionByIdResponse = DeleteModelVersionByIdResponse
+
+class PromoteModelVersionToBuildRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_VERSION_SPEC_FIELD_NUMBER: builtins.int
+    MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
+    BUILD_SPEC_FIELD_NUMBER: builtins.int
+    @property
+    def model_version_spec(self) -> frogml_proto.jfml.model_version.v1.model_version_pb2.ModelVersionSpec:
+        """The model_version spec, we will use from it only the model's id and the model version's name"""
+    model_version_id: builtins.str
+    """The model_version id"""
+    @property
+    def build_spec(self) -> frogml_proto.jfml.model_version.v1.build_spec_pb2.BuildSpec:
+        """The build specifications"""
+    def __init__(
+        self,
+        *,
+        model_version_spec: frogml_proto.jfml.model_version.v1.model_version_pb2.ModelVersionSpec | None = ...,
+        model_version_id: builtins.str = ...,
+        build_spec: frogml_proto.jfml.model_version.v1.build_spec_pb2.BuildSpec | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec", "model_version_id", b"model_version_id", "model_version_identifier", b"model_version_identifier", "model_version_spec", b"model_version_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec", "model_version_id", b"model_version_id", "model_version_identifier", b"model_version_identifier", "model_version_spec", b"model_version_spec"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["model_version_identifier", b"model_version_identifier"]) -> typing_extensions.Literal["model_version_spec", "model_version_id"] | None: ...
+
+global___PromoteModelVersionToBuildRequest = PromoteModelVersionToBuildRequest
+
+class PromoteModelVersionToBuildResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_ID_FIELD_NUMBER: builtins.int
+    build_id: builtins.str
+    """The Build that was created"""
+    def __init__(
+        self,
+        *,
+        build_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
+
+global___PromoteModelVersionToBuildResponse = PromoteModelVersionToBuildResponse

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

@@ -44,6 +44,11 @@ class ModelVersionManagerServiceStub(object):
                 request_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.DeleteModelVersionByIdRequest.SerializeToString,
                 response_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.DeleteModelVersionByIdResponse.FromString,
                 )
+        self.PromoteModelVersionToBuild = channel.unary_unary(
+                '/jfml.model_version.v1.ModelVersionManagerService/PromoteModelVersionToBuild',
+                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,
+                )
 
 
 class ModelVersionManagerServiceServicer(object):
@@ -91,6 +96,13 @@ class ModelVersionManagerServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def PromoteModelVersionToBuild(self, request, context):
+        """Promote a model_version to a build
+        """
+        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 = {
@@ -124,6 +136,11 @@ def add_ModelVersionManagerServiceServicer_to_server(servicer, server):
                     request_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.DeleteModelVersionByIdRequest.FromString,
                     response_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.DeleteModelVersionByIdResponse.SerializeToString,
             ),
+            'PromoteModelVersionToBuild': grpc.unary_unary_rpc_method_handler(
+                    servicer.PromoteModelVersionToBuild,
+                    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,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'jfml.model_version.v1.ModelVersionManagerService', rpc_method_handlers)
@@ -235,3 +252,20 @@ class ModelVersionManagerService(object):
             jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.DeleteModelVersionByIdResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def PromoteModelVersionToBuild(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/PromoteModelVersionToBuild',
+            jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.PromoteModelVersionToBuildRequest.SerializeToString,
+            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)