frogml-core 0.0.14__tar.gz → 0.0.16__tar.gz

{frogml_core-0.0.14 → frogml_core-0.0.16}/PKG-INFO

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

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

{frogml_core-0.0.14 → frogml_core-0.0.16}/frogml_core/clients/model_version_manager/client.py

@@ -1,11 +1,15 @@
 from platform import python_version
+from typing import List, Optional, Dict, Union, Tuple
 
 from dependency_injector.wiring import Provide
+from frogml_storage.entity_manifest import Artifact
 from grpc import RpcError
 
 from frogml_core.exceptions import FrogmlException
 from frogml_core.inner.di_configuration import FrogmlContainer
 from frogml_core.utils.model_utils import get_model_id_from_model_name
+from frogml_core.utils.proto_utils import ProtoUtils
+from frogml_proto.jfml.model_version.v1.artifact_pb2 import Artifact as ArtifactProto
 from frogml_proto.jfml.model_version.v1.model_repository_spec_pb2 import (
     ModelRepositorySpec,
 )
@@ -21,7 +25,12 @@ from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2 import
 from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
     ModelVersionManagerServiceStub,
 )
-from frogml_proto.jfml.model_version.v1.model_version_pb2 import ModelVersionSpec
+from frogml_proto.jfml.model_version.v1.model_version_pb2 import (
+    ModelVersionSpec,
+    MlBomData,
+    ParameterValue,
+    MetricValue,
+)
 
 
 class ModelVersionManagerClient:
@@ -33,14 +42,54 @@ class ModelVersionManagerClient:
         self.__model_version_manager_stub = ModelVersionManagerServiceStub(grpc_channel)
 
     @staticmethod
+    def __build_mlbom_data_kwargs(
+        model_artifact: List[ArtifactProto],
+        code_artifacts: Optional[List[ArtifactProto]] = None,
+        dependency_artifacts: Optional[List[ArtifactProto]] = None,
+        metrics: Optional[Dict[str, str]] = None,
+        parameters: Optional[Dict[str, str]] = None,
+    ) -> Dict[
+        str, Union[List[ArtifactProto], Dict[str, Union[ParameterValue, MetricValue]]]
+    ]:
+        kwargs = {"model_artifact": model_artifact}
+
+        if dependency_artifacts:
+            kwargs["dependency_artifacts"] = dependency_artifacts
+
+        if code_artifacts:
+            kwargs["code_artifacts"] = code_artifacts
+
+        if parameters:
+            kwargs["parameters"] = {
+                key: ParameterValue(value=str(value))
+                for key, value in parameters.items()
+            }
+
+        if metrics:
+            kwargs["metrics"] = {
+                key: MetricValue(value=str(value)) for key, value in metrics.items()
+            }
+
+        return kwargs
+
     def __build_create_model_version_request(
+        self,
         project_key: str,
         repository_key: str,
         model_name: str,
         model_version_name: str,
         model_version_framework: ModelVersionFramework,
         dry_run: bool,
+        model_artifacts: List[ArtifactProto],
+        dependency_artifacts: Optional[List[ArtifactProto]] = None,
+        code_artifacts: Optional[List[ArtifactProto]] = None,
+        parameters: Optional[Dict[str, str]] = None,
+        metrics: Optional[Dict[str, str]] = None,
     ) -> CreateModelVersionRequest:
+        kwargs: dict = self.__build_mlbom_data_kwargs(
+            model_artifacts, code_artifacts, dependency_artifacts, metrics, parameters
+        )
+
         return CreateModelVersionRequest(
             dry_run=dry_run,
             model_version=ModelVersionSpec(
@@ -54,6 +103,7 @@ class ModelVersionManagerClient:
                 framework=model_version_framework,
                 python_version=python_version(),
             ),
+            ml_bom_data=MlBomData(**kwargs),
         )
 
     def validate_create_model_version(
@@ -63,8 +113,21 @@ class ModelVersionManagerClient:
         model_name: str,
         model_version_name: str,
         model_version_framework: ModelVersionFramework,
+        model_artifact: List[Artifact],
+        dependency_artifacts: Optional[List[Artifact]] = None,
+        code_artifacts: Optional[List[Artifact]] = None,
+        parameters: Optional[Dict[str, str]] = None,
+        metrics: Optional[Dict[str, str]] = None,
     ):
         try:
