frogml-core 0.0.90__tar.gz → 0.0.92__tar.gz

frogml_core-0.0.92/PKG-INFO

@@ -0,0 +1,46 @@
+Metadata-Version: 2.3
+Name: frogml-core
+Version: 0.0.92
+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
+Author: Jfrog
+Requires-Python: >=3.9,<3.12
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Provides-Extra: feature-store
+Requires-Dist: PyYAML
+Requires-Dist: cachetools
+Requires-Dist: chevron (==0.14.0)
+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.11.2)
+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)
+Requires-Dist: protobuf (>=3.10,<4) ; python_version < "3.10"
+Requires-Dist: protobuf (>=4.21.6) ; python_version >= "3.10"
+Requires-Dist: pyarrow (>=6.0.0) ; extra == "feature-store"
+Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
+Requires-Dist: pyspark (==3.4.2) ; extra == "feature-store"
+Requires-Dist: python-jose[cryptography] (>=3.4.0)
+Requires-Dist: python-json-logger (>=2.0.2)
+Requires-Dist: requests
+Requires-Dist: retrying (==1.3.4)
+Requires-Dist: typeguard (>=2,<3)
+Project-URL: Home page, https://www.jfrog.com/
+Description-Content-Type: text/markdown
+
+# Frogml Core
+
+Frogml is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.
+Frogml Core contains all the objects and tools necessary to use the Frogml Platform
+

frogml_core-0.0.92/README.md

@@ -0,0 +1,4 @@
+# Frogml Core
+
+Frogml is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.
+Frogml Core contains all the objects and tools necessary to use the Frogml Platform

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/__init__.py

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

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/clients/instance_template/client.py

@@ -50,12 +50,10 @@ class InstanceTemplateManagementClient:
 
     def list_instance_templates(self) -> List[InstanceTemplateSpec]:
         try:
-            result: ListInstanceTemplatesResponse = (
-                self._instance_template_service.ListInstanceTemplates(
-                    ListInstanceTemplatesRequest(
-                        optional_instance_filter=InstanceFilter(
-                            instance_type_filter=InstanceTypeFilter.INSTANCE_TYPE_FILTER_ALL
-                        )
+            result: ListInstanceTemplatesResponse = self._instance_template_service.ListInstanceTemplates(
+                ListInstanceTemplatesRequest(
+                    optional_instance_filter=InstanceFilter(
+                        instance_type_filter=InstanceTypeFilter.INSTANCE_TYPE_FILTER_ALL
                     )
                 )
             )

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/clients/model_version_manager/client.py

@@ -210,9 +210,9 @@ class ModelVersionManagerClient:
         Optional[List[ArtifactProto]],
         List[ArtifactProto],
     ]:
-        model_artifact_proto: List[ArtifactProto] = (
-            ProtoUtils.convert_artifacts_to_artifacts_proto(model_artifact)
-        )
+        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

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/clients/prompt_manager/model_descriptor_mapper.py

@@ -57,10 +57,10 @@ class ModelDescriptorMapper:
         model_id: str, openai_chat_params: ProtoOpenAIChatModelParams
     ) -> OpenAIChat:
         p = openai_chat_params
-        _tool_choice: Union[str, ChatCompletionNamedToolChoiceParam] = (
-            ModelDescriptorMapper._from_tool_choice(
-                openai_chat_params=openai_chat_params
-            )
+        _tool_choice: Union[
+            str, ChatCompletionNamedToolChoiceParam
+        ] = ModelDescriptorMapper._from_tool_choice(
+            openai_chat_params=openai_chat_params
         )
         _tools: List[ChatCompletionToolParam] = []
 

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/feature_store/_common/artifact_utils.py

@@ -45,9 +45,9 @@ class ArtifactsUploader:
         featureset_name: str,
         __instance_module_path__: str,
     ) -> Optional[ArtifactSpec]:
-        transformation_functions: Optional[List[Callable[..., Any]]] = (
-            transformation.get_functions()
-        )
+        transformation_functions: Optional[
+            List[Callable[..., Any]]
+        ] = transformation.get_functions()
         if transformation_functions is not None and transformation_functions:
             return ArtifactSpec(
                 artifact_name=featureset_name,

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/feature_store/data_sources/batch/athena.py

@@ -130,9 +130,9 @@ class AthenaSource(JdbcSource):
             )
         )
 
-        time_partition_columns: Optional[TimePartitionColumns] = (
-            AthenaSource._extract_partition_column(proto_athena_source)
-        )
+        time_partition_columns: Optional[
+            TimePartitionColumns
+        ] = AthenaSource._extract_partition_column(proto_athena_source)
         workgroup: Optional[str] = (
             proto_athena_source.workgroup
             if proto_athena_source.HasField("workgroup")

frogml_core-0.0.92/frogml_core/feature_store/feature_sets/context.py

@@ -0,0 +1,13 @@
+from dataclasses import dataclass
+from datetime import datetime
+from typing import Union
+
+
+@dataclass
+class Context:
+    start_time: Union[
+        str, datetime
+    ] = "${qwak_ingestion_start_timestamp}"  # todo mlops-2312 - rename?
+    end_time: Union[
+        str, datetime
+    ] = "${qwak_ingestion_end_timestamp}"  # todo mlops-2312 - rename?

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/feature_store/feature_sets/streaming.py

@@ -131,9 +131,9 @@ def feature_set(
             offline_scheduling_policy=offline_scheduling_policy,
         )
 
-        streaming_backfill: Optional[StreamingBackfill] = (
-            StreamingBackfill.get_streaming_backfill_from_function(function=function)
-        )
+        streaming_backfill: Optional[
+            StreamingBackfill
+        ] = StreamingBackfill.get_streaming_backfill_from_function(function=function)
 
         fs_name = name or function.__name__
         streaming_feature_set = StreamingFeatureSet(

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/feature_store/feature_sets/streaming_backfill.py

@@ -230,7 +230,7 @@ class StreamingBackfill:
 
     @staticmethod
     def _get_normalized_backfill_sources_spec(
-        data_sources: Union[List[str], List[DataSourceBackfillSpec]],
+        data_sources: Union[List[str], List[DataSourceBackfillSpec]]
     ) -> List[DataSourceBackfillSpec]:
         # reformat all data source specs to 'DataSourceBackfillSpec'
         return [

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/feature_store/online/client.py

@@ -197,12 +197,12 @@ class OnlineClient:
         )
         ordered_entities = [entity[0] for entity in ordered_entities_tuple]
 
-        request_chunks: List[Tuple[RequestedEntitiesMatrix, pd.DataFrame]] = (
-            OnlineClient._split_entities(
-                entity_names=ordered_entities,
-                population_df=df,
-                max_entities_per_split=max_keys_per_request,
-            )
+        request_chunks: List[
+            Tuple[RequestedEntitiesMatrix, pd.DataFrame]
+        ] = OnlineClient._split_entities(
+            entity_names=ordered_entities,
+            population_df=df,
+            max_entities_per_split=max_keys_per_request,
         )
 
         results: List[pd.DataFrame] = []

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/feature_store/sinks/streaming/factory.py

@@ -25,7 +25,7 @@ class StreamingSinkFactory:
     def get_streaming_sink(proto_streaming_sink: ProtoStreamingSink) -> BaseSink:
         sink_type = proto_streaming_sink.WhichOneof("sink_type")
 
-        auth_conf: BaseAuthentication  # noqa: F842
+        auth_conf: BaseAuthentication
         if sink_type == "kafka_sink":
             proto_kafka_sink: ProtoKafkaSink = proto_streaming_sink.kafka_sink
             auth_configuration: BaseAuthentication = cast(

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/inner/build_logic/phases/phase_010_fetch_model/fetch_strategy_manager/strategy/git/git_strategy.py

@@ -161,9 +161,9 @@ def make_ssh_key_file(git_ssh_key: str) -> str:
 
 
 def add_ssh_file_to_env(ssh_key_file_path: str) -> None:
-    os.environ["GIT_SSH_COMMAND"] = (
-        f"ssh -i {ssh_key_file_path} -o StrictHostKeyChecking=no"
-    )
+    os.environ[
+        "GIT_SSH_COMMAND"
+    ] = f"ssh -i {ssh_key_file_path} -o StrictHostKeyChecking=no"
     os.environ["GIT_SSH"] = f"ssh -i {ssh_key_file_path} -o StrictHostKeyChecking=no"
 
 

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/llmops/provider/openai/provider.py

@@ -30,9 +30,9 @@ class OpenAIProvider:
         self.client = OpenAIClient()
 
     def _get_random_openai_api_key(self) -> Optional[str]:
-        openai_api_keys: List[OpenAIApiKeySystemSecret] = (
-            IntegrationUtils().get_openai_api_keys()
-        )
+        openai_api_keys: List[
+            OpenAIApiKeySystemSecret
+        ] = IntegrationUtils().get_openai_api_keys()
         if len(openai_api_keys) == 0:
             return None
 

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/model/tools/adapters/output.py

@@ -46,7 +46,7 @@ def get_output_adapter(
 
             first_result = (
                 return_result[0]
-                if isinstance(return_result, list) and len(return_result) > 0
+                if type(return_result) == list and len(return_result) > 0
                 else return_result
             )
             if issubclass(type(first_result), Message):

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_core/tools/logger/logger.py

@@ -276,7 +276,7 @@ def set_file_handler_log_file(
     logger: logging.Logger, handler_name: str, log_file: Path
 ):
     existing_handler = get_handler_from_logger(logger, handler_name)
-    if not isinstance(existing_handler, RotatingFileHandler):
+    if type(existing_handler) != RotatingFileHandler:
         raise FrogmlException(
             f"Error in setting log file. Error message: handler of name {handler_name} is not a file logger handler"
         )

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_proto/qwak/builds/builds_orchestrator_service_pb2.py

@@ -16,7 +16,7 @@ from frogml_proto.qwak.builds import build_pb2 as qwak_dot_builds_dot_build__pb2
 from frogml_proto.qwak.builds import build_url_pb2 as qwak_dot_builds_dot_build__url__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/builds/builds_orchestrator_service.proto\x12\x18qwak.builds.orchestrator\x1a\x17qwak/builds/build.proto\x1a\x1bqwak/builds/build_url.proto\"R\n\x11\x42uildModelRequest\x12=\n\nbuild_spec\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.RemoteBuildSpec\"\x14\n\x12\x42uildModelResponse\"+\n\x17\x43\x61ncelBuildModelRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"\x1a\n\x18\x43\x61ncelBuildModelResponse\"*\n\x16\x43reateUploadURLRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"C\n\x17\x43reateUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x14\n\x0c\x64ownload_url\x18\x02 \x01(\t\"h\n\"GetBuildVersioningUploadURLRequest\x12\x42\n\x06params\x18\x01 \x01(\x0b\x32\x32.qwak.builds.orchestrator.BuildVersioningUrlParams\"\xa7\x02\n#GetBuildVersioningUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\x12Q\n\x15\x61uthentication_detail\x18\x03 \x01(\x0b\x32..qwak.builds.orchestrator.AuthenticationDetailB\x02\x18\x01\x12[\n\x07headers\x18\x04 \x03(\x0b\x32J.qwak.builds.orchestrator.GetBuildVersioningUploadURLResponse.HeadersEntry\x1a.\n\x0cHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x86\x01\n\x14\x41uthenticationDetail\x12Z\n\x1bjfrog_authentication_detail\x18\x01 \x01(\x0b\x32\x33.qwak.builds.orchestrator.JfrogAuthenticationDetailH\x00\x42\x12\n\x10integration_type\"<\n\x19JfrogAuthenticationDetail\x12\x10\n\x08username\x18\x01 \x01(\t\x12\r\n\x05token\x18\x02 \x01(\t\"j\n$GetBuildVersioningDownloadURLRequest\x12\x42\n\x06params\x18\x01 \x01(\x0b\x32\x32.qwak.builds.orchestrator.BuildVersioningUrlParams\"\xdf\x01\n%GetBuildVersioningDownloadURLResponse\x12\x14\n\x0c\x64ownload_url\x18\x01 \x01(\t\x12\x11\n\tfile_size\x18\x02 \x01(\x03\x12]\n\x07headers\x18\x03 \x03(\x0b\x32L.qwak.builds.orchestrator.GetBuildVersioningDownloadURLResponse.HeadersEntry\x1a.\n\x0cHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"n\n\x1eListBuildVersioningTagsRequest\x12L\n\x15\x62uild_versioning_tags\x18\x01 \x01(\x0b\x32-.qwak.builds.orchestrator.BuildVersioningTags\"\x9c\x01\n\x1fListBuildVersioningTagsResponse\x12\x61\n build_versioning_tags_properties\x18\x01 \x03(\x0b\x32\x37.qwak.builds.orchestrator.BuildVersioningTagsProperties\x12\x16\n\x0eui_direct_link\x18\x02 \x01(\t\"\x95\x01\n\x16\x43reateDataTableRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x0b\n\x03tag\x18\x03 \x01(\t\x12<\n\x05table\x18\x04 \x01(\x0b\x32-.qwak.builds.orchestrator.DataTableDefinition\x12\x0c\n\x04path\x18\x05 \x01(\t\"\x19\n\x17\x43reateDataTableResponse\"n\n\x1dGetBaseDockerImageNameRequest\x12M\n\x16\x62\x61se_docker_image_type\x18\x01 \x01(\x0e\x32-.qwak.builds.orchestrator.BaseDockerImageType\"@\n\x1eGetBaseDockerImageNameResponse\x12\x1e\n\x16\x62\x61se_docker_image_name\x18\x01 \x01(\t\"f\n\x17SaveArtifactPathRequest\x12K\n\x12\x61rtifact_path_spec\x18\x01 \x01(\x0b\x32/.qwak.builds.orchestrator.BuildArtifactPathSpec\"\x1a\n\x18SaveArtifactPathResponse\"-\n\x19ListBuildArtifactsRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"X\n\x1aListBuildArtifactsResponse\x12:\n\tartifacts\x18\x01 \x03(\x0b\x32\'.qwak.builds.orchestrator.BuildArtifact2\xcf\n\n\x19\x42uildsOrchestratorService\x12g\n\nBuildModel\x12+.qwak.builds.orchestrator.BuildModelRequest\x1a,.qwak.builds.orchestrator.BuildModelResponse\x12y\n\x10\x43\x61ncelBuildModel\x12\x31.qwak.builds.orchestrator.CancelBuildModelRequest\x1a\x32.qwak.builds.orchestrator.CancelBuildModelResponse\x12{\n\x0f\x43reateUploadURL\x12\x30.qwak.builds.orchestrator.CreateUploadURLRequest\x1a\x31.qwak.builds.orchestrator.CreateUploadURLResponse\"\x03\x88\x02\x01\x12\x9a\x01\n\x1bGetBuildVersioningUploadURL\x12<.qwak.builds.orchestrator.GetBuildVersioningUploadURLRequest\x1a=.qwak.builds.orchestrator.GetBuildVersioningUploadURLResponse\x12\xa0\x01\n\x1dGetBuildVersioningDownloadURL\x12>.qwak.builds.orchestrator.GetBuildVersioningDownloadURLRequest\x1a?.qwak.builds.orchestrator.GetBuildVersioningDownloadURLResponse\x12\x8e\x01\n\x17ListBuildVersioningTags\x12\x38.qwak.builds.orchestrator.ListBuildVersioningTagsRequest\x1a\x39.qwak.builds.orchestrator.ListBuildVersioningTagsResponse\x12v\n\x0f\x43reateDataTable\x12\x30.qwak.builds.orchestrator.CreateDataTableRequest\x1a\x31.qwak.builds.orchestrator.CreateDataTableResponse\x12\x8b\x01\n\x16GetBaseDockerImageName\x12\x37.qwak.builds.orchestrator.GetBaseDockerImageNameRequest\x1a\x38.qwak.builds.orchestrator.GetBaseDockerImageNameResponse\x12y\n\x10SaveArtifactPath\x12\x31.qwak.builds.orchestrator.SaveArtifactPathRequest\x1a\x32.qwak.builds.orchestrator.SaveArtifactPathResponse\x12\x7f\n\x12ListBuildArtifacts\x12\x33.qwak.builds.orchestrator.ListBuildArtifactsRequest\x1a\x34.qwak.builds.orchestrator.ListBuildArtifactsResponseB\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/builds/builds_orchestrator_service.proto\x12\x18qwak.builds.orchestrator\x1a\x17qwak/builds/build.proto\x1a\x1bqwak/builds/build_url.proto\"R\n\x11\x42uildModelRequest\x12=\n\nbuild_spec\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.RemoteBuildSpec\"\x14\n\x12\x42uildModelResponse\"+\n\x17\x43\x61ncelBuildModelRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"\x1a\n\x18\x43\x61ncelBuildModelResponse\"*\n\x16\x43reateUploadURLRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"C\n\x17\x43reateUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x14\n\x0c\x64ownload_url\x18\x02 \x01(\t\"h\n\"GetBuildVersioningUploadURLRequest\x12\x42\n\x06params\x18\x01 \x01(\x0b\x32\x32.qwak.builds.orchestrator.BuildVersioningUrlParams\"\xa7\x02\n#GetBuildVersioningUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\x12Q\n\x15\x61uthentication_detail\x18\x03 \x01(\x0b\x32..qwak.builds.orchestrator.AuthenticationDetailB\x02\x18\x01\x12[\n\x07headers\x18\x04 \x03(\x0b\x32J.qwak.builds.orchestrator.GetBuildVersioningUploadURLResponse.HeadersEntry\x1a.\n\x0cHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x86\x01\n\x14\x41uthenticationDetail\x12Z\n\x1bjfrog_authentication_detail\x18\x01 \x01(\x0b\x32\x33.qwak.builds.orchestrator.JfrogAuthenticationDetailH\x00\x42\x12\n\x10integration_type\"<\n\x19JfrogAuthenticationDetail\x12\x10\n\x08username\x18\x01 \x01(\t\x12\r\n\x05token\x18\x02 \x01(\t\"j\n$GetBuildVersioningDownloadURLRequest\x12\x42\n\x06params\x18\x01 \x01(\x0b\x32\x32.qwak.builds.orchestrator.BuildVersioningUrlParams\"\xdf\x01\n%GetBuildVersioningDownloadURLResponse\x12\x14\n\x0c\x64ownload_url\x18\x01 \x01(\t\x12\x11\n\tfile_size\x18\x02 \x01(\x03\x12]\n\x07headers\x18\x03 \x03(\x0b\x32L.qwak.builds.orchestrator.GetBuildVersioningDownloadURLResponse.HeadersEntry\x1a.\n\x0cHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"n\n\x1eListBuildVersioningTagsRequest\x12L\n\x15\x62uild_versioning_tags\x18\x01 \x01(\x0b\x32-.qwak.builds.orchestrator.BuildVersioningTags\"\x9c\x01\n\x1fListBuildVersioningTagsResponse\x12\x61\n build_versioning_tags_properties\x18\x01 \x03(\x0b\x32\x37.qwak.builds.orchestrator.BuildVersioningTagsProperties\x12\x16\n\x0eui_direct_link\x18\x02 \x01(\t\"\x95\x01\n\x16\x43reateDataTableRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x0b\n\x03tag\x18\x03 \x01(\t\x12<\n\x05table\x18\x04 \x01(\x0b\x32-.qwak.builds.orchestrator.DataTableDefinition\x12\x0c\n\x04path\x18\x05 \x01(\t\"\x19\n\x17\x43reateDataTableResponse\"n\n\x1dGetBaseDockerImageNameRequest\x12M\n\x16\x62\x61se_docker_image_type\x18\x01 \x01(\x0e\x32-.qwak.builds.orchestrator.BaseDockerImageType\"@\n\x1eGetBaseDockerImageNameResponse\x12\x1e\n\x16\x62\x61se_docker_image_name\x18\x01 \x01(\t\"f\n\x17SaveArtifactPathRequest\x12K\n\x12\x61rtifact_path_spec\x18\x01 \x01(\x0b\x32/.qwak.builds.orchestrator.BuildArtifactPathSpec\"\x1a\n\x18SaveArtifactPathResponse\"-\n\x19ListBuildArtifactsRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"X\n\x1aListBuildArtifactsResponse\x12:\n\tartifacts\x18\x01 \x03(\x0b\x32\'.qwak.builds.orchestrator.BuildArtifact\"A\n$DetachBuildsFromModelVersionsRequest\x12\x19\n\x11model_version_ids\x18\x01 \x03(\t\"\'\n%DetachBuildsFromModelVersionsResponse2\xf2\x0b\n\x19\x42uildsOrchestratorService\x12g\n\nBuildModel\x12+.qwak.builds.orchestrator.BuildModelRequest\x1a,.qwak.builds.orchestrator.BuildModelResponse\x12y\n\x10\x43\x61ncelBuildModel\x12\x31.qwak.builds.orchestrator.CancelBuildModelRequest\x1a\x32.qwak.builds.orchestrator.CancelBuildModelResponse\x12{\n\x0f\x43reateUploadURL\x12\x30.qwak.builds.orchestrator.CreateUploadURLRequest\x1a\x31.qwak.builds.orchestrator.CreateUploadURLResponse\"\x03\x88\x02\x01\x12\x9a\x01\n\x1bGetBuildVersioningUploadURL\x12<.qwak.builds.orchestrator.GetBuildVersioningUploadURLRequest\x1a=.qwak.builds.orchestrator.GetBuildVersioningUploadURLResponse\x12\xa0\x01\n\x1dGetBuildVersioningDownloadURL\x12>.qwak.builds.orchestrator.GetBuildVersioningDownloadURLRequest\x1a?.qwak.builds.orchestrator.GetBuildVersioningDownloadURLResponse\x12\x8e\x01\n\x17ListBuildVersioningTags\x12\x38.qwak.builds.orchestrator.ListBuildVersioningTagsRequest\x1a\x39.qwak.builds.orchestrator.ListBuildVersioningTagsResponse\x12v\n\x0f\x43reateDataTable\x12\x30.qwak.builds.orchestrator.CreateDataTableRequest\x1a\x31.qwak.builds.orchestrator.CreateDataTableResponse\x12\x8b\x01\n\x16GetBaseDockerImageName\x12\x37.qwak.builds.orchestrator.GetBaseDockerImageNameRequest\x1a\x38.qwak.builds.orchestrator.GetBaseDockerImageNameResponse\x12y\n\x10SaveArtifactPath\x12\x31.qwak.builds.orchestrator.SaveArtifactPathRequest\x1a\x32.qwak.builds.orchestrator.SaveArtifactPathResponse\x12\x7f\n\x12ListBuildArtifacts\x12\x33.qwak.builds.orchestrator.ListBuildArtifactsRequest\x1a\x34.qwak.builds.orchestrator.ListBuildArtifactsResponse\x12\xa0\x01\n\x1d\x44\x65tachBuildsFromModelVersions\x12>.qwak.builds.orchestrator.DetachBuildsFromModelVersionsRequest\x1a?.qwak.builds.orchestrator.DetachBuildsFromModelVersionsResponseB\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
 
 
 
@@ -44,6 +44,8 @@ _SAVEARTIFACTPATHREQUEST = DESCRIPTOR.message_types_by_name['SaveArtifactPathReq
 _SAVEARTIFACTPATHRESPONSE = DESCRIPTOR.message_types_by_name['SaveArtifactPathResponse']
 _LISTBUILDARTIFACTSREQUEST = DESCRIPTOR.message_types_by_name['ListBuildArtifactsRequest']
 _LISTBUILDARTIFACTSRESPONSE = DESCRIPTOR.message_types_by_name['ListBuildArtifactsResponse']
+_DETACHBUILDSFROMMODELVERSIONSREQUEST = DESCRIPTOR.message_types_by_name['DetachBuildsFromModelVersionsRequest']
+_DETACHBUILDSFROMMODELVERSIONSRESPONSE = DESCRIPTOR.message_types_by_name['DetachBuildsFromModelVersionsResponse']
 BuildModelRequest = _reflection.GeneratedProtocolMessageType('BuildModelRequest', (_message.Message,), {
   'DESCRIPTOR' : _BUILDMODELREQUEST,
   '__module__' : 'qwak.builds.builds_orchestrator_service_pb2'
@@ -214,6 +216,20 @@ ListBuildArtifactsResponse = _reflection.GeneratedProtocolMessageType('ListBuild
   })
 _sym_db.RegisterMessage(ListBuildArtifactsResponse)
 
+DetachBuildsFromModelVersionsRequest = _reflection.GeneratedProtocolMessageType('DetachBuildsFromModelVersionsRequest', (_message.Message,), {
+  'DESCRIPTOR' : _DETACHBUILDSFROMMODELVERSIONSREQUEST,
+  '__module__' : 'qwak.builds.builds_orchestrator_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.builds.orchestrator.DetachBuildsFromModelVersionsRequest)
+  })
+_sym_db.RegisterMessage(DetachBuildsFromModelVersionsRequest)
+
+DetachBuildsFromModelVersionsResponse = _reflection.GeneratedProtocolMessageType('DetachBuildsFromModelVersionsResponse', (_message.Message,), {
+  'DESCRIPTOR' : _DETACHBUILDSFROMMODELVERSIONSRESPONSE,
+  '__module__' : 'qwak.builds.builds_orchestrator_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.builds.orchestrator.DetachBuildsFromModelVersionsResponse)
+  })
+_sym_db.RegisterMessage(DetachBuildsFromModelVersionsResponse)
+
 _BUILDSORCHESTRATORSERVICE = DESCRIPTOR.services_by_name['BuildsOrchestratorService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
@@ -275,6 +291,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _LISTBUILDARTIFACTSREQUEST._serialized_end=2163
   _LISTBUILDARTIFACTSRESPONSE._serialized_start=2165
   _LISTBUILDARTIFACTSRESPONSE._serialized_end=2253
-  _BUILDSORCHESTRATORSERVICE._serialized_start=2256
-  _BUILDSORCHESTRATORSERVICE._serialized_end=3615
+  _DETACHBUILDSFROMMODELVERSIONSREQUEST._serialized_start=2255
+  _DETACHBUILDSFROMMODELVERSIONSREQUEST._serialized_end=2320
+  _DETACHBUILDSFROMMODELVERSIONSRESPONSE._serialized_start=2322
+  _DETACHBUILDSFROMMODELVERSIONSRESPONSE._serialized_end=2361
+  _BUILDSORCHESTRATORSERVICE._serialized_start=2364
+  _BUILDSORCHESTRATORSERVICE._serialized_end=3886
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_proto/qwak/builds/builds_orchestrator_service_pb2.pyi

@@ -417,3 +417,28 @@ class ListBuildArtifactsResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["artifacts", b"artifacts"]) -> None: ...
 
 global___ListBuildArtifactsResponse = ListBuildArtifactsResponse
+
+class DetachBuildsFromModelVersionsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_VERSION_IDS_FIELD_NUMBER: builtins.int
+    @property
+    def model_version_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """The model versions to detach"""
+    def __init__(
+        self,
+        *,
+        model_version_ids: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["model_version_ids", b"model_version_ids"]) -> None: ...
+
+global___DetachBuildsFromModelVersionsRequest = DetachBuildsFromModelVersionsRequest
+
+class DetachBuildsFromModelVersionsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___DetachBuildsFromModelVersionsResponse = DetachBuildsFromModelVersionsResponse

{frogml_core-0.0.90 → frogml_core-0.0.92}/frogml_proto/qwak/builds/builds_orchestrator_service_pb2_grpc.py

@@ -64,6 +64,11 @@ class BuildsOrchestratorServiceStub(object):
                 request_serializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsRequest.SerializeToString,
                 response_deserializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsResponse.FromString,
                 )
+        self.DetachBuildsFromModelVersions = channel.unary_unary(
+                '/qwak.builds.orchestrator.BuildsOrchestratorService/DetachBuildsFromModelVersions',
+                request_serializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsRequest.SerializeToString,
+                response_deserializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsResponse.FromString,
+                )
 
 
 class BuildsOrchestratorServiceServicer(object):
@@ -139,6 +144,12 @@ class BuildsOrchestratorServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def DetachBuildsFromModelVersions(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_BuildsOrchestratorServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -192,6 +203,11 @@ def add_BuildsOrchestratorServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsRequest.FromString,
                     response_serializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsResponse.SerializeToString,
             ),
+            'DetachBuildsFromModelVersions': grpc.unary_unary_rpc_method_handler(
+                    servicer.DetachBuildsFromModelVersions,
+                    request_deserializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsRequest.FromString,
+                    response_serializer=qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'qwak.builds.orchestrator.BuildsOrchestratorService', rpc_method_handlers)
@@ -371,3 +387,20 @@ class BuildsOrchestratorService(object):
             qwak_dot_builds_dot_builds__orchestrator__service__pb2.ListBuildArtifactsResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def DetachBuildsFromModelVersions(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, '/qwak.builds.orchestrator.BuildsOrchestratorService/DetachBuildsFromModelVersions',
+            qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsRequest.SerializeToString,
+            qwak_dot_builds_dot_builds__orchestrator__service__pb2.DetachBuildsFromModelVersionsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

{frogml_core-0.0.90 → frogml_core-0.0.92}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "frogml-core"
-version = "0.0.90"
+version = "0.0.92"
 description = "frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform"
 authors = ["Jfrog"]
 readme = "README.md"
@@ -9,13 +9,11 @@ packages = [
     { include = "frogml_core", format = ["sdist", "wheel"] },
     { include = "frogml_proto", format = ["sdist", "wheel"] },
     { include = "frogml_services_mock", format = ["sdist", "wheel"] },
-    { include = "frogml_storage", format = ["sdist", "wheel"] },
 ]
 include = [
     {path = "frogml_core/**/*", format = ["sdist", "wheel"]},
     {path = "frogml_proto/**/*", format = ["sdist", "wheel"]},
-    {path = "frogml_services_mock/**/*", format = ["sdist", "wheel"]},
-    {path = "frogml_storage/**/*", format = ["sdist", "wheel"]},
+    {path = "frogml_services_mock/**/*", format = ["sdist", "wheel"]}
 ]
 classifiers = [
     "Operating System :: OS Independent",
@@ -54,20 +52,18 @@ retrying = "1.3.4"
 chevron = "0.14.0"
 dacite = "1.8.1"
 cachetools = "*"
-pydantic = "*"
-typer = "*"
-tqdm = "*"
+frogml-storage = ">=0.11.2"
 
 
 [tool.poetry.group.dev.dependencies]
 pyarrow = { version = ">=6.0.0" }
-pytest = "<8"
+pytest = "7.2.2"
 pyspark = "3.4.2"
 pytest-httpserver = "1.0.6"
-black = ">=24.3.0"
-flake8 = ">=7.0.0"
+black = "23.1.0"
+flake8 = "5.0.0"
 bandit = { version = "^1.7.5", extras = ["toml"] }
-coverage = ">=7.4.0"
+coverage = "7.2.1"
 typing_extensions = ">=4.4.0"
 bump2version = "1.0.1"
 grpcio-tools = [
@@ -100,17 +96,10 @@ tensorflow-io-gcs-filesystem = [
 imageio = "^2.16.0"
 QwakBentoML = { version = "0.13.1+26.g506a197", source = "qwak", python = ">=3.7.1,<3.10" }
 croniter = "1.4.1"
-mypy = ">=1.9.0"
-isort = ">=5.13.0"
-types-requests = ">=2.28.0"
-pre-commit = "^4.2.0"
 
 [tool.poetry.extras]
 feature-store = ["pyarrow", "pyathena", "cloudpickle", "pyspark" ,"pandas"]
 
-[tool.poetry.scripts]
-frogml = "frogml_storage.cli._frogml_cli:main"
-
 [tool.black]
 extend-exclude = '''
 (