frogml-core 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

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/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/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.2.dist-info → frogml_core-0.0.4.dist-info}/METADATA

@@ -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.dist-info → frogml_core-0.0.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=eo4alKPpl6QZmpXAVU5_iaxr9CqRANJnwT4hCWeHpzA,776
+frogml_core/__init__.py,sha256=FEH4fn3sgb5td_T6APitX5ZU7tYrNApGiJUwKXfuQP4,776
 frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
 frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
 frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -415,10 +415,11 @@ frogml_core/testing/fixtures.py,sha256=tjWIvdZ2nIfNPs6VtUeGx5coJepQVMUWemKGtqUYP
 frogml_core/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/tools/logger/__init__.py,sha256=3FfLlYKajB15QA7d2AeDUTfv_UPGz7s2hucPRe_GXdA,115
 frogml_core/tools/logger/logger.py,sha256=m38m6niR7QJgcd90t7_5N6ZTKnO-cP2cRmshcbLRa5k,9677
-frogml_core/tools/logger/logging.yml,sha256=UWC2i3NVKT3j5S8_SapzqClDzXLIEAurzNIXa2tS4UA,1941
+frogml_core/tools/logger/logging.yml,sha256=E3wtIVo7JV5DPP0EjT2Ang1zhdntW9s7uMVWCygv92w,1929
 frogml_core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/utils/datetime_utils.py,sha256=3zK7PUrerMlwB7U6WeuYwMAclVMfPqBNZihNRyIVMs4,581
 frogml_core/utils/dict_utils.py,sha256=YsOSQeGpmrsm3CMnx2ZbqvzEcybk3m-0TCMXbBdBXaY,263
+frogml_core/utils/proto_utils.py,sha256=QXEOf-pmqRLaijqCVt4IHb7_b8bmG1_ghfgMmqzHrg0,1730
 frogml_core/vector_store/__init__.py,sha256=9gR2B5A1MXb2ks5VUzqrPASGpVTDuJs4uq7N7FNWOY4,120
 frogml_core/vector_store/client.py,sha256=mmchLHjR5aOZOLIz1eibmsoXSMHvcHjCtaWB24QZoAY,6067
 frogml_core/vector_store/collection.py,sha256=KKeEww8iogmHdfEeqgTHqIV9PizeiUl1coWwHJY8Glc,17150
@@ -432,10 +433,10 @@ frogml_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.py,sha256=llPIiXM7XrcYZMOb68gJNnzwMiE9UQDSalcAW8ztNlU,1656
 frogml_proto/jfml/model_version/v1/model_repository_spec_pb2.pyi,sha256=UXjFPw49S9jVVx0WXOdXv8aQMj74RGpBii6Kw1vrlls,1470
 frogml_proto/jfml/model_version/v1/model_repository_spec_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/jfml/model_version/v1/model_version_framework_pb2.py,sha256=WRhAv0s3hzW9PMoYOE5ZjT-EbStB831aMgP4cvbDPlo,2172
-frogml_proto/jfml/model_version/v1/model_version_framework_pb2.pyi,sha256=EfF5a33YUtmP7tozDHmFGUFLNCk2Uu8yurvY-U2TBdc,1329
+frogml_proto/jfml/model_version/v1/model_version_framework_pb2.py,sha256=iRVMQsfWbuKLTGW4tnOqvOaGWsd1M5Jbs0jQ9lFhd30,5336
+frogml_proto/jfml/model_version/v1/model_version_framework_pb2.pyi,sha256=st3v0aHdDXF8onrr4HuiY02GGXTNchk5UNmE9or6y9g,3648
 frogml_proto/jfml/model_version/v1/model_version_framework_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py,sha256=fPag8F2MfHRC4VSz9jpHybldNegOlYJFRDek1cRv0A8,11128
+frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py,sha256=LNsBpYeMk8TBr1Kgmzv97cKE0UYR35PB6JpgOw4tFKY,10923
 frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi,sha256=Qw0iqfYADeSmAzZTTlwO04inSKtwwF3gDYRG-0kGLg4,9477
 frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py,sha256=VKFmhXUTpz1tnnQIRaozTEEaChquv7hG0LwngkQAAs4,13302
 frogml_proto/jfml/model_version/v1/model_version_pb2.py,sha256=87CsbasbiwvPWGI1b6AfGrje_wzDCZFD8Hi7meguDjU,5797
@@ -861,8 +862,8 @@ frogml_proto/qwak/logging/log_source_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXH
 frogml_proto/qwak/model_descriptor/open_ai_descriptor_pb2.py,sha256=yArUUX9SsaOuTmFfhxGIxe_ZEFC6u39oqYBAiPro5mo,6764
 frogml_proto/qwak/model_descriptor/open_ai_descriptor_pb2.pyi,sha256=G5CaQGczrZ9kGsMbmJXMFWAwnqMZ4jPSl9KQpSq0yMo,12345
 frogml_proto/qwak/model_descriptor/open_ai_descriptor_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/qwak/models/models_pb2.py,sha256=vFkGZfot9GipWUJSkY1HWNaTQ9cLnaCJjnGvM_MYJ10,29834
-frogml_proto/qwak/models/models_pb2.pyi,sha256=bcFrmhKv6Vn593WHP3vgCts2yd3T7-ThUvTxjs4QIMs,35371
+frogml_proto/qwak/models/models_pb2.py,sha256=7JC60OCmr3_3LwG85hHo3eBPKgdMVH-5WvSArtPUIcc,30108
+frogml_proto/qwak/models/models_pb2.pyi,sha256=974RCUcyp6vHiuBlMyzdyG-j5KCqHmHRYsqhmcYhsEA,35930
 frogml_proto/qwak/models/models_pb2_grpc.py,sha256=KV9Ez2InTVv7CwFgzNzNmjmMlWQkRgSVAS1iFTrUBh8,19831
 frogml_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2.py,sha256=skfQrw_RUDMu-F0A1jUmkKrpF694hBgTC1jSNGjeuyg,10759
 frogml_proto/qwak/monitoring/v0/alerting_channel_management_service_pb2.pyi,sha256=enqAQBr8RW_FLVL1BkZ_MGjeklQkhIZZ2OU2jczePdk,6907
@@ -885,8 +886,11 @@ frogml_proto/qwak/offline/serving/v1/options_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw
 frogml_proto/qwak/offline/serving/v1/population_pb2.py,sha256=ZdZbhDf89YPYRMFdKbSZBjA7cakkSJNgz0pgv1qMisM,4073
 frogml_proto/qwak/offline/serving/v1/population_pb2.pyi,sha256=O9khD516sbdaY5FZHcDOGxTFSoOOUa__SiJN8wkteLs,3906
 frogml_proto/qwak/offline/serving/v1/population_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/qwak/projects/projects_pb2.py,sha256=nsbWAyTIqn5d9xcc6vvIdSDwIhCO8OtA26Ls9Ap1kH8,15786
-frogml_proto/qwak/projects/projects_pb2.pyi,sha256=MYjPFj0JRQ4XmbPa0r67z3ANoNnTQK5OXvXmLW2S5s8,18665
+frogml_proto/qwak/projects/jfrog_project_spec_pb2.py,sha256=dWhM1EfkKTZgmNggxMGstI9o-jEa9_hjRTn4PiOg1z0,1617
+frogml_proto/qwak/projects/jfrog_project_spec_pb2.pyi,sha256=Ovtqc19hCDGuOaaXKfMplDFnuMjVnzI7GX1lep8NrjA,1279
+frogml_proto/qwak/projects/jfrog_project_spec_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml_proto/qwak/projects/projects_pb2.py,sha256=XUqz4Q4wXoY02H1YeFRkfTymJS1hrzDGHoh_UEqHTEY,15273
+frogml_proto/qwak/projects/projects_pb2.pyi,sha256=jTswNOXBUxEBzDiLBInc64IR3ldEma-wXrZ-h9UcysM,17854
 frogml_proto/qwak/projects/projects_pb2_grpc.py,sha256=Lzofyas6Bwx2z-5kgB5MGwxp_CJ_au2k8-QgNzzunqo,9643
 frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.py,sha256=JEzeZ9_DoENW1BJgOFseqCqj3xccZSK0Wfj9y7m1HtA,19872
 frogml_proto/qwak/prompt/v1/prompt/prompt_manager_service_pb2.pyi,sha256=SPZ0n_Jc29nhIkU2UKZQlq9M5XRz-tNVCqdIHooUjm0,16539
@@ -992,10 +996,10 @@ frogml_services_mock/mocks/job_registry_service_api.py,sha256=Zd5lVM6h4jFfKHxnQA
 frogml_services_mock/mocks/kube_captain_service_api.py,sha256=WVCaoOHY-kFdS73bd7kuOssr1RAK1F6MUlJ-NO0eLfY,1596
 frogml_services_mock/mocks/logging_service.py,sha256=JRTCjJRIKlJQeTHr3Qt5gbkfxMLUPblo0Sp50COspxE,7385
 frogml_services_mock/mocks/model_management_service.py,sha256=YKNgZ65AcKTMFs1zkrfVfGOBjqPBpodTgD6r358iRMg,4167
-frogml_services_mock/mocks/model_version_manager_service.py,sha256=izF1ke3swXCIM4S5Wv_8HVZGrLK2Pn_po38YHjiT628,3665
+frogml_services_mock/mocks/model_version_manager_service.py,sha256=GoipKHH6-aN_ulcOzOoIotgT5eJM69SFMUmw4zpcs4M,3745
 frogml_services_mock/mocks/project_manager_service.py,sha256=dxjhZwqN2UjDpHybArB8O6JOBke2JnQ9OMxy_1lLLJM,3093
 frogml_services_mock/mocks/prompt_manager_service.py,sha256=dJufY5oO4SwdO3rmbz9KaOzZKz25H5Ggx3grUQ3sIHg,11369
-frogml_services_mock/mocks/repository_service_mock.py,sha256=jct78Re6rcOAoBIMi9kTVwzwNCuHh-_9tA_DaIXRUFU,1331
+frogml_services_mock/mocks/repository_service_mock.py,sha256=2KoXSZ4l1Q5kvLns_eoH_yDZQBRzgCk50yTzNFTL3Sc,1405
 frogml_services_mock/mocks/secret_service.py,sha256=8MNvM8Us5OwImLijR__Q7iphsVLOfn-E6qcL3K7fi_I,1409
 frogml_services_mock/mocks/self_service_user_service.py,sha256=3JTvekeE04KoiAuS0r4w4nI7yRQxsqDd3UJ4viJDhdc,1207
 frogml_services_mock/mocks/system_secret_service.py,sha256=IXcjXoN0L7jbxAF-7g6HO4LIdq0NZeR35DES6JorA80,1900
@@ -1008,6 +1012,6 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=_34z6rFCHFwcSni-9eyJlrH264xsL_QDNmG-EBv0zMc,20281
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_core-0.0.2.dist-info/METADATA,sha256=0Kjz_KRsvTG-AIFzlruQ2qF952q9V65nzJf1nciWTxM,2221
-frogml_core-0.0.2.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-frogml_core-0.0.2.dist-info/RECORD,,
+frogml_core-0.0.4.dist-info/METADATA,sha256=p-i36cxAoVFMZ9lytMKXYpNxNTJ-NP0y0UKLWH33dd4,2228
+frogml_core-0.0.4.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+frogml_core-0.0.4.dist-info/RECORD,,

frogml_proto/jfml/model_version/v1/model_version_framework_pb2.py

@@ -14,12 +14,17 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3jfml/model_version/v1/model_version_framework.proto\x12\x15jfml.model_version.v1\"]\n\x15ModelVersionFramework\x12<\n\x08\x63\x61tboost\x18\x01 \x01(\x0b\x32(.jfml.model_version.v1.CatboostFrameworkH\x00\x42\x06\n\x04name\"\x13\n\x11\x43\x61tboostFrameworkB%\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\"\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'
@@ -34,12 +39,57 @@ CatboostFramework = _reflection.GeneratedProtocolMessageType('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=78
-  _MODELVERSIONFRAMEWORK._serialized_end=171
-  _CATBOOSTFRAMEWORK._serialized_start=173
-  _CATBOOSTFRAMEWORK._serialized_end=192
+  _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_proto/jfml/model_version/v1/model_version_framework_pb2.pyi

@@ -18,16 +18,36 @@ 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", "name", b"name"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["catboost", b"catboost", "name", b"name"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["name", b"name"]) -> typing_extensions.Literal["catboost"] | 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
 
@@ -39,3 +59,48 @@ class CatboostFramework(google.protobuf.message.Message):
     ) -> 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_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_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_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
 

frogml_proto/qwak/projects/jfrog_project_spec_pb2.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.qwak.projects/jfrog_project_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()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&qwak/projects/jfrog_project_spec.proto\x12\x18qwak.projects.management\"S\n\x18ModelRepositoryJFrogSpec\x12\x1b\n\x11jfrog_project_key\x18\x01 \x01(\tH\x00\x42\x1a\n\x18jfrog_project_identifierB\'\n#com.qwak.ai.management.projects.apiP\x01\x62\x06proto3')
+
+
+
+_MODELREPOSITORYJFROGSPEC = DESCRIPTOR.message_types_by_name['ModelRepositoryJFrogSpec']
+ModelRepositoryJFrogSpec = _reflection.GeneratedProtocolMessageType('ModelRepositoryJFrogSpec', (_message.Message,), {
+  'DESCRIPTOR' : _MODELREPOSITORYJFROGSPEC,
+  '__module__' : 'qwak.projects.jfrog_project_spec_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.projects.management.ModelRepositoryJFrogSpec)
+  })
+_sym_db.RegisterMessage(ModelRepositoryJFrogSpec)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n#com.qwak.ai.management.projects.apiP\001'
+  _MODELREPOSITORYJFROGSPEC._serialized_start=68
+  _MODELREPOSITORYJFROGSPEC._serialized_end=151
+# @@protoc_insertion_point(module_scope)

frogml_proto/qwak/projects/jfrog_project_spec_pb2.pyi

@@ -0,0 +1,32 @@
+"""
+@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 ModelRepositoryJFrogSpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    JFROG_PROJECT_KEY_FIELD_NUMBER: builtins.int
+    jfrog_project_key: builtins.str
+    """JFrog project key"""
+    def __init__(
+        self,
+        *,
+        jfrog_project_key: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["jfrog_project_identifier", b"jfrog_project_identifier", "jfrog_project_key", b"jfrog_project_key"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["jfrog_project_identifier", b"jfrog_project_identifier", "jfrog_project_key", b"jfrog_project_key"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["jfrog_project_identifier", b"jfrog_project_identifier"]) -> typing_extensions.Literal["jfrog_project_key"] | None: ...
+
+global___ModelRepositoryJFrogSpec = ModelRepositoryJFrogSpec

frogml_proto/qwak/projects/jfrog_project_spec_pb2_grpc.py

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

frogml_proto/qwak/projects/projects_pb2.py

@@ -13,10 +13,11 @@ _sym_db = _symbol_database.Default()
 
 
 from frogml_proto.qwak.models import models_pb2 as qwak_dot_models_dot_models__pb2
+from frogml_proto.qwak.projects import jfrog_project_spec_pb2 as qwak_dot_projects_dot_jfrog__project__spec__pb2
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cqwak/projects/projects.proto\x12\x18qwak.projects.management\x1a\x18qwak/models/models.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x91\x01\n\x14\x43reateProjectRequest\x12\x14\n\x0cproject_name\x18\x01 \x01(\t\x12\x1b\n\x13project_description\x18\x02 \x01(\t\x12\x46\n\njfrog_spec\x18\x03 \x01(\x0b\x32\x32.qwak.projects.management.ModelRepositoryJFrogSpec\"O\n\x15\x43reateProjectResponse\x12\x36\n\x07project\x18\x01 \x01(\x0b\x32%.qwak.projects.management.ProjectSpec\"\x15\n\x13ListProjectsRequest\"O\n\x14ListProjectsResponse\x12\x37\n\x08projects\x18\x01 \x03(\x0b\x32%.qwak.projects.management.ProjectSpec\"=\n\x11GetProjectRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\x12\x14\n\x0cproject_name\x18\x02 \x01(\t\"H\n\x12GetProjectResponse\x12\x32\n\x07project\x18\x01 \x01(\x0b\x32!.qwak.projects.management.Project\"*\n\x14\x44\x65leteProjectRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"%\n\x15\x44\x65leteProjectResponse\x12\x0c\n\x04info\x18\x01 \x01(\t\"{\n\x14UpdateProjectRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\x12O\n\x13project_spec_config\x18\x02 \x01(\x0b\x32\x32.qwak.projects.management.ProjectSpecConfiguration\"T\n\x15UpdateProjectResponse\x12;\n\x0cproject_spec\x18\x01 \x01(\x0b\x32%.qwak.projects.management.ProjectSpec\"\xac\x01\n\x07Project\x12\x33\n\x04spec\x18\x01 \x01(\x0b\x32%.qwak.projects.management.ProjectSpec\x12\x31\n\x06models\x18\x02 \x03(\x0b\x32\x1d.qwak.models.management.ModelB\x02\x18\x01\x12\x39\n\rmodels_briefs\x18\x03 \x03(\x0b\x32\".qwak.models.management.ModelBrief\"\xeb\x03\n\x0bProjectSpec\x12\x12\n\nproject_id\x18\x01 \x01(\t\x12\x14\n\x0cproject_name\x18\x02 \x01(\t\x12\x1b\n\x13project_description\x18\x03 \x01(\t\x12\x44\n\x0eproject_status\x18\x04 \x01(\x0e\x32,.qwak.projects.management.ProjectSpec.Status\x12.\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x34\n\x10last_modified_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x08 \x01(\t\x12\x14\n\x0cmodels_count\x18\t \x01(\x05\x12\x15\n\rmodels_active\x18\n \x01(\x05\x12N\n\x0ejfrog_settings\x18\x0b \x01(\x0b\x32\x36.qwak.projects.management.ModelRepositoryJFrogSettings\">\n\x06Status\x12\x0b\n\x07INVALID\x10\x00\x12\x0c\n\x08\x44ISABLED\x10\x01\x12\r\n\tSUSPENDED\x10\x02\x12\n\n\x06\x41\x43TIVE\x10\x03\"M\n\x18ProjectSpecConfiguration\x12\x14\n\x0cproject_name\x18\x01 \x01(\t\x12\x1b\n\x13project_description\x18\x02 \x01(\t\"\xb1\x02\n\x1cModelRepositoryJFrogSettings\x12S\n\x11pull_image_secret\x18\x01 \x01(\x0b\x32\x38.qwak.projects.management.ModelRepositoryPullImageSecret\x12[\n\x15jfrog_project_details\x18\x02 \x01(\x0b\x32<.qwak.projects.management.ModelRepositoryJFrogProjectDetails\x12_\n\x1ajfrog_repositories_details\x18\x03 \x01(\x0b\x32;.qwak.projects.management.ModelRepositoryArtifactoryDetails\"@\n\x1eModelRepositoryPullImageSecret\x12\x1e\n\x16pull_image_secret_name\x18\x01 \x01(\t\"?\n\"ModelRepositoryJFrogProjectDetails\x12\x19\n\x11jfrog_project_key\x18\x01 \x01(\t\"S\n\x18ModelRepositoryJFrogSpec\x12\x1b\n\x11jfrog_project_key\x18\x01 \x01(\tH\x00\x42\x1a\n\x18jfrog_project_identifier\"\xb7\x02\n!ModelRepositoryArtifactoryDetails\x12[\n\x1b\x61rtifact_repository_details\x18\x01 \x01(\x0b\x32\x36.qwak.projects.management.ArtifactoryRepositoryDetails\x12Z\n\x1a\x64\x61taset_repository_details\x18\x02 \x01(\x0b\x32\x36.qwak.projects.management.ArtifactoryRepositoryDetails\x12Y\n\x19\x64ocker_repository_details\x18\x03 \x01(\x0b\x32\x36.qwak.projects.management.ArtifactoryRepositoryDetails\"6\n\x1c\x41rtifactoryRepositoryDetails\x12\x16\n\x0erepository_key\x18\x01 \x01(\t2\xc9\x04\n\x19ProjectsManagementService\x12p\n\rCreateProject\x12..qwak.projects.management.CreateProjectRequest\x1a/.qwak.projects.management.CreateProjectResponse\x12m\n\x0cListProjects\x12-.qwak.projects.management.ListProjectsRequest\x1a..qwak.projects.management.ListProjectsResponse\x12g\n\nGetProject\x12+.qwak.projects.management.GetProjectRequest\x1a,.qwak.projects.management.GetProjectResponse\x12p\n\rDeleteProject\x12..qwak.projects.management.DeleteProjectRequest\x1a/.qwak.projects.management.DeleteProjectResponse\x12p\n\rUpdateProject\x12..qwak.projects.management.UpdateProjectRequest\x1a/.qwak.projects.management.UpdateProjectResponseB\'\n#com.qwak.ai.management.projects.apiP\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1cqwak/projects/projects.proto\x12\x18qwak.projects.management\x1a\x18qwak/models/models.proto\x1a&qwak/projects/jfrog_project_spec.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x91\x01\n\x14\x43reateProjectRequest\x12\x14\n\x0cproject_name\x18\x01 \x01(\t\x12\x1b\n\x13project_description\x18\x02 \x01(\t\x12\x46\n\njfrog_spec\x18\x03 \x01(\x0b\x32\x32.qwak.projects.management.ModelRepositoryJFrogSpec\"O\n\x15\x43reateProjectResponse\x12\x36\n\x07project\x18\x01 \x01(\x0b\x32%.qwak.projects.management.ProjectSpec\"\x15\n\x13ListProjectsRequest\"O\n\x14ListProjectsResponse\x12\x37\n\x08projects\x18\x01 \x03(\x0b\x32%.qwak.projects.management.ProjectSpec\"=\n\x11GetProjectRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\x12\x14\n\x0cproject_name\x18\x02 \x01(\t\"H\n\x12GetProjectResponse\x12\x32\n\x07project\x18\x01 \x01(\x0b\x32!.qwak.projects.management.Project\"*\n\x14\x44\x65leteProjectRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"%\n\x15\x44\x65leteProjectResponse\x12\x0c\n\x04info\x18\x01 \x01(\t\"{\n\x14UpdateProjectRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\x12O\n\x13project_spec_config\x18\x02 \x01(\x0b\x32\x32.qwak.projects.management.ProjectSpecConfiguration\"T\n\x15UpdateProjectResponse\x12;\n\x0cproject_spec\x18\x01 \x01(\x0b\x32%.qwak.projects.management.ProjectSpec\"\xac\x01\n\x07Project\x12\x33\n\x04spec\x18\x01 \x01(\x0b\x32%.qwak.projects.management.ProjectSpec\x12\x31\n\x06models\x18\x02 \x03(\x0b\x32\x1d.qwak.models.management.ModelB\x02\x18\x01\x12\x39\n\rmodels_briefs\x18\x03 \x03(\x0b\x32\".qwak.models.management.ModelBrief\"\xeb\x03\n\x0bProjectSpec\x12\x12\n\nproject_id\x18\x01 \x01(\t\x12\x14\n\x0cproject_name\x18\x02 \x01(\t\x12\x1b\n\x13project_description\x18\x03 \x01(\t\x12\x44\n\x0eproject_status\x18\x04 \x01(\x0e\x32,.qwak.projects.management.ProjectSpec.Status\x12.\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x34\n\x10last_modified_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x08 \x01(\t\x12\x14\n\x0cmodels_count\x18\t \x01(\x05\x12\x15\n\rmodels_active\x18\n \x01(\x05\x12N\n\x0ejfrog_settings\x18\x0b \x01(\x0b\x32\x36.qwak.projects.management.ModelRepositoryJFrogSettings\">\n\x06Status\x12\x0b\n\x07INVALID\x10\x00\x12\x0c\n\x08\x44ISABLED\x10\x01\x12\r\n\tSUSPENDED\x10\x02\x12\n\n\x06\x41\x43TIVE\x10\x03\"M\n\x18ProjectSpecConfiguration\x12\x14\n\x0cproject_name\x18\x01 \x01(\t\x12\x1b\n\x13project_description\x18\x02 \x01(\t\"\xb1\x02\n\x1cModelRepositoryJFrogSettings\x12S\n\x11pull_image_secret\x18\x01 \x01(\x0b\x32\x38.qwak.projects.management.ModelRepositoryPullImageSecret\x12[\n\x15jfrog_project_details\x18\x02 \x01(\x0b\x32<.qwak.projects.management.ModelRepositoryJFrogProjectDetails\x12_\n\x1ajfrog_repositories_details\x18\x03 \x01(\x0b\x32;.qwak.projects.management.ModelRepositoryArtifactoryDetails\"@\n\x1eModelRepositoryPullImageSecret\x12\x1e\n\x16pull_image_secret_name\x18\x01 \x01(\t\"?\n\"ModelRepositoryJFrogProjectDetails\x12\x19\n\x11jfrog_project_key\x18\x01 \x01(\t\"\xb7\x02\n!ModelRepositoryArtifactoryDetails\x12[\n\x1b\x61rtifact_repository_details\x18\x01 \x01(\x0b\x32\x36.qwak.projects.management.ArtifactoryRepositoryDetails\x12Z\n\x1a\x64\x61taset_repository_details\x18\x02 \x01(\x0b\x32\x36.qwak.projects.management.ArtifactoryRepositoryDetails\x12Y\n\x19\x64ocker_repository_details\x18\x03 \x01(\x0b\x32\x36.qwak.projects.management.ArtifactoryRepositoryDetails\"6\n\x1c\x41rtifactoryRepositoryDetails\x12\x16\n\x0erepository_key\x18\x01 \x01(\t2\xc9\x04\n\x19ProjectsManagementService\x12p\n\rCreateProject\x12..qwak.projects.management.CreateProjectRequest\x1a/.qwak.projects.management.CreateProjectResponse\x12m\n\x0cListProjects\x12-.qwak.projects.management.ListProjectsRequest\x1a..qwak.projects.management.ListProjectsResponse\x12g\n\nGetProject\x12+.qwak.projects.management.GetProjectRequest\x1a,.qwak.projects.management.GetProjectResponse\x12p\n\rDeleteProject\x12..qwak.projects.management.DeleteProjectRequest\x1a/.qwak.projects.management.DeleteProjectResponse\x12p\n\rUpdateProject\x12..qwak.projects.management.UpdateProjectRequest\x1a/.qwak.projects.management.UpdateProjectResponseB\'\n#com.qwak.ai.management.projects.apiP\x01\x62\x06proto3')
 
 
 
@@ -36,7 +37,6 @@ _PROJECTSPECCONFIGURATION = DESCRIPTOR.message_types_by_name['ProjectSpecConfigu
 _MODELREPOSITORYJFROGSETTINGS = DESCRIPTOR.message_types_by_name['ModelRepositoryJFrogSettings']
 _MODELREPOSITORYPULLIMAGESECRET = DESCRIPTOR.message_types_by_name['ModelRepositoryPullImageSecret']
 _MODELREPOSITORYJFROGPROJECTDETAILS = DESCRIPTOR.message_types_by_name['ModelRepositoryJFrogProjectDetails']
-_MODELREPOSITORYJFROGSPEC = DESCRIPTOR.message_types_by_name['ModelRepositoryJFrogSpec']
 _MODELREPOSITORYARTIFACTORYDETAILS = DESCRIPTOR.message_types_by_name['ModelRepositoryArtifactoryDetails']
 _ARTIFACTORYREPOSITORYDETAILS = DESCRIPTOR.message_types_by_name['ArtifactoryRepositoryDetails']
 _PROJECTSPEC_STATUS = _PROJECTSPEC.enum_types_by_name['Status']
@@ -152,13 +152,6 @@ ModelRepositoryJFrogProjectDetails = _reflection.GeneratedProtocolMessageType('M
   })
 _sym_db.RegisterMessage(ModelRepositoryJFrogProjectDetails)
 
-ModelRepositoryJFrogSpec = _reflection.GeneratedProtocolMessageType('ModelRepositoryJFrogSpec', (_message.Message,), {
-  'DESCRIPTOR' : _MODELREPOSITORYJFROGSPEC,
-  '__module__' : 'qwak.projects.projects_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.projects.management.ModelRepositoryJFrogSpec)
-  })
-_sym_db.RegisterMessage(ModelRepositoryJFrogSpec)
-
 ModelRepositoryArtifactoryDetails = _reflection.GeneratedProtocolMessageType('ModelRepositoryArtifactoryDetails', (_message.Message,), {
   'DESCRIPTOR' : _MODELREPOSITORYARTIFACTORYDETAILS,
   '__module__' : 'qwak.projects.projects_pb2'
@@ -180,46 +173,44 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   DESCRIPTOR._serialized_options = b'\n#com.qwak.ai.management.projects.apiP\001'
   _PROJECT.fields_by_name['models']._options = None
   _PROJECT.fields_by_name['models']._serialized_options = b'\030\001'
-  _CREATEPROJECTREQUEST._serialized_start=118
-  _CREATEPROJECTREQUEST._serialized_end=263
-  _CREATEPROJECTRESPONSE._serialized_start=265
-  _CREATEPROJECTRESPONSE._serialized_end=344
-  _LISTPROJECTSREQUEST._serialized_start=346
-  _LISTPROJECTSREQUEST._serialized_end=367
-  _LISTPROJECTSRESPONSE._serialized_start=369
-  _LISTPROJECTSRESPONSE._serialized_end=448
-  _GETPROJECTREQUEST._serialized_start=450
-  _GETPROJECTREQUEST._serialized_end=511
-  _GETPROJECTRESPONSE._serialized_start=513
-  _GETPROJECTRESPONSE._serialized_end=585
-  _DELETEPROJECTREQUEST._serialized_start=587
-  _DELETEPROJECTREQUEST._serialized_end=629
-  _DELETEPROJECTRESPONSE._serialized_start=631
-  _DELETEPROJECTRESPONSE._serialized_end=668
-  _UPDATEPROJECTREQUEST._serialized_start=670
-  _UPDATEPROJECTREQUEST._serialized_end=793
-  _UPDATEPROJECTRESPONSE._serialized_start=795
-  _UPDATEPROJECTRESPONSE._serialized_end=879
-  _PROJECT._serialized_start=882
-  _PROJECT._serialized_end=1054
-  _PROJECTSPEC._serialized_start=1057
-  _PROJECTSPEC._serialized_end=1548
-  _PROJECTSPEC_STATUS._serialized_start=1486
-  _PROJECTSPEC_STATUS._serialized_end=1548
-  _PROJECTSPECCONFIGURATION._serialized_start=1550
-  _PROJECTSPECCONFIGURATION._serialized_end=1627
-  _MODELREPOSITORYJFROGSETTINGS._serialized_start=1630
-  _MODELREPOSITORYJFROGSETTINGS._serialized_end=1935
-  _MODELREPOSITORYPULLIMAGESECRET._serialized_start=1937
-  _MODELREPOSITORYPULLIMAGESECRET._serialized_end=2001
-  _MODELREPOSITORYJFROGPROJECTDETAILS._serialized_start=2003
-  _MODELREPOSITORYJFROGPROJECTDETAILS._serialized_end=2066
-  _MODELREPOSITORYJFROGSPEC._serialized_start=2068
-  _MODELREPOSITORYJFROGSPEC._serialized_end=2151
-  _MODELREPOSITORYARTIFACTORYDETAILS._serialized_start=2154
-  _MODELREPOSITORYARTIFACTORYDETAILS._serialized_end=2465
-  _ARTIFACTORYREPOSITORYDETAILS._serialized_start=2467
-  _ARTIFACTORYREPOSITORYDETAILS._serialized_end=2521
-  _PROJECTSMANAGEMENTSERVICE._serialized_start=2524
-  _PROJECTSMANAGEMENTSERVICE._serialized_end=3109
+  _CREATEPROJECTREQUEST._serialized_start=158
+  _CREATEPROJECTREQUEST._serialized_end=303
+  _CREATEPROJECTRESPONSE._serialized_start=305
+  _CREATEPROJECTRESPONSE._serialized_end=384
+  _LISTPROJECTSREQUEST._serialized_start=386
+  _LISTPROJECTSREQUEST._serialized_end=407
+  _LISTPROJECTSRESPONSE._serialized_start=409
+  _LISTPROJECTSRESPONSE._serialized_end=488
+  _GETPROJECTREQUEST._serialized_start=490
+  _GETPROJECTREQUEST._serialized_end=551
+  _GETPROJECTRESPONSE._serialized_start=553
+  _GETPROJECTRESPONSE._serialized_end=625
+  _DELETEPROJECTREQUEST._serialized_start=627
+  _DELETEPROJECTREQUEST._serialized_end=669
+  _DELETEPROJECTRESPONSE._serialized_start=671
+  _DELETEPROJECTRESPONSE._serialized_end=708
+  _UPDATEPROJECTREQUEST._serialized_start=710
+  _UPDATEPROJECTREQUEST._serialized_end=833
+  _UPDATEPROJECTRESPONSE._serialized_start=835
+  _UPDATEPROJECTRESPONSE._serialized_end=919
+  _PROJECT._serialized_start=922
+  _PROJECT._serialized_end=1094
+  _PROJECTSPEC._serialized_start=1097
+  _PROJECTSPEC._serialized_end=1588
+  _PROJECTSPEC_STATUS._serialized_start=1526
+  _PROJECTSPEC_STATUS._serialized_end=1588
+  _PROJECTSPECCONFIGURATION._serialized_start=1590
+  _PROJECTSPECCONFIGURATION._serialized_end=1667
+  _MODELREPOSITORYJFROGSETTINGS._serialized_start=1670
+  _MODELREPOSITORYJFROGSETTINGS._serialized_end=1975
+  _MODELREPOSITORYPULLIMAGESECRET._serialized_start=1977
+  _MODELREPOSITORYPULLIMAGESECRET._serialized_end=2041
+  _MODELREPOSITORYJFROGPROJECTDETAILS._serialized_start=2043
+  _MODELREPOSITORYJFROGPROJECTDETAILS._serialized_end=2106
+  _MODELREPOSITORYARTIFACTORYDETAILS._serialized_start=2109
+  _MODELREPOSITORYARTIFACTORYDETAILS._serialized_end=2420
+  _ARTIFACTORYREPOSITORYDETAILS._serialized_start=2422
+  _ARTIFACTORYREPOSITORYDETAILS._serialized_end=2476
+  _PROJECTSMANAGEMENTSERVICE._serialized_start=2479
+  _PROJECTSMANAGEMENTSERVICE._serialized_end=3064
 # @@protoc_insertion_point(module_scope)

frogml_proto/qwak/projects/projects_pb2.pyi

@@ -10,6 +10,7 @@ import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
 import google.protobuf.timestamp_pb2
 import frogml_proto.qwak.models.models_pb2
+import frogml_proto.qwak.projects.jfrog_project_spec_pb2
 import sys
 import typing
 
@@ -31,14 +32,14 @@ class CreateProjectRequest(google.protobuf.message.Message):
     project_description: builtins.str
     """Project description"""
     @property
-    def jfrog_spec(self) -> global___ModelRepositoryJFrogSpec:
+    def jfrog_spec(self) -> frogml_proto.qwak.projects.jfrog_project_spec_pb2.ModelRepositoryJFrogSpec:
         """jfrog spec"""
     def __init__(
         self,
         *,
         project_name: builtins.str = ...,
         project_description: builtins.str = ...,
-        jfrog_spec: global___ModelRepositoryJFrogSpec | None = ...,
+        jfrog_spec: frogml_proto.qwak.projects.jfrog_project_spec_pb2.ModelRepositoryJFrogSpec | None = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["jfrog_spec", b"jfrog_spec"]) -> builtins.bool: ...
     def ClearField(self, field_name: typing_extensions.Literal["jfrog_spec", b"jfrog_spec", "project_description", b"project_description", "project_name", b"project_name"]) -> None: ...
@@ -376,23 +377,6 @@ class ModelRepositoryJFrogProjectDetails(google.protobuf.message.Message):
 
 global___ModelRepositoryJFrogProjectDetails = ModelRepositoryJFrogProjectDetails
 
-class ModelRepositoryJFrogSpec(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    JFROG_PROJECT_KEY_FIELD_NUMBER: builtins.int
-    jfrog_project_key: builtins.str
-    """JFrog project key"""
-    def __init__(
-        self,
-        *,
-        jfrog_project_key: builtins.str = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["jfrog_project_identifier", b"jfrog_project_identifier", "jfrog_project_key", b"jfrog_project_key"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["jfrog_project_identifier", b"jfrog_project_identifier", "jfrog_project_key", b"jfrog_project_key"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["jfrog_project_identifier", b"jfrog_project_identifier"]) -> typing_extensions.Literal["jfrog_project_key"] | None: ...
-
-global___ModelRepositoryJFrogSpec = ModelRepositoryJFrogSpec
-
 class ModelRepositoryArtifactoryDetails(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

frogml_services_mock/mocks/model_version_manager_service.py

@@ -24,7 +24,7 @@ from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc i
 from frogml_proto.jfml.model_version.v1.model_version_pb2 import (
     ModelVersion,
     ModelVersionCompletedStatus,
-    ModelVersionSpec,
+    ModelVersionSpec, ModelVersionStatus,
 )
 
 timestamp = Timestamp()
@@ -41,7 +41,7 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
             model_version=ModelVersion(
                 model_version_id=str(uuid.uuid4()),
                 spec=request.model_version,
-                status=ModelVersionCompletedStatus(),
+                status=ModelVersionStatus(completed=ModelVersionCompletedStatus()),
                 created_by="mock",
                 created_at=timestamp,
                 updated_at=timestamp,
@@ -62,7 +62,7 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
                     framework=CatboostFramework(),
                     python_version="3.9",
                 ),
-                status=ModelVersionCompletedStatus(),
+                status=ModelVersionStatus(completed=ModelVersionCompletedStatus()),
                 created_by="mock",
                 created_at=timestamp,
                 updated_at=timestamp,

frogml_services_mock/mocks/repository_service_mock.py

@@ -36,4 +36,6 @@ class RepositoryServiceMock(RepositoryServiceServicer):
                 FrogmlNotFoundException(f"Repository {repository_key} not found"),
             )
 
-        return self.repositories[repository_key]
+        return GetRepositoryConfigurationResponse(
+            repository_spec=self.repositories[repository_key]
+        )