PyPI - frogml-core - Versions diffs - 0.0.11__py3-none-any.whl → 0.0.13__py3-none-any.whl - Mend

frogml-core 0.0.11py3-none-any.whl → 0.0.13py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (9) hide show

frogml_core/__init__.py CHANGED Viewed

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

frogml_core/clients/model_version_manager/client.py CHANGED Viewed

@@ -3,6 +3,8 @@ from platform import python_version
 from dependency_injector.wiring import Provide
 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_proto.jfml.model_version.v1.model_repository_spec_pb2 import (
     ModelRepositorySpec,
@@ -13,13 +15,13 @@ from frogml_proto.jfml.model_version.v1.model_version_framework_pb2 import (
 from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2 import (
     CreateModelVersionRequest,
     CreateModelVersionResponse,
+    GetMlBomModelVersionByIdRequest,
+    GetMlBomModelVersionByIdResponse,
 )
 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_core.exceptions import FrogmlException
-from frogml_core.inner.di_configuration import FrogmlContainer
 class ModelVersionManagerClient:
@@ -111,3 +113,27 @@ class ModelVersionManagerClient:
         except Exception as e:
             message = f"Failed to validate model version, details [{e}]"
             raise FrogmlException(message)
+    def get_mlbom_by_model_version_id(
+        self, model_version_id: str
+    ) -> GetMlBomModelVersionByIdResponse:
+        try:
+            request = GetMlBomModelVersionByIdRequest(model_version_id=model_version_id)
+            create_model_version_response: GetMlBomModelVersionByIdResponse = (
+                self.__model_version_manager_stub.GetMlBomModelVersionById(request)
+            )
+            return create_model_version_response
+        except RpcError as e:
+            message = (
+                f"Failed to get MLBOM for model version {model_version_id}, "
+                f"details [{e.details()}]"
+            )
+            raise FrogmlException(message)
+        except Exception as e:
+            message = (
+                f"Failed to get MLBOM for model version {model_version_id}, "
+                f"details [{e}]"
+            )
+            raise FrogmlException(message)

{frogml_core-0.0.11.dist-info → frogml_core-0.0.13.dist-info}/METADATA RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.11
+Version: 0.0.13
 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
@@ -22,7 +22,7 @@ Requires-Dist: cloudpickle (==2.2.1) ; extra == "feature-store"
 Requires-Dist: dacite (==1.8.1)
 Requires-Dist: dependency-injector (>=4.0)
 Requires-Dist: filelock
-Requires-Dist: frogml-storage (>=0.9.0,<0.10.0)
+Requires-Dist: frogml-storage (>=0.10.0)
 Requires-Dist: grpcio (>=1.57.0)
 Requires-Dist: joblib (>=1.3.2,<2.0.0)
 Requires-Dist: marshmallow-dataclass (>=8.5.8,<9.0.0)

{frogml_core-0.0.11.dist-info → frogml_core-0.0.13.dist-info}/RECORD RENAMED Viewed

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=KITB_jkjEatVYfNOLDBRgtpKoSR-7CpgUjW2y8p97Lk,777
+frogml_core/__init__.py,sha256=umEfVMl-NfQbfz2tjGhxXMjUgtPEdV2iaW7NIHF-OPw,777
 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
@@ -73,7 +73,7 @@ frogml_core/clients/logging_client/client.py,sha256=A7qQJWW54Ve7O0DuJX9gmSJlUQyR
 frogml_core/clients/model_management/__init__.py,sha256=vjWVP8MjmK4_A70WOgJqa6x24AeLK-ABjGJtogGzw9w,43
 frogml_core/clients/model_management/client.py,sha256=rXnzcP3-pfXCVG0KlcCwbdeIJ1iFzV84JYtyNC8n-x8,4575
 frogml_core/clients/model_version_manager/__init__.py,sha256=4Pnfzj4Egps48__dProdbSKQl5pNip0hGJd75w67BfY,46
-frogml_core/clients/model_version_manager/client.py,sha256=6WIvzWY1IWuUINvKzHPs8klMKaRJuyIN3DDdSsAryCQ,4214
+frogml_core/clients/model_version_manager/client.py,sha256=MKw3ek_dwvm6AhjCaAWtm_s_N3LueN9-Rer4P-ItN9Y,5194
 frogml_core/clients/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/clients/project/client.py,sha256=VvnUk7mo-UkE9Gkj9jVkCPGTVx8qmuekMz7oH9Uzv2Y,2159
 frogml_core/clients/prompt_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -440,9 +440,9 @@ frogml_proto/jfml/model_version/v1/model_repository_spec_pb2_grpc.py,sha256=1obo
 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=X0KJkIj8eg41SVWD48AuJSMw1P_1R5lfD4X_L_rSajs,13816
-frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi,sha256=kLV9okhcSx5LKAwTBUknA3e0sotbkDMa_JnOwcUzrRc,12416
-frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py,sha256=2v-qLZmVDKKE2J6c9Sz1WAXPLO9Ngtw9dFnehH7Kc5U,17508
+frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py,sha256=HJRR-ms9oBSTDOZSTkDfuYv2A7I5GbtNXbqqD-GGKX8,14583
+frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi,sha256=TfXaNDVcoSgX0gUTcvHaiEcazyXR9JKkHiHFKkjHzCw,12969
+frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py,sha256=5b77ei6ijZs9-z0frMZ6MngPziapHRYJAqU6CzZFMFs,17535
 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
@@ -1000,7 +1000,7 @@ 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=GoipKHH6-aN_ulcOzOoIotgT5eJM69SFMUmw4zpcs4M,3745
+frogml_services_mock/mocks/model_version_manager_service.py,sha256=nBSB-bL0X8AR1e_3J8R_bHuRaGQvf9P8aFskjNUvEo8,4814
 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=2KoXSZ4l1Q5kvLns_eoH_yDZQBRzgCk50yTzNFTL3Sc,1405
@@ -1016,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.11.dist-info/METADATA,sha256=zbNFPfeKgRHN1cs_QG8dykfYYBIfBv_pMB7SZ5-9nik,2106
-frogml_core-0.0.11.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-frogml_core-0.0.11.dist-info/RECORD,,
+frogml_core-0.0.13.dist-info/METADATA,sha256=LKymVJGiUdTllY1I79mIhz80vN8555SXVKR3WPqVS3c,2099
+frogml_core-0.0.13.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+frogml_core-0.0.13.dist-info/RECORD,,

frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.py CHANGED Viewed

@@ -17,7 +17,7 @@ from frogml_proto.jfml.model_version.v1 import model_repository_spec_pb2 as jfml
 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\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(\t\";\n\x1cGetMLBOMModelVersionResponse\x12\x1b\n\x13mlbom_model_version\x18\x01 \x01(\t2\xa2\x08\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.PromoteModelVersionToBuildResponse\x12\x82\x01\n\x18GetMlBomModelVersionById\x12\x31.jfml.model_version.v1.GetModelVersionByIdRequest\x1a\x33.jfml.model_version.v1.GetMLBOMModelVersionResponseB%\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(\t\";\n\x1fGetMlBomModelVersionByIdRequest\x12\x18\n\x10model_version_id\x18\x01 \x01(\t\"?\n GetMlBomModelVersionByIdResponse\x12\x1b\n\x13mlbom_model_version\x18\x01 \x01(\t2\xab\x08\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.PromoteModelVersionToBuildResponse\x12\x8b\x01\n\x18GetMlBomModelVersionById\x12\x36.jfml.model_version.v1.GetMlBomModelVersionByIdRequest\x1a\x37.jfml.model_version.v1.GetMlBomModelVersionByIdResponseB%\n!com.jfrog.ml.model_version.api.v1P\x01\x62\x06proto3')
@@ -35,7 +35,8 @@ _DELETEMODELVERSIONBYIDREQUEST = DESCRIPTOR.message_types_by_name['DeleteModelVe
 _DELETEMODELVERSIONBYIDRESPONSE = DESCRIPTOR.message_types_by_name['DeleteModelVersionByIdResponse']
 _PROMOTEMODELVERSIONTOBUILDREQUEST = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildRequest']
 _PROMOTEMODELVERSIONTOBUILDRESPONSE = DESCRIPTOR.message_types_by_name['PromoteModelVersionToBuildResponse']
-_GETMLBOMMODELVERSIONRESPONSE = DESCRIPTOR.message_types_by_name['GetMLBOMModelVersionResponse']
+_GETMLBOMMODELVERSIONBYIDREQUEST = DESCRIPTOR.message_types_by_name['GetMlBomModelVersionByIdRequest']
+_GETMLBOMMODELVERSIONBYIDRESPONSE = DESCRIPTOR.message_types_by_name['GetMlBomModelVersionByIdResponse']
 CreateModelVersionRequest = _reflection.GeneratedProtocolMessageType('CreateModelVersionRequest', (_message.Message,), {
   'DESCRIPTOR' : _CREATEMODELVERSIONREQUEST,
   '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
@@ -134,12 +135,19 @@ PromoteModelVersionToBuildResponse = _reflection.GeneratedProtocolMessageType('P
   })
 _sym_db.RegisterMessage(PromoteModelVersionToBuildResponse)
-GetMLBOMModelVersionResponse = _reflection.GeneratedProtocolMessageType('GetMLBOMModelVersionResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETMLBOMMODELVERSIONRESPONSE,
+GetMlBomModelVersionByIdRequest = _reflection.GeneratedProtocolMessageType('GetMlBomModelVersionByIdRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETMLBOMMODELVERSIONBYIDREQUEST,
   '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
-  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.GetMLBOMModelVersionResponse)
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.GetMlBomModelVersionByIdRequest)
   })
-_sym_db.RegisterMessage(GetMLBOMModelVersionResponse)
+_sym_db.RegisterMessage(GetMlBomModelVersionByIdRequest)
+GetMlBomModelVersionByIdResponse = _reflection.GeneratedProtocolMessageType('GetMlBomModelVersionByIdResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETMLBOMMODELVERSIONBYIDRESPONSE,
+  '__module__' : 'jfml.model_version.v1.model_version_manager_service_pb2'
+  # @@protoc_insertion_point(class_scope:jfml.model_version.v1.GetMlBomModelVersionByIdResponse)
+  })
+_sym_db.RegisterMessage(GetMlBomModelVersionByIdResponse)
 _MODELVERSIONMANAGERSERVICE = DESCRIPTOR.services_by_name['ModelVersionManagerService']
 if _descriptor._USE_C_DESCRIPTORS == False:
@@ -174,8 +182,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _PROMOTEMODELVERSIONTOBUILDREQUEST._serialized_end=1411
   _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_start=1413
   _PROMOTEMODELVERSIONTOBUILDRESPONSE._serialized_end=1467
-  _GETMLBOMMODELVERSIONRESPONSE._serialized_start=1469
-  _GETMLBOMMODELVERSIONRESPONSE._serialized_end=1528
-  _MODELVERSIONMANAGERSERVICE._serialized_start=1531
-  _MODELVERSIONMANAGERSERVICE._serialized_end=2589
+  _GETMLBOMMODELVERSIONBYIDREQUEST._serialized_start=1469
+  _GETMLBOMMODELVERSIONBYIDREQUEST._serialized_end=1528
+  _GETMLBOMMODELVERSIONBYIDRESPONSE._serialized_start=1530
+  _GETMLBOMMODELVERSIONBYIDRESPONSE._serialized_end=1593
+  _MODELVERSIONMANAGERSERVICE._serialized_start=1596
+  _MODELVERSIONMANAGERSERVICE._serialized_end=2663
 # @@protoc_insertion_point(module_scope)

frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2.pyi CHANGED Viewed

@@ -272,7 +272,22 @@ class PromoteModelVersionToBuildResponse(google.protobuf.message.Message):
 global___PromoteModelVersionToBuildResponse = PromoteModelVersionToBuildResponse
-class GetMLBOMModelVersionResponse(google.protobuf.message.Message):
+class GetMlBomModelVersionByIdRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
+    model_version_id: builtins.str
+    """ModelVersion's id"""
+    def __init__(
+        self,
+        *,
+        model_version_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["model_version_id", b"model_version_id"]) -> None: ...
+global___GetMlBomModelVersionByIdRequest = GetMlBomModelVersionByIdRequest
+class GetMlBomModelVersionByIdResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
     MLBOM_MODEL_VERSION_FIELD_NUMBER: builtins.int
@@ -285,4 +300,4 @@ class GetMLBOMModelVersionResponse(google.protobuf.message.Message):
     ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["mlbom_model_version", b"mlbom_model_version"]) -> None: ...
-global___GetMLBOMModelVersionResponse = GetMLBOMModelVersionResponse
+global___GetMlBomModelVersionByIdResponse = GetMlBomModelVersionByIdResponse

frogml_proto/jfml/model_version/v1/model_version_manager_service_pb2_grpc.py CHANGED Viewed

@@ -51,8 +51,8 @@ class ModelVersionManagerServiceStub(object):
                 )
         self.GetMlBomModelVersionById = channel.unary_unary(
                 '/jfml.model_version.v1.ModelVersionManagerService/GetMlBomModelVersionById',
-                request_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetModelVersionByIdRequest.SerializeToString,
-                response_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMLBOMModelVersionResponse.FromString,
+                request_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdRequest.SerializeToString,
+                response_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdResponse.FromString,
                 )
