frogml-core 0.0.2__tar.gz → 0.0.4__tar.gz

{frogml_core-0.0.2 → frogml_core-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.2
+Version: 0.0.4
 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
@@ -34,7 +34,7 @@ Requires-Dist: pyarrow (>=6.0.0) ; extra == "feature-store"
 Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
 Requires-Dist: pyspark (==3.4.2) ; extra == "feature-store"
 Requires-Dist: python-jose
-Requires-Dist: python-json-logger (>=2.0.2)
+Requires-Dist: python-json-logger (>=3.2.1,<4.0.0)
 Requires-Dist: requests
 Requires-Dist: retrying (==1.3.4)
 Requires-Dist: typeguard (>=2,<3)

{frogml_core-0.0.2 → frogml_core-0.0.4}/frogml_core/__init__.py

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

{frogml_core-0.0.2 → frogml_core-0.0.4}/frogml_core/tools/logger/logging.yml

@@ -11,11 +11,11 @@ formatters:
     format: '%(asctime)s %(levelname)-8s %(module)s - %(lineno)d - %(message)s'
     datefmt: '%Y-%m-%d %H:%M:%S'
   remote:
-    class: pythonjsonlogger.jsonlogger.JsonFormatter
+    class: pythonjsonlogger.json.JsonFormatter
     format: '%(asctime)s %(process)d %(levelname)s %(module)s %(lineno)d %(message)s'
     datefmt: '%Y-%m-%dT%H:%M:%S'
   json:
-    class: pythonjsonlogger.jsonlogger.JsonFormatter
+    class: pythonjsonlogger.json.JsonFormatter
     format: '%(asctime)s %(process)d %(levelname)s %(module)s %(lineno)d %(message)s'
     datefmt: '%Y-%m-%dT%H:%M:%S'
 

frogml_core-0.0.4/frogml_core/utils/proto_utils.py

@@ -0,0 +1,42 @@
+from frogml_proto.jfml.model_version.v1.model_version_framework_pb2 import (
+    ModelVersionFramework,
+    CatboostFramework,
+    HuggingFaceFramework,
+    OnnxFramework,
+    PythonPickleFramework,
+    PytorchFramework,
+    ScikitLearnFramework,
+)
+
+CATBOOST_SERIALIZED_TYPE = "cbm"
+HUGGINGFACE_FRAMEWORK_FORMAT = "pretrained_model"
+ONNX_FRAMEWORK_FORMAT = "onnx"
+PYTHON_FRAMEWORK_FORMAT = "pkl"
+PYTORCH_FRAMEWORK_FORMAT = "pth"
+SCIKIT_LEARN_FRAMEWORK_FORMAT = "joblib"
+
+
+class ProtoUtils:
+    @staticmethod
+    def model_framework_from_file_format(
+        serialization_format: str,
+    ) -> ModelVersionFramework:
+        framework_to_define: dict
+        stripped_and_lowered_format = serialization_format.strip().lower()
+
+        if stripped_and_lowered_format == CATBOOST_SERIALIZED_TYPE:
+            framework_to_define = {"catboost": CatboostFramework()}
+        elif stripped_and_lowered_format == HUGGINGFACE_FRAMEWORK_FORMAT:
+            framework_to_define = {"hugging_face": HuggingFaceFramework()}
+        elif stripped_and_lowered_format == ONNX_FRAMEWORK_FORMAT:
+            framework_to_define = {"onnx": OnnxFramework()}
+        elif stripped_and_lowered_format == PYTHON_FRAMEWORK_FORMAT:
+            framework_to_define = {"python_pickle": PythonPickleFramework()}
+        elif stripped_and_lowered_format == PYTORCH_FRAMEWORK_FORMAT:
+            framework_to_define = {"pytorch": PytorchFramework()}
+        elif stripped_and_lowered_format == SCIKIT_LEARN_FRAMEWORK_FORMAT:
+            framework_to_define = {"scikit_learn": ScikitLearnFramework()}
+        else:
+            raise ValueError(f"Format {serialization_format} is not supported yet")
+
+        return ModelVersionFramework(**framework_to_define)

frogml_core-0.0.4/frogml_proto/jfml/model_version/v1/model_version_framework_pb2.py

@@ -0,0 +1,95 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.jfml.model_version/v1/model_version_framework.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'\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')
+
+
+
+_MODELVERSIONFRAMEWORK = DESCRIPTOR.message_types_by_name['ModelVersionFramework']
+_CATBOOSTFRAMEWORK = DESCRIPTOR.message_types_by_name['CatboostFramework']
+_HUGGINGFACEFRAMEWORK = DESCRIPTOR.message_types_by_name['HuggingFaceFramework']
+_ONNXFRAMEWORK = DESCRIPTOR.message_types_by_name['OnnxFramework']
+_PYTHONPICKLEFRAMEWORK = DESCRIPTOR.message_types_by_name['PythonPickleFramework']
+_PYTORCHFRAMEWORK = DESCRIPTOR.message_types_by_name['PytorchFramework']
+_SCIKITLEARNFRAMEWORK = DESCRIPTOR.message_types_by_name['ScikitLearnFramework']
+ModelVersionFramework = _reflection.GeneratedProtocolMessageType('ModelVersionFramework', (_message.Message,), {
+  'DESCRIPTOR' : _MODELVERSIONFRAMEWORK,
+  '__module__' : 'jfml.model_version.v1.model_version_framework_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.ModelVersionFramework)
+  })
+_sym_db.RegisterMessage(ModelVersionFramework)
+
+CatboostFramework = _reflection.GeneratedProtocolMessageType('CatboostFramework', (_message.Message,), {
+  'DESCRIPTOR' : _CATBOOSTFRAMEWORK,
+  '__module__' : 'jfml.model_version.v1.model_version_framework_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.CatboostFramework)
+  })
+_sym_db.RegisterMessage(CatboostFramework)
+
+HuggingFaceFramework = _reflection.GeneratedProtocolMessageType('HuggingFaceFramework', (_message.Message,), {
+  'DESCRIPTOR' : _HUGGINGFACEFRAMEWORK,
+  '__module__' : 'jfml.model_version.v1.model_version_framework_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.HuggingFaceFramework)
+  })
+_sym_db.RegisterMessage(HuggingFaceFramework)
+
+OnnxFramework = _reflection.GeneratedProtocolMessageType('OnnxFramework', (_message.Message,), {
+  'DESCRIPTOR' : _ONNXFRAMEWORK,
+  '__module__' : 'jfml.model_version.v1.model_version_framework_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.OnnxFramework)
+  })
+_sym_db.RegisterMessage(OnnxFramework)
+
+PythonPickleFramework = _reflection.GeneratedProtocolMessageType('PythonPickleFramework', (_message.Message,), {
+  'DESCRIPTOR' : _PYTHONPICKLEFRAMEWORK,
+  '__module__' : 'jfml.model_version.v1.model_version_framework_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.PythonPickleFramework)
+  })
+_sym_db.RegisterMessage(PythonPickleFramework)
+
+PytorchFramework = _reflection.GeneratedProtocolMessageType('PytorchFramework', (_message.Message,), {
+  'DESCRIPTOR' : _PYTORCHFRAMEWORK,
+  '__module__' : 'jfml.model_version.v1.model_version_framework_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.PytorchFramework)
+  })
+_sym_db.RegisterMessage(PytorchFramework)
+
+ScikitLearnFramework = _reflection.GeneratedProtocolMessageType('ScikitLearnFramework', (_message.Message,), {
+  'DESCRIPTOR' : _SCIKITLEARNFRAMEWORK,
+  '__module__' : 'jfml.model_version.v1.model_version_framework_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.ScikitLearnFramework)
+  })
+_sym_db.RegisterMessage(ScikitLearnFramework)
+
+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
+# @@protoc_insertion_point(module_scope)

