frogml 1.2.26__tar.gz → 1.2.34__tar.gz

{frogml-1.2.26 → frogml-1.2.34}/PKG-INFO

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

{frogml-1.2.26 → frogml-1.2.34}/frogml/__init__.py

@@ -1,5 +1,5 @@
 __author__ = "jfrog"
-__version__ = "1.2.26"
+__version__ = "1.2.34"
 
 from frogml.sdk.model.decorators.api import api_decorator as api
 from frogml.sdk.model_loggers.model_logger import load_model, log_model

frogml-1.2.34/frogml/_proto/qwak/builds/build_pb2.py

@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.builds/build.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@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
+from frogml._proto.qwak.builds import build_values_pb2 as qwak_dot_builds_dot_build__values__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17qwak/builds/build.proto\x12\x18qwak.builds.orchestrator\x1a/qwak/user_application/common/v0/resources.proto\x1a\"qwak/fitness_service/fitness.proto\x1a\x1eqwak/builds/build_values.proto\"\xd9\x06\n\x0fRemoteBuildSpec\x12\x43\n\x10\x62uild_properties\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.BuildProperties\x12\x35\n\tbuild_env\x18\x02 \x01(\x0b\x32\".qwak.builds.orchestrator.BuildEnv\x12\x43\n\rcpu_resources\x18\x03 \x01(\x0b\x32&.qwak.builds.orchestrator.CpuResourcesB\x02\x18\x01H\x00\x12\x43\n\rgpu_resources\x18\x07 \x01(\x0b\x32&.qwak.builds.orchestrator.GpuResourcesB\x02\x18\x01H\x00\x12\x62\n\x1c\x63lient_pod_compute_resources\x18\r \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResourcesH\x00\x12\x0f\n\x07verbose\x18\x04 \x01(\x05\x12\x17\n\x0f\x62uild_code_path\x18\x05 \x01(\t\x12\x14\n\x0c\x62uild_config\x18\x06 \x01(\t\x12\x15\n\rbuild_v1_flag\x18\x08 \x01(\x08\x12H\n\x13\x62uild_properties_v1\x18\t \x01(\x0b\x32+.qwak.builds.orchestrator.BuildPropertiesV1\x12\x13\n\x0b\x62uild_steps\x18\n \x03(\t\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x13\n\x0bsdk_version\x18\x0c \x01(\t\x12=\n\x0fpurchase_option\x18\x0e \x01(\x0e\x32$.qwak.fitness.service.PurchaseOption\x12\x1c\n\x14\x62uild_destined_image\x18\x0f \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x10 \x01(\x05\x12\x16\n\x0ejfrog_token_id\x18\x11 \x01(\t\x12\x1f\n\x17\x61rtifactory_project_key\x18\x12 \x01(\t\x12\x18\n\x10model_version_id\x18\x13 \x01(\t\x12\x18\n\x10model_group_name\x18\x14 \x01(\tB\x0b\n\tResources\"g\n\x11\x42uildPropertiesV1\x12\x18\n\x10\x62uild_config_url\x18\x01 \x01(\t\x12\x1a\n\x12qwak_sdk_wheel_url\x18\x02 \x01(\t\x12\x1c\n\x14qwak_sdk_version_url\x18\x03 \x01(\t\"\xba\x01\n\x0f\x42uildProperties\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x0e\n\x06\x62ranch\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12\x39\n\tmodel_uri\x18\x05 \x01(\x0b\x32&.qwak.builds.orchestrator.ModelUriSpec\x12\x16\n\x0egpu_compatible\x18\x06 \x01(\x08\x12\x12\n\nbuild_name\x18\x07 \x01(\t\"\xb2\x01\n\x0cModelUriSpec\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x12\n\ngit_branch\x18\x02 \x01(\t\x12\x17\n\x0fgit_credentials\x18\x03 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x04 \x01(\t\x12\x11\n\tcommit_id\x18\x05 \x01(\t\x12\x10\n\x08main_dir\x18\x06 \x01(\t\x12#\n\x1b\x64\x65pendency_required_folders\x18\x07 \x03(\t\"|\n\x08\x42uildEnv\x12\x37\n\ndocker_env\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.DockerEnv\x12\x37\n\npython_env\x18\x02 \x01(\x0b\x32#.qwak.builds.orchestrator.PythonEnv\"\xcc\x01\n\tDockerEnv\x12\x12\n\nbase_image\x18\x01 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\x02 \x03(\t\x12\x10\n\x08no_cache\x18\x03 \x01(\x08\x12\x1c\n\x14\x61ssumed_iam_role_arn\x18\x04 \x01(\t\x12\'\n\x1fservice_account_key_secret_name\x18\x05 \x01(\t\x12@\n\x10\x63ontainer_engine\x18\x06 \x01(\x0b\x32&.qwak.build.management.ContainerEngine\"\xb8\x01\n\tPythonEnv\x12\x17\n\x0fgit_credentials\x18\x01 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x02 \x01(\t\x12\x18\n\x10qwak_sdk_version\x18\x03 \x01(\t\x12\x1c\n\x14qwak_sdk_extra_index\x18\x04 \x01(\t\x12\x1a\n\x12\x66rogml_cli_version\x18\x05 \x01(\t\x12\x1e\n\x16\x66rogml_cli_extra_index\x18\x06 \x01(\t\"n\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12:\n\x0cmemory_units\x18\x03 \x01(\x0e\x32$.qwak.builds.orchestrator.MemoryUnit\"W\n\x0cGpuResources\x12\x33\n\x08gpu_type\x18\x01 \x01(\x0e\x32!.qwak.builds.orchestrator.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"\x91\x01\n\x13\x44\x61taTableDefinition\x12?\n\x07\x63olumns\x18\x01 \x03(\x0b\x32..qwak.builds.orchestrator.DataColumnDefinition\x12\x39\n\x0b\x64\x61ta_format\x18\x02 \x01(\x0b\x32$.qwak.builds.orchestrator.DataFormat\"\\\n\x14\x44\x61taColumnDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x36\n\x04type\x18\x02 \x01(\x0e\x32(.qwak.builds.orchestrator.DataColumnType\"J\n\nDataFormat\x12\x32\n\x03\x63sv\x18\x01 \x01(\x0b\x32#.qwak.builds.orchestrator.CsvFormatH\x00\x42\x08\n\x06\x66ormat\"G\n\tCsvFormat\x12\x11\n\tdelimiter\x18\x01 \x01(\t\x12\x12\n\nquote_char\x18\x02 \x01(\t\x12\x13\n\x0b\x65scape_char\x18\x03 \x01(\t\"\xa5\x01\n\x0e\x42uildInitiator\x12<\n\x04user\x18\x01 \x01(\x0b\x32,.qwak.builds.orchestrator.UserBuildInitiatorH\x00\x12H\n\nautomation\x18\x02 \x01(\x0b\x32\x32.qwak.builds.orchestrator.AutomationBuildInitiatorH\x00\x42\x0b\n\tInitiator\"\x14\n\x12UserBuildInitiator\"`\n\x18\x41utomationBuildInitiator\x12\x15\n\rautomation_id\x18\x01 \x01(\t\x12\x14\n\x0c\x65xecution_id\x18\x02 \x01(\t\x12\x17\n\x0f\x61utomation_name\x18\x03 \x01(\t*7\n\nMemoryUnit\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x82\x03\n\x07GpuType\x12\x0b\n\x07INVALID\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06\x12\x14\n\x10NVIDIA_T4_1_4_15\x10\x07\x12\x14\n\x10NVIDIA_T4_1_8_30\x10\x08\x12\x15\n\x11NVIDIA_T4_1_16_60\x10\t\x12\x1e\n\x1aNVIDIA_A100_80GB_8_96_1360\x10\n\x12\x16\n\x12NVIDIA_V100_1_8_52\x10\x0b\x12\x18\n\x14NVIDIA_V100_4_32_208\x10\x0c\x12\x13\n\x0fNVIDIA_H100_NVL\x10\r\x12\x0e\n\nNVIDIA_A10\x10\x0e\x12\x13\n\x0fNVIDIA_H100_SXM\x10\x0f\x12\x14\n\x10NVIDIA_H100_40GB\x10\x10\x12\x14\n\x10NVIDIA_H100_80GB\x10\x11\x12\x0e\n\nNVIDIA_M60\x10\x12*d\n\x0e\x44\x61taColumnType\x12\x17\n\x13INVALID_COLUMN_TYPE\x10\x00\x12\n\n\x06OBJECT\x10\x01\x12\x07\n\x03INT\x10\x02\x12\t\n\x05\x46LOAT\x10\x03\x12\x0b\n\x07\x42OOLEAN\x10\x04\x12\x0c\n\x08\x44\x41TETIME\x10\x05*K\n\x13\x42\x61seDockerImageType\x12\"\n\x1eUNKNOWN_BASE_DOCKER_IMAGE_TYPE\x10\x00\x12\x07\n\x03\x43PU\x10\x01\x12\x07\n\x03GPU\x10\x02\x42\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.builds.build_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n#com.qwak.ai.builds.orchestrator.apiP\001'
+  _globals['_REMOTEBUILDSPEC'].fields_by_name['cpu_resources']._options = None
+  _globals['_REMOTEBUILDSPEC'].fields_by_name['cpu_resources']._serialized_options = b'\030\001'
+  _globals['_REMOTEBUILDSPEC'].fields_by_name['gpu_resources']._options = None
+  _globals['_REMOTEBUILDSPEC'].fields_by_name['gpu_resources']._serialized_options = b'\030\001'
+  _globals['_MEMORYUNIT']._serialized_start=2905
+  _globals['_MEMORYUNIT']._serialized_end=2960
+  _globals['_GPUTYPE']._serialized_start=2963
+  _globals['_GPUTYPE']._serialized_end=3349
+  _globals['_DATACOLUMNTYPE']._serialized_start=3351
+  _globals['_DATACOLUMNTYPE']._serialized_end=3451
+  _globals['_BASEDOCKERIMAGETYPE']._serialized_start=3453
+  _globals['_BASEDOCKERIMAGETYPE']._serialized_end=3528
+  _globals['_REMOTEBUILDSPEC']._serialized_start=171
+  _globals['_REMOTEBUILDSPEC']._serialized_end=1028
+  _globals['_BUILDPROPERTIESV1']._serialized_start=1030
+  _globals['_BUILDPROPERTIESV1']._serialized_end=1133
+  _globals['_BUILDPROPERTIES']._serialized_start=1136
+  _globals['_BUILDPROPERTIES']._serialized_end=1322
+  _globals['_MODELURISPEC']._serialized_start=1325
+  _globals['_MODELURISPEC']._serialized_end=1503
+  _globals['_BUILDENV']._serialized_start=1505
+  _globals['_BUILDENV']._serialized_end=1629
+  _globals['_DOCKERENV']._serialized_start=1632
+  _globals['_DOCKERENV']._serialized_end=1836
+  _globals['_PYTHONENV']._serialized_start=1839
+  _globals['_PYTHONENV']._serialized_end=2023
+  _globals['_CPURESOURCES']._serialized_start=2025
+  _globals['_CPURESOURCES']._serialized_end=2135
+  _globals['_GPURESOURCES']._serialized_start=2137
+  _globals['_GPURESOURCES']._serialized_end=2224
+  _globals['_DATATABLEDEFINITION']._serialized_start=2227
+  _globals['_DATATABLEDEFINITION']._serialized_end=2372
+  _globals['_DATACOLUMNDEFINITION']._serialized_start=2374
+  _globals['_DATACOLUMNDEFINITION']._serialized_end=2466
+  _globals['_DATAFORMAT']._serialized_start=2468
+  _globals['_DATAFORMAT']._serialized_end=2542
+  _globals['_CSVFORMAT']._serialized_start=2544
+  _globals['_CSVFORMAT']._serialized_end=2615
+  _globals['_BUILDINITIATOR']._serialized_start=2618
+  _globals['_BUILDINITIATOR']._serialized_end=2783
+  _globals['_USERBUILDINITIATOR']._serialized_start=2785
+  _globals['_USERBUILDINITIATOR']._serialized_end=2805
+  _globals['_AUTOMATIONBUILDINITIATOR']._serialized_start=2807
+  _globals['_AUTOMATIONBUILDINITIATOR']._serialized_end=2903
+# @@protoc_insertion_point(module_scope)

{frogml-1.2.26 → frogml-1.2.34}/frogml/_proto/qwak/builds/build_pb2.pyi

@@ -180,6 +180,7 @@ class RemoteBuildSpec(google.protobuf.message.Message):
     JFROG_TOKEN_ID_FIELD_NUMBER: builtins.int
     ARTIFACTORY_PROJECT_KEY_FIELD_NUMBER: builtins.int
     MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
     verbose: builtins.int
     """Logs verbose mode 0: warning, 1: info, 2: debug (by qwak.tools.logger.py VERBOSITY_LEVEL_MAPPING from frogml._proto.qwak.sdk)"""
     build_code_path: builtins.str
@@ -203,6 +204,10 @@ class RemoteBuildSpec(google.protobuf.message.Message):
     """Artifactory project key"""
     model_version_id: builtins.str
     """Optional - The model version's id used for the build"""
+    model_group_name: builtins.str
+    """Human readable model group identifier (e.g., "my-model-group").
+    This is the display name, not the UUID.
+    """
     @property
     def build_properties(self) -> global___BuildProperties: ...
     @property
@@ -249,9 +254,10 @@ class RemoteBuildSpec(google.protobuf.message.Message):
         jfrog_token_id: builtins.str = ...,
         artifactory_project_key: builtins.str = ...,
         model_version_id: builtins.str = ...,
+        model_group_name: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing.Literal["Resources", b"Resources", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["Resources", b"Resources", "artifactory_project_key", b"artifactory_project_key", "build_code_path", b"build_code_path", "build_config", b"build_config", "build_destined_image", b"build_destined_image", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "build_steps", b"build_steps", "build_v1_flag", b"build_v1_flag", "client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "environment_id", b"environment_id", "gpu_resources", b"gpu_resources", "jfrog_token_id", b"jfrog_token_id", "model_version_id", b"model_version_id", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "sdk_version", b"sdk_version", "verbose", b"verbose"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["Resources", b"Resources", "artifactory_project_key", b"artifactory_project_key", "build_code_path", b"build_code_path", "build_config", b"build_config", "build_destined_image", b"build_destined_image", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "build_steps", b"build_steps", "build_v1_flag", b"build_v1_flag", "client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "environment_id", b"environment_id", "gpu_resources", b"gpu_resources", "jfrog_token_id", b"jfrog_token_id", "model_group_name", b"model_group_name", "model_version_id", b"model_version_id", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "sdk_version", b"sdk_version", "verbose", b"verbose"]) -> None: ...
     def WhichOneof(self, oneof_group: typing.Literal["Resources", b"Resources"]) -> typing.Literal["cpu_resources", "gpu_resources", "client_pod_compute_resources"] | None: ...
 
 global___RemoteBuildSpec = RemoteBuildSpec

frogml-1.2.34/frogml/_proto/qwak/builds/build_values_pb2.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.builds/build_values.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@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
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1eqwak/builds/build_values.proto\x12\x15qwak.build.management\x1a/qwak/user_application/common/v0/resources.proto\"\xc3\x03\n\x10ModelBuildValues\x12\x34\n\nbuild_info\x18\x01 \x01(\x0b\x32 .qwak.build.management.BuildInfo\x12<\n\x0emodel_metadata\x18\x02 \x01(\x0b\x32$.qwak.build.management.ModelMetadata\x12\x45\n\x13model_code_location\x18\x03 \x01(\x0b\x32(.qwak.build.management.ModelCodeLocation\x12G\n\x14\x62uild_execution_spec\x18\x04 \x01(\x0b\x32).qwak.build.management.BuildExecutionSpec\x12\x30\n\x08sdk_spec\x18\x05 \x01(\x0b\x32\x1e.qwak.build.management.SdkSpec\x12G\n\x14jfrog_ecosystem_spec\x18\x06 \x01(\x0b\x32).qwak.build.management.JfrogEcosystemSpec\x12\x30\n\x08job_spec\x18\x07 \x01(\x0b\x32\x1e.qwak.build.management.JobSpec\"\xa3\x01\n\tBuildInfo\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x12\n\nbuild_tags\x18\x03 \x03(\t\x12\x14\n\x0c\x62uild_config\x18\x04 \x01(\t\x12\x45\n\x13\x62uild_properties_v1\x18\x05 \x01(\x0b\x32(.qwak.build.management.BuildPropertiesV1\"]\n\x11\x42uildPropertiesV1\x12\x18\n\x10\x62uild_config_url\x18\x01 \x01(\t\x12\x15\n\rsdk_wheel_url\x18\x02 \x01(\t\x12\x17\n\x0fsdk_version_url\x18\x03 \x01(\t\"\xa8\x01\n\rModelMetadata\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\naccount_id\x18\x02 \x01(\t\x12\x0f\n\x07user_id\x18\x03 \x01(\t\x12\x46\n\x13\x61rtifactory_details\x18\x04 \x01(\x0b\x32).qwak.build.management.ArtifactoryDetails\x12\x18\n\x10model_group_name\x18\x05 \x01(\t\")\n\x12\x41rtifactoryDetails\x12\x13\n\x0bproject_key\x18\x01 \x01(\t\"\xa0\x01\n\x11ModelCodeLocation\x12\x1c\n\x14main_model_directory\x18\x01 \x01(\t\x12\x34\n\tlocal_uri\x18\x02 \x01(\x0b\x32\x1f.qwak.build.management.LocalUriH\x00\x12\x30\n\x07git_uri\x18\x03 \x01(\x0b\x32\x1d.qwak.build.management.GitUriH\x00\x42\x05\n\x03uri\"\x18\n\x08LocalUri\x12\x0c\n\x04path\x18\x01 \x01(\t\"M\n\x06GitUri\x12\x0b\n\x03url\x18\x01 \x01(\t\x12\x36\n\x0bgit_details\x18\x02 \x01(\x0b\x32!.qwak.build.management.GitDetails\"|\n\nGitDetails\x12\x0e\n\x06\x62ranch\x18\x01 \x01(\t\x12\x11\n\tcommit_id\x18\x02 \x01(\t\x12\x15\n\x0b\x63redentials\x18\x03 \x01(\tH\x00\x12!\n\x17\x63redentials_secret_name\x18\x04 \x01(\tH\x00\x42\x11\n\x0fgit_credentials\"\xd0\x02\n\x12\x42uildExecutionSpec\x12\x1f\n\x17\x62uild_image_destination\x18\x01 \x01(\t\x12\"\n\x1ashould_build_without_cache\x18\x02 \x01(\x08\x12G\n\x14logs_verbosity_level\x18\x03 \x01(\x0e\x32).qwak.build.management.LogsVerbosityLevel\x12\x12\n\nbase_image\x18\x04 \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x05 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\x06 \x03(\t\x12\'\n\x1fremote_builder_auth_secret_name\x18\x07 \x01(\t\x12@\n\x10\x63ontainer_engine\x18\x08 \x01(\x0b\x32&.qwak.build.management.ContainerEngine\"a\n\x07SdkSpec\x12\x13\n\x0bsdk_version\x18\x01 \x01(\t\x12\x1b\n\x13sdk_extra_index_url\x18\x02 \x01(\t\x12$\n\x1csdk_runtime_version_override\x18\x03 \x01(\t\"k\n\x12JfrogEcosystemSpec\x12\x0f\n\x07\x62\x61seUrl\x18\x01 \x01(\t\x12\x1d\n\x15\x61\x63\x63\x65ssTokenSecretName\x18\x02 \x01(\t\x12%\n\x1d\x61rtifactorySettingsSecretName\x18\x03 \x01(\t\"\xb5\x04\n\x07JobSpec\x12\x11\n\tnamespace\x18\x01 \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x02 \x01(\x05\x12>\n\x0fpurchase_option\x18\x03 \x01(\x0e\x32%.qwak.build.management.PurchaseOption\x12$\n\x1cis_node_optimization_enabled\x18\x04 \x01(\x08\x12\"\n\x1ashould_use_build_isolation\x18\x05 \x01(\x08\x12#\n\x1bis_multi_tenant_environment\x18\x06 \x01(\x08\x12\x1a\n\x12is_self_hosted_jpd\x18\n \x01(\x08\x12 \n\x18\x62uild_executor_image_url\x18\x07 \x01(\t\x12<\n\x0e\x63loud_provider\x18\x08 \x01(\x0e\x32$.qwak.build.management.CloudProvider\x12\x30\n\x08job_size\x18\t \x01(\x0b\x32\x1e.qwak.build.management.JobSize\x12Y\n\x17\x62uild_executor_env_vars\x18\x0b \x03(\x0b\x32\x38.qwak.build.management.JobSpec.BuildExecutorEnvVarsEntry\x1a;\n\x19\x42uildExecutorEnvVarsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x9e\x02\n\x07JobSize\x12\x0f\n\x03\x63pu\x18\x01 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x02 \x01(\x05\x42\x02\x18\x01\x12;\n\x0cmemory_units\x18\x03 \x01(\x0e\x32!.qwak.build.management.MemoryUnitB\x02\x18\x01\x12H\n\rgpu_resources\x18\x04 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResourcesB\x02\x18\x01\x12`\n\x1c\x63lient_pod_compute_resources\x18\x05 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\"\x9d\x01\n\x0f\x43ontainerEngine\x12>\n\x06\x64ocker\x18\x01 \x01(\x0b\x32,.qwak.build.management.DockerContainerEngineH\x00\x12\x42\n\x06podman\x18\x02 \x01(\x0b\x32,.qwak.build.management.PodmanContainerEngineB\x02\x18\x01H\x00\x42\x06\n\x04type\"\x17\n\x15\x44ockerContainerEngine\"\x17\n\x15PodmanContainerEngine*e\n\x0ePurchaseOption\x12\x1b\n\x17INVALID_PURCHASE_OPTION\x10\x00\x12\x18\n\x14SPOT_PURCHASE_OPTION\x10\x01\x12\x1c\n\x18ONDEMAND_PURCHASE_OPTION\x10\x02*H\n\rCloudProvider\x12\x1a\n\x16INVALID_CLOUD_PROVIDER\x10\x00\x12\x07\n\x03\x41WS\x10\x01\x12\x07\n\x03GCP\x10\x02\x12\t\n\x05\x41ZURE\x10\x03*7\n\nMemoryUnit\x12\x17\n\x13INVALID_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*S\n\x12LogsVerbosityLevel\x12\x1a\n\x16INVALID_LOGS_VERBOSITY\x10\x00\x12\t\n\x05\x44\x45\x42UG\x10\n\x12\x08\n\x04INFO\x10\x14\x12\x0c\n\x08\x43RITICAL\x10\x32\x42\x36\n com.qwak.ai.management.build.apiP\x01Z\x10qwak/build;buildb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.builds.build_values_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n com.qwak.ai.management.build.apiP\001Z\020qwak/build;build'
+  _globals['_JOBSPEC_BUILDEXECUTORENVVARSENTRY']._options = None
+  _globals['_JOBSPEC_BUILDEXECUTORENVVARSENTRY']._serialized_options = b'8\001'
+  _globals['_JOBSIZE'].fields_by_name['cpu']._options = None
+  _globals['_JOBSIZE'].fields_by_name['cpu']._serialized_options = b'\030\001'
+  _globals['_JOBSIZE'].fields_by_name['memory_amount']._options = None
+  _globals['_JOBSIZE'].fields_by_name['memory_amount']._serialized_options = b'\030\001'
+  _globals['_JOBSIZE'].fields_by_name['memory_units']._options = None
+  _globals['_JOBSIZE'].fields_by_name['memory_units']._serialized_options = b'\030\001'
+  _globals['_JOBSIZE'].fields_by_name['gpu_resources']._options = None
+  _globals['_JOBSIZE'].fields_by_name['gpu_resources']._serialized_options = b'\030\001'
+  _globals['_CONTAINERENGINE'].fields_by_name['podman']._options = None
+  _globals['_CONTAINERENGINE'].fields_by_name['podman']._serialized_options = b'\030\001'
+  _globals['_PURCHASEOPTION']._serialized_start=3043
+  _globals['_PURCHASEOPTION']._serialized_end=3144
+  _globals['_CLOUDPROVIDER']._serialized_start=3146
+  _globals['_CLOUDPROVIDER']._serialized_end=3218
+  _globals['_MEMORYUNIT']._serialized_start=3220
+  _globals['_MEMORYUNIT']._serialized_end=3275
+  _globals['_LOGSVERBOSITYLEVEL']._serialized_start=3277
+  _globals['_LOGSVERBOSITYLEVEL']._serialized_end=3360
+  _globals['_MODELBUILDVALUES']._serialized_start=107
+  _globals['_MODELBUILDVALUES']._serialized_end=558
+  _globals['_BUILDINFO']._serialized_start=561
+  _globals['_BUILDINFO']._serialized_end=724
+  _globals['_BUILDPROPERTIESV1']._serialized_start=726
+  _globals['_BUILDPROPERTIESV1']._serialized_end=819
+  _globals['_MODELMETADATA']._serialized_start=822
+  _globals['_MODELMETADATA']._serialized_end=990
+  _globals['_ARTIFACTORYDETAILS']._serialized_start=992
+  _globals['_ARTIFACTORYDETAILS']._serialized_end=1033
+  _globals['_MODELCODELOCATION']._serialized_start=1036
+  _globals['_MODELCODELOCATION']._serialized_end=1196
+  _globals['_LOCALURI']._serialized_start=1198
+  _globals['_LOCALURI']._serialized_end=1222
+  _globals['_GITURI']._serialized_start=1224
+  _globals['_GITURI']._serialized_end=1301
+  _globals['_GITDETAILS']._serialized_start=1303
+  _globals['_GITDETAILS']._serialized_end=1427
+  _globals['_BUILDEXECUTIONSPEC']._serialized_start=1430
+  _globals['_BUILDEXECUTIONSPEC']._serialized_end=1766
+  _globals['_SDKSPEC']._serialized_start=1768
+  _globals['_SDKSPEC']._serialized_end=1865
+  _globals['_JFROGECOSYSTEMSPEC']._serialized_start=1867
+  _globals['_JFROGECOSYSTEMSPEC']._serialized_end=1974
+  _globals['_JOBSPEC']._serialized_start=1977
+  _globals['_JOBSPEC']._serialized_end=2542
+  _globals['_JOBSPEC_BUILDEXECUTORENVVARSENTRY']._serialized_start=2483
+  _globals['_JOBSPEC_BUILDEXECUTORENVVARSENTRY']._serialized_end=2542
+  _globals['_JOBSIZE']._serialized_start=2545
+  _globals['_JOBSIZE']._serialized_end=2831
+  _globals['_CONTAINERENGINE']._serialized_start=2834
+  _globals['_CONTAINERENGINE']._serialized_end=2991
+  _globals['_DOCKERCONTAINERENGINE']._serialized_start=2993
+  _globals['_DOCKERCONTAINERENGINE']._serialized_end=3016
+  _globals['_PODMANCONTAINERENGINE']._serialized_start=3018
+  _globals['_PODMANCONTAINERENGINE']._serialized_end=3041
+# @@protoc_insertion_point(module_scope)

{frogml-1.2.26 → frogml-1.2.34}/frogml/_proto/qwak/builds/build_values_pb2.pyi

@@ -217,9 +217,14 @@ class ModelMetadata(google.protobuf.message.Message):
     ACCOUNT_ID_FIELD_NUMBER: builtins.int
     USER_ID_FIELD_NUMBER: builtins.int
     ARTIFACTORY_DETAILS_FIELD_NUMBER: builtins.int
+    MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
     model_id: builtins.str
     account_id: builtins.str
     user_id: builtins.str
+    model_group_name: builtins.str
+    """Human readable model group identifier (e.g., "my-model-group").
+    This is the display name, not the UUID.
+    """
     @property
     def artifactory_details(self) -> global___ArtifactoryDetails: ...
     def __init__(
@@ -229,9 +234,10 @@ class ModelMetadata(google.protobuf.message.Message):
         account_id: builtins.str = ...,
         user_id: builtins.str = ...,
         artifactory_details: global___ArtifactoryDetails | None = ...,
+        model_group_name: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing.Literal["artifactory_details", b"artifactory_details"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["account_id", b"account_id", "artifactory_details", b"artifactory_details", "model_id", b"model_id", "user_id", b"user_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["account_id", b"account_id", "artifactory_details", b"artifactory_details", "model_group_name", b"model_group_name", "model_id", b"model_id", "user_id", b"user_id"]) -> None: ...
 
 global___ModelMetadata = ModelMetadata
 

frogml-1.2.34/frogml/_proto/qwak/fitness_service/fitness_pb2.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.fitness_service/fitness.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"qwak/fitness_service/fitness.proto\x12\x14qwak.fitness.service*e\n\x0ePurchaseOption\x12\x1b\n\x17INVALID_PURCHASE_OPTION\x10\x00\x12\x18\n\x14SPOT_PURCHASE_OPTION\x10\x01\x12\x1c\n\x18ONDEMAND_PURCHASE_OPTION\x10\x02\x42#\n\x1f\x63om.qwak.ai.fitness.service.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.fitness_service.fitness_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
+  _globals['_PURCHASEOPTION']._serialized_start=60
+  _globals['_PURCHASEOPTION']._serialized_end=161
+# @@protoc_insertion_point(module_scope)

frogml-1.2.34/frogml/_proto/qwak/fitness_service/fitness_pb2.pyi

@@ -0,0 +1,34 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _PurchaseOption:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _PurchaseOptionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PurchaseOption.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    INVALID_PURCHASE_OPTION: _PurchaseOption.ValueType  # 0
+    SPOT_PURCHASE_OPTION: _PurchaseOption.ValueType  # 1
+    ONDEMAND_PURCHASE_OPTION: _PurchaseOption.ValueType  # 2
+
+class PurchaseOption(_PurchaseOption, metaclass=_PurchaseOptionEnumTypeWrapper): ...
+
+INVALID_PURCHASE_OPTION: PurchaseOption.ValueType  # 0
+SPOT_PURCHASE_OPTION: PurchaseOption.ValueType  # 1
+ONDEMAND_PURCHASE_OPTION: PurchaseOption.ValueType  # 2
+global___PurchaseOption = PurchaseOption

frogml-1.2.34/frogml/_proto/qwak/fitness_service/status_pb2.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.fitness_service/status.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!qwak/fitness_service/status.proto\x12\x14qwak.fitness.service\"\xa2\x01\n\tPodStatus\x12\x41\n\x12\x63ontainer_statuses\x18\x01 \x03(\x0b\x32%.qwak.fitness.service.ContainerStatus\x12\x0f\n\x07message\x18\x02 \x01(\t\x12\r\n\x05phase\x18\x03 \x01(\t\x12\x0e\n\x06reason\x18\x04 \x01(\t\x12\x12\n\nstart_time\x18\x05 \x01(\t\x12\x0e\n\x06pod_id\x18\x06 \x01(\t\"\x9d\x01\n\x0f\x43ontainerStatus\x12\x14\n\x0c\x63ontainer_id\x18\x01 \x01(\t\x12\r\n\x05image\x18\x02 \x01(\t\x12\x15\n\rrestart_count\x18\x03 \x01(\x05\x12\x0f\n\x07started\x18\x04 \x01(\x08\x12=\n\x0f\x63ontainer_state\x18\x05 \x01(\x0b\x32$.qwak.fitness.service.ContainerState\"\x8f\x02\n\x0e\x43ontainerState\x12N\n\x17\x63ontainer_state_running\x18\x01 \x01(\x0b\x32+.qwak.fitness.service.ContainerStateRunningH\x00\x12T\n\x1a\x63ontainer_state_terminated\x18\x02 \x01(\x0b\x32..qwak.fitness.service.ContainerStateTerminatedH\x00\x12N\n\x17\x63ontainer_state_waiting\x18\x03 \x01(\x0b\x32+.qwak.fitness.service.ContainerStateWaitingH\x00\x42\x07\n\x05state\"+\n\x15\x43ontainerStateRunning\x12\x12\n\nstarted_at\x18\x01 \x01(\t\"\x9d\x01\n\x18\x43ontainerStateTerminated\x12\x14\n\x0c\x63ontainer_id\x18\x01 \x01(\t\x12\x11\n\texit_code\x18\x02 \x01(\x05\x12\x13\n\x0b\x66inished_at\x18\x03 \x01(\t\x12\x0f\n\x07message\x18\x04 \x01(\t\x12\x0e\n\x06reason\x18\x05 \x01(\t\x12\x0e\n\x06signal\x18\x06 \x01(\x05\x12\x12\n\nstarted_at\x18\x07 \x01(\t\"8\n\x15\x43ontainerStateWaiting\x12\x0f\n\x07message\x18\x01 \x01(\t\x12\x0e\n\x06reason\x18\x02 \x01(\tB#\n\x1f\x63om.qwak.ai.fitness.service.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.fitness_service.status_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
+  _globals['_PODSTATUS']._serialized_start=60
+  _globals['_PODSTATUS']._serialized_end=222
+  _globals['_CONTAINERSTATUS']._serialized_start=225
+  _globals['_CONTAINERSTATUS']._serialized_end=382
+  _globals['_CONTAINERSTATE']._serialized_start=385
+  _globals['_CONTAINERSTATE']._serialized_end=656
+  _globals['_CONTAINERSTATERUNNING']._serialized_start=658
+  _globals['_CONTAINERSTATERUNNING']._serialized_end=701
+  _globals['_CONTAINERSTATETERMINATED']._serialized_start=704
+  _globals['_CONTAINERSTATETERMINATED']._serialized_end=861
+  _globals['_CONTAINERSTATEWAITING']._serialized_start=863
+  _globals['_CONTAINERSTATEWAITING']._serialized_end=919
+# @@protoc_insertion_point(module_scope)

{frogml-1.2.26 → frogml-1.2.34}/frogml/_proto/qwak/fitness_service/status_pb2.pyi

@@ -7,139 +7,11 @@ import builtins
 import collections.abc
 import google.protobuf.descriptor
 import google.protobuf.internal.containers
-import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
-import sys
 import typing
 
-if sys.version_info >= (3, 10):
-    import typing as typing_extensions
-else:
-    import typing_extensions
-
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
-class _RemoteBuildingStatusCode:
-    ValueType = typing.NewType("ValueType", builtins.int)
-    V: typing_extensions.TypeAlias = ValueType
-
-class _RemoteBuildingStatusCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RemoteBuildingStatusCode.ValueType], builtins.type):
-    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    INVALID_STATUS: _RemoteBuildingStatusCode.ValueType  # 0
-    """Invalid remote build status"""
-    INITIALIZING: _RemoteBuildingStatusCode.ValueType  # 1
-    """Setting up remote build"""
-    IN_PROGRESS: _RemoteBuildingStatusCode.ValueType  # 2
-    """Running remote build"""
-    SUCCESSFUL: _RemoteBuildingStatusCode.ValueType  # 3
-    """Successful remote build"""
-    FAILED: _RemoteBuildingStatusCode.ValueType  # 4
-    """Failed remote build"""
-    CANCELLED: _RemoteBuildingStatusCode.ValueType  # 5
-    """Cancelled remote build"""
-    TIMED_OUT: _RemoteBuildingStatusCode.ValueType  # 6
-    """Timed out while pending remote build"""
-    UNKNOWN: _RemoteBuildingStatusCode.ValueType  # 7
-    """The status is unknown"""
-
-class RemoteBuildingStatusCode(_RemoteBuildingStatusCode, metaclass=_RemoteBuildingStatusCodeEnumTypeWrapper): ...
-
-INVALID_STATUS: RemoteBuildingStatusCode.ValueType  # 0
-"""Invalid remote build status"""
-INITIALIZING: RemoteBuildingStatusCode.ValueType  # 1
-"""Setting up remote build"""
-IN_PROGRESS: RemoteBuildingStatusCode.ValueType  # 2
-"""Running remote build"""
-SUCCESSFUL: RemoteBuildingStatusCode.ValueType  # 3
-"""Successful remote build"""
-FAILED: RemoteBuildingStatusCode.ValueType  # 4
-"""Failed remote build"""
-CANCELLED: RemoteBuildingStatusCode.ValueType  # 5
-"""Cancelled remote build"""
-TIMED_OUT: RemoteBuildingStatusCode.ValueType  # 6
-"""Timed out while pending remote build"""
-UNKNOWN: RemoteBuildingStatusCode.ValueType  # 7
-"""The status is unknown"""
-global___RemoteBuildingStatusCode = RemoteBuildingStatusCode
-
-class _RemoteBuildingFailureReasonCode:
-    ValueType = typing.NewType("ValueType", builtins.int)
-    V: typing_extensions.TypeAlias = ValueType
-
-class _RemoteBuildingFailureReasonCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RemoteBuildingFailureReasonCode.ValueType], builtins.type):
-    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
-    INVALID_CODE: _RemoteBuildingFailureReasonCode.ValueType  # 0
-    """Invalid reason"""
-    UNKNOWN_REASON: _RemoteBuildingFailureReasonCode.ValueType  # 1
-    """Unknown reason"""
-    INSUFFICIENT_RESOURCES: _RemoteBuildingFailureReasonCode.ValueType  # 2
-    """Insufficient resources - for unscheduled pods"""
-    MISSING_DOCKER_IMAGE: _RemoteBuildingFailureReasonCode.ValueType  # 3
-    """The docker image is missing"""
-    MEMORY_LIMIT_EXCEEDED: _RemoteBuildingFailureReasonCode.ValueType  # 4
-    """Out of memory"""
-
-class RemoteBuildingFailureReasonCode(_RemoteBuildingFailureReasonCode, metaclass=_RemoteBuildingFailureReasonCodeEnumTypeWrapper): ...
-
-INVALID_CODE: RemoteBuildingFailureReasonCode.ValueType  # 0
-"""Invalid reason"""
-UNKNOWN_REASON: RemoteBuildingFailureReasonCode.ValueType  # 1
-"""Unknown reason"""
-INSUFFICIENT_RESOURCES: RemoteBuildingFailureReasonCode.ValueType  # 2
-"""Insufficient resources - for unscheduled pods"""
-MISSING_DOCKER_IMAGE: RemoteBuildingFailureReasonCode.ValueType  # 3
-"""The docker image is missing"""
-MEMORY_LIMIT_EXCEEDED: RemoteBuildingFailureReasonCode.ValueType  # 4
-"""Out of memory"""
-global___RemoteBuildingFailureReasonCode = RemoteBuildingFailureReasonCode
-
-@typing.final
-class RemoteBuildingStatus(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    REMOTE_BUILDING_STATUS_CODE_FIELD_NUMBER: builtins.int
-    FAILURE_DETAILS_FIELD_NUMBER: builtins.int
-    POD_STATUS_FIELD_NUMBER: builtins.int
-    remote_building_status_code: global___RemoteBuildingStatusCode.ValueType
-    """The status of the building"""
-    @property
-    def failure_details(self) -> global___RemoteBuildingFailureReasonDetails:
-        """Failure details"""
-
-    @property
-    def pod_status(self) -> global___PodStatus: ...
-    def __init__(
-        self,
-        *,
-        remote_building_status_code: global___RemoteBuildingStatusCode.ValueType = ...,
-        failure_details: global___RemoteBuildingFailureReasonDetails | None = ...,
-        pod_status: global___PodStatus | None = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing.Literal["failure_details", b"failure_details", "pod_status", b"pod_status"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["failure_details", b"failure_details", "pod_status", b"pod_status", "remote_building_status_code", b"remote_building_status_code"]) -> None: ...
-
-global___RemoteBuildingStatus = RemoteBuildingStatus
-
-@typing.final
-class RemoteBuildingFailureReasonDetails(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    FAILURE_CODE_FIELD_NUMBER: builtins.int
-    TECHNICAL_DETAILS_FIELD_NUMBER: builtins.int
-    failure_code: global___RemoteBuildingFailureReasonCode.ValueType
-    """failure reason code"""
-    technical_details: builtins.str
-    """Extra messages"""
-    def __init__(
-        self,
-        *,
-        failure_code: global___RemoteBuildingFailureReasonCode.ValueType = ...,
-        technical_details: builtins.str = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["failure_code", b"failure_code", "technical_details", b"technical_details"]) -> None: ...
-
-global___RemoteBuildingFailureReasonDetails = RemoteBuildingFailureReasonDetails
-
 @typing.final
 class PodStatus(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor

{frogml-1.2.26 → frogml-1.2.34}/frogml/core/clients/model_version_manager/client.py

@@ -6,18 +6,8 @@ from dependency_injector.wiring import Provide
 from grpc import RpcError
 from typing_extensions import Self
 
-from frogml.core.clients.model_version_manager.build_model_version_dto import (
-    BuildConfigDTO,
-)
-from frogml.core.clients.model_version_manager.build_model_version_request_mapper import (
-    map_build_conf_to_build_spec,
-)
-from frogml.core.exceptions import FrogmlException
-from frogml.core.inner.di_configuration import FrogmlContainer
-from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
-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.build_spec_pb2 import BuildSpec
 from frogml._proto.jfml.model_version.v1.model_repository_spec_pb2 import (
     ModelRepositorySpec,
 )
@@ -39,10 +29,19 @@ from frogml._proto.jfml.model_version.v1.model_version_pb2 import (
     ParameterValue,
     MetricValue,
 )
-from frogml._proto.jfml.model_version.v1.build_spec_pb2 import BuildSpec
+from frogml.core.clients.model_version_manager.build_model_version_dto import (
+    BuildConfigDTO,
+)
+from frogml.core.clients.model_version_manager.build_model_version_request_mapper import (
+    map_build_conf_to_build_spec,
+)
+from frogml.core.exceptions import FrogmlException
+from frogml.core.inner.di_configuration import FrogmlContainer
+from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
+from frogml.core.utils.model_utils import get_model_id_from_model_name
+from frogml.core.utils.proto_utils import ProtoUtils
 from frogml.storage.models.entity_manifest import Artifact
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -87,7 +86,6 @@ class ModelVersionManagerClient:
 
     def __build_create_model_version_request(
         self,
-        project_key: str,
         repository_key: str,
         model_name: str,
         model_version_name: str,
@@ -107,7 +105,6 @@ class ModelVersionManagerClient:
             dry_run=dry_run,
             model_version=ModelVersionSpec(
                 repository_spec=ModelRepositorySpec(
-                    project_key=project_key,
                     repository_key=repository_key,
                     model_id=get_model_id_from_model_name(model_name),
                     model_name=model_name,
@@ -121,7 +118,6 @@ class ModelVersionManagerClient:
 
     def validate_create_model_version(
         self,
-        project_key: str,
         repository_key: str,
         model_name: str,
         model_version_name: str,
@@ -142,7 +138,6 @@ class ModelVersionManagerClient:
             )
 
             create_model_request = self.__build_create_model_version_request(
-                project_key=project_key,
                 repository_key=repository_key,
                 model_name=model_name,
                 model_version_name=model_version_name,
@@ -165,7 +160,6 @@ class ModelVersionManagerClient:
 
     def create_model_version(
         self,
-        project_key: str,
         repository_key: str,
         model_name: str,
         model_version_name: str,
@@ -186,7 +180,6 @@ class ModelVersionManagerClient:
             )
 
             create_model_request = self.__build_create_model_version_request(
-                project_key=project_key,
                 repository_key=repository_key,
                 model_name=model_name,
                 model_version_name=model_version_name,