@@ -155,8 +155,8 @@ def add_ModelVersionManagerServiceServicer_to_server(servicer, server):
             ),
             'GetMlBomModelVersionById': grpc.unary_unary_rpc_method_handler(
                     servicer.GetMlBomModelVersionById,
-                    request_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetModelVersionByIdRequest.FromString,
-                    response_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMLBOMModelVersionResponse.SerializeToString,
+                    request_deserializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdRequest.FromString,
+                    response_serializer=jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdResponse.SerializeToString,
             ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
@@ -299,7 +299,7 @@ class ModelVersionManagerService(object):
             timeout=None,
             metadata=None):
         return grpc.experimental.unary_unary(request, target, '/jfml.model_version.v1.ModelVersionManagerService/GetMlBomModelVersionById',
-            jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetModelVersionByIdRequest.SerializeToString,
-            jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMLBOMModelVersionResponse.FromString,
+            jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdRequest.SerializeToString,
+            jfml_dot_model__version_dot_v1_dot_model__version__manager__service__pb2.GetMlBomModelVersionByIdResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

frogml_services_mock/mocks/model_version_manager_service.py CHANGED Viewed

@@ -17,6 +17,16 @@ from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2 import
     GetModelVersionByIdRequest,
     GetModelVersionByNameRequest,
     GetModelVersionByNameResponse,
+    InitPaginationRequest,
+    InitPaginationResponse,
+    ListModelVersionsRequest,
+    ListModelVersionsResponse,
+    DeleteModelVersionByIdRequest,
+    DeleteModelVersionByIdResponse,
+    PromoteModelVersionToBuildRequest,
+    PromoteModelVersionToBuildResponse,
+    GetMlBomModelVersionByIdRequest,
+    GetMlBomModelVersionByIdResponse,
 )
 from frogml_proto.jfml.model_version.v1.model_version_manager_service_pb2_grpc import (
     ModelVersionManagerServiceServicer,
@@ -24,7 +34,8 @@ 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, ModelVersionStatus,
+    ModelVersionSpec,
+    ModelVersionStatus,
 )
 timestamp = Timestamp()