+            (
+                code_artifacts_proto,
+                dependency_artifacts_proto,
+                model_artifact_proto,
+            ) = self.convert_artifacts(
+                model_artifact, dependency_artifacts, code_artifacts
+            )
+
             create_model_request = self.__build_create_model_version_request(
                 project_key=project_key,
                 repository_key=repository_key,
@@ -72,6 +135,11 @@ class ModelVersionManagerClient:
                 model_version_name=model_version_name,
                 model_version_framework=model_version_framework,
                 dry_run=True,
+                model_artifacts=model_artifact_proto,
+                dependency_artifacts=dependency_artifacts_proto,
+                code_artifacts=code_artifacts_proto,
+                parameters=parameters,
+                metrics=metrics,
             )
             self.__model_version_manager_stub.CreateModelVersion(create_model_request)
         except RpcError as e:
@@ -89,8 +157,21 @@ class ModelVersionManagerClient:
         model_name: str,
         model_version_name: str,
         model_version_framework: ModelVersionFramework,
+        model_artifact: List[Artifact],
+        dependency_artifacts: Optional[List[Artifact]] = None,
+        code_artifacts: Optional[List[Artifact]] = None,
+        parameters: Optional[Dict[str, str]] = None,
+        metrics: Optional[Dict[str, str]] = None,
     ) -> CreateModelVersionResponse:
         try:
+            (
+                code_artifacts_proto,
+                dependency_artifacts_proto,
+                model_artifact_proto,
+            ) = self.convert_artifacts(
+                model_artifact, dependency_artifacts, code_artifacts
+            )
+
             create_model_request = self.__build_create_model_version_request(
                 project_key=project_key,
                 repository_key=repository_key,
@@ -98,6 +179,11 @@ class ModelVersionManagerClient:
                 model_version_name=model_version_name,
                 model_version_framework=model_version_framework,
                 dry_run=False,
+                model_artifacts=model_artifact_proto,
+                dependency_artifacts=dependency_artifacts_proto,
+                code_artifacts=code_artifacts_proto,
+                parameters=parameters,
+                metrics=metrics,
             )
             create_model_version_response: CreateModelVersionResponse = (
                 self.__model_version_manager_stub.CreateModelVersion(
@@ -114,6 +200,31 @@ class ModelVersionManagerClient:
             message = f"Failed to validate model version, details [{e}]"
             raise FrogmlException(message)
 
+    @staticmethod
+    def convert_artifacts(
+        model_artifact: List[Artifact],
+        dependency_artifacts: Optional[List[Artifact]] = None,
+        code_artifacts: Optional[List[Artifact]] = None,
+    ) -> Tuple[
+        Optional[List[ArtifactProto]],
+        Optional[List[ArtifactProto]],
+        List[ArtifactProto],
+    ]:
+        model_artifact_proto: List[
+            ArtifactProto
+        ] = ProtoUtils.convert_artifacts_to_artifacts_proto(model_artifact)
+        dependency_artifacts_proto = (
+            ProtoUtils.convert_artifacts_to_artifacts_proto(dependency_artifacts)
+            if dependency_artifacts
+            else None
+        )
+        code_artifacts_proto = (
+            ProtoUtils.convert_artifacts_to_artifacts_proto(code_artifacts)
+            if code_artifacts
+            else None
+        )
+        return code_artifacts_proto, dependency_artifacts_proto, model_artifact_proto
+
     def get_mlbom_by_model_version_id(
         self, model_version_id: str
     ) -> GetMlBomModelVersionByIdResponse:

{frogml_core-0.0.14 → frogml_core-0.0.16}/frogml_core/utils/proto_utils.py

@@ -1,3 +1,11 @@
+from typing import List
+
+from frogml_storage.entity_manifest import Artifact
+
+from frogml_proto.jfml.model_version.v1.artifact_pb2 import (
+    Artifact as ArtifactProto,
+    Checksums as ChecksumsProto,
+)
 from frogml_proto.jfml.model_version.v1.model_version_framework_pb2 import (
     ModelVersionFramework,
     CatboostFramework,
@@ -40,3 +48,16 @@ class ProtoUtils:
             raise ValueError(f"Format {serialization_format} is not supported yet")
 
         return ModelVersionFramework(**framework_to_define)
+
+    @staticmethod
+    def convert_artifacts_to_artifacts_proto(
+        artifacts: List[Artifact],
+    ) -> List[ArtifactProto]:
+        return [
+            ArtifactProto(
+                artifact_path=artifact.artifact_path,
+                size=artifact.size,
+                checksums=ChecksumsProto(sha2=artifact.checksums.sha2),
+            )
+            for artifact in artifacts
+        ]

{frogml_core-0.0.14 → frogml_core-0.0.16}/frogml_proto/jfml/model_version/v1/model_version_pb2.py

@@ -18,7 +18,7 @@ from frogml_proto.jfml.model_version.v1 import model_repository_spec_pb2 as jfml
 from frogml_proto.jfml.model_version.v1 import artifact_pb2 as jfml_dot_model__version_dot_v1_dot_artifact__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)jfml/model_version/v1/model_version.proto\x12\x15jfml.model_version.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x33jfml/model_version/v1/model_version_framework.proto\x1a\x31jfml/model_version/v1/model_repository_spec.proto\x1a$jfml/model_version/v1/artifact.proto\"\x8e\x02\n\x0cModelVersion\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\x12\x35\n\x04spec\x18\x02 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpec\x12\x39\n\x06status\x18\x03 \x01(\x0b\x32).jfml.model_version.v1.ModelVersionStatus\x12\x12\n\ncreated_by\x18\x04 \x01(\t\x12.\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\xbe\x01\n\x10ModelVersionSpec\x12\x43\n\x0frepository_spec\x18\x01 \x01(\x0b\x32*.jfml.model_version.v1.ModelRepositorySpec\x12\x0c\n\x04name\x18\x02 \x01(\t\x12?\n\tframework\x18\x03 \x01(\x0b\x32,.jfml.model_version.v1.ModelVersionFramework\x12\x16\n\x0epython_version\x18\x04 \x01(\t\"\xa5\x03\n\tMlBomData\x12\x37\n\x0emodel_artifact\x18\x01 \x03(\x0b\x32\x1f.jfml.model_version.v1.Artifact\x12\x37\n\x0e\x63ode_artifacts\x18\x02 \x03(\x0b\x32\x1f.jfml.model_version.v1.Artifact\x12=\n\x14\x64\x65pendency_artifacts\x18\x03 \x03(\x0b\x32\x1f.jfml.model_version.v1.Artifact\x12\x44\n\nparameters\x18\x04 \x03(\x0b\x32\x30.jfml.model_version.v1.MlBomData.ParametersEntry\x12>\n\x07metrics\x18\x05 \x03(\x0b\x32-.jfml.model_version.v1.MlBomData.MetricsEntry\x1a\x31\n\x0fParametersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a.\n\x0cMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xf6\x01\n\x12ModelVersionStatus\x12J\n\x0bin_progress\x18\x01 \x01(\x0b\x32\x33.jfml.model_version.v1.ModelVersionInProgressStatusH\x00\x12G\n\tcompleted\x18\x02 \x01(\x0b\x32\x32.jfml.model_version.v1.ModelVersionCompletedStatusH\x00\x12\x41\n\x06\x66\x61iled\x18\x03 \x01(\x0b\x32/.jfml.model_version.v1.ModelVersionFailedStatusH\x00\x42\x08\n\x06status\"\x1e\n\x1cModelVersionInProgressStatus\"\x1d\n\x1bModelVersionCompletedStatus\"\x1a\n\x18ModelVersionFailedStatusB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)jfml/model_version/v1/model_version.proto\x12\x15jfml.model_version.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x33jfml/model_version/v1/model_version_framework.proto\x1a\x31jfml/model_version/v1/model_repository_spec.proto\x1a$jfml/model_version/v1/artifact.proto\"\x8e\x02\n\x0cModelVersion\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\x12\x35\n\x04spec\x18\x02 \x01(\x0b\x32\'.jfml.model_version.v1.ModelVersionSpec\x12\x39\n\x06status\x18\x03 \x01(\x0b\x32).jfml.model_version.v1.ModelVersionStatus\x12\x12\n\ncreated_by\x18\x04 \x01(\t\x12.\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\xbe\x01\n\x10ModelVersionSpec\x12\x43\n\x0frepository_spec\x18\x01 \x01(\x0b\x32*.jfml.model_version.v1.ModelRepositorySpec\x12\x0c\n\x04name\x18\x02 \x01(\t\x12?\n\tframework\x18\x03 \x01(\x0b\x32,.jfml.model_version.v1.ModelVersionFramework\x12\x16\n\x0epython_version\x18\x04 \x01(\t\"\xf0\x03\n\tMlBomData\x12\x37\n\x0emodel_artifact\x18\x01 \x03(\x0b\x32\x1f.jfml.model_version.v1.Artifact\x12\x37\n\x0e\x63ode_artifacts\x18\x02 \x03(\x0b\x32\x1f.jfml.model_version.v1.Artifact\x12=\n\x14\x64\x65pendency_artifacts\x18\x03 \x03(\x0b\x32\x1f.jfml.model_version.v1.Artifact\x12\x44\n\nparameters\x18\x04 \x03(\x0b\x32\x30.jfml.model_version.v1.MlBomData.ParametersEntry\x12>\n\x07metrics\x18\x05 \x03(\x0b\x32-.jfml.model_version.v1.MlBomData.MetricsEntry\x1aX\n\x0fParametersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.jfml.model_version.v1.ParameterValue:\x02\x38\x01\x1aR\n\x0cMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x31\n\x05value\x18\x02 \x01(\x0b\x32\".jfml.model_version.v1.MetricValue:\x02\x38\x01\"\x1f\n\x0eParameterValue\x12\r\n\x05value\x18\x01 \x01(\t\"\x1c\n\x0bMetricValue\x12\r\n\x05value\x18\x01 \x01(\t\"\xf6\x01\n\x12ModelVersionStatus\x12J\n\x0bin_progress\x18\x01 \x01(\x0b\x32\x33.jfml.model_version.v1.ModelVersionInProgressStatusH\x00\x12G\n\tcompleted\x18\x02 \x01(\x0b\x32\x32.jfml.model_version.v1.ModelVersionCompletedStatusH\x00\x12\x41\n\x06\x66\x61iled\x18\x03 \x01(\x0b\x32/.jfml.model_version.v1.ModelVersionFailedStatusH\x00\x42\x08\n\x06status\"\x1e\n\x1cModelVersionInProgressStatus\"\x1d\n\x1bModelVersionCompletedStatus\"\x1a\n\x18ModelVersionFailedStatusB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
 
 
 
@@ -27,6 +27,8 @@ _MODELVERSIONSPEC = DESCRIPTOR.message_types_by_name['ModelVersionSpec']
 _MLBOMDATA = DESCRIPTOR.message_types_by_name['MlBomData']
 _MLBOMDATA_PARAMETERSENTRY = _MLBOMDATA.nested_types_by_name['ParametersEntry']
 _MLBOMDATA_METRICSENTRY = _MLBOMDATA.nested_types_by_name['MetricsEntry']
+_PARAMETERVALUE = DESCRIPTOR.message_types_by_name['ParameterValue']
+_METRICVALUE = DESCRIPTOR.message_types_by_name['MetricValue']
 _MODELVERSIONSTATUS = DESCRIPTOR.message_types_by_name['ModelVersionStatus']
 _MODELVERSIONINPROGRESSSTATUS = DESCRIPTOR.message_types_by_name['ModelVersionInProgressStatus']
 _MODELVERSIONCOMPLETEDSTATUS = DESCRIPTOR.message_types_by_name['ModelVersionCompletedStatus']
@@ -68,6 +70,20 @@ _sym_db.RegisterMessage(MlBomData)
 _sym_db.RegisterMessage(MlBomData.ParametersEntry)
 _sym_db.RegisterMessage(MlBomData.MetricsEntry)
 
+ParameterValue = _reflection.GeneratedProtocolMessageType('ParameterValue', (_message.Message,), {
+  'DESCRIPTOR' : _PARAMETERVALUE,
+  '__module__' : 'jfml.model_version.v1.model_version_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.ParameterValue)
+  })
+_sym_db.RegisterMessage(ParameterValue)
+
+MetricValue = _reflection.GeneratedProtocolMessageType('MetricValue', (_message.Message,), {
+  'DESCRIPTOR' : _METRICVALUE,
+  '__module__' : 'jfml.model_version.v1.model_version_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.MetricValue)
+  })
+_sym_db.RegisterMessage(MetricValue)
+
 ModelVersionStatus = _reflection.GeneratedProtocolMessageType('ModelVersionStatus', (_message.Message,), {
   'DESCRIPTOR' : _MODELVERSIONSTATUS,
   '__module__' : 'jfml.model_version.v1.model_version_pb2'
@@ -109,17 +125,21 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _MODELVERSIONSPEC._serialized_start=517
   _MODELVERSIONSPEC._serialized_end=707
   _MLBOMDATA._serialized_start=710
-  _MLBOMDATA._serialized_end=1131
+  _MLBOMDATA._serialized_end=1206
   _MLBOMDATA_PARAMETERSENTRY._serialized_start=1034
-  _MLBOMDATA_PARAMETERSENTRY._serialized_end=1083
-  _MLBOMDATA_METRICSENTRY._serialized_start=1085
-  _MLBOMDATA_METRICSENTRY._serialized_end=1131
-  _MODELVERSIONSTATUS._serialized_start=1134
-  _MODELVERSIONSTATUS._serialized_end=1380
-  _MODELVERSIONINPROGRESSSTATUS._serialized_start=1382
-  _MODELVERSIONINPROGRESSSTATUS._serialized_end=1412
-  _MODELVERSIONCOMPLETEDSTATUS._serialized_start=1414
-  _MODELVERSIONCOMPLETEDSTATUS._serialized_end=1443
-  _MODELVERSIONFAILEDSTATUS._serialized_start=1445
-  _MODELVERSIONFAILEDSTATUS._serialized_end=1471
+  _MLBOMDATA_PARAMETERSENTRY._serialized_end=1122
+  _MLBOMDATA_METRICSENTRY._serialized_start=1124
+  _MLBOMDATA_METRICSENTRY._serialized_end=1206
+  _PARAMETERVALUE._serialized_start=1208
+  _PARAMETERVALUE._serialized_end=1239
+  _METRICVALUE._serialized_start=1241
+  _METRICVALUE._serialized_end=1269
+  _MODELVERSIONSTATUS._serialized_start=1272
+  _MODELVERSIONSTATUS._serialized_end=1518
+  _MODELVERSIONINPROGRESSSTATUS._serialized_start=1520
+  _MODELVERSIONINPROGRESSSTATUS._serialized_end=1550
+  _MODELVERSIONCOMPLETEDSTATUS._serialized_start=1552
+  _MODELVERSIONCOMPLETEDSTATUS._serialized_end=1581
+  _MODELVERSIONFAILEDSTATUS._serialized_start=1583
+  _MODELVERSIONFAILEDSTATUS._serialized_end=1609
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.14 → frogml_core-0.0.16}/frogml_proto/jfml/model_version/v1/model_version_pb2.pyi

@@ -92,13 +92,15 @@ class MlBomData(google.protobuf.message.Message):
         KEY_FIELD_NUMBER: builtins.int
         VALUE_FIELD_NUMBER: builtins.int
         key: builtins.str
-        value: builtins.str
+        @property
+        def value(self) -> global___ParameterValue: ...
         def __init__(
             self,
             *,
             key: builtins.str = ...,
-            value: builtins.str = ...,
+            value: global___ParameterValue | None = ...,
         ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
         def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
 
     class MetricsEntry(google.protobuf.message.Message):
@@ -107,13 +109,15 @@ class MlBomData(google.protobuf.message.Message):
         KEY_FIELD_NUMBER: builtins.int
         VALUE_FIELD_NUMBER: builtins.int
         key: builtins.str
-        value: builtins.str
+        @property
+        def value(self) -> global___MetricValue: ...
         def __init__(
             self,
             *,
             key: builtins.str = ...,
-            value: builtins.str = ...,
+            value: global___MetricValue | None = ...,
         ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
         def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
 
     MODEL_ARTIFACT_FIELD_NUMBER: builtins.int
@@ -131,10 +135,10 @@ class MlBomData(google.protobuf.message.Message):
     def dependency_artifacts(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[jfml.model_version.v1.artifact_pb2.Artifact]:
         """The model's dependency artifacts"""
     @property
-    def parameters(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+    def parameters(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___ParameterValue]:
         """The model's parameters"""
     @property
-    def metrics(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+    def metrics(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___MetricValue]:
         """The model's metrics"""
     def __init__(
         self,
@@ -142,13 +146,41 @@ class MlBomData(google.protobuf.message.Message):
         model_artifact: collections.abc.Iterable[jfml.model_version.v1.artifact_pb2.Artifact] | None = ...,
         code_artifacts: collections.abc.Iterable[jfml.model_version.v1.artifact_pb2.Artifact] | None = ...,
         dependency_artifacts: collections.abc.Iterable[jfml.model_version.v1.artifact_pb2.Artifact] | None = ...,
-        parameters: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
-        metrics: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+        parameters: collections.abc.Mapping[builtins.str, global___ParameterValue] | None = ...,
+        metrics: collections.abc.Mapping[builtins.str, global___MetricValue] | None = ...,
     ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["code_artifacts", b"code_artifacts", "dependency_artifacts", b"dependency_artifacts", "metrics", b"metrics", "model_artifact", b"model_artifact", "parameters", b"parameters"]) -> None: ...
 
 global___MlBomData = MlBomData
 
+class ParameterValue(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VALUE_FIELD_NUMBER: builtins.int
+    value: builtins.str
+    def __init__(
+        self,
+        *,
+        value: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["value", b"value"]) -> None: ...
+
+global___ParameterValue = ParameterValue
+
+class MetricValue(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VALUE_FIELD_NUMBER: builtins.int
+    value: builtins.str
+    def __init__(
+        self,
+        *,
+        value: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["value", b"value"]) -> None: ...
+
+global___MetricValue = MetricValue
+
 class ModelVersionStatus(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

{frogml_core-0.0.14 → frogml_core-0.0.16}/pyproject.toml

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