frogml_core-0.0.4/frogml_proto/jfml/model_version/v1/model_version_framework_pb2.pyi

@@ -0,0 +1,106 @@
+"""
+@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 ModelVersionFramework(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CATBOOST_FIELD_NUMBER: builtins.int
+    HUGGING_FACE_FIELD_NUMBER: builtins.int
+    ONNX_FIELD_NUMBER: builtins.int
+    PYTHON_PICKLE_FIELD_NUMBER: builtins.int
+    PYTORCH_FIELD_NUMBER: builtins.int
+    SCIKIT_LEARN_FIELD_NUMBER: builtins.int
+    @property
+    def catboost(self) -> global___CatboostFramework: ...
+    @property
+    def hugging_face(self) -> global___HuggingFaceFramework: ...
+    @property
+    def onnx(self) -> global___OnnxFramework: ...
+    @property
+    def python_pickle(self) -> global___PythonPickleFramework: ...
+    @property
+    def pytorch(self) -> global___PytorchFramework: ...
+    @property
+    def scikit_learn(self) -> global___ScikitLearnFramework: ...
+    def __init__(
+        self,
+        *,
+        catboost: global___CatboostFramework | None = ...,
+        hugging_face: global___HuggingFaceFramework | None = ...,
+        onnx: global___OnnxFramework | None = ...,
+        python_pickle: global___PythonPickleFramework | None = ...,
+        pytorch: global___PytorchFramework | None = ...,
+        scikit_learn: global___ScikitLearnFramework | None = ...,
+    ) -> 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 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
+
+class CatboostFramework(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___CatboostFramework = CatboostFramework
+
+class HuggingFaceFramework(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___HuggingFaceFramework = HuggingFaceFramework
+
+class OnnxFramework(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___OnnxFramework = OnnxFramework
+
+class PythonPickleFramework(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___PythonPickleFramework = PythonPickleFramework
+
+class PytorchFramework(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___PytorchFramework = PytorchFramework
+
+class ScikitLearnFramework(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ScikitLearnFramework = ScikitLearnFramework

{frogml_core-0.0.2 → frogml_core-0.0.4}/frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py

@@ -14,10 +14,9 @@ _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 model_version_framework_pb2 as jfml_dot_model__version_dot_v1_dot_model__version__framework__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\x33jfml/model_version/v1/model_version_framework.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\"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')
 
 
 
@@ -122,30 +121,30 @@ 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=231
-  _CREATEMODELVERSIONREQUEST._serialized_end=339
-  _CREATEMODELVERSIONRESPONSE._serialized_start=341
-  _CREATEMODELVERSIONRESPONSE._serialized_end=429
-  _GETMODELVERSIONBYIDREQUEST._serialized_start=431
-  _GETMODELVERSIONBYIDREQUEST._serialized_end=485
-  _GETMODELVERSIONBYIDRESPONSE._serialized_start=487
-  _GETMODELVERSIONBYIDRESPONSE._serialized_end=576
-  _GETMODELVERSIONBYNAMEREQUEST._serialized_start=578
-  _GETMODELVERSIONBYNAMEREQUEST._serialized_end=654
-  _GETMODELVERSIONBYNAMERESPONSE._serialized_start=656
-  _GETMODELVERSIONBYNAMERESPONSE._serialized_end=747
-  _INITPAGINATIONREQUEST._serialized_start=749
-  _INITPAGINATIONREQUEST._serialized_end=790
-  _INITPAGINATIONRESPONSE._serialized_start=792
-  _INITPAGINATIONRESPONSE._serialized_end=861
-  _LISTMODELVERSIONSREQUEST._serialized_start=864
-  _LISTMODELVERSIONSREQUEST._serialized_end=1022
-  _LISTMODELVERSIONSRESPONSE._serialized_start=1024
-  _LISTMODELVERSIONSRESPONSE._serialized_end=1112
-  _DELETEMODELVERSIONBYIDREQUEST._serialized_start=1114
-  _DELETEMODELVERSIONBYIDREQUEST._serialized_end=1171
-  _DELETEMODELVERSIONBYIDRESPONSE._serialized_start=1173
-  _DELETEMODELVERSIONBYIDRESPONSE._serialized_end=1205
-  _MODELVERSIONMANAGERSERVICE._serialized_start=1208
-  _MODELVERSIONMANAGERSERVICE._serialized_end=1985
+  _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
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.2 → frogml_core-0.0.4}/frogml_proto/qwak/models/models_pb2.py

@@ -15,12 +15,13 @@ _sym_db = _symbol_database.Default()
 
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 from frogml_proto.qwak.builds import builds_pb2 as qwak_dot_builds_dot_builds__pb2
+from frogml_proto.qwak.projects import jfrog_project_spec_pb2 as qwak_dot_projects_dot_jfrog__project__spec__pb2
 from frogml_proto.qwak.audience.v1 import audience_pb2 as qwak_dot_audience_dot_v1_dot_audience__pb2
 from frogml_proto.qwak.deployment import deployment_pb2 as qwak_dot_deployment_dot_deployment__pb2
 from frogml_proto.qwak.build.v1 import build_pb2 as qwak_dot_build_dot_v1_dot_build__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18qwak/models/models.proto\x12\x16qwak.models.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x18qwak/builds/builds.proto\x1a\x1fqwak/audience/v1/audience.proto\x1a qwak/deployment/deployment.proto\x1a\x19qwak/build/v1/build.proto\"K\n\x12\x43reateModelRequest\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\"\'\n\x13\x43reateModelResponse\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\'\n\x11ListModelsRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"C\n\x12ListModelsResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\"I\n\x0fGetModelRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"@\n\x10GetModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\":\n\x12\x44\x65leteModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nproject_id\x18\x02 \x01(\t\"\x15\n\x13\x44\x65leteModelResponse\"q\n\x12UpdateModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12I\n\x11model_spec_config\x18\x02 \x01(\x0b\x32..qwak.models.management.ModelSpecConfiguration\"L\n\x13UpdateModelResponse\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\"<\n\x13\x43reateBranchRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\")\n\x14\x43reateBranchResponse\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"(\n\x13\x44\x65leteBranchRequest\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"\x16\n\x14\x44\x65leteBranchResponse\"N\n\x14IsModelExistsRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"\'\n\x15IsModelExistsResponse\x12\x0e\n\x06\x65xists\x18\x01 \x01(\x08\"\xda\x01\n\tModelSpec\x12\x16\n\x0emodel_named_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x14\n\nproject_id\x18\x03 \x01(\tH\x00\x12\x16\n\x0cproject_name\x18\x06 \x01(\tH\x00\x12\x19\n\x11model_description\x18\x04 \x01(\t\x12@\n\x10model_build_type\x18\x05 \x01(\x0b\x32&.qwak.models.management.ModelBuildTypeB\x14\n\x12project_identifier\"\xc7\x02\n\nModelBrief\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12N\n\x17\x64\x65ployment_model_status\x18\n \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\"3\n\x16ModelSpecConfiguration\x12\x19\n\x11model_description\x18\x04 \x01(\t\"\xf4\x04\n\x05Model\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x39\n\x0cmodel_status\x18\x07 \x01(\x0e\x32#.qwak.models.management.ModelStatus\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12\x30\n\x08\x62ranches\x18\n \x03(\x0b\x32\x1e.qwak.models.management.Branch\x12N\n\x17\x64\x65ployment_model_status\x18\x0b \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\x12\x0c\n\x04uuid\x18\x0c \x01(\t\x12\x42\n\nmodel_type\x18\r \x01(\x0e\x32..qwak.deployment.management.KubeDeploymentType\x12\x14\n\x0cproject_name\x18\x0e \x01(\t\x12@\n\x10model_build_type\x18\x0f \x01(\x0b\x32&.qwak.models.management.ModelBuildType\x12\x19\n\x11jfrog_project_key\x18\x10 \x01(\t\"\x96\x01\n\x0eModelBuildType\x12\x30\n\x06source\x18\x01 \x01(\x0b\x32\x1e.qwak.models.management.SourceH\x00\x12J\n\x14model_registry_image\x18\x02 \x01(\x0b\x32*.qwak.models.management.ModelRegistryImageH\x00\x42\x06\n\x04type\"\x08\n\x06Source\"3\n\x12ModelRegistryImage\x12\x1d\n\x15last_image_identifier\x18\x01 \x01(\t\"\xeb\x01\n\x06\x42ranch\x12\x11\n\tbranch_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x42\n\rbranch_status\x18\x04 \x01(\x0e\x32+.qwak.models.management.Branch.BranchStatus\x12\x10\n\x08model_id\x18\x05 \x01(\t\x12\x1d\n\x11\x64\x65ployed_build_id\x18\x06 \x01(\tB\x02\x18\x01\"D\n\x0c\x42ranchStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01\"B\n\x1bGetDeploymentDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x11\n\tbranch_id\x18\x02 \x01(\t\"e\n\x1cGetDeploymentDetailsResponse\x12\x45\n\x12\x64\x65ployment_details\x18\x01 \x01(\x0b\x32).qwak.models.management.DeploymentDetails\"\x93\x02\n\x11\x44\x65ploymentDetails\x12\x31\n\rlast_deployed\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x16\n\x0enumber_of_pods\x18\x03 \x01(\x05\x12\x14\n\x0c\x63pu_fraction\x18\x04 \x01(\x02\x12\x15\n\rmemory_amount\x18\x05 \x01(\x05\x12\x38\n\x0cmemory_units\x18\x06 \x01(\x0e\x32\".qwak.models.management.MemoryUnit\x12\x19\n\x11number_of_workers\x18\x07 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x08 \x01(\x05\"/\n\x19ListModelsMetadataRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"[\n\x1aListModelsMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x03(\x0b\x32%.qwak.models.management.ModelMetadata\"+\n\x17GetModelMetadataRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"Y\n\x18GetModelMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x01(\x0b\x32%.qwak.models.management.ModelMetadata\"h\n\x11\x45nvironmentRoutes\x12\x13\n\x0b\x65nvironment\x18\x01 \x01(\t\x12>\n\x0f\x61udience_routes\x18\x02 \x03(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry\"\xb8\x05\n\rModelMetadata\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\x12I\n\x12\x64\x65ployment_details\x18\x02 \x03(\x0b\x32-.qwak.deployment.management.DeploymentDetails\x12r\n\x1e\x61udience_routes_by_environment\x18\x03 \x03(\x0b\x32\x46.qwak.models.management.ModelMetadata.AudienceRoutesByEnvironmentEntryB\x02\x18\x01\x12\x30\n\x05\x62uild\x18\x04 \x03(\x0b\x32\x1d.qwak.builds.management.BuildB\x02\x18\x01\x12}\n&audience_routes_grouped_by_environment\x18\x05 \x03(\x0b\x32M.qwak.models.management.ModelMetadata.AudienceRoutesGroupedByEnvironmentEntry\x12(\n\x06\x62uilds\x18\x06 \x03(\x0b\x32\x18.com.qwak.build.v1.Build\x1ai\n AudienceRoutesByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry:\x02\x38\x01\x1at\n\'AudienceRoutesGroupedByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).qwak.models.management.EnvironmentRoutes:\x02\x38\x01*O\n\x15\x44\x65ploymentModelStatus\x12\x1d\n\x19\x44\x45PLOYMENT_STATUS_INVALID\x10\x00\x12\n\n\x06ONLINE\x10\x01\x12\x0b\n\x07OFFLINE\x10\x02*C\n\x0bModelStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01*7\n\nMemoryUnit\x12\x17\n\x13MEMORY_UNIT_UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x32\xe0\t\n\x17ModelsManagementService\x12\x66\n\x0b\x43reateModel\x12*.qwak.models.management.CreateModelRequest\x1a+.qwak.models.management.CreateModelResponse\x12\x63\n\nListModels\x12).qwak.models.management.ListModelsRequest\x1a*.qwak.models.management.ListModelsResponse\x12]\n\x08GetModel\x12\'.qwak.models.management.GetModelRequest\x1a(.qwak.models.management.GetModelResponse\x12\x66\n\x0b\x44\x65leteModel\x12*.qwak.models.management.DeleteModelRequest\x1a+.qwak.models.management.DeleteModelResponse\x12\x66\n\x0bUpdateModel\x12*.qwak.models.management.UpdateModelRequest\x1a+.qwak.models.management.UpdateModelResponse\x12n\n\x0c\x43reateBranch\x12+.qwak.models.management.CreateBranchRequest\x1a,.qwak.models.management.CreateBranchResponse\"\x03\x88\x02\x01\x12n\n\x0c\x44\x65leteBranch\x12+.qwak.models.management.DeleteBranchRequest\x1a,.qwak.models.management.DeleteBranchResponse\"\x03\x88\x02\x01\x12\x86\x01\n\x14GetDeploymentDetails\x12\x33.qwak.models.management.GetDeploymentDetailsRequest\x1a\x34.qwak.models.management.GetDeploymentDetailsResponse\"\x03\x88\x02\x01\x12l\n\rIsModelExists\x12,.qwak.models.management.IsModelExistsRequest\x1a-.qwak.models.management.IsModelExistsResponse\x12{\n\x12ListModelsMetadata\x12\x31.qwak.models.management.ListModelsMetadataRequest\x1a\x32.qwak.models.management.ListModelsMetadataResponse\x12u\n\x10GetModelMetadata\x12/.qwak.models.management.GetModelMetadataRequest\x1a\x30.qwak.models.management.GetModelMetadataResponseB%\n!com.qwak.ai.management.models.apiP\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18qwak/models/models.proto\x12\x16qwak.models.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x18qwak/builds/builds.proto\x1a&qwak/projects/jfrog_project_spec.proto\x1a\x1fqwak/audience/v1/audience.proto\x1a qwak/deployment/deployment.proto\x1a\x19qwak/build/v1/build.proto\"\x9b\x01\n\x12\x43reateModelRequest\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\x12N\n\x12jfrog_project_spec\x18\x02 \x01(\x0b\x32\x32.qwak.projects.management.ModelRepositoryJFrogSpec\"\'\n\x13\x43reateModelResponse\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\'\n\x11ListModelsRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"C\n\x12ListModelsResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\"I\n\x0fGetModelRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"@\n\x10GetModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\":\n\x12\x44\x65leteModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nproject_id\x18\x02 \x01(\t\"\x15\n\x13\x44\x65leteModelResponse\"q\n\x12UpdateModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12I\n\x11model_spec_config\x18\x02 \x01(\x0b\x32..qwak.models.management.ModelSpecConfiguration\"L\n\x13UpdateModelResponse\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\"<\n\x13\x43reateBranchRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\")\n\x14\x43reateBranchResponse\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"(\n\x13\x44\x65leteBranchRequest\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"\x16\n\x14\x44\x65leteBranchResponse\"N\n\x14IsModelExistsRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"\'\n\x15IsModelExistsResponse\x12\x0e\n\x06\x65xists\x18\x01 \x01(\x08\"\xda\x01\n\tModelSpec\x12\x16\n\x0emodel_named_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x14\n\nproject_id\x18\x03 \x01(\tH\x00\x12\x16\n\x0cproject_name\x18\x06 \x01(\tH\x00\x12\x19\n\x11model_description\x18\x04 \x01(\t\x12@\n\x10model_build_type\x18\x05 \x01(\x0b\x32&.qwak.models.management.ModelBuildTypeB\x14\n\x12project_identifier\"\xc7\x02\n\nModelBrief\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12N\n\x17\x64\x65ployment_model_status\x18\n \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\"3\n\x16ModelSpecConfiguration\x12\x19\n\x11model_description\x18\x04 \x01(\t\"\xf4\x04\n\x05Model\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x39\n\x0cmodel_status\x18\x07 \x01(\x0e\x32#.qwak.models.management.ModelStatus\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12\x30\n\x08\x62ranches\x18\n \x03(\x0b\x32\x1e.qwak.models.management.Branch\x12N\n\x17\x64\x65ployment_model_status\x18\x0b \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\x12\x0c\n\x04uuid\x18\x0c \x01(\t\x12\x42\n\nmodel_type\x18\r \x01(\x0e\x32..qwak.deployment.management.KubeDeploymentType\x12\x14\n\x0cproject_name\x18\x0e \x01(\t\x12@\n\x10model_build_type\x18\x0f \x01(\x0b\x32&.qwak.models.management.ModelBuildType\x12\x19\n\x11jfrog_project_key\x18\x10 \x01(\t\"\x96\x01\n\x0eModelBuildType\x12\x30\n\x06source\x18\x01 \x01(\x0b\x32\x1e.qwak.models.management.SourceH\x00\x12J\n\x14model_registry_image\x18\x02 \x01(\x0b\x32*.qwak.models.management.ModelRegistryImageH\x00\x42\x06\n\x04type\"\x08\n\x06Source\"3\n\x12ModelRegistryImage\x12\x1d\n\x15last_image_identifier\x18\x01 \x01(\t\"\xeb\x01\n\x06\x42ranch\x12\x11\n\tbranch_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x42\n\rbranch_status\x18\x04 \x01(\x0e\x32+.qwak.models.management.Branch.BranchStatus\x12\x10\n\x08model_id\x18\x05 \x01(\t\x12\x1d\n\x11\x64\x65ployed_build_id\x18\x06 \x01(\tB\x02\x18\x01\"D\n\x0c\x42ranchStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01\"B\n\x1bGetDeploymentDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x11\n\tbranch_id\x18\x02 \x01(\t\"e\n\x1cGetDeploymentDetailsResponse\x12\x45\n\x12\x64\x65ployment_details\x18\x01 \x01(\x0b\x32).qwak.models.management.DeploymentDetails\"\x93\x02\n\x11\x44\x65ploymentDetails\x12\x31\n\rlast_deployed\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x16\n\x0enumber_of_pods\x18\x03 \x01(\x05\x12\x14\n\x0c\x63pu_fraction\x18\x04 \x01(\x02\x12\x15\n\rmemory_amount\x18\x05 \x01(\x05\x12\x38\n\x0cmemory_units\x18\x06 \x01(\x0e\x32\".qwak.models.management.MemoryUnit\x12\x19\n\x11number_of_workers\x18\x07 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x08 \x01(\x05\"/\n\x19ListModelsMetadataRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"[\n\x1aListModelsMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x03(\x0b\x32%.qwak.models.management.ModelMetadata\"+\n\x17GetModelMetadataRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"Y\n\x18GetModelMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x01(\x0b\x32%.qwak.models.management.ModelMetadata\"h\n\x11\x45nvironmentRoutes\x12\x13\n\x0b\x65nvironment\x18\x01 \x01(\t\x12>\n\x0f\x61udience_routes\x18\x02 \x03(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry\"\xb8\x05\n\rModelMetadata\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\x12I\n\x12\x64\x65ployment_details\x18\x02 \x03(\x0b\x32-.qwak.deployment.management.DeploymentDetails\x12r\n\x1e\x61udience_routes_by_environment\x18\x03 \x03(\x0b\x32\x46.qwak.models.management.ModelMetadata.AudienceRoutesByEnvironmentEntryB\x02\x18\x01\x12\x30\n\x05\x62uild\x18\x04 \x03(\x0b\x32\x1d.qwak.builds.management.BuildB\x02\x18\x01\x12}\n&audience_routes_grouped_by_environment\x18\x05 \x03(\x0b\x32M.qwak.models.management.ModelMetadata.AudienceRoutesGroupedByEnvironmentEntry\x12(\n\x06\x62uilds\x18\x06 \x03(\x0b\x32\x18.com.qwak.build.v1.Build\x1ai\n AudienceRoutesByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry:\x02\x38\x01\x1at\n\'AudienceRoutesGroupedByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).qwak.models.management.EnvironmentRoutes:\x02\x38\x01*O\n\x15\x44\x65ploymentModelStatus\x12\x1d\n\x19\x44\x45PLOYMENT_STATUS_INVALID\x10\x00\x12\n\n\x06ONLINE\x10\x01\x12\x0b\n\x07OFFLINE\x10\x02*C\n\x0bModelStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01*7\n\nMemoryUnit\x12\x17\n\x13MEMORY_UNIT_UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x32\xe0\t\n\x17ModelsManagementService\x12\x66\n\x0b\x43reateModel\x12*.qwak.models.management.CreateModelRequest\x1a+.qwak.models.management.CreateModelResponse\x12\x63\n\nListModels\x12).qwak.models.management.ListModelsRequest\x1a*.qwak.models.management.ListModelsResponse\x12]\n\x08GetModel\x12\'.qwak.models.management.GetModelRequest\x1a(.qwak.models.management.GetModelResponse\x12\x66\n\x0b\x44\x65leteModel\x12*.qwak.models.management.DeleteModelRequest\x1a+.qwak.models.management.DeleteModelResponse\x12\x66\n\x0bUpdateModel\x12*.qwak.models.management.UpdateModelRequest\x1a+.qwak.models.management.UpdateModelResponse\x12n\n\x0c\x43reateBranch\x12+.qwak.models.management.CreateBranchRequest\x1a,.qwak.models.management.CreateBranchResponse\"\x03\x88\x02\x01\x12n\n\x0c\x44\x65leteBranch\x12+.qwak.models.management.DeleteBranchRequest\x1a,.qwak.models.management.DeleteBranchResponse\"\x03\x88\x02\x01\x12\x86\x01\n\x14GetDeploymentDetails\x12\x33.qwak.models.management.GetDeploymentDetailsRequest\x1a\x34.qwak.models.management.GetDeploymentDetailsResponse\"\x03\x88\x02\x01\x12l\n\rIsModelExists\x12,.qwak.models.management.IsModelExistsRequest\x1a-.qwak.models.management.IsModelExistsResponse\x12{\n\x12ListModelsMetadata\x12\x31.qwak.models.management.ListModelsMetadataRequest\x1a\x32.qwak.models.management.ListModelsMetadataResponse\x12u\n\x10GetModelMetadata\x12/.qwak.models.management.GetModelMetadataRequest\x1a\x30.qwak.models.management.GetModelMetadataResponseB%\n!com.qwak.ai.management.models.apiP\x01\x62\x06proto3')
 
 _DEPLOYMENTMODELSTATUS = DESCRIPTOR.enum_types_by_name['DeploymentModelStatus']
 DeploymentModelStatus = enum_type_wrapper.EnumTypeWrapper(_DEPLOYMENTMODELSTATUS)
@@ -344,84 +345,84 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _MODELSMANAGEMENTSERVICE.methods_by_name['DeleteBranch']._serialized_options = b'\210\002\001'
   _MODELSMANAGEMENTSERVICE.methods_by_name['GetDeploymentDetails']._options = None
   _MODELSMANAGEMENTSERVICE.methods_by_name['GetDeploymentDetails']._serialized_options = b'\210\002\001'
-  _DEPLOYMENTMODELSTATUS._serialized_start=4363
-  _DEPLOYMENTMODELSTATUS._serialized_end=4442
-  _MODELSTATUS._serialized_start=4444
-  _MODELSTATUS._serialized_end=4511
-  _MEMORYUNIT._serialized_start=4513
-  _MEMORYUNIT._serialized_end=4568
-  _CREATEMODELREQUEST._serialized_start=205
-  _CREATEMODELREQUEST._serialized_end=280
-  _CREATEMODELRESPONSE._serialized_start=282
-  _CREATEMODELRESPONSE._serialized_end=321
-  _LISTMODELSREQUEST._serialized_start=323
-  _LISTMODELSREQUEST._serialized_end=362
-  _LISTMODELSRESPONSE._serialized_start=364
-  _LISTMODELSRESPONSE._serialized_end=431
-  _GETMODELREQUEST._serialized_start=433
-  _GETMODELREQUEST._serialized_end=506
-  _GETMODELRESPONSE._serialized_start=508
-  _GETMODELRESPONSE._serialized_end=572
-  _DELETEMODELREQUEST._serialized_start=574
-  _DELETEMODELREQUEST._serialized_end=632
-  _DELETEMODELRESPONSE._serialized_start=634
-  _DELETEMODELRESPONSE._serialized_end=655
-  _UPDATEMODELREQUEST._serialized_start=657
-  _UPDATEMODELREQUEST._serialized_end=770
-  _UPDATEMODELRESPONSE._serialized_start=772
-  _UPDATEMODELRESPONSE._serialized_end=848
-  _CREATEBRANCHREQUEST._serialized_start=850
-  _CREATEBRANCHREQUEST._serialized_end=910
-  _CREATEBRANCHRESPONSE._serialized_start=912
-  _CREATEBRANCHRESPONSE._serialized_end=953
-  _DELETEBRANCHREQUEST._serialized_start=955
-  _DELETEBRANCHREQUEST._serialized_end=995
-  _DELETEBRANCHRESPONSE._serialized_start=997
-  _DELETEBRANCHRESPONSE._serialized_end=1019
-  _ISMODELEXISTSREQUEST._serialized_start=1021
-  _ISMODELEXISTSREQUEST._serialized_end=1099
-  _ISMODELEXISTSRESPONSE._serialized_start=1101
-  _ISMODELEXISTSRESPONSE._serialized_end=1140
-  _MODELSPEC._serialized_start=1143
-  _MODELSPEC._serialized_end=1361
-  _MODELBRIEF._serialized_start=1364
-  _MODELBRIEF._serialized_end=1691
-  _MODELSPECCONFIGURATION._serialized_start=1693
-  _MODELSPECCONFIGURATION._serialized_end=1744
-  _MODEL._serialized_start=1747
-  _MODEL._serialized_end=2375
-  _MODELBUILDTYPE._serialized_start=2378
-  _MODELBUILDTYPE._serialized_end=2528
-  _SOURCE._serialized_start=2530
-  _SOURCE._serialized_end=2538
-  _MODELREGISTRYIMAGE._serialized_start=2540
-  _MODELREGISTRYIMAGE._serialized_end=2591
-  _BRANCH._serialized_start=2594
-  _BRANCH._serialized_end=2829
-  _BRANCH_BRANCHSTATUS._serialized_start=2761
-  _BRANCH_BRANCHSTATUS._serialized_end=2829
-  _GETDEPLOYMENTDETAILSREQUEST._serialized_start=2831
-  _GETDEPLOYMENTDETAILSREQUEST._serialized_end=2897
-  _GETDEPLOYMENTDETAILSRESPONSE._serialized_start=2899
-  _GETDEPLOYMENTDETAILSRESPONSE._serialized_end=3000
-  _DEPLOYMENTDETAILS._serialized_start=3003
-  _DEPLOYMENTDETAILS._serialized_end=3278
-  _LISTMODELSMETADATAREQUEST._serialized_start=3280
-  _LISTMODELSMETADATAREQUEST._serialized_end=3327
-  _LISTMODELSMETADATARESPONSE._serialized_start=3329
-  _LISTMODELSMETADATARESPONSE._serialized_end=3420
-  _GETMODELMETADATAREQUEST._serialized_start=3422
-  _GETMODELMETADATAREQUEST._serialized_end=3465
-  _GETMODELMETADATARESPONSE._serialized_start=3467
-  _GETMODELMETADATARESPONSE._serialized_end=3556
-  _ENVIRONMENTROUTES._serialized_start=3558
-  _ENVIRONMENTROUTES._serialized_end=3662
-  _MODELMETADATA._serialized_start=3665
-  _MODELMETADATA._serialized_end=4361
-  _MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY._serialized_start=4138
-  _MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY._serialized_end=4243
-  _MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY._serialized_start=4245
-  _MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY._serialized_end=4361
-  _MODELSMANAGEMENTSERVICE._serialized_start=4571
-  _MODELSMANAGEMENTSERVICE._serialized_end=5819
+  _DEPLOYMENTMODELSTATUS._serialized_start=4484
+  _DEPLOYMENTMODELSTATUS._serialized_end=4563
+  _MODELSTATUS._serialized_start=4565
+  _MODELSTATUS._serialized_end=4632
+  _MEMORYUNIT._serialized_start=4634
+  _MEMORYUNIT._serialized_end=4689
+  _CREATEMODELREQUEST._serialized_start=246
+  _CREATEMODELREQUEST._serialized_end=401
+  _CREATEMODELRESPONSE._serialized_start=403
+  _CREATEMODELRESPONSE._serialized_end=442
+  _LISTMODELSREQUEST._serialized_start=444
+  _LISTMODELSREQUEST._serialized_end=483
+  _LISTMODELSRESPONSE._serialized_start=485
+  _LISTMODELSRESPONSE._serialized_end=552
+  _GETMODELREQUEST._serialized_start=554
+  _GETMODELREQUEST._serialized_end=627
+  _GETMODELRESPONSE._serialized_start=629
+  _GETMODELRESPONSE._serialized_end=693
+  _DELETEMODELREQUEST._serialized_start=695
+  _DELETEMODELREQUEST._serialized_end=753
+  _DELETEMODELRESPONSE._serialized_start=755
+  _DELETEMODELRESPONSE._serialized_end=776
+  _UPDATEMODELREQUEST._serialized_start=778
+  _UPDATEMODELREQUEST._serialized_end=891
+  _UPDATEMODELRESPONSE._serialized_start=893
+  _UPDATEMODELRESPONSE._serialized_end=969
+  _CREATEBRANCHREQUEST._serialized_start=971
+  _CREATEBRANCHREQUEST._serialized_end=1031
+  _CREATEBRANCHRESPONSE._serialized_start=1033
+  _CREATEBRANCHRESPONSE._serialized_end=1074
+  _DELETEBRANCHREQUEST._serialized_start=1076
+  _DELETEBRANCHREQUEST._serialized_end=1116
+  _DELETEBRANCHRESPONSE._serialized_start=1118
+  _DELETEBRANCHRESPONSE._serialized_end=1140
+  _ISMODELEXISTSREQUEST._serialized_start=1142
+  _ISMODELEXISTSREQUEST._serialized_end=1220
+  _ISMODELEXISTSRESPONSE._serialized_start=1222
+  _ISMODELEXISTSRESPONSE._serialized_end=1261
+  _MODELSPEC._serialized_start=1264
+  _MODELSPEC._serialized_end=1482
+  _MODELBRIEF._serialized_start=1485
+  _MODELBRIEF._serialized_end=1812
+  _MODELSPECCONFIGURATION._serialized_start=1814
+  _MODELSPECCONFIGURATION._serialized_end=1865
+  _MODEL._serialized_start=1868
+  _MODEL._serialized_end=2496
+  _MODELBUILDTYPE._serialized_start=2499
+  _MODELBUILDTYPE._serialized_end=2649
+  _SOURCE._serialized_start=2651
+  _SOURCE._serialized_end=2659
+  _MODELREGISTRYIMAGE._serialized_start=2661
+  _MODELREGISTRYIMAGE._serialized_end=2712
+  _BRANCH._serialized_start=2715
+  _BRANCH._serialized_end=2950
+  _BRANCH_BRANCHSTATUS._serialized_start=2882
+  _BRANCH_BRANCHSTATUS._serialized_end=2950
+  _GETDEPLOYMENTDETAILSREQUEST._serialized_start=2952
+  _GETDEPLOYMENTDETAILSREQUEST._serialized_end=3018
+  _GETDEPLOYMENTDETAILSRESPONSE._serialized_start=3020
+  _GETDEPLOYMENTDETAILSRESPONSE._serialized_end=3121
+  _DEPLOYMENTDETAILS._serialized_start=3124
+  _DEPLOYMENTDETAILS._serialized_end=3399
+  _LISTMODELSMETADATAREQUEST._serialized_start=3401
+  _LISTMODELSMETADATAREQUEST._serialized_end=3448
+  _LISTMODELSMETADATARESPONSE._serialized_start=3450
+  _LISTMODELSMETADATARESPONSE._serialized_end=3541
+  _GETMODELMETADATAREQUEST._serialized_start=3543
+  _GETMODELMETADATAREQUEST._serialized_end=3586
+  _GETMODELMETADATARESPONSE._serialized_start=3588
+  _GETMODELMETADATARESPONSE._serialized_end=3677
+  _ENVIRONMENTROUTES._serialized_start=3679
+  _ENVIRONMENTROUTES._serialized_end=3783
+  _MODELMETADATA._serialized_start=3786
+  _MODELMETADATA._serialized_end=4482
+  _MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY._serialized_start=4259
+  _MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY._serialized_end=4364
+  _MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY._serialized_start=4366
+  _MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY._serialized_end=4482
+  _MODELSMANAGEMENTSERVICE._serialized_start=4692
+  _MODELSMANAGEMENTSERVICE._serialized_end=5940
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.2 → frogml_core-0.0.4}/frogml_proto/qwak/models/models_pb2.pyi

@@ -13,6 +13,7 @@ import frogml_proto.qwak.audience.v1.audience_pb2
 import frogml_proto.qwak.build.v1.build_pb2
 import frogml_proto.qwak.builds.builds_pb2
 import frogml_proto.qwak.deployment.deployment_pb2
+import frogml_proto.qwak.projects.jfrog_project_spec_pb2
 import sys
 import typing
 
@@ -85,16 +86,21 @@ class CreateModelRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     MODEL_SPEC_FIELD_NUMBER: builtins.int
+    JFROG_PROJECT_SPEC_FIELD_NUMBER: builtins.int
     @property
     def model_spec(self) -> global___ModelSpec:
         """Model spec to add"""
+    @property
+    def jfrog_project_spec(self) -> frogml_proto.qwak.projects.jfrog_project_spec_pb2.ModelRepositoryJFrogSpec:
+        """JFrog project spec to add in case we want to associate the created model to another project than default"""
     def __init__(
         self,
         *,
         model_spec: global___ModelSpec | None = ...,
+        jfrog_project_spec: frogml_proto.qwak.projects.jfrog_project_spec_pb2.ModelRepositoryJFrogSpec | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["model_spec", b"model_spec"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["model_spec", b"model_spec"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["jfrog_project_spec", b"jfrog_project_spec", "model_spec", b"model_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["jfrog_project_spec", b"jfrog_project_spec", "model_spec", b"model_spec"]) -> None: ...
 
 global___CreateModelRequest = CreateModelRequest