@@ -36,7 +47,9 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
         super(ModelVersionManagerServiceMock, self).__init__()
         self.model_versions: Dict[str, ModelVersion] = {}
-    def CreateModelVersion(self, request: CreateModelVersionRequest, context):
+    def CreateModelVersion(
+        self, request: CreateModelVersionRequest, context
+    ) -> CreateModelVersionResponse:
         return CreateModelVersionResponse(
             model_version=ModelVersion(
                 model_version_id=str(uuid.uuid4()),
@@ -48,7 +61,9 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
             )
         )
-    def GetModelVersionById(self, request: GetModelVersionByIdRequest, context):
+    def GetModelVersionById(
+        self, request: GetModelVersionByIdRequest, context
+    ) -> GetModelVersionByIdResponse:
         return GetModelVersionByIdResponse(
             model_version=ModelVersion(
                 model_version_id=request.model_version_id,
@@ -90,11 +105,27 @@ class ModelVersionManagerServiceMock(ModelVersionManagerServiceServicer):
             )
         )
-    def InitPagination(self, request, context):
+    def InitPagination(
+        self, request: InitPaginationRequest, context
+    ) -> InitPaginationResponse:
         return super().InitPagination(request, context)
-    def ListModelVersions(self, request, context):
+    def ListModelVersions(
+        self, request: ListModelVersionsRequest, context
+    ) -> ListModelVersionsResponse:
         return super().ListModelVersions(request, context)
-    def DeleteModelVersionById(self, request, context):
+    def DeleteModelVersionById(
+        self, request: DeleteModelVersionByIdRequest, context
+    ) -> DeleteModelVersionByIdResponse:
         return super().DeleteModelVersionById(request, context)
+    def PromoteModelVersionToBuild(
+        self, request: PromoteModelVersionToBuildRequest, context
+    ) -> PromoteModelVersionToBuildResponse:
+        return super().PromoteModelVersionToBuild(request, context)
+    def GetMlBomModelVersionById(
+        self, request: GetMlBomModelVersionByIdRequest, context
+    ) -> GetMlBomModelVersionByIdResponse:
+        return super().GetMlBomModelVersionById(request, context)

{frogml_core-0.0.11.dist-info → frogml_core-0.0.13.dist-info}/WHEEL RENAMED Viewed

File without changes

frogml-core 0.0.11__py3-none-any.whl → 0.0.13__py3-none-any.whl

frogml-core 0.0.11py3-none-any.whl → 0.0.13py3-none-any.whl