frogml-core 0.0.6__py3-none-any.whl → 0.0.8__py3-none-any.whl

frogml_core/__init__.py

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

frogml_core/utils/proto_utils.py

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

{frogml_core-0.0.6.dist-info → frogml_core-0.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.6
+Version: 0.0.8
 Summary: frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{frogml_core-0.0.6.dist-info → frogml_core-0.0.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=YrxSzDS2w-mX_wxtVwIoK6vNGQ32nDj-A7Pv5GAq97A,776
+frogml_core/__init__.py,sha256=JhwoSbF47ceA8YU8-lN8O9gtjsC_Eymxflr02nb2ioI,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
@@ -420,7 +420,7 @@ frogml_core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 frogml_core/utils/datetime_utils.py,sha256=3zK7PUrerMlwB7U6WeuYwMAclVMfPqBNZihNRyIVMs4,581
 frogml_core/utils/dict_utils.py,sha256=YsOSQeGpmrsm3CMnx2ZbqvzEcybk3m-0TCMXbBdBXaY,263
 frogml_core/utils/model_utils.py,sha256=fLdYIScmVUfChGGxOacH2mID00ZUOPtS4ewUimEDGhg,132
-frogml_core/utils/proto_utils.py,sha256=QXEOf-pmqRLaijqCVt4IHb7_b8bmG1_ghfgMmqzHrg0,1730
+frogml_core/utils/proto_utils.py,sha256=p0ry5k7iwNT6btEdSBedcfvAKRBmC5X5oKmUz27_E6k,1779
 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
@@ -431,15 +431,18 @@ frogml_core/vector_store/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 frogml_core/vector_store/utils/filter_utils.py,sha256=n80IlmS7lGelCNEpZ8o3rAHzrZpHc-3EykEd8tZoeAE,854
 frogml_core/vector_store/utils/upsert_utils.py,sha256=vqnyAJOl46WqpZMUiOa4I8Zf2CDIXQxR7wM4Z8rgooc,7517
 frogml_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+frogml_proto/jfml/model_version/v1/build_spec_pb2.py,sha256=7FeOfvGtH2v4swm0FGl8ALvUgR4ckIcn-pSkzYd-uhQ,3779
+frogml_proto/jfml/model_version/v1/build_spec_pb2.pyi,sha256=CIBEscG5MQoOGddvsfGqxD2K4U342IS-kFpzyYFiFD8,4923
+frogml_proto/jfml/model_version/v1/build_spec_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 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=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.py,sha256=Vk4NO9dlhXkNVgxdLGDfh20s-mBfcoOOZEEr6hUtLag,5370
+frogml_proto/jfml/model_version/v1/model_version_framework_pb2.pyi,sha256=JgOE0Im69ir0PSgpbg6Zz4scRK_u7WAxWTtWcdzfyAw,3773
 frogml_proto/jfml/model_version/v1/model_version_framework_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py,sha256=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_manager_service_pb2.py,sha256=-QakaXaKJLZX68AuiYqyBGNFEy2i6EralNxddhFQy7Y,12964
+frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi,sha256=LE4_HRvT_zDjubYt8cRgGrYqtHcAEh1TVSmA8VzsUFc,11863
+frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py,sha256=qx5iX5FK7k8ICpD-T13T4wGDX7dSyVyOPe_O8U6TXSg,15434
 frogml_proto/jfml/model_version/v1/model_version_pb2.py,sha256=87CsbasbiwvPWGI1b6AfGrje_wzDCZFD8Hi7meguDjU,5797
 frogml_proto/jfml/model_version/v1/model_version_pb2.pyi,sha256=NqHepmIm1HiKBYn77JDFOoBAEVwCbyFsj0Wiurhsxkc,5664
 frogml_proto/jfml/model_version/v1/model_version_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -686,7 +689,7 @@ frogml_proto/qwak/feature_store/features/feature_set_state_pb2_grpc.py,sha256=1o
 frogml_proto/qwak/feature_store/features/feature_set_state_service_pb2.py,sha256=GMvhHAT4EyiAzkXuGBdrLIe_9vXOJqHynfP7LcoHnQY,10058
 frogml_proto/qwak/feature_store/features/feature_set_state_service_pb2.pyi,sha256=B4n0NG9y_johGTypJLSvl8k2_DtNCI8ZsDzGQ2oVkaw,6911
 frogml_proto/qwak/feature_store/features/feature_set_state_service_pb2_grpc.py,sha256=cZRumlpDjHGjb_6IvqgVM45ns3U7yoW-0KX8_XJn628,11648
-frogml_proto/qwak/feature_store/features/feature_set_types_pb2.py,sha256=TuilWjwfxIJ-1mGSLs7jQ8nWfQTMca2odr7LMgwVolo,29514
+frogml_proto/qwak/feature_store/features/feature_set_types_pb2.py,sha256=urSMcPxapWM_trfNcGumwCHxcR2FMva03pevhiUuK4s,29808
 frogml_proto/qwak/feature_store/features/feature_set_types_pb2.pyi,sha256=Q8Xc-gAwsKZm0UjAYo58lYSZdsavM6QY1eTsjboF1To,43855
 frogml_proto/qwak/feature_store/features/feature_set_types_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml_proto/qwak/feature_store/features/monitoring_pb2.py,sha256=IVPbVitd3_qYTKaVbxYAUPk6oN4BtkXvMpBuk2TA5ME,2885
@@ -1013,6 +1016,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.6.dist-info/METADATA,sha256=Z9dDG7CFuEb9D-pxmi7Lkcv-ORCfXBNMqdOY8TW_T8Q,2105
-frogml_core-0.0.6.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-frogml_core-0.0.6.dist-info/RECORD,,
+frogml_core-0.0.8.dist-info/METADATA,sha256=il6AZcJlu5LVDij3EQQSeO-Yriah3rXnwXByh6PJyIc,2105
+frogml_core-0.0.8.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+frogml_core-0.0.8.dist-info/RECORD,,

frogml_proto/jfml/model_version/v1/build_spec_pb2.py

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

frogml_proto/jfml/model_version/v1/build_spec_pb2.pyi

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

frogml_proto/jfml/model_version/v1/build_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/jfml/model_version/v1/model_version_framework_pb2.py

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

frogml_proto/jfml/model_version/v1/model_version_framework_pb2.pyi

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

frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py

@@ -14,9 +14,10 @@ _sym_db = _symbol_database.Default()
 
 from frogml_proto.jfml.model_version.v1 import model_version_pb2 as jfml_dot_model__version_dot_v1_dot_model__version__pb2
 from frogml_proto.jfml.model_version.v1 import model_repository_spec_pb2 as jfml_dot_model__version_dot_v1_dot_model__repository__spec__pb2
+from frogml_proto.jfml.model_version.v1 import build_spec_pb2 as jfml_dot_model__version_dot_v1_dot_build__spec__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n9jfml/model_version/v1/model_version_manager_service.proto\x12\x15jfml.model_version.v1\x1a)jfml/model_version/v1/model_version.proto\x1a\x31jfml/model_version/v1/model_repository_spec.proto\"l\n\x19\x43reateModelVersionRequest\x12\x0f\n\x07\x64ry_run\x18\x01 \x01(\x08\x12>\n\rmodel_version\x18\x02 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpec\"X\n\x1a\x43reateModelVersionResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\"6\n\x1aGetModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\"Y\n\x1bGetModelVersionByIdResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\"L\n\x1cGetModelVersionByNameRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x1a\n\x12model_version_name\x18\x02 \x01(\t\"[\n\x1dGetModelVersionByNameResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\")\n\x15InitPaginationRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"E\n\x16InitPaginationResponse\x12\x14\n\x0crecord_count\x18\x01 \x01(\x03\x12\x15\n\rmax_record_id\x18\x02 \x01(\x03\"\x9e\x01\n\x18ListModelVersionsRequest\x12\x43\n\x0frepository_spec\x18\x01 \x01(\x0b\x32*.jfml.model_version.v1.ModelRepositorySpec\x12\x11\n\tpage_size\x18\x02 \x01(\x05\x12\x13\n\x0bpage_number\x18\x03 \x01(\x05\x12\x15\n\rmax_record_id\x18\x04 \x01(\x03\"X\n\x19ListModelVersionsResponse\x12;\n\x0emodel_versions\x18\x01 \x03(\x0b\x32#.jfml.model_version.v1.ModelVersion\"9\n\x1d\x44\x65leteModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\" \n\x1e\x44\x65leteModelVersionByIdResponse2\x89\x06\n\x1aModelVersionManagerService\x12y\n\x12\x43reateModelVersion\x12\x30.jfml.model_version.v1.CreateModelVersionRequest\x1a\x31.jfml.model_version.v1.CreateModelVersionResponse\x12|\n\x13GetModelVersionById\x12\x31.jfml.model_version.v1.GetModelVersionByIdRequest\x1a\x32.jfml.model_version.v1.GetModelVersionByIdResponse\x12\x82\x01\n\x15GetModelVersionByName\x12\x33.jfml.model_version.v1.GetModelVersionByNameRequest\x1a\x34.jfml.model_version.v1.GetModelVersionByNameResponse\x12m\n\x0eInitPagination\x12,.jfml.model_version.v1.InitPaginationRequest\x1a-.jfml.model_version.v1.InitPaginationResponse\x12v\n\x11ListModelVersions\x12/.jfml.model_version.v1.ListModelVersionsRequest\x1a\x30.jfml.model_version.v1.ListModelVersionsResponse\x12\x85\x01\n\x16\x44\x65leteModelVersionById\x12\x34.jfml.model_version.v1.DeleteModelVersionByIdRequest\x1a\x35.jfml.model_version.v1.DeleteModelVersionByIdResponseB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n9jfml/model_version/v1/model_version_manager_service.proto\x12\x15jfml.model_version.v1\x1a)jfml/model_version/v1/model_version.proto\x1a\x31jfml/model_version/v1/model_repository_spec.proto\x1a&jfml/model_version/v1/build_spec.proto\"l\n\x19\x43reateModelVersionRequest\x12\x0f\n\x07\x64ry_run\x18\x01 \x01(\x08\x12>\n\rmodel_version\x18\x02 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpec\"X\n\x1a\x43reateModelVersionResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\"6\n\x1aGetModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\"Y\n\x1bGetModelVersionByIdResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\"L\n\x1cGetModelVersionByNameRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x1a\n\x12model_version_name\x18\x02 \x01(\t\"[\n\x1dGetModelVersionByNameResponse\x12:\n\rmodel_version\x18\x01 \x01(\x0b\x32#.jfml.model_version.v1.ModelVersion\")\n\x15InitPaginationRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"E\n\x16InitPaginationResponse\x12\x14\n\x0crecord_count\x18\x01 \x01(\x03\x12\x15\n\rmax_record_id\x18\x02 \x01(\x03\"\x9e\x01\n\x18ListModelVersionsRequest\x12\x43\n\x0frepository_spec\x18\x01 \x01(\x0b\x32*.jfml.model_version.v1.ModelRepositorySpec\x12\x11\n\tpage_size\x18\x02 \x01(\x05\x12\x13\n\x0bpage_number\x18\x03 \x01(\x05\x12\x15\n\rmax_record_id\x18\x04 \x01(\x03\"X\n\x19ListModelVersionsResponse\x12;\n\x0emodel_versions\x18\x01 \x03(\x0b\x32#.jfml.model_version.v1.ModelVersion\"9\n\x1d\x44\x65leteModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\" \n\x1e\x44\x65leteModelVersionByIdResponse\"\xd8\x01\n!PromoteModelVersionToBuildRequest\x12\x45\n\x12model_version_spec\x18\x01 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpecH\x00\x12\x1a\n\x10model_version_id\x18\x02 \x01(\tH\x00\x12\x34\n\nbuild_spec\x18\x03 \x01(\x0b\x32 .jfml.model_version.v1.BuildSpecB\x1a\n\x18model_version_identifier\"6\n\"PromoteModelVersionToBuildResponse\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t2\x9d\x07\n\x1aModelVersionManagerService\x12y\n\x12\x43reateModelVersion\x12\x30.jfml.model_version.v1.CreateModelVersionRequest\x1a\x31.jfml.model_version.v1.CreateModelVersionResponse\x12|\n\x13GetModelVersionById\x12\x31.jfml.model_version.v1.GetModelVersionByIdRequest\x1a\x32.jfml.model_version.v1.GetModelVersionByIdResponse\x12\x82\x01\n\x15GetModelVersionByName\x12\x33.jfml.model_version.v1.GetModelVersionByNameRequest\x1a\x34.jfml.model_version.v1.GetModelVersionByNameResponse\x12m\n\x0eInitPagination\x12,.jfml.model_version.v1.InitPaginationRequest\x1a-.jfml.model_version.v1.InitPaginationResponse\x12v\n\x11ListModelVersions\x12/.jfml.model_version.v1.ListModelVersionsRequest\x1a\x30.jfml.model_version.v1.ListModelVersionsResponse\x12\x85\x01\n\x16\x44\x65leteModelVersionById\x12\x34.jfml.model_version.v1.DeleteModelVersionByIdRequest\x1a\x35.jfml.model_version.v1.DeleteModelVersionByIdResponse\x12\x91\x01\n\x1aPromoteModelVersionToBuild\x12\x38.jfml.model_version.v1.PromoteModelVersionToBuildRequest\x1a\x39.jfml.model_version.v1.PromoteModelVersionToBuildResponseB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
 
 
 
@@ -32,6 +33,8 @@ _LISTMODELVERSIONSREQUEST = DESCRIPTOR.message_types_by_name['ListModelVersionsR
 _LISTMODELVERSIONSRESPONSE = DESCRIPTOR.message_types_by_name['ListModelVersionsResponse']
 _DELETEMODELVERSIONBYIDREQUEST = DESCRIPTOR.message_types_by_name['DeleteModelVersionByIdRequest']
 _DELETEMODELVERSIONBYIDRESPONSE = DESCRIPTOR.message_types_by_name['DeleteModelVersionByIdResponse']
+_PROMOTEMODELVERSIONTOBUILDREQUEST = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildRequest']
+_PROMOTEMODELVERSIONTOBUILDRESPONSE = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildResponse']
 CreateModelVersionRequest = _reflection.GeneratedProtocolMessageType('CreateModelVersionRequest', (_message.Message,), {
   'DESCRIPTOR' : _CREATEMODELVERSIONREQUEST,
   '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
@@ -116,35 +119,53 @@ DeleteModelVersionByIdResponse = _reflection.GeneratedProtocolMessageType('Delet
   })
 _sym_db.RegisterMessage(DeleteModelVersionByIdResponse)
 
+PromoteModelVersionToBuildRequest = _reflection.GeneratedProtocolMessageType('PromoteModelVersionToBuildRequest', (_message.Message,), {
+  'DESCRIPTOR' : _PROMOTEMODELVERSIONTOBUILDREQUEST,
+  '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.PromoteModelVersionToBuildRequest)
+  })
+_sym_db.RegisterMessage(PromoteModelVersionToBuildRequest)
+
+PromoteModelVersionToBuildResponse = _reflection.GeneratedProtocolMessageType('PromoteModelVersionToBuildResponse', (_message.Message,), {
+  'DESCRIPTOR' : _PROMOTEMODELVERSIONTOBUILDRESPONSE,
+  '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.PromoteModelVersionToBuildResponse)
+  })
+_sym_db.RegisterMessage(PromoteModelVersionToBuildResponse)
+
 _MODELVERSIONMANAGERSERVICE = DESCRIPTOR.services_by_name['ModelVersionManagerService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
-  _CREATEMODELVERSIONREQUEST._serialized_start=178
-  _CREATEMODELVERSIONREQUEST._serialized_end=286
-  _CREATEMODELVERSIONRESPONSE._serialized_start=288
-  _CREATEMODELVERSIONRESPONSE._serialized_end=376
-  _GETMODELVERSIONBYIDREQUEST._serialized_start=378
-  _GETMODELVERSIONBYIDREQUEST._serialized_end=432
-  _GETMODELVERSIONBYIDRESPONSE._serialized_start=434
-  _GETMODELVERSIONBYIDRESPONSE._serialized_end=523
-  _GETMODELVERSIONBYNAMEREQUEST._serialized_start=525
-  _GETMODELVERSIONBYNAMEREQUEST._serialized_end=601
-  _GETMODELVERSIONBYNAMERESPONSE._serialized_start=603
-  _GETMODELVERSIONBYNAMERESPONSE._serialized_end=694
-  _INITPAGINATIONREQUEST._serialized_start=696
-  _INITPAGINATIONREQUEST._serialized_end=737
-  _INITPAGINATIONRESPONSE._serialized_start=739
-  _INITPAGINATIONRESPONSE._serialized_end=808
-  _LISTMODELVERSIONSREQUEST._serialized_start=811
-  _LISTMODELVERSIONSREQUEST._serialized_end=969
-  _LISTMODELVERSIONSRESPONSE._serialized_start=971
-  _LISTMODELVERSIONSRESPONSE._serialized_end=1059
-  _DELETEMODELVERSIONBYIDREQUEST._serialized_start=1061
-  _DELETEMODELVERSIONBYIDREQUEST._serialized_end=1118
-  _DELETEMODELVERSIONBYIDRESPONSE._serialized_start=1120
-  _DELETEMODELVERSIONBYIDRESPONSE._serialized_end=1152
-  _MODELVERSIONMANAGERSERVICE._serialized_start=1155
-  _MODELVERSIONMANAGERSERVICE._serialized_end=1932
+  _CREATEMODELVERSIONREQUEST._serialized_start=218
+  _CREATEMODELVERSIONREQUEST._serialized_end=326
+  _CREATEMODELVERSIONRESPONSE._serialized_start=328
+  _CREATEMODELVERSIONRESPONSE._serialized_end=416
+  _GETMODELVERSIONBYIDREQUEST._serialized_start=418
+  _GETMODELVERSIONBYIDREQUEST._serialized_end=472
+  _GETMODELVERSIONBYIDRESPONSE._serialized_start=474
+  _GETMODELVERSIONBYIDRESPONSE._serialized_end=563
+  _GETMODELVERSIONBYNAMEREQUEST._serialized_start=565
+  _GETMODELVERSIONBYNAMEREQUEST._serialized_end=641
+  _GETMODELVERSIONBYNAMERESPONSE._serialized_start=643
+  _GETMODELVERSIONBYNAMERESPONSE._serialized_end=734
+  _INITPAGINATIONREQUEST._serialized_start=736
+  _INITPAGINATIONREQUEST._serialized_end=777
+  _INITPAGINATIONRESPONSE._serialized_start=779
+  _INITPAGINATIONRESPONSE._serialized_end=848
+  _LISTMODELVERSIONSREQUEST._serialized_start=851
+  _LISTMODELVERSIONSREQUEST._serialized_end=1009
+  _LISTMODELVERSIONSRESPONSE._serialized_start=1011
+  _LISTMODELVERSIONSRESPONSE._serialized_end=1099
+  _DELETEMODELVERSIONBYIDREQUEST._serialized_start=1101
+  _DELETEMODELVERSIONBYIDREQUEST._serialized_end=1158
+  _DELETEMODELVERSIONBYIDRESPONSE._serialized_start=1160
+  _DELETEMODELVERSIONBYIDRESPONSE._serialized_end=1192
+  _PROMOTEMODELVERSIONTOBUILDREQUEST._serialized_start=1195
+  _PROMOTEMODELVERSIONTOBUILDREQUEST._serialized_end=1411
+  _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_start=1413
+  _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_end=1467
+  _MODELVERSIONMANAGERSERVICE._serialized_start=1470
+  _MODELVERSIONMANAGERSERVICE._serialized_end=2395
 # @@protoc_insertion_point(module_scope)

frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi

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

frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py

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

frogml_proto/qwak/feature_store/features/feature_set_types_pb2.py

@@ -23,7 +23,7 @@ from frogml_proto.qwak.feature_store.features import real_time_feature_extractor
 from frogml_proto.qwak.feature_store.sinks import sink_pb2 as qwak_dot_feature__store_dot_sinks_dot_sink__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3qwak/feature_store/features/feature_set_types.proto\x12\x1bqwak.feature.store.features\x1a\x1fgoogle/protobuf/timestamp.proto\x1a&qwak/feature_store/sources/batch.proto\x1a*qwak/feature_store/sources/streaming.proto\x1a+qwak/feature_store/features/execution.proto\x1a-qwak/feature_store/features/aggregation.proto\x1a,qwak/feature_store/features/monitoring.proto\x1a=qwak/feature_store/features/real_time_feature_extractor.proto\x1a#qwak/feature_store/sinks/sink.proto\"\x9f\x04\n\x0e\x46\x65\x61tureSetType\x12I\n\x11\x62\x61tch_feature_set\x18\x05 \x01(\x0b\x32,.qwak.feature.store.features.BatchFeatureSetH\x00\x12Q\n\x16on_the_fly_feature_set\x18\x06 \x01(\x0b\x32/.qwak.feature.store.features.OnTheFlyFeatureSetH\x00\x12Q\n\x15streaming_feature_set\x18\x07 \x01(\x0b\x32\x30.qwak.feature.store.features.StreamingFeatureSetH\x00\x12V\n\x18streaming_feature_set_v1\x18\x08 \x01(\x0b\x32\x32.qwak.feature.store.features.StreamingFeatureSetV1H\x00\x12N\n\x14\x62\x61tch_feature_set_v1\x18\t \x01(\x0b\x32..qwak.feature.store.features.BatchFeatureSetV1H\x00\x12h\n!streaming_aggregation_feature_set\x18\n \x01(\x0b\x32;.qwak.feature.store.features.StreamingAggregationFeatureSetH\x00\x42\n\n\x08set_type\"\x9d\x03\n\x0f\x42\x61tchFeatureSet\x12\x19\n\x11scheduling_policy\x18\x01 \x01(\t\x12\x41\n\x0c\x64\x61ta_sources\x18\x02 \x03(\x0b\x32\'.qwak.feature.store.sources.BatchSourceB\x02\x18\x01\x12\x37\n\x08\x62\x61\x63kfill\x18\x03 \x01(\x0b\x32%.qwak.feature.store.features.Backfill\x12\x37\n\x08\x66unction\x18\x04 \x01(\x0b\x32%.qwak.feature.store.features.Function\x12U\n\x19\x66\x65\x61ture_set_batch_sources\x18\x05 \x03(\x0b\x32\x32.qwak.feature.store.features.FeatureSetBatchSource\x12\x63\n\x19monitoring_configurations\x18\x06 \x01(\x0b\x32@.qwak.feature.store.features.monitoring.MonitoringConfigurations\"\xd8\x05\n\x11\x42\x61tchFeatureSetV1\x12\x19\n\x11scheduling_policy\x18\x01 \x01(\t\x12\x37\n\x08\x62\x61\x63kfill\x18\x02 \x01(\x0b\x32%.qwak.feature.store.features.Backfill\x12;\n\x08\x66unction\x18\x03 \x01(\x0b\x32%.qwak.feature.store.features.FunctionB\x02\x18\x01\x12U\n\x19\x66\x65\x61ture_set_batch_sources\x18\x04 \x03(\x0b\x32\x32.qwak.feature.store.features.FeatureSetBatchSource\x12L\n\x0e\x65xecution_spec\x18\x05 \x01(\x0b\x32\x34.qwak.feature.store.features.execution.ExecutionSpec\x12\x14\n\x0coffline_sink\x18\x06 \x01(\x08\x12\x13\n\x0bonline_sink\x18\x07 \x01(\x08\x12\x63\n\x19monitoring_configurations\x18\x08 \x01(\x0b\x32@.qwak.feature.store.features.monitoring.MonitoringConfigurations\x12\x1d\n\x15timestamp_column_name\x18\t \x01(\t\x12\x43\n\x0etransformation\x18\n \x01(\x0b\x32+.qwak.feature.store.features.Transformation\x12&\n\x1eqwak_internal_protocol_version\x18\x0b \x01(\x05\x12\\\n\x1breal_time_feature_extractor\x18\x0c \x01(\x0b\x32\x35.qwak.feature.store.features.RealTimeFeatureExtractorH\x00\x42\x13\n\x11\x66\x65\x61ture_extractor\"\x9d\x01\n\x15\x46\x65\x61tureSetBatchSource\x12<\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.sources.BatchSource\x12\x46\n\x0bread_policy\x18\x02 \x01(\x0b\x32\x31.qwak.feature.store.features.DataSourceReadPolicy\"X\n\x13StreamingFeatureSet\x12\x41\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32+.qwak.feature.store.sources.StreamingSource\"\xc3\x03\n\x15StreamingFeatureSetV1\x12\x41\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32+.qwak.feature.store.sources.StreamingSource\x12\x43\n\x0etransformation\x18\x02 \x01(\x0b\x32+.qwak.feature.store.features.Transformation\x12U\n\x0e\x65xecution_spec\x18\x03 \x01(\x0b\x32=.qwak.feature.store.features.execution.StreamingExecutionSpec\x12\x1d\n\x15timestamp_column_name\x18\x04 \x01(\t\x12\x1f\n\x17online_trigger_interval\x18\x05 \x01(\x05\x12!\n\x19offline_scheduling_policy\x18\x06 \x01(\t\x12&\n\x1eqwak_internal_protocol_version\x18\x07 \x01(\x05\x12@\n\x0f\x61uxiliary_sinks\x18\x08 \x03(\x0b\x32\'.qwak.feature.store.sinks.StreamingSink\"\x97\x04\n\x1eStreamingAggregationFeatureSet\x12\x41\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32+.qwak.feature.store.sources.StreamingSource\x12\x43\n\x0etransformation\x18\x02 \x01(\x0b\x32+.qwak.feature.store.features.Transformation\x12U\n\x0e\x65xecution_spec\x18\x03 \x01(\x0b\x32=.qwak.feature.store.features.execution.StreamingExecutionSpec\x12\x1d\n\x15timestamp_column_name\x18\x04 \x01(\t\x12\x1f\n\x17online_trigger_interval\x18\x05 \x01(\x05\x12$\n\x1c\x63ompaction_scheduling_policy\x18\x06 \x01(\t\x12\x46\n\x10\x61ggregation_spec\x18\x07 \x01(\x0b\x32,.qwak.feature.store.features.AggregationSpec\x12@\n\rbackfill_spec\x18\x08 \x01(\x0b\x32).qwak.feature.store.features.BackfillSpec\x12&\n\x1eqwak_internal_protocol_version\x18\t \x01(\x05\"\xc3\x01\n\x1b\x42\x61\x63kfillBatchDataSourceSpec\x12<\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.sources.BatchSource\x12\x33\n\x0fstart_timestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x31\n\rend_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"|\n\x16\x42\x61\x63kfillDataSourceSpec\x12Z\n\x16\x62\x61tch_data_source_spec\x18\x01 \x01(\x0b\x32\x38.qwak.feature.store.features.BackfillBatchDataSourceSpecH\x00\x42\x06\n\x04type\"\xe1\x02\n\x0c\x42\x61\x63kfillSpec\x12T\n\x0e\x65xecution_spec\x18\x01 \x01(\x0b\x32<.qwak.feature.store.features.execution.BackfillExecutionSpec\x12\x33\n\x0fstart_timestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x31\n\rend_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x43\n\x0etransformation\x18\x04 \x01(\x0b\x32+.qwak.feature.store.features.Transformation\x12N\n\x11\x64\x61ta_source_specs\x18\x05 \x03(\x0b\x32\x33.qwak.feature.store.features.BackfillDataSourceSpec\"\x9f\x01\n\x0f\x41ggregationSpec\x12O\n\x0c\x61ggregations\x18\x01 \x03(\x0b\x32\x39.qwak.feature.store.features.aggregation.AggregationField\x12\x15\n\rslide_seconds\x18\x02 \x01(\x05\x12$\n\x1c\x61llowed_late_arrival_seconds\x18\x03 \x01(\x05\"\xd9\x03\n\x0eTransformation\x12L\n\x12udf_transformation\x18\x01 \x01(\x0b\x32..qwak.feature.store.features.UdfTransformationH\x00\x12L\n\x12sql_transformation\x18\x02 \x01(\x0b\x32..qwak.feature.store.features.SqlTransformationH\x00\x12R\n\x15koalas_transformation\x18\x03 \x01(\x0b\x32\x31.qwak.feature.store.features.KoalasTransformationH\x00\x12T\n\x16pyspark_transformation\x18\x05 \x01(\x0b\x32\x32.qwak.feature.store.features.PySparkTransformationH\x00\x12\x62\n\x1epandas_on_spark_transformation\x18\x06 \x01(\x0b\x32\x38.qwak.feature.store.features.PandasOnSparkTransformationH\x00\x12\x15\n\rartifact_path\x18\x04 \x01(\tB\x06\n\x04type\"*\n\x11UdfTransformation\x12\x15\n\rfunction_name\x18\x01 \x01(\t\"8\n\x11SqlTransformation\x12\x0b\n\x03sql\x18\x01 \x01(\t\x12\x16\n\x0e\x66unction_names\x18\x02 \x03(\t\"\x7f\n\x14KoalasTransformation\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x41\n\x06qwargs\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.TransformArgumentsH\x00\x42\r\n\x0b\x61rgs_option\"\x80\x01\n\x15PySparkTransformation\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x41\n\x06qwargs\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.TransformArgumentsH\x00\x42\r\n\x0b\x61rgs_option\"\x86\x01\n\x1bPandasOnSparkTransformation\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x41\n\x06qwargs\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.TransformArgumentsH\x00\x42\r\n\x0b\x61rgs_option\"\x90\x01\n\x12TransformArguments\x12K\n\x06qwargs\x18\x01 \x03(\x0b\x32;.qwak.feature.store.features.TransformArguments.QwargsEntry\x1a-\n\x0bQwargsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"f\n\x12OnTheFlyFeatureSet\x12:\n\x08\x66unction\x18\x01 \x01(\x0b\x32(.qwak.feature.store.features.UdfFunction\x12\x14\n\x0crequirements\x18\x02 \x01(\x0c\"\x96\x01\n\x08\x46unction\x12@\n\x0csql_function\x18\x01 \x01(\x0b\x32(.qwak.feature.store.features.SqlFunctionH\x00\x12@\n\x0cudf_function\x18\x02 \x01(\x0b\x32(.qwak.feature.store.features.UdfFunctionH\x00\x42\x06\n\x04type\"\xb5\x01\n\x08\x42\x61\x63kfill\x12.\n\nstart_date\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12I\n\rfillup_method\x18\x02 \x01(\x0e\x32\x32.qwak.feature.store.features.Backfill.FillUpMethod\".\n\x0c\x46illUpMethod\x12\x10\n\x0c\x41S_SCHEDULED\x10\x00\x12\x0c\n\x08SNAPSHOT\x10\x01\"\x1a\n\x0bSqlFunction\x12\x0b\n\x03sql\x18\x01 \x01(\t\"\x1b\n\x0bUdfFunction\x12\x0c\n\x04\x63ode\x18\x01 \x01(\x0c\"\t\n\x07NewOnly\"\xa0\x01\n\tTimeFrame\x12\x0f\n\x07minutes\x18\x01 \x01(\x05\x12\x37\n\x07vanilla\x18\x02 \x01(\x0b\x32$.qwak.feature.store.features.VanillaH\x00\x12?\n\x0b\x61ggregation\x18\x03 \x01(\x0b\x32(.qwak.feature.store.features.AggregationH\x00\x42\x08\n\x06\x66lavor\"k\n\x0b\x41ggregation\x12T\n\x16\x61ggregation_population\x18\x01 \x01(\x0b\x32\x32.qwak.feature.store.features.AggregationPopulationH\x00\x42\x06\n\x04type\"\x17\n\x15\x41ggregationPopulation\"\x1c\n\x1aPopulationTimeframeNewOnly\"j\n\x13PopulationTimeframe\x12K\n\x08new_only\x18\x01 \x01(\x0b\x32\x37.qwak.feature.store.features.PopulationTimeframeNewOnlyH\x00\x42\x06\n\x04type\"\t\n\x07Vanilla\"\x9f\x01\n\x08\x46ullRead\x12\x37\n\x07\x64\x65\x66\x61ult\x18\x01 \x01(\x0b\x32$.qwak.feature.store.features.VanillaH\x00\x12P\n\x14population_timeframe\x18\x02 \x01(\x0b\x32\x30.qwak.feature.store.features.PopulationTimeframeH\x00\x42\x08\n\x06\x66lavor\"\xd2\x01\n\x14\x44\x61taSourceReadPolicy\x12\x38\n\x08new_only\x18\x01 \x01(\x0b\x32$.qwak.feature.store.features.NewOnlyH\x00\x12<\n\ntime_frame\x18\x02 \x01(\x0b\x32&.qwak.feature.store.features.TimeFrameH\x00\x12:\n\tfull_read\x18\x03 \x01(\x0b\x32%.qwak.feature.store.features.FullReadH\x00\x42\x06\n\x04type*}\n\x12\x46\x65\x61tureSetTypeView\x12!\n\x1d\x46\x45\x41TURE_SET_TYPE_VIEW_INVALID\x10\x00\x12\x1f\n\x1b\x46\x45\x41TURE_SET_TYPE_VIEW_BATCH\x10\x01\x12#\n\x1f\x46\x45\x41TURE_SET_TYPE_VIEW_STREAMING\x10\x02\x42[\n&com.qwak.ai.feature.store.features.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3qwak/feature_store/features/feature_set_types.proto\x12\x1bqwak.feature.store.features\x1a\x1fgoogle/protobuf/timestamp.proto\x1a&qwak/feature_store/sources/batch.proto\x1a*qwak/feature_store/sources/streaming.proto\x1a+qwak/feature_store/features/execution.proto\x1a-qwak/feature_store/features/aggregation.proto\x1a,qwak/feature_store/features/monitoring.proto\x1a=qwak/feature_store/features/real_time_feature_extractor.proto\x1a#qwak/feature_store/sinks/sink.proto\"\x9f\x04\n\x0e\x46\x65\x61tureSetType\x12I\n\x11\x62\x61tch_feature_set\x18\x05 \x01(\x0b\x32,.qwak.feature.store.features.BatchFeatureSetH\x00\x12Q\n\x16on_the_fly_feature_set\x18\x06 \x01(\x0b\x32/.qwak.feature.store.features.OnTheFlyFeatureSetH\x00\x12Q\n\x15streaming_feature_set\x18\x07 \x01(\x0b\x32\x30.qwak.feature.store.features.StreamingFeatureSetH\x00\x12V\n\x18streaming_feature_set_v1\x18\x08 \x01(\x0b\x32\x32.qwak.feature.store.features.StreamingFeatureSetV1H\x00\x12N\n\x14\x62\x61tch_feature_set_v1\x18\t \x01(\x0b\x32..qwak.feature.store.features.BatchFeatureSetV1H\x00\x12h\n!streaming_aggregation_feature_set\x18\n \x01(\x0b\x32;.qwak.feature.store.features.StreamingAggregationFeatureSetH\x00\x42\n\n\x08set_type\"\x9d\x03\n\x0f\x42\x61tchFeatureSet\x12\x19\n\x11scheduling_policy\x18\x01 \x01(\t\x12\x41\n\x0c\x64\x61ta_sources\x18\x02 \x03(\x0b\x32\'.qwak.feature.store.sources.BatchSourceB\x02\x18\x01\x12\x37\n\x08\x62\x61\x63kfill\x18\x03 \x01(\x0b\x32%.qwak.feature.store.features.Backfill\x12\x37\n\x08\x66unction\x18\x04 \x01(\x0b\x32%.qwak.feature.store.features.Function\x12U\n\x19\x66\x65\x61ture_set_batch_sources\x18\x05 \x03(\x0b\x32\x32.qwak.feature.store.features.FeatureSetBatchSource\x12\x63\n\x19monitoring_configurations\x18\x06 \x01(\x0b\x32@.qwak.feature.store.features.monitoring.MonitoringConfigurations\"\xd8\x05\n\x11\x42\x61tchFeatureSetV1\x12\x19\n\x11scheduling_policy\x18\x01 \x01(\t\x12\x37\n\x08\x62\x61\x63kfill\x18\x02 \x01(\x0b\x32%.qwak.feature.store.features.Backfill\x12;\n\x08\x66unction\x18\x03 \x01(\x0b\x32%.qwak.feature.store.features.FunctionB\x02\x18\x01\x12U\n\x19\x66\x65\x61ture_set_batch_sources\x18\x04 \x03(\x0b\x32\x32.qwak.feature.store.features.FeatureSetBatchSource\x12L\n\x0e\x65xecution_spec\x18\x05 \x01(\x0b\x32\x34.qwak.feature.store.features.execution.ExecutionSpec\x12\x14\n\x0coffline_sink\x18\x06 \x01(\x08\x12\x13\n\x0bonline_sink\x18\x07 \x01(\x08\x12\x63\n\x19monitoring_configurations\x18\x08 \x01(\x0b\x32@.qwak.feature.store.features.monitoring.MonitoringConfigurations\x12\x1d\n\x15timestamp_column_name\x18\t \x01(\t\x12\x43\n\x0etransformation\x18\n \x01(\x0b\x32+.qwak.feature.store.features.Transformation\x12&\n\x1eqwak_internal_protocol_version\x18\x0b \x01(\x05\x12\\\n\x1breal_time_feature_extractor\x18\x0c \x01(\x0b\x32\x35.qwak.feature.store.features.RealTimeFeatureExtractorH\x00\x42\x13\n\x11\x66\x65\x61ture_extractor\"\x9d\x01\n\x15\x46\x65\x61tureSetBatchSource\x12<\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.sources.BatchSource\x12\x46\n\x0bread_policy\x18\x02 \x01(\x0b\x32\x31.qwak.feature.store.features.DataSourceReadPolicy\"X\n\x13StreamingFeatureSet\x12\x41\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32+.qwak.feature.store.sources.StreamingSource\"\xc3\x03\n\x15StreamingFeatureSetV1\x12\x41\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32+.qwak.feature.store.sources.StreamingSource\x12\x43\n\x0etransformation\x18\x02 \x01(\x0b\x32+.qwak.feature.store.features.Transformation\x12U\n\x0e\x65xecution_spec\x18\x03 \x01(\x0b\x32=.qwak.feature.store.features.execution.StreamingExecutionSpec\x12\x1d\n\x15timestamp_column_name\x18\x04 \x01(\t\x12\x1f\n\x17online_trigger_interval\x18\x05 \x01(\x05\x12!\n\x19offline_scheduling_policy\x18\x06 \x01(\t\x12&\n\x1eqwak_internal_protocol_version\x18\x07 \x01(\x05\x12@\n\x0f\x61uxiliary_sinks\x18\x08 \x03(\x0b\x32\'.qwak.feature.store.sinks.StreamingSink\"\x97\x04\n\x1eStreamingAggregationFeatureSet\x12\x41\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32+.qwak.feature.store.sources.StreamingSource\x12\x43\n\x0etransformation\x18\x02 \x01(\x0b\x32+.qwak.feature.store.features.Transformation\x12U\n\x0e\x65xecution_spec\x18\x03 \x01(\x0b\x32=.qwak.feature.store.features.execution.StreamingExecutionSpec\x12\x1d\n\x15timestamp_column_name\x18\x04 \x01(\t\x12\x1f\n\x17online_trigger_interval\x18\x05 \x01(\x05\x12$\n\x1c\x63ompaction_scheduling_policy\x18\x06 \x01(\t\x12\x46\n\x10\x61ggregation_spec\x18\x07 \x01(\x0b\x32,.qwak.feature.store.features.AggregationSpec\x12@\n\rbackfill_spec\x18\x08 \x01(\x0b\x32).qwak.feature.store.features.BackfillSpec\x12&\n\x1eqwak_internal_protocol_version\x18\t \x01(\x05\"\xc3\x01\n\x1b\x42\x61\x63kfillBatchDataSourceSpec\x12<\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.sources.BatchSource\x12\x33\n\x0fstart_timestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x31\n\rend_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"|\n\x16\x42\x61\x63kfillDataSourceSpec\x12Z\n\x16\x62\x61tch_data_source_spec\x18\x01 \x01(\x0b\x32\x38.qwak.feature.store.features.BackfillBatchDataSourceSpecH\x00\x42\x06\n\x04type\"\xe1\x02\n\x0c\x42\x61\x63kfillSpec\x12T\n\x0e\x65xecution_spec\x18\x01 \x01(\x0b\x32<.qwak.feature.store.features.execution.BackfillExecutionSpec\x12\x33\n\x0fstart_timestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x31\n\rend_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x43\n\x0etransformation\x18\x04 \x01(\x0b\x32+.qwak.feature.store.features.Transformation\x12N\n\x11\x64\x61ta_source_specs\x18\x05 \x03(\x0b\x32\x33.qwak.feature.store.features.BackfillDataSourceSpec\"\x9f\x01\n\x0f\x41ggregationSpec\x12O\n\x0c\x61ggregations\x18\x01 \x03(\x0b\x32\x39.qwak.feature.store.features.aggregation.AggregationField\x12\x15\n\rslide_seconds\x18\x02 \x01(\x05\x12$\n\x1c\x61llowed_late_arrival_seconds\x18\x03 \x01(\x05\"\xdd\x03\n\x0eTransformation\x12L\n\x12udf_transformation\x18\x01 \x01(\x0b\x32..qwak.feature.store.features.UdfTransformationH\x00\x12L\n\x12sql_transformation\x18\x02 \x01(\x0b\x32..qwak.feature.store.features.SqlTransformationH\x00\x12V\n\x15koalas_transformation\x18\x03 \x01(\x0b\x32\x31.qwak.feature.store.features.KoalasTransformationB\x02\x18\x01H\x00\x12T\n\x16pyspark_transformation\x18\x05 \x01(\x0b\x32\x32.qwak.feature.store.features.PySparkTransformationH\x00\x12\x62\n\x1epandas_on_spark_transformation\x18\x06 \x01(\x0b\x32\x38.qwak.feature.store.features.PandasOnSparkTransformationH\x00\x12\x15\n\rartifact_path\x18\x04 \x01(\tB\x06\n\x04type\"*\n\x11UdfTransformation\x12\x15\n\rfunction_name\x18\x01 \x01(\t\"8\n\x11SqlTransformation\x12\x0b\n\x03sql\x18\x01 \x01(\t\x12\x16\n\x0e\x66unction_names\x18\x02 \x03(\t\"\x83\x01\n\x14KoalasTransformation\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x41\n\x06qwargs\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.TransformArgumentsH\x00:\x02\x18\x01\x42\r\n\x0b\x61rgs_option\"\x80\x01\n\x15PySparkTransformation\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x41\n\x06qwargs\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.TransformArgumentsH\x00\x42\r\n\x0b\x61rgs_option\"\x86\x01\n\x1bPandasOnSparkTransformation\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x41\n\x06qwargs\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.TransformArgumentsH\x00\x42\r\n\x0b\x61rgs_option\"\x90\x01\n\x12TransformArguments\x12K\n\x06qwargs\x18\x01 \x03(\x0b\x32;.qwak.feature.store.features.TransformArguments.QwargsEntry\x1a-\n\x0bQwargsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"f\n\x12OnTheFlyFeatureSet\x12:\n\x08\x66unction\x18\x01 \x01(\x0b\x32(.qwak.feature.store.features.UdfFunction\x12\x14\n\x0crequirements\x18\x02 \x01(\x0c\"\x96\x01\n\x08\x46unction\x12@\n\x0csql_function\x18\x01 \x01(\x0b\x32(.qwak.feature.store.features.SqlFunctionH\x00\x12@\n\x0cudf_function\x18\x02 \x01(\x0b\x32(.qwak.feature.store.features.UdfFunctionH\x00\x42\x06\n\x04type\"\xb5\x01\n\x08\x42\x61\x63kfill\x12.\n\nstart_date\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12I\n\rfillup_method\x18\x02 \x01(\x0e\x32\x32.qwak.feature.store.features.Backfill.FillUpMethod\".\n\x0c\x46illUpMethod\x12\x10\n\x0c\x41S_SCHEDULED\x10\x00\x12\x0c\n\x08SNAPSHOT\x10\x01\"\x1a\n\x0bSqlFunction\x12\x0b\n\x03sql\x18\x01 \x01(\t\"\x1b\n\x0bUdfFunction\x12\x0c\n\x04\x63ode\x18\x01 \x01(\x0c\"\t\n\x07NewOnly\"\xa0\x01\n\tTimeFrame\x12\x0f\n\x07minutes\x18\x01 \x01(\x05\x12\x37\n\x07vanilla\x18\x02 \x01(\x0b\x32$.qwak.feature.store.features.VanillaH\x00\x12?\n\x0b\x61ggregation\x18\x03 \x01(\x0b\x32(.qwak.feature.store.features.AggregationH\x00\x42\x08\n\x06\x66lavor\"k\n\x0b\x41ggregation\x12T\n\x16\x61ggregation_population\x18\x01 \x01(\x0b\x32\x32.qwak.feature.store.features.AggregationPopulationH\x00\x42\x06\n\x04type\"\x17\n\x15\x41ggregationPopulation\"\x1c\n\x1aPopulationTimeframeNewOnly\"j\n\x13PopulationTimeframe\x12K\n\x08new_only\x18\x01 \x01(\x0b\x32\x37.qwak.feature.store.features.PopulationTimeframeNewOnlyH\x00\x42\x06\n\x04type\"\t\n\x07Vanilla\"\x9f\x01\n\x08\x46ullRead\x12\x37\n\x07\x64\x65\x66\x61ult\x18\x01 \x01(\x0b\x32$.qwak.feature.store.features.VanillaH\x00\x12P\n\x14population_timeframe\x18\x02 \x01(\x0b\x32\x30.qwak.feature.store.features.PopulationTimeframeH\x00\x42\x08\n\x06\x66lavor\"\xd2\x01\n\x14\x44\x61taSourceReadPolicy\x12\x38\n\x08new_only\x18\x01 \x01(\x0b\x32$.qwak.feature.store.features.NewOnlyH\x00\x12<\n\ntime_frame\x18\x02 \x01(\x0b\x32&.qwak.feature.store.features.TimeFrameH\x00\x12:\n\tfull_read\x18\x03 \x01(\x0b\x32%.qwak.feature.store.features.FullReadH\x00\x42\x06\n\x04type*}\n\x12\x46\x65\x61tureSetTypeView\x12!\n\x1d\x46\x45\x41TURE_SET_TYPE_VIEW_INVALID\x10\x00\x12\x1f\n\x1b\x46\x45\x41TURE_SET_TYPE_VIEW_BATCH\x10\x01\x12#\n\x1f\x46\x45\x41TURE_SET_TYPE_VIEW_STREAMING\x10\x02\x42[\n&com.qwak.ai.feature.store.features.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
 
 _FEATURESETTYPEVIEW = DESCRIPTOR.enum_types_by_name['FeatureSetTypeView']
 FeatureSetTypeView = enum_type_wrapper.EnumTypeWrapper(_FEATURESETTYPEVIEW)
@@ -306,10 +306,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _BATCHFEATURESET.fields_by_name['data_sources']._serialized_options = b'\030\001'
   _BATCHFEATURESETV1.fields_by_name['function']._options = None
   _BATCHFEATURESETV1.fields_by_name['function']._serialized_options = b'\030\001'
+  _TRANSFORMATION.fields_by_name['koalas_transformation']._options = None
+  _TRANSFORMATION.fields_by_name['koalas_transformation']._serialized_options = b'\030\001'
+  _KOALASTRANSFORMATION._options = None
+  _KOALASTRANSFORMATION._serialized_options = b'\030\001'
   _TRANSFORMARGUMENTS_QWARGSENTRY._options = None
   _TRANSFORMARGUMENTS_QWARGSENTRY._serialized_options = b'8\001'
-  _FEATURESETTYPEVIEW._serialized_start=6668
-  _FEATURESETTYPEVIEW._serialized_end=6793
+  _FEATURESETTYPEVIEW._serialized_start=6677
+  _FEATURESETTYPEVIEW._serialized_end=6802
   _FEATURESETTYPE._serialized_start=440
   _FEATURESETTYPE._serialized_end=983
   _BATCHFEATURESET._serialized_start=986
@@ -333,49 +337,49 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _AGGREGATIONSPEC._serialized_start=4055
   _AGGREGATIONSPEC._serialized_end=4214
   _TRANSFORMATION._serialized_start=4217
-  _TRANSFORMATION._serialized_end=4690
-  _UDFTRANSFORMATION._serialized_start=4692
-  _UDFTRANSFORMATION._serialized_end=4734
-  _SQLTRANSFORMATION._serialized_start=4736
-  _SQLTRANSFORMATION._serialized_end=4792
-  _KOALASTRANSFORMATION._serialized_start=4794
-  _KOALASTRANSFORMATION._serialized_end=4921
-  _PYSPARKTRANSFORMATION._serialized_start=4924
-  _PYSPARKTRANSFORMATION._serialized_end=5052
-  _PANDASONSPARKTRANSFORMATION._serialized_start=5055
-  _PANDASONSPARKTRANSFORMATION._serialized_end=5189
-  _TRANSFORMARGUMENTS._serialized_start=5192
-  _TRANSFORMARGUMENTS._serialized_end=5336
-  _TRANSFORMARGUMENTS_QWARGSENTRY._serialized_start=5291
-  _TRANSFORMARGUMENTS_QWARGSENTRY._serialized_end=5336
-  _ONTHEFLYFEATURESET._serialized_start=5338
-  _ONTHEFLYFEATURESET._serialized_end=5440
-  _FUNCTION._serialized_start=5443
-  _FUNCTION._serialized_end=5593
-  _BACKFILL._serialized_start=5596
-  _BACKFILL._serialized_end=5777
-  _BACKFILL_FILLUPMETHOD._serialized_start=5731
-  _BACKFILL_FILLUPMETHOD._serialized_end=5777
-  _SQLFUNCTION._serialized_start=5779
-  _SQLFUNCTION._serialized_end=5805
-  _UDFFUNCTION._serialized_start=5807
-  _UDFFUNCTION._serialized_end=5834
-  _NEWONLY._serialized_start=5836
-  _NEWONLY._serialized_end=5845
-  _TIMEFRAME._serialized_start=5848
-  _TIMEFRAME._serialized_end=6008
-  _AGGREGATION._serialized_start=6010
-  _AGGREGATION._serialized_end=6117
-  _AGGREGATIONPOPULATION._serialized_start=6119
-  _AGGREGATIONPOPULATION._serialized_end=6142
-  _POPULATIONTIMEFRAMENEWONLY._serialized_start=6144
-  _POPULATIONTIMEFRAMENEWONLY._serialized_end=6172
-  _POPULATIONTIMEFRAME._serialized_start=6174
-  _POPULATIONTIMEFRAME._serialized_end=6280
-  _VANILLA._serialized_start=6282
-  _VANILLA._serialized_end=6291
-  _FULLREAD._serialized_start=6294
-  _FULLREAD._serialized_end=6453
-  _DATASOURCEREADPOLICY._serialized_start=6456
-  _DATASOURCEREADPOLICY._serialized_end=6666
+  _TRANSFORMATION._serialized_end=4694
+  _UDFTRANSFORMATION._serialized_start=4696
+  _UDFTRANSFORMATION._serialized_end=4738
+  _SQLTRANSFORMATION._serialized_start=4740
+  _SQLTRANSFORMATION._serialized_end=4796
+  _KOALASTRANSFORMATION._serialized_start=4799
+  _KOALASTRANSFORMATION._serialized_end=4930
+  _PYSPARKTRANSFORMATION._serialized_start=4933
+  _PYSPARKTRANSFORMATION._serialized_end=5061
+  _PANDASONSPARKTRANSFORMATION._serialized_start=5064
+  _PANDASONSPARKTRANSFORMATION._serialized_end=5198
+  _TRANSFORMARGUMENTS._serialized_start=5201
+  _TRANSFORMARGUMENTS._serialized_end=5345
+  _TRANSFORMARGUMENTS_QWARGSENTRY._serialized_start=5300
+  _TRANSFORMARGUMENTS_QWARGSENTRY._serialized_end=5345
+  _ONTHEFLYFEATURESET._serialized_start=5347
+  _ONTHEFLYFEATURESET._serialized_end=5449
+  _FUNCTION._serialized_start=5452
+  _FUNCTION._serialized_end=5602
+  _BACKFILL._serialized_start=5605
+  _BACKFILL._serialized_end=5786
+  _BACKFILL_FILLUPMETHOD._serialized_start=5740
+  _BACKFILL_FILLUPMETHOD._serialized_end=5786
+  _SQLFUNCTION._serialized_start=5788
+  _SQLFUNCTION._serialized_end=5814
+  _UDFFUNCTION._serialized_start=5816
+  _UDFFUNCTION._serialized_end=5843
+  _NEWONLY._serialized_start=5845
+  _NEWONLY._serialized_end=5854
+  _TIMEFRAME._serialized_start=5857
+  _TIMEFRAME._serialized_end=6017
+  _AGGREGATION._serialized_start=6019
+  _AGGREGATION._serialized_end=6126
+  _AGGREGATIONPOPULATION._serialized_start=6128
+  _AGGREGATIONPOPULATION._serialized_end=6151
+  _POPULATIONTIMEFRAMENEWONLY._serialized_start=6153
+  _POPULATIONTIMEFRAMENEWONLY._serialized_end=6181
+  _POPULATIONTIMEFRAME._serialized_start=6183
+  _POPULATIONTIMEFRAME._serialized_end=6289
+  _VANILLA._serialized_start=6291
+  _VANILLA._serialized_end=6300
+  _FULLREAD._serialized_start=6303
+  _FULLREAD._serialized_end=6462
+  _DATASOURCEREADPOLICY._serialized_start=6465
+  _DATASOURCEREADPOLICY._serialized_end=6675
 # @@protoc_insertion_point(module_scope)