frogml 1.2.35__py3-none-any.whl → 1.2.41__py3-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 (46) hide show
  1. frogml/__init__.py +1 -1
  2. frogml/_proto/com/jfrog/ml/model/deployment/v1/auto_scaling_pb2.py +43 -0
  3. frogml/_proto/com/jfrog/ml/model/deployment/v1/auto_scaling_pb2.pyi +218 -0
  4. frogml/_proto/com/jfrog/ml/model/deployment/v1/auto_scaling_pb2_grpc.py +4 -0
  5. frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_pb2.py +53 -0
  6. frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_pb2.pyi +258 -0
  7. frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_pb2_grpc.py +4 -0
  8. frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_service_pb2.py +58 -0
  9. frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_service_pb2.pyi +224 -0
  10. frogml/_proto/com/jfrog/ml/model/deployment/v1/deployment_service_pb2_grpc.py +206 -0
  11. frogml/_proto/com/jfrog/ml/model/deployment/v1/environment_variables_configuration_pb2.py +33 -0
  12. frogml/_proto/com/jfrog/ml/model/deployment/v1/environment_variables_configuration_pb2.pyi +99 -0
  13. frogml/_proto/com/jfrog/ml/model/deployment/v1/environment_variables_configuration_pb2_grpc.py +4 -0
  14. frogml/_proto/com/jfrog/ml/model/deployment/v1/model_artifact_identifier_pb2.py +31 -0
  15. frogml/_proto/com/jfrog/ml/model/deployment/v1/model_artifact_identifier_pb2.pyi +99 -0
  16. frogml/_proto/com/jfrog/ml/model/deployment/v1/model_artifact_identifier_pb2_grpc.py +4 -0
  17. frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_brief_pb2.py +30 -0
  18. frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_brief_pb2.pyi +55 -0
  19. frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_brief_pb2_grpc.py +4 -0
  20. frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_filter_pb2.py +42 -0
  21. frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_filter_pb2.pyi +204 -0
  22. frogml/_proto/com/jfrog/ml/model/deployment/v1/model_deployment_filter_pb2_grpc.py +4 -0
  23. frogml/_proto/com/jfrog/ml/model/deployment/v1/probes_configuration_pb2.py +29 -0
  24. frogml/_proto/com/jfrog/ml/model/deployment/v1/probes_configuration_pb2.pyi +60 -0
  25. frogml/_proto/com/jfrog/ml/model/deployment/v1/probes_configuration_pb2_grpc.py +4 -0
  26. frogml/_proto/com/jfrog/ml/model/deployment/v1/realtime_deployment_pb2.py +28 -0
  27. frogml/_proto/com/jfrog/ml/model/deployment/v1/realtime_deployment_pb2.pyi +47 -0
  28. frogml/_proto/com/jfrog/ml/model/deployment/v1/realtime_deployment_pb2_grpc.py +4 -0
  29. frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_configuration_pb2.py +27 -0
  30. frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_configuration_pb2.pyi +29 -0
  31. frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_configuration_pb2_grpc.py +4 -0
  32. frogml/_proto/qwak/logging/log_source_pb2.py +31 -25
  33. frogml/_proto/qwak/logging/log_source_pb2.pyi +28 -14
  34. frogml/_proto/qwak/models/models_pb2.py +81 -79
  35. frogml/_proto/qwak/models/models_pb2.pyi +24 -0
  36. frogml/_proto/qwak/models/models_pb2_grpc.py +34 -0
  37. frogml/core/clients/batch_job_management/client.py +14 -6
  38. frogml/core/clients/logging_client/client.py +118 -33
  39. frogml/core/clients/model_deployment_manager/__init__.py +1 -0
  40. frogml/core/clients/model_deployment_manager/client.py +103 -0
  41. {frogml-1.2.35.dist-info → frogml-1.2.41.dist-info}/METADATA +1 -1
  42. {frogml-1.2.35.dist-info → frogml-1.2.41.dist-info}/RECORD +46 -13
  43. frogml_services_mock/mocks/frogml_mocks.py +11 -3
  44. frogml_services_mock/mocks/model_deployment_manager_service_mock.py +255 -0
  45. frogml_services_mock/services_mock.py +18 -2
  46. {frogml-1.2.35.dist-info → frogml-1.2.41.dist-info}/WHEEL +0 -0
frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_configuration_pb2.py ADDED
@@ -0,0 +1,27 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: com/jfrog/ml/model/deployment/v1/resource_configuration.proto
4
+ # Protobuf Python Version: 4.25.1
5
+ """Generated protocol buffer code."""
6
+ from google.protobuf import descriptor as _descriptor
7
+ from google.protobuf import descriptor_pool as _descriptor_pool
8
+ from google.protobuf import symbol_database as _symbol_database
9
+ from google.protobuf.internal import builder as _builder
10
+ # @@protoc_insertion_point(imports)
11
+
12
+ _sym_db = _symbol_database.Default()
13
+
14
+
15
+
16
+
17
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n=com/jfrog/ml/model/deployment/v1/resource_configuration.proto\x12 com.jfrog.ml.model.deployment.v1\"U\n\x15ResourceConfiguration\x12 \n\x16template_specification\x18\x01 \x01(\tH\x00\x42\x1a\n\x18resource_descriptor_typeB(\n$com.jfrog.ml.model.deployment.v1.apiP\x01\x62\x06proto3')
18
+
19
+ _globals = globals()
20
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
21
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'com.jfrog.ml.model.deployment.v1.resource_configuration_pb2', _globals)
22
+ if _descriptor._USE_C_DESCRIPTORS == False:
23
+ _globals['DESCRIPTOR']._options = None
24
+ _globals['DESCRIPTOR']._serialized_options = b'\n$com.jfrog.ml.model.deployment.v1.apiP\001'
25
+ _globals['_RESOURCECONFIGURATION']._serialized_start=99
26
+ _globals['_RESOURCECONFIGURATION']._serialized_end=184
27
+ # @@protoc_insertion_point(module_scope)
frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_configuration_pb2.pyi ADDED
@@ -0,0 +1,29 @@
1
+ """
2
+ @generated by mypy-protobuf. Do not edit manually!
3
+ isort:skip_file
4
+ """
5
+
6
+ import builtins
7
+ import google.protobuf.descriptor
8
+ import google.protobuf.message
9
+ import typing
10
+
11
+ DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
12
+
13
+ @typing.final
14
+ class ResourceConfiguration(google.protobuf.message.Message):
15
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
16
+
17
+ TEMPLATE_SPECIFICATION_FIELD_NUMBER: builtins.int
18
+ template_specification: builtins.str
19
+ """A legacy field for describing a compute resources allocation using templated resource sizes."""
20
+ def __init__(
21
+ self,
22
+ *,
23
+ template_specification: builtins.str = ...,
24
+ ) -> None: ...
25
+ def HasField(self, field_name: typing.Literal["resource_descriptor_type", b"resource_descriptor_type", "template_specification", b"template_specification"]) -> builtins.bool: ...
26
+ def ClearField(self, field_name: typing.Literal["resource_descriptor_type", b"resource_descriptor_type", "template_specification", b"template_specification"]) -> None: ...
27
+ def WhichOneof(self, oneof_group: typing.Literal["resource_descriptor_type", b"resource_descriptor_type"]) -> typing.Literal["template_specification"] | None: ...
28
+
29
+ global___ResourceConfiguration = ResourceConfiguration
frogml/_proto/com/jfrog/ml/model/deployment/v1/resource_configuration_pb2_grpc.py ADDED
@@ -0,0 +1,4 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+
frogml/_proto/qwak/logging/log_source_pb2.py CHANGED
@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
14
14
 
15
15
 
16
16
 
17
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dqwak/logging/log_source.proto\x12\x14qwak.logging.service\"\xca\x04\n\tLogSource\x12\x41\n\rmodel_runtime\x18\x01 \x01(\x0b\x32(.qwak.logging.service.ModelRuntimeSourceH\x00\x12?\n\x0cremote_build\x18\x02 \x01(\x0b\x32\'.qwak.logging.service.RemoteBuildSourceH\x00\x12M\n\x13inference_execution\x18\x03 \x01(\x0b\x32..qwak.logging.service.InferenceExecutionSourceH\x00\x12P\n\x15streaming_feature_set\x18\x04 \x01(\x0b\x32/.qwak.logging.service.StreamingFeatureSetSourceH\x00\x12g\n!streaming_aggregation_feature_set\x18\x05 \x01(\x0b\x32:.qwak.logging.service.StreamingAggregationFeatureSetSourceH\x00\x12H\n\x11\x62\x61tch_feature_set\x18\x06 \x01(\x0b\x32+.qwak.logging.service.BatchFeatureSetSourceH\x00\x12[\n\x1breal_time_feature_extractor\x18\x07 \x01(\x0b\x32\x34.qwak.logging.service.RealTimeFeatureExtractorSourceH\x00\x42\x08\n\x06source\"f\n\x11RemoteBuildSource\x12\x18\n\x10model_group_name\x18\x03 \x01(\t\x12\x12\n\nmodel_name\x18\x04 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x11\n\tphase_ids\x18\x02 \x03(\t\"|\n\x12ModelRuntimeSource\x12\x18\n\x10model_group_name\x18\x03 \x01(\t\x12\x12\n\nmodel_name\x18\x04 \x01(\t\x12\x17\n\rdeployment_id\x18\x01 \x01(\tH\x00\x12\x12\n\x08\x62uild_id\x18\x02 \x01(\tH\x00\x42\x0b\n\tsearch_by\"\x8e\x01\n\x18InferenceExecutionSource\x12\x18\n\x10model_group_name\x18\x03 \x01(\t\x12\x12\n\nmodel_name\x18\x04 \x01(\t\x12\x1a\n\x10inference_job_id\x18\x01 \x01(\tH\x00\x12\x1b\n\x11inference_task_id\x18\x02 \x01(\tH\x00\x42\x0b\n\tsearch_by\"\xe4\x01\n\x19StreamingFeatureSetSource\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x05 \x01(\t\x12\x34\n\x07offline\x18\x03 \x01(\x0b\x32!.qwak.logging.service.OfflineTypeH\x00\x12\x32\n\x06online\x18\x04 \x01(\x0b\x32 .qwak.logging.service.OnlineTypeH\x00\x42\x06\n\x04type\"\xa0\x01\n\x15\x42\x61tchFeatureSetSource\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x04 \x01(\t\x12\"\n\x1a\x66ilter_by_feature_set_name\x18\x05 \x01(\x08\x12\x0e\n\x06run_id\x18\x03 \x01(\t\"\xdb\x01\n$StreamingAggregationFeatureSetSource\x12\x37\n\trow_level\x18\x01 \x01(\x0b\x32\".qwak.logging.service.RowLevelTypeH\x00\x12:\n\ncompaction\x18\x02 \x01(\x0b\x32$.qwak.logging.service.CompactionTypeH\x00\x12\x36\n\x08\x62\x61\x63kfill\x18\x03 \x01(\x0b\x32\".qwak.logging.service.BackfillTypeH\x00\x42\x06\n\x04type\"c\n\nOnlineType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x03 \x01(\t\"t\n\x0bOfflineType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x04 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\t\"e\n\x0cRowLevelType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x03 \x01(\t\"w\n\x0e\x43ompactionType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x04 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\t\"e\n\x0c\x42\x61\x63kfillType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x03 \x01(\t\"H\n\x1eRealTimeFeatureExtractorSource\x12&\n\x1ereal_time_feature_extractor_id\x18\x01 \x01(\tB\"\n\x1e\x63om.qwak.ai.logging.reader.apiP\x01\x62\x06proto3')
17
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dqwak/logging/log_source.proto\x12\x14qwak.logging.service\"\xca\x04\n\tLogSource\x12\x41\n\rmodel_runtime\x18\x01 \x01(\x0b\x32(.qwak.logging.service.ModelRuntimeSourceH\x00\x12?\n\x0cremote_build\x18\x02 \x01(\x0b\x32\'.qwak.logging.service.RemoteBuildSourceH\x00\x12M\n\x13inference_execution\x18\x03 \x01(\x0b\x32..qwak.logging.service.InferenceExecutionSourceH\x00\x12P\n\x15streaming_feature_set\x18\x04 \x01(\x0b\x32/.qwak.logging.service.StreamingFeatureSetSourceH\x00\x12g\n!streaming_aggregation_feature_set\x18\x05 \x01(\x0b\x32:.qwak.logging.service.StreamingAggregationFeatureSetSourceH\x00\x12H\n\x11\x62\x61tch_feature_set\x18\x06 \x01(\x0b\x32+.qwak.logging.service.BatchFeatureSetSourceH\x00\x12[\n\x1breal_time_feature_extractor\x18\x07 \x01(\x0b\x32\x34.qwak.logging.service.RealTimeFeatureExtractorSourceH\x00\x42\x08\n\x06source\"\x92\x01\n\x11RemoteBuildSource\x12\x18\n\x10model_group_name\x18\x03 \x01(\t\x12\x16\n\nmodel_name\x18\x04 \x01(\tB\x02\x18\x01\x12\x12\n\x08model_id\x18\x05 \x01(\tH\x00\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x11\n\tphase_ids\x18\x02 \x03(\tB\x12\n\x10model_identifier\"\xa8\x01\n\x12ModelRuntimeSource\x12\x18\n\x10model_group_name\x18\x03 \x01(\t\x12\x16\n\nmodel_name\x18\x04 \x01(\tB\x02\x18\x01\x12\x12\n\x08model_id\x18\x05 \x01(\tH\x00\x12\x17\n\rdeployment_id\x18\x01 \x01(\tH\x01\x12\x12\n\x08\x62uild_id\x18\x02 \x01(\tH\x01\x42\x12\n\x10model_identifierB\x0b\n\tsearch_by\"\xba\x01\n\x18InferenceExecutionSource\x12\x18\n\x10model_group_name\x18\x03 \x01(\t\x12\x16\n\nmodel_name\x18\x04 \x01(\tB\x02\x18\x01\x12\x12\n\x08model_id\x18\x05 \x01(\tH\x00\x12\x1a\n\x10inference_job_id\x18\x01 \x01(\tH\x01\x12\x1b\n\x11inference_task_id\x18\x02 \x01(\tH\x01\x42\x12\n\x10model_identifierB\x0b\n\tsearch_by\"\xe4\x01\n\x19StreamingFeatureSetSource\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x05 \x01(\t\x12\x34\n\x07offline\x18\x03 \x01(\x0b\x32!.qwak.logging.service.OfflineTypeH\x00\x12\x32\n\x06online\x18\x04 \x01(\x0b\x32 .qwak.logging.service.OnlineTypeH\x00\x42\x06\n\x04type\"\xa0\x01\n\x15\x42\x61tchFeatureSetSource\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x04 \x01(\t\x12\"\n\x1a\x66ilter_by_feature_set_name\x18\x05 \x01(\x08\x12\x0e\n\x06run_id\x18\x03 \x01(\t\"\xdb\x01\n$StreamingAggregationFeatureSetSource\x12\x37\n\trow_level\x18\x01 \x01(\x0b\x32\".qwak.logging.service.RowLevelTypeH\x00\x12:\n\ncompaction\x18\x02 \x01(\x0b\x32$.qwak.logging.service.CompactionTypeH\x00\x12\x36\n\x08\x62\x61\x63kfill\x18\x03 \x01(\x0b\x32\".qwak.logging.service.BackfillTypeH\x00\x42\x06\n\x04type\"c\n\nOnlineType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x03 \x01(\t\"t\n\x0bOfflineType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x04 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\t\"e\n\x0cRowLevelType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x03 \x01(\t\"w\n\x0e\x43ompactionType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x04 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\t\"e\n\x0c\x42\x61\x63kfillType\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12#\n\x1b\x66\x65\x61ture_set_repository_name\x18\x03 \x01(\t\"H\n\x1eRealTimeFeatureExtractorSource\x12&\n\x1ereal_time_feature_extractor_id\x18\x01 \x01(\tB\"\n\x1e\x63om.qwak.ai.logging.reader.apiP\x01\x62\x06proto3')
18
18
 
19
19
  _globals = globals()
20
20
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -22,30 +22,36 @@ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.logging.log_source_pb2
22
22
  if _descriptor._USE_C_DESCRIPTORS == False:
23
23
  _globals['DESCRIPTOR']._options = None
24
24
  _globals['DESCRIPTOR']._serialized_options = b'\n\036com.qwak.ai.logging.reader.apiP\001'
25
+ _globals['_REMOTEBUILDSOURCE'].fields_by_name['model_name']._options = None
26
+ _globals['_REMOTEBUILDSOURCE'].fields_by_name['model_name']._serialized_options = b'\030\001'
27
+ _globals['_MODELRUNTIMESOURCE'].fields_by_name['model_name']._options = None
28
+ _globals['_MODELRUNTIMESOURCE'].fields_by_name['model_name']._serialized_options = b'\030\001'
29
+ _globals['_INFERENCEEXECUTIONSOURCE'].fields_by_name['model_name']._options = None
30
+ _globals['_INFERENCEEXECUTIONSOURCE'].fields_by_name['model_name']._serialized_options = b'\030\001'
25
31
  _globals['_LOGSOURCE']._serialized_start=56
26
32
  _globals['_LOGSOURCE']._serialized_end=642
27
- _globals['_REMOTEBUILDSOURCE']._serialized_start=644
28
- _globals['_REMOTEBUILDSOURCE']._serialized_end=746
29
- _globals['_MODELRUNTIMESOURCE']._serialized_start=748
30
- _globals['_MODELRUNTIMESOURCE']._serialized_end=872
31
- _globals['_INFERENCEEXECUTIONSOURCE']._serialized_start=875
32
- _globals['_INFERENCEEXECUTIONSOURCE']._serialized_end=1017
33
- _globals['_STREAMINGFEATURESETSOURCE']._serialized_start=1020
34
- _globals['_STREAMINGFEATURESETSOURCE']._serialized_end=1248
35
- _globals['_BATCHFEATURESETSOURCE']._serialized_start=1251
36
- _globals['_BATCHFEATURESETSOURCE']._serialized_end=1411
37
- _globals['_STREAMINGAGGREGATIONFEATURESETSOURCE']._serialized_start=1414
38
- _globals['_STREAMINGAGGREGATIONFEATURESETSOURCE']._serialized_end=1633
39
- _globals['_ONLINETYPE']._serialized_start=1635
40
- _globals['_ONLINETYPE']._serialized_end=1734
41
- _globals['_OFFLINETYPE']._serialized_start=1736
42
- _globals['_OFFLINETYPE']._serialized_end=1852
43
- _globals['_ROWLEVELTYPE']._serialized_start=1854
44
- _globals['_ROWLEVELTYPE']._serialized_end=1955
45
- _globals['_COMPACTIONTYPE']._serialized_start=1957
46
- _globals['_COMPACTIONTYPE']._serialized_end=2076
47
- _globals['_BACKFILLTYPE']._serialized_start=2078
48
- _globals['_BACKFILLTYPE']._serialized_end=2179
49
- _globals['_REALTIMEFEATUREEXTRACTORSOURCE']._serialized_start=2181
50
- _globals['_REALTIMEFEATUREEXTRACTORSOURCE']._serialized_end=2253
33
+ _globals['_REMOTEBUILDSOURCE']._serialized_start=645
34
+ _globals['_REMOTEBUILDSOURCE']._serialized_end=791
35
+ _globals['_MODELRUNTIMESOURCE']._serialized_start=794
36
+ _globals['_MODELRUNTIMESOURCE']._serialized_end=962
37
+ _globals['_INFERENCEEXECUTIONSOURCE']._serialized_start=965
38
+ _globals['_INFERENCEEXECUTIONSOURCE']._serialized_end=1151
39
+ _globals['_STREAMINGFEATURESETSOURCE']._serialized_start=1154
40
+ _globals['_STREAMINGFEATURESETSOURCE']._serialized_end=1382
41
+ _globals['_BATCHFEATURESETSOURCE']._serialized_start=1385
42
+ _globals['_BATCHFEATURESETSOURCE']._serialized_end=1545
43
+ _globals['_STREAMINGAGGREGATIONFEATURESETSOURCE']._serialized_start=1548
44
+ _globals['_STREAMINGAGGREGATIONFEATURESETSOURCE']._serialized_end=1767
45
+ _globals['_ONLINETYPE']._serialized_start=1769
46
+ _globals['_ONLINETYPE']._serialized_end=1868
47
+ _globals['_OFFLINETYPE']._serialized_start=1870
48
+ _globals['_OFFLINETYPE']._serialized_end=1986
49
+ _globals['_ROWLEVELTYPE']._serialized_start=1988
50
+ _globals['_ROWLEVELTYPE']._serialized_end=2089
51
+ _globals['_COMPACTIONTYPE']._serialized_start=2091
52
+ _globals['_COMPACTIONTYPE']._serialized_end=2210
53
+ _globals['_BACKFILLTYPE']._serialized_start=2212
54
+ _globals['_BACKFILLTYPE']._serialized_end=2313
55
+ _globals['_REALTIMEFEATUREEXTRACTORSOURCE']._serialized_start=2315
56
+ _globals['_REALTIMEFEATUREEXTRACTORSOURCE']._serialized_end=2387
51
57
  # @@protoc_insertion_point(module_scope)
frogml/_proto/qwak/logging/log_source_pb2.pyi CHANGED
@@ -64,6 +64,7 @@ class RemoteBuildSource(google.protobuf.message.Message):
64
64
 
65
65
  MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
66
66
  MODEL_NAME_FIELD_NUMBER: builtins.int
67
+ MODEL_ID_FIELD_NUMBER: builtins.int
67
68
  BUILD_ID_FIELD_NUMBER: builtins.int
68
69
  PHASE_IDS_FIELD_NUMBER: builtins.int
69
70
  model_group_name: builtins.str
@@ -71,9 +72,9 @@ class RemoteBuildSource(google.protobuf.message.Message):
71
72
  This field is mandatory
72
73
  """
73
74
  model_name: builtins.str
74
- """Model name - for verifying user permissions
75
- This field is mandatory
76
- """
75
+ """This field is identical to model_id, use model_id instead"""
76
+ model_id: builtins.str
77
+ """Model id (human readable name)"""
77
78
  build_id: builtins.str
78
79
  """Filter by build id"""
79
80
  @property
@@ -85,10 +86,13 @@ class RemoteBuildSource(google.protobuf.message.Message):
85
86
  *,
86
87
  model_group_name: builtins.str = ...,
87
88
  model_name: builtins.str = ...,
89
+ model_id: builtins.str = ...,
88
90
  build_id: builtins.str = ...,
89
91
  phase_ids: collections.abc.Iterable[builtins.str] | None = ...,
90
92
  ) -> None: ...
91
- def ClearField(self, field_name: typing.Literal["build_id", b"build_id", "model_group_name", b"model_group_name", "model_name", b"model_name", "phase_ids", b"phase_ids"]) -> None: ...
93
+ def HasField(self, field_name: typing.Literal["model_id", b"model_id", "model_identifier", b"model_identifier"]) -> builtins.bool: ...
94
+ def ClearField(self, field_name: typing.Literal["build_id", b"build_id", "model_group_name", b"model_group_name", "model_id", b"model_id", "model_identifier", b"model_identifier", "model_name", b"model_name", "phase_ids", b"phase_ids"]) -> None: ...
95
+ def WhichOneof(self, oneof_group: typing.Literal["model_identifier", b"model_identifier"]) -> typing.Literal["model_id"] | None: ...
92
96
 
93
97
  global___RemoteBuildSource = RemoteBuildSource
94
98
 
@@ -100,6 +104,7 @@ class ModelRuntimeSource(google.protobuf.message.Message):
100
104
 
101
105
  MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
102
106
  MODEL_NAME_FIELD_NUMBER: builtins.int
107
+ MODEL_ID_FIELD_NUMBER: builtins.int
103
108
  DEPLOYMENT_ID_FIELD_NUMBER: builtins.int
104
109
  BUILD_ID_FIELD_NUMBER: builtins.int
105
110
  model_group_name: builtins.str
@@ -107,9 +112,9 @@ class ModelRuntimeSource(google.protobuf.message.Message):
107
112
  This field is mandatory
108
113
  """
109
114
  model_name: builtins.str
110
- """Model name - for verifying user permissions
111
- This field is mandatory
112
- """
115
+ """This field is identical to model_id, use model_id instead"""
116
+ model_id: builtins.str
117
+ """Model id (human readable name)"""
113
118
  deployment_id: builtins.str
114
119
  """Filter by deployment id"""
115
120
  build_id: builtins.str
@@ -119,11 +124,15 @@ class ModelRuntimeSource(google.protobuf.message.Message):
119
124
  *,
120
125
  model_group_name: builtins.str = ...,
121
126
  model_name: builtins.str = ...,
127
+ model_id: builtins.str = ...,
122
128
  deployment_id: builtins.str = ...,
123
129
  build_id: builtins.str = ...,
124
130
  ) -> None: ...
125
- def HasField(self, field_name: typing.Literal["build_id", b"build_id", "deployment_id", b"deployment_id", "search_by", b"search_by"]) -> builtins.bool: ...
126
- def ClearField(self, field_name: typing.Literal["build_id", b"build_id", "deployment_id", b"deployment_id", "model_group_name", b"model_group_name", "model_name", b"model_name", "search_by", b"search_by"]) -> None: ...
131
+ def HasField(self, field_name: typing.Literal["build_id", b"build_id", "deployment_id", b"deployment_id", "model_id", b"model_id", "model_identifier", b"model_identifier", "search_by", b"search_by"]) -> builtins.bool: ...
132
+ def ClearField(self, field_name: typing.Literal["build_id", b"build_id", "deployment_id", b"deployment_id", "model_group_name", b"model_group_name", "model_id", b"model_id", "model_identifier", b"model_identifier", "model_name", b"model_name", "search_by", b"search_by"]) -> None: ...
133
+ @typing.overload
134
+ def WhichOneof(self, oneof_group: typing.Literal["model_identifier", b"model_identifier"]) -> typing.Literal["model_id"] | None: ...
135
+ @typing.overload
127
136
  def WhichOneof(self, oneof_group: typing.Literal["search_by", b"search_by"]) -> typing.Literal["deployment_id", "build_id"] | None: ...
128
137
 
129
138
  global___ModelRuntimeSource = ModelRuntimeSource
@@ -134,6 +143,7 @@ class InferenceExecutionSource(google.protobuf.message.Message):
134
143
 
135
144
  MODEL_GROUP_NAME_FIELD_NUMBER: builtins.int
136
145
  MODEL_NAME_FIELD_NUMBER: builtins.int
146
+ MODEL_ID_FIELD_NUMBER: builtins.int
137
147
  INFERENCE_JOB_ID_FIELD_NUMBER: builtins.int
138
148
  INFERENCE_TASK_ID_FIELD_NUMBER: builtins.int
139
149
  model_group_name: builtins.str
@@ -141,9 +151,9 @@ class InferenceExecutionSource(google.protobuf.message.Message):
141
151
  This field is mandatory
142
152
  """
143
153
  model_name: builtins.str
144
- """Model name - for verifying user permissions
145
- This field is mandatory
146
- """
154
+ """This field is identical to model_id, use model_id instead"""
155
+ model_id: builtins.str
156
+ """Model id (human readable name)"""
147
157
  inference_job_id: builtins.str
148
158
  """Filter by deployment id"""
149
159
  inference_task_id: builtins.str
@@ -153,11 +163,15 @@ class InferenceExecutionSource(google.protobuf.message.Message):
153
163
  *,
154
164
  model_group_name: builtins.str = ...,
155
165
  model_name: builtins.str = ...,
166
+ model_id: builtins.str = ...,
156
167
  inference_job_id: builtins.str = ...,
157
168
  inference_task_id: builtins.str = ...,
158
169
  ) -> None: ...
159
- def HasField(self, field_name: typing.Literal["inference_job_id", b"inference_job_id", "inference_task_id", b"inference_task_id", "search_by", b"search_by"]) -> builtins.bool: ...
160
- def ClearField(self, field_name: typing.Literal["inference_job_id", b"inference_job_id", "inference_task_id", b"inference_task_id", "model_group_name", b"model_group_name", "model_name", b"model_name", "search_by", b"search_by"]) -> None: ...
170
+ def HasField(self, field_name: typing.Literal["inference_job_id", b"inference_job_id", "inference_task_id", b"inference_task_id", "model_id", b"model_id", "model_identifier", b"model_identifier", "search_by", b"search_by"]) -> builtins.bool: ...
171
+ def ClearField(self, field_name: typing.Literal["inference_job_id", b"inference_job_id", "inference_task_id", b"inference_task_id", "model_group_name", b"model_group_name", "model_id", b"model_id", "model_identifier", b"model_identifier", "model_name", b"model_name", "search_by", b"search_by"]) -> None: ...
172
+ @typing.overload
173
+ def WhichOneof(self, oneof_group: typing.Literal["model_identifier", b"model_identifier"]) -> typing.Literal["model_id"] | None: ...
174
+ @typing.overload
161
175
  def WhichOneof(self, oneof_group: typing.Literal["search_by", b"search_by"]) -> typing.Literal["inference_job_id", "inference_task_id"] | None: ...
162
176
 
163
177
  global___InferenceExecutionSource = InferenceExecutionSource
frogml/_proto/qwak/models/models_pb2.py CHANGED
@@ -21,7 +21,7 @@ from frogml._proto.qwak.deployment import deployment_pb2 as qwak_dot_deployment_
21
21
  from frogml._proto.qwak.build.v1 import build_pb2 as qwak_dot_build_dot_v1_dot_build__pb2
22
22
 
23
23
 
24
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18qwak/models/models.proto\x12\x16qwak.models.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x18qwak/builds/builds.proto\x1a&qwak/projects/jfrog_project_spec.proto\x1a\x1eqwak/models/models_query.proto\x1a\x1fqwak/audience/v1/audience.proto\x1a qwak/deployment/deployment.proto\x1a\x19qwak/build/v1/build.proto\"\x9b\x01\n\x12\x43reateModelRequest\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\x12N\n\x12jfrog_project_spec\x18\x02 \x01(\x0b\x32\x32.qwak.projects.management.ModelRepositoryJFrogSpec\"\'\n\x13\x43reateModelResponse\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\'\n\x11ListModelsRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"C\n\x12ListModelsResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\"\xf0\x01\n\x1aListModelsPaginatedRequest\x12\x13\n\tpage_size\x18\x01 \x01(\x05H\x00\x12\x10\n\x06\x63ursor\x18\x02 \x01(\tH\x01\x12\x39\n\x0bsort_models\x18\x03 \x01(\x0b\x32\".qwak.models.management.SortModelsH\x02\x12\x30\n\x06\x66ilter\x18\x04 \x01(\x0b\x32\x1e.qwak.models.management.FilterH\x03\x42\x0f\n\rpage_size_optB\x0c\n\ncursor_optB\x11\n\x0fsort_models_optB\x0c\n\nfilter_opt\"\x8a\x01\n\x1bListModelsPaginatedResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\x12<\n\x08metadata\x18\x02 \x01(\x0b\x32*.qwak.models.management.PaginationMetadata\"I\n\x0fGetModelRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"@\n\x10GetModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\":\n\x12\x44\x65leteModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nproject_id\x18\x02 \x01(\t\"\x15\n\x13\x44\x65leteModelResponse\"q\n\x12UpdateModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12I\n\x11model_spec_config\x18\x02 \x01(\x0b\x32..qwak.models.management.ModelSpecConfiguration\"L\n\x13UpdateModelResponse\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\"<\n\x13\x43reateBranchRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\")\n\x14\x43reateBranchResponse\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"(\n\x13\x44\x65leteBranchRequest\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"\x16\n\x14\x44\x65leteBranchResponse\"N\n\x14IsModelExistsRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"\'\n\x15IsModelExistsResponse\x12\x0e\n\x06\x65xists\x18\x01 \x01(\x08\"\xda\x01\n\tModelSpec\x12\x16\n\x0emodel_named_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x14\n\nproject_id\x18\x03 \x01(\tH\x00\x12\x16\n\x0cproject_name\x18\x06 \x01(\tH\x00\x12\x19\n\x11model_description\x18\x04 \x01(\t\x12@\n\x10model_build_type\x18\x05 \x01(\x0b\x32&.qwak.models.management.ModelBuildTypeB\x14\n\x12project_identifier\"\xc7\x02\n\nModelBrief\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12N\n\x17\x64\x65ployment_model_status\x18\n \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\"3\n\x16ModelSpecConfiguration\x12\x19\n\x11model_description\x18\x04 \x01(\t\"\xf4\x04\n\x05Model\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x39\n\x0cmodel_status\x18\x07 \x01(\x0e\x32#.qwak.models.management.ModelStatus\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12\x30\n\x08\x62ranches\x18\n \x03(\x0b\x32\x1e.qwak.models.management.Branch\x12N\n\x17\x64\x65ployment_model_status\x18\x0b \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\x12\x0c\n\x04uuid\x18\x0c \x01(\t\x12\x42\n\nmodel_type\x18\r \x01(\x0e\x32..qwak.deployment.management.KubeDeploymentType\x12\x14\n\x0cproject_name\x18\x0e \x01(\t\x12@\n\x10model_build_type\x18\x0f \x01(\x0b\x32&.qwak.models.management.ModelBuildType\x12\x19\n\x11jfrog_project_key\x18\x10 \x01(\t\"\x96\x01\n\x0eModelBuildType\x12\x30\n\x06source\x18\x01 \x01(\x0b\x32\x1e.qwak.models.management.SourceH\x00\x12J\n\x14model_registry_image\x18\x02 \x01(\x0b\x32*.qwak.models.management.ModelRegistryImageH\x00\x42\x06\n\x04type\"\x08\n\x06Source\"3\n\x12ModelRegistryImage\x12\x1d\n\x15last_image_identifier\x18\x01 \x01(\t\"\xeb\x01\n\x06\x42ranch\x12\x11\n\tbranch_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x42\n\rbranch_status\x18\x04 \x01(\x0e\x32+.qwak.models.management.Branch.BranchStatus\x12\x10\n\x08model_id\x18\x05 \x01(\t\x12\x1d\n\x11\x64\x65ployed_build_id\x18\x06 \x01(\tB\x02\x18\x01\"D\n\x0c\x42ranchStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01\"B\n\x1bGetDeploymentDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x11\n\tbranch_id\x18\x02 \x01(\t\"e\n\x1cGetDeploymentDetailsResponse\x12\x45\n\x12\x64\x65ployment_details\x18\x01 \x01(\x0b\x32).qwak.models.management.DeploymentDetails\"\x93\x02\n\x11\x44\x65ploymentDetails\x12\x31\n\rlast_deployed\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x16\n\x0enumber_of_pods\x18\x03 \x01(\x05\x12\x14\n\x0c\x63pu_fraction\x18\x04 \x01(\x02\x12\x15\n\rmemory_amount\x18\x05 \x01(\x05\x12\x38\n\x0cmemory_units\x18\x06 \x01(\x0e\x32\".qwak.models.management.MemoryUnit\x12\x19\n\x11number_of_workers\x18\x07 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x08 \x01(\x05\"/\n\x19ListModelsMetadataRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"[\n\x1aListModelsMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x03(\x0b\x32%.qwak.models.management.ModelMetadata\"+\n\x17GetModelMetadataRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"Y\n\x18GetModelMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x01(\x0b\x32%.qwak.models.management.ModelMetadata\"h\n\x11\x45nvironmentRoutes\x12\x13\n\x0b\x65nvironment\x18\x01 \x01(\t\x12>\n\x0f\x61udience_routes\x18\x02 \x03(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry\"\xb8\x05\n\rModelMetadata\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\x12I\n\x12\x64\x65ployment_details\x18\x02 \x03(\x0b\x32-.qwak.deployment.management.DeploymentDetails\x12r\n\x1e\x61udience_routes_by_environment\x18\x03 \x03(\x0b\x32\x46.qwak.models.management.ModelMetadata.AudienceRoutesByEnvironmentEntryB\x02\x18\x01\x12\x30\n\x05\x62uild\x18\x04 \x03(\x0b\x32\x1d.qwak.builds.management.BuildB\x02\x18\x01\x12}\n&audience_routes_grouped_by_environment\x18\x05 \x03(\x0b\x32M.qwak.models.management.ModelMetadata.AudienceRoutesGroupedByEnvironmentEntry\x12(\n\x06\x62uilds\x18\x06 \x03(\x0b\x32\x18.com.qwak.build.v1.Build\x1ai\n AudienceRoutesByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry:\x02\x38\x01\x1at\n\'AudienceRoutesGroupedByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).qwak.models.management.EnvironmentRoutes:\x02\x38\x01\"i\n CheckModelCreationAllowedRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x18\n\x10model_group_name\x18\x02 \x01(\t\x12\x19\n\x11jfrog_project_key\x18\x03 \x01(\t\"\x9b\x02\n!CheckModelCreationAllowedResponse\x12\x14\n\x0cis_creatable\x18\x01 \x01(\x08\x12;\n\x0cmodel_exists\x18\x02 \x01(\x0b\x32#.qwak.models.management.ModelExistsH\x00\x12\x42\n\x10model_not_exists\x18\x03 \x01(\x0b\x32&.qwak.models.management.ModelNotExistsH\x00\x12U\n\x1amodel_exists_with_mismatch\x18\x04 \x01(\x0b\x32/.qwak.models.management.ModelExistsWithMismatchH\x00\x42\x08\n\x06status\"\r\n\x0bModelExists\"\x10\n\x0eModelNotExists\"*\n\x17ModelExistsWithMismatch\x12\x0f\n\x07\x64\x65tails\x18\x01 \x03(\t*O\n\x15\x44\x65ploymentModelStatus\x12\x1d\n\x19\x44\x45PLOYMENT_STATUS_INVALID\x10\x00\x12\n\n\x06ONLINE\x10\x01\x12\x0b\n\x07OFFLINE\x10\x02*C\n\x0bModelStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01*7\n\nMemoryUnit\x12\x17\n\x13MEMORY_UNIT_UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x32\xf3\x0b\n\x17ModelsManagementService\x12\x66\n\x0b\x43reateModel\x12*.qwak.models.management.CreateModelRequest\x1a+.qwak.models.management.CreateModelResponse\x12\x63\n\nListModels\x12).qwak.models.management.ListModelsRequest\x1a*.qwak.models.management.ListModelsResponse\x12~\n\x13ListModelsPaginated\x12\x32.qwak.models.management.ListModelsPaginatedRequest\x1a\x33.qwak.models.management.ListModelsPaginatedResponse\x12]\n\x08GetModel\x12\'.qwak.models.management.GetModelRequest\x1a(.qwak.models.management.GetModelResponse\x12\x66\n\x0b\x44\x65leteModel\x12*.qwak.models.management.DeleteModelRequest\x1a+.qwak.models.management.DeleteModelResponse\x12\x66\n\x0bUpdateModel\x12*.qwak.models.management.UpdateModelRequest\x1a+.qwak.models.management.UpdateModelResponse\x12n\n\x0c\x43reateBranch\x12+.qwak.models.management.CreateBranchRequest\x1a,.qwak.models.management.CreateBranchResponse\"\x03\x88\x02\x01\x12n\n\x0c\x44\x65leteBranch\x12+.qwak.models.management.DeleteBranchRequest\x1a,.qwak.models.management.DeleteBranchResponse\"\x03\x88\x02\x01\x12\x86\x01\n\x14GetDeploymentDetails\x12\x33.qwak.models.management.GetDeploymentDetailsRequest\x1a\x34.qwak.models.management.GetDeploymentDetailsResponse\"\x03\x88\x02\x01\x12l\n\rIsModelExists\x12,.qwak.models.management.IsModelExistsRequest\x1a-.qwak.models.management.IsModelExistsResponse\x12{\n\x12ListModelsMetadata\x12\x31.qwak.models.management.ListModelsMetadataRequest\x1a\x32.qwak.models.management.ListModelsMetadataResponse\x12u\n\x10GetModelMetadata\x12/.qwak.models.management.GetModelMetadataRequest\x1a\x30.qwak.models.management.GetModelMetadataResponse\x12\x90\x01\n\x19\x43heckModelCreationAllowed\x12\x38.qwak.models.management.CheckModelCreationAllowedRequest\x1a\x39.qwak.models.management.CheckModelCreationAllowedResponseB%\n!com.qwak.ai.management.models.apiP\x01\x62\x06proto3')
24
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18qwak/models/models.proto\x12\x16qwak.models.management\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x18qwak/builds/builds.proto\x1a&qwak/projects/jfrog_project_spec.proto\x1a\x1eqwak/models/models_query.proto\x1a\x1fqwak/audience/v1/audience.proto\x1a qwak/deployment/deployment.proto\x1a\x19qwak/build/v1/build.proto\"\x9b\x01\n\x12\x43reateModelRequest\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\x12N\n\x12jfrog_project_spec\x18\x02 \x01(\x0b\x32\x32.qwak.projects.management.ModelRepositoryJFrogSpec\"\'\n\x13\x43reateModelResponse\x12\x10\n\x08model_id\x18\x01 \x01(\t\"\'\n\x11ListModelsRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"C\n\x12ListModelsResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\"\xf0\x01\n\x1aListModelsPaginatedRequest\x12\x13\n\tpage_size\x18\x01 \x01(\x05H\x00\x12\x10\n\x06\x63ursor\x18\x02 \x01(\tH\x01\x12\x39\n\x0bsort_models\x18\x03 \x01(\x0b\x32\".qwak.models.management.SortModelsH\x02\x12\x30\n\x06\x66ilter\x18\x04 \x01(\x0b\x32\x1e.qwak.models.management.FilterH\x03\x42\x0f\n\rpage_size_optB\x0c\n\ncursor_optB\x11\n\x0fsort_models_optB\x0c\n\nfilter_opt\"\x8a\x01\n\x1bListModelsPaginatedResponse\x12-\n\x06models\x18\x01 \x03(\x0b\x32\x1d.qwak.models.management.Model\x12<\n\x08metadata\x18\x02 \x01(\x0b\x32*.qwak.models.management.PaginationMetadata\"I\n\x0fGetModelRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"@\n\x10GetModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\"E\n\x1bListModelsByModelIdsRequest\x12\x11\n\tmodel_ids\x18\x01 \x03(\t\x12\x13\n\x0bmodel_uuids\x18\x02 \x03(\t\":\n\x12\x44\x65leteModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nproject_id\x18\x02 \x01(\t\"\x15\n\x13\x44\x65leteModelResponse\"q\n\x12UpdateModelRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12I\n\x11model_spec_config\x18\x02 \x01(\x0b\x32..qwak.models.management.ModelSpecConfiguration\"L\n\x13UpdateModelResponse\x12\x35\n\nmodel_spec\x18\x01 \x01(\x0b\x32!.qwak.models.management.ModelSpec\"<\n\x13\x43reateBranchRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\")\n\x14\x43reateBranchResponse\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"(\n\x13\x44\x65leteBranchRequest\x12\x11\n\tbranch_id\x18\x01 \x01(\t\"\x16\n\x14\x44\x65leteBranchResponse\"N\n\x14IsModelExistsRequest\x12\x12\n\x08model_id\x18\x01 \x01(\tH\x00\x12\x14\n\nmodel_uuid\x18\x02 \x01(\tH\x00\x42\x0c\n\nidentifier\"\'\n\x15IsModelExistsResponse\x12\x0e\n\x06\x65xists\x18\x01 \x01(\x08\"\xda\x01\n\tModelSpec\x12\x16\n\x0emodel_named_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x14\n\nproject_id\x18\x03 \x01(\tH\x00\x12\x16\n\x0cproject_name\x18\x06 \x01(\tH\x00\x12\x19\n\x11model_description\x18\x04 \x01(\t\x12@\n\x10model_build_type\x18\x05 \x01(\x0b\x32&.qwak.models.management.ModelBuildTypeB\x14\n\x12project_identifier\"\xc7\x02\n\nModelBrief\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12N\n\x17\x64\x65ployment_model_status\x18\n \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\"3\n\x16ModelSpecConfiguration\x12\x19\n\x11model_description\x18\x04 \x01(\t\"\xf4\x04\n\x05Model\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12.\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x34\n\x10last_modified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x05 \x01(\t\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12\x39\n\x0cmodel_status\x18\x07 \x01(\x0e\x32#.qwak.models.management.ModelStatus\x12\x19\n\x11model_description\x18\x08 \x01(\t\x12\x12\n\nproject_id\x18\t \x01(\t\x12\x30\n\x08\x62ranches\x18\n \x03(\x0b\x32\x1e.qwak.models.management.Branch\x12N\n\x17\x64\x65ployment_model_status\x18\x0b \x01(\x0e\x32-.qwak.models.management.DeploymentModelStatus\x12\x0c\n\x04uuid\x18\x0c \x01(\t\x12\x42\n\nmodel_type\x18\r \x01(\x0e\x32..qwak.deployment.management.KubeDeploymentType\x12\x14\n\x0cproject_name\x18\x0e \x01(\t\x12@\n\x10model_build_type\x18\x0f \x01(\x0b\x32&.qwak.models.management.ModelBuildType\x12\x19\n\x11jfrog_project_key\x18\x10 \x01(\t\"\x96\x01\n\x0eModelBuildType\x12\x30\n\x06source\x18\x01 \x01(\x0b\x32\x1e.qwak.models.management.SourceH\x00\x12J\n\x14model_registry_image\x18\x02 \x01(\x0b\x32*.qwak.models.management.ModelRegistryImageH\x00\x42\x06\n\x04type\"\x08\n\x06Source\"3\n\x12ModelRegistryImage\x12\x1d\n\x15last_image_identifier\x18\x01 \x01(\t\"\xeb\x01\n\x06\x42ranch\x12\x11\n\tbranch_id\x18\x01 \x01(\t\x12\x13\n\x0b\x62ranch_name\x18\x02 \x01(\t\x12\x42\n\rbranch_status\x18\x04 \x01(\x0e\x32+.qwak.models.management.Branch.BranchStatus\x12\x10\n\x08model_id\x18\x05 \x01(\t\x12\x1d\n\x11\x64\x65ployed_build_id\x18\x06 \x01(\tB\x02\x18\x01\"D\n\x0c\x42ranchStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01\"B\n\x1bGetDeploymentDetailsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x11\n\tbranch_id\x18\x02 \x01(\t\"e\n\x1cGetDeploymentDetailsResponse\x12\x45\n\x12\x64\x65ployment_details\x18\x01 \x01(\x0b\x32).qwak.models.management.DeploymentDetails\"\x93\x02\n\x11\x44\x65ploymentDetails\x12\x31\n\rlast_deployed\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x16\n\x0enumber_of_pods\x18\x03 \x01(\x05\x12\x14\n\x0c\x63pu_fraction\x18\x04 \x01(\x02\x12\x15\n\rmemory_amount\x18\x05 \x01(\x05\x12\x38\n\x0cmemory_units\x18\x06 \x01(\x0e\x32\".qwak.models.management.MemoryUnit\x12\x19\n\x11number_of_workers\x18\x07 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x08 \x01(\x05\"/\n\x19ListModelsMetadataRequest\x12\x12\n\nproject_id\x18\x01 \x01(\t\"[\n\x1aListModelsMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x03(\x0b\x32%.qwak.models.management.ModelMetadata\"+\n\x17GetModelMetadataRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\"Y\n\x18GetModelMetadataResponse\x12=\n\x0emodel_metadata\x18\x01 \x01(\x0b\x32%.qwak.models.management.ModelMetadata\"h\n\x11\x45nvironmentRoutes\x12\x13\n\x0b\x65nvironment\x18\x01 \x01(\t\x12>\n\x0f\x61udience_routes\x18\x02 \x03(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry\"\xb8\x05\n\rModelMetadata\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x1d.qwak.models.management.Model\x12I\n\x12\x64\x65ployment_details\x18\x02 \x03(\x0b\x32-.qwak.deployment.management.DeploymentDetails\x12r\n\x1e\x61udience_routes_by_environment\x18\x03 \x03(\x0b\x32\x46.qwak.models.management.ModelMetadata.AudienceRoutesByEnvironmentEntryB\x02\x18\x01\x12\x30\n\x05\x62uild\x18\x04 \x03(\x0b\x32\x1d.qwak.builds.management.BuildB\x02\x18\x01\x12}\n&audience_routes_grouped_by_environment\x18\x05 \x03(\x0b\x32M.qwak.models.management.ModelMetadata.AudienceRoutesGroupedByEnvironmentEntry\x12(\n\x06\x62uilds\x18\x06 \x03(\x0b\x32\x18.com.qwak.build.v1.Build\x1ai\n AudienceRoutesByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x34\n\x05value\x18\x02 \x01(\x0b\x32%.qwak.audience.v1.AudienceRoutesEntry:\x02\x38\x01\x1at\n\'AudienceRoutesGroupedByEnvironmentEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32).qwak.models.management.EnvironmentRoutes:\x02\x38\x01\"i\n CheckModelCreationAllowedRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x18\n\x10model_group_name\x18\x02 \x01(\t\x12\x19\n\x11jfrog_project_key\x18\x03 \x01(\t\"\x9b\x02\n!CheckModelCreationAllowedResponse\x12\x14\n\x0cis_creatable\x18\x01 \x01(\x08\x12;\n\x0cmodel_exists\x18\x02 \x01(\x0b\x32#.qwak.models.management.ModelExistsH\x00\x12\x42\n\x10model_not_exists\x18\x03 \x01(\x0b\x32&.qwak.models.management.ModelNotExistsH\x00\x12U\n\x1amodel_exists_with_mismatch\x18\x04 \x01(\x0b\x32/.qwak.models.management.ModelExistsWithMismatchH\x00\x42\x08\n\x06status\"\r\n\x0bModelExists\"\x10\n\x0eModelNotExists\"*\n\x17ModelExistsWithMismatch\x12\x0f\n\x07\x64\x65tails\x18\x01 \x03(\t*O\n\x15\x44\x65ploymentModelStatus\x12\x1d\n\x19\x44\x45PLOYMENT_STATUS_INVALID\x10\x00\x12\n\n\x06ONLINE\x10\x01\x12\x0b\n\x07OFFLINE\x10\x02*C\n\x0bModelStatus\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x03\x12\r\n\tSUSPENDED\x10\x02\x12\x0c\n\x08\x44ISABLED\x10\x01*7\n\nMemoryUnit\x12\x17\n\x13MEMORY_UNIT_UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02\x32\xec\x0c\n\x17ModelsManagementService\x12\x66\n\x0b\x43reateModel\x12*.qwak.models.management.CreateModelRequest\x1a+.qwak.models.management.CreateModelResponse\x12\x63\n\nListModels\x12).qwak.models.management.ListModelsRequest\x1a*.qwak.models.management.ListModelsResponse\x12~\n\x13ListModelsPaginated\x12\x32.qwak.models.management.ListModelsPaginatedRequest\x1a\x33.qwak.models.management.ListModelsPaginatedResponse\x12]\n\x08GetModel\x12\'.qwak.models.management.GetModelRequest\x1a(.qwak.models.management.GetModelResponse\x12w\n\x14ListModelsByModelIds\x12\x33.qwak.models.management.ListModelsByModelIdsRequest\x1a*.qwak.models.management.ListModelsResponse\x12\x66\n\x0b\x44\x65leteModel\x12*.qwak.models.management.DeleteModelRequest\x1a+.qwak.models.management.DeleteModelResponse\x12\x66\n\x0bUpdateModel\x12*.qwak.models.management.UpdateModelRequest\x1a+.qwak.models.management.UpdateModelResponse\x12n\n\x0c\x43reateBranch\x12+.qwak.models.management.CreateBranchRequest\x1a,.qwak.models.management.CreateBranchResponse\"\x03\x88\x02\x01\x12n\n\x0c\x44\x65leteBranch\x12+.qwak.models.management.DeleteBranchRequest\x1a,.qwak.models.management.DeleteBranchResponse\"\x03\x88\x02\x01\x12\x86\x01\n\x14GetDeploymentDetails\x12\x33.qwak.models.management.GetDeploymentDetailsRequest\x1a\x34.qwak.models.management.GetDeploymentDetailsResponse\"\x03\x88\x02\x01\x12l\n\rIsModelExists\x12,.qwak.models.management.IsModelExistsRequest\x1a-.qwak.models.management.IsModelExistsResponse\x12{\n\x12ListModelsMetadata\x12\x31.qwak.models.management.ListModelsMetadataRequest\x1a\x32.qwak.models.management.ListModelsMetadataResponse\x12u\n\x10GetModelMetadata\x12/.qwak.models.management.GetModelMetadataRequest\x1a\x30.qwak.models.management.GetModelMetadataResponse\x12\x90\x01\n\x19\x43heckModelCreationAllowed\x12\x38.qwak.models.management.CheckModelCreationAllowedRequest\x1a\x39.qwak.models.management.CheckModelCreationAllowedResponseB%\n!com.qwak.ai.management.models.apiP\x01\x62\x06proto3')
25
25
 
26
26
  _globals = globals()
27
27
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -45,12 +45,12 @@ if _descriptor._USE_C_DESCRIPTORS == False:
45
45
  _globals['_MODELSMANAGEMENTSERVICE'].methods_by_name['DeleteBranch']._serialized_options = b'\210\002\001'
46
46
  _globals['_MODELSMANAGEMENTSERVICE'].methods_by_name['GetDeploymentDetails']._options = None
47
47
  _globals['_MODELSMANAGEMENTSERVICE'].methods_by_name['GetDeploymentDetails']._serialized_options = b'\210\002\001'
48
- _globals['_DEPLOYMENTMODELSTATUS']._serialized_start=5370
49
- _globals['_DEPLOYMENTMODELSTATUS']._serialized_end=5449
50
- _globals['_MODELSTATUS']._serialized_start=5451
51
- _globals['_MODELSTATUS']._serialized_end=5518
52
- _globals['_MEMORYUNIT']._serialized_start=5520
53
- _globals['_MEMORYUNIT']._serialized_end=5575
48
+ _globals['_DEPLOYMENTMODELSTATUS']._serialized_start=5441
49
+ _globals['_DEPLOYMENTMODELSTATUS']._serialized_end=5520
50
+ _globals['_MODELSTATUS']._serialized_start=5522
51
+ _globals['_MODELSTATUS']._serialized_end=5589
52
+ _globals['_MEMORYUNIT']._serialized_start=5591
53
+ _globals['_MEMORYUNIT']._serialized_end=5646
54
54
  _globals['_CREATEMODELREQUEST']._serialized_start=278
55
55
  _globals['_CREATEMODELREQUEST']._serialized_end=433
56
56
  _globals['_CREATEMODELRESPONSE']._serialized_start=435
@@ -67,76 +67,78 @@ if _descriptor._USE_C_DESCRIPTORS == False:
67
67
  _globals['_GETMODELREQUEST']._serialized_end=1043
68
68
  _globals['_GETMODELRESPONSE']._serialized_start=1045
69
69
  _globals['_GETMODELRESPONSE']._serialized_end=1109
70
- _globals['_DELETEMODELREQUEST']._serialized_start=1111
71
- _globals['_DELETEMODELREQUEST']._serialized_end=1169
72
- _globals['_DELETEMODELRESPONSE']._serialized_start=1171
73
- _globals['_DELETEMODELRESPONSE']._serialized_end=1192
74
- _globals['_UPDATEMODELREQUEST']._serialized_start=1194
75
- _globals['_UPDATEMODELREQUEST']._serialized_end=1307
76
- _globals['_UPDATEMODELRESPONSE']._serialized_start=1309
77
- _globals['_UPDATEMODELRESPONSE']._serialized_end=1385
78
- _globals['_CREATEBRANCHREQUEST']._serialized_start=1387
79
- _globals['_CREATEBRANCHREQUEST']._serialized_end=1447
80
- _globals['_CREATEBRANCHRESPONSE']._serialized_start=1449
81
- _globals['_CREATEBRANCHRESPONSE']._serialized_end=1490
82
- _globals['_DELETEBRANCHREQUEST']._serialized_start=1492
83
- _globals['_DELETEBRANCHREQUEST']._serialized_end=1532
84
- _globals['_DELETEBRANCHRESPONSE']._serialized_start=1534
85
- _globals['_DELETEBRANCHRESPONSE']._serialized_end=1556
86
- _globals['_ISMODELEXISTSREQUEST']._serialized_start=1558
87
- _globals['_ISMODELEXISTSREQUEST']._serialized_end=1636
88
- _globals['_ISMODELEXISTSRESPONSE']._serialized_start=1638
89
- _globals['_ISMODELEXISTSRESPONSE']._serialized_end=1677
90
- _globals['_MODELSPEC']._serialized_start=1680
91
- _globals['_MODELSPEC']._serialized_end=1898
92
- _globals['_MODELBRIEF']._serialized_start=1901
93
- _globals['_MODELBRIEF']._serialized_end=2228
94
- _globals['_MODELSPECCONFIGURATION']._serialized_start=2230
95
- _globals['_MODELSPECCONFIGURATION']._serialized_end=2281
96
- _globals['_MODEL']._serialized_start=2284
97
- _globals['_MODEL']._serialized_end=2912
98
- _globals['_MODELBUILDTYPE']._serialized_start=2915
99
- _globals['_MODELBUILDTYPE']._serialized_end=3065
100
- _globals['_SOURCE']._serialized_start=3067
101
- _globals['_SOURCE']._serialized_end=3075
102
- _globals['_MODELREGISTRYIMAGE']._serialized_start=3077
103
- _globals['_MODELREGISTRYIMAGE']._serialized_end=3128
104
- _globals['_BRANCH']._serialized_start=3131
105
- _globals['_BRANCH']._serialized_end=3366
106
- _globals['_BRANCH_BRANCHSTATUS']._serialized_start=3298
107
- _globals['_BRANCH_BRANCHSTATUS']._serialized_end=3366
108
- _globals['_GETDEPLOYMENTDETAILSREQUEST']._serialized_start=3368
109
- _globals['_GETDEPLOYMENTDETAILSREQUEST']._serialized_end=3434
110
- _globals['_GETDEPLOYMENTDETAILSRESPONSE']._serialized_start=3436
111
- _globals['_GETDEPLOYMENTDETAILSRESPONSE']._serialized_end=3537
112
- _globals['_DEPLOYMENTDETAILS']._serialized_start=3540
113
- _globals['_DEPLOYMENTDETAILS']._serialized_end=3815
114
- _globals['_LISTMODELSMETADATAREQUEST']._serialized_start=3817
115
- _globals['_LISTMODELSMETADATAREQUEST']._serialized_end=3864
116
- _globals['_LISTMODELSMETADATARESPONSE']._serialized_start=3866
117
- _globals['_LISTMODELSMETADATARESPONSE']._serialized_end=3957
118
- _globals['_GETMODELMETADATAREQUEST']._serialized_start=3959
119
- _globals['_GETMODELMETADATAREQUEST']._serialized_end=4002
120
- _globals['_GETMODELMETADATARESPONSE']._serialized_start=4004
121
- _globals['_GETMODELMETADATARESPONSE']._serialized_end=4093
122
- _globals['_ENVIRONMENTROUTES']._serialized_start=4095
123
- _globals['_ENVIRONMENTROUTES']._serialized_end=4199
124
- _globals['_MODELMETADATA']._serialized_start=4202
125
- _globals['_MODELMETADATA']._serialized_end=4898
126
- _globals['_MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY']._serialized_start=4675
127
- _globals['_MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY']._serialized_end=4780
128
- _globals['_MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY']._serialized_start=4782
129
- _globals['_MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY']._serialized_end=4898
130
- _globals['_CHECKMODELCREATIONALLOWEDREQUEST']._serialized_start=4900
131
- _globals['_CHECKMODELCREATIONALLOWEDREQUEST']._serialized_end=5005
132
- _globals['_CHECKMODELCREATIONALLOWEDRESPONSE']._serialized_start=5008
133
- _globals['_CHECKMODELCREATIONALLOWEDRESPONSE']._serialized_end=5291
134
- _globals['_MODELEXISTS']._serialized_start=5293
135
- _globals['_MODELEXISTS']._serialized_end=5306
136
- _globals['_MODELNOTEXISTS']._serialized_start=5308
137
- _globals['_MODELNOTEXISTS']._serialized_end=5324
138
- _globals['_MODELEXISTSWITHMISMATCH']._serialized_start=5326
139
- _globals['_MODELEXISTSWITHMISMATCH']._serialized_end=5368
140
- _globals['_MODELSMANAGEMENTSERVICE']._serialized_start=5578
141
- _globals['_MODELSMANAGEMENTSERVICE']._serialized_end=7101
70
+ _globals['_LISTMODELSBYMODELIDSREQUEST']._serialized_start=1111
71
+ _globals['_LISTMODELSBYMODELIDSREQUEST']._serialized_end=1180
72
+ _globals['_DELETEMODELREQUEST']._serialized_start=1182
73
+ _globals['_DELETEMODELREQUEST']._serialized_end=1240
74
+ _globals['_DELETEMODELRESPONSE']._serialized_start=1242
75
+ _globals['_DELETEMODELRESPONSE']._serialized_end=1263
76
+ _globals['_UPDATEMODELREQUEST']._serialized_start=1265
77
+ _globals['_UPDATEMODELREQUEST']._serialized_end=1378
78
+ _globals['_UPDATEMODELRESPONSE']._serialized_start=1380
79
+ _globals['_UPDATEMODELRESPONSE']._serialized_end=1456
80
+ _globals['_CREATEBRANCHREQUEST']._serialized_start=1458
81
+ _globals['_CREATEBRANCHREQUEST']._serialized_end=1518
82
+ _globals['_CREATEBRANCHRESPONSE']._serialized_start=1520
83
+ _globals['_CREATEBRANCHRESPONSE']._serialized_end=1561
84
+ _globals['_DELETEBRANCHREQUEST']._serialized_start=1563
85
+ _globals['_DELETEBRANCHREQUEST']._serialized_end=1603
86
+ _globals['_DELETEBRANCHRESPONSE']._serialized_start=1605
87
+ _globals['_DELETEBRANCHRESPONSE']._serialized_end=1627
88
+ _globals['_ISMODELEXISTSREQUEST']._serialized_start=1629
89
+ _globals['_ISMODELEXISTSREQUEST']._serialized_end=1707
90
+ _globals['_ISMODELEXISTSRESPONSE']._serialized_start=1709
91
+ _globals['_ISMODELEXISTSRESPONSE']._serialized_end=1748
92
+ _globals['_MODELSPEC']._serialized_start=1751
93
+ _globals['_MODELSPEC']._serialized_end=1969
94
+ _globals['_MODELBRIEF']._serialized_start=1972
95
+ _globals['_MODELBRIEF']._serialized_end=2299
96
+ _globals['_MODELSPECCONFIGURATION']._serialized_start=2301
97
+ _globals['_MODELSPECCONFIGURATION']._serialized_end=2352
98
+ _globals['_MODEL']._serialized_start=2355
99
+ _globals['_MODEL']._serialized_end=2983
100
+ _globals['_MODELBUILDTYPE']._serialized_start=2986
101
+ _globals['_MODELBUILDTYPE']._serialized_end=3136
102
+ _globals['_SOURCE']._serialized_start=3138
103
+ _globals['_SOURCE']._serialized_end=3146
104
+ _globals['_MODELREGISTRYIMAGE']._serialized_start=3148
105
+ _globals['_MODELREGISTRYIMAGE']._serialized_end=3199
106
+ _globals['_BRANCH']._serialized_start=3202
107
+ _globals['_BRANCH']._serialized_end=3437
108
+ _globals['_BRANCH_BRANCHSTATUS']._serialized_start=3369
109
+ _globals['_BRANCH_BRANCHSTATUS']._serialized_end=3437
110
+ _globals['_GETDEPLOYMENTDETAILSREQUEST']._serialized_start=3439
111
+ _globals['_GETDEPLOYMENTDETAILSREQUEST']._serialized_end=3505
112
+ _globals['_GETDEPLOYMENTDETAILSRESPONSE']._serialized_start=3507
113
+ _globals['_GETDEPLOYMENTDETAILSRESPONSE']._serialized_end=3608
114
+ _globals['_DEPLOYMENTDETAILS']._serialized_start=3611
115
+ _globals['_DEPLOYMENTDETAILS']._serialized_end=3886
116
+ _globals['_LISTMODELSMETADATAREQUEST']._serialized_start=3888
117
+ _globals['_LISTMODELSMETADATAREQUEST']._serialized_end=3935
118
+ _globals['_LISTMODELSMETADATARESPONSE']._serialized_start=3937
119
+ _globals['_LISTMODELSMETADATARESPONSE']._serialized_end=4028
120
+ _globals['_GETMODELMETADATAREQUEST']._serialized_start=4030
121
+ _globals['_GETMODELMETADATAREQUEST']._serialized_end=4073
122
+ _globals['_GETMODELMETADATARESPONSE']._serialized_start=4075
123
+ _globals['_GETMODELMETADATARESPONSE']._serialized_end=4164
124
+ _globals['_ENVIRONMENTROUTES']._serialized_start=4166
125
+ _globals['_ENVIRONMENTROUTES']._serialized_end=4270
126
+ _globals['_MODELMETADATA']._serialized_start=4273
127
+ _globals['_MODELMETADATA']._serialized_end=4969
128
+ _globals['_MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY']._serialized_start=4746
129
+ _globals['_MODELMETADATA_AUDIENCEROUTESBYENVIRONMENTENTRY']._serialized_end=4851
130
+ _globals['_MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY']._serialized_start=4853
131
+ _globals['_MODELMETADATA_AUDIENCEROUTESGROUPEDBYENVIRONMENTENTRY']._serialized_end=4969
132
+ _globals['_CHECKMODELCREATIONALLOWEDREQUEST']._serialized_start=4971
133
+ _globals['_CHECKMODELCREATIONALLOWEDREQUEST']._serialized_end=5076
134
+ _globals['_CHECKMODELCREATIONALLOWEDRESPONSE']._serialized_start=5079
135
+ _globals['_CHECKMODELCREATIONALLOWEDRESPONSE']._serialized_end=5362
136
+ _globals['_MODELEXISTS']._serialized_start=5364
137
+ _globals['_MODELEXISTS']._serialized_end=5377
138
+ _globals['_MODELNOTEXISTS']._serialized_start=5379
139
+ _globals['_MODELNOTEXISTS']._serialized_end=5395
140
+ _globals['_MODELEXISTSWITHMISMATCH']._serialized_start=5397
141
+ _globals['_MODELEXISTSWITHMISMATCH']._serialized_end=5439
142
+ _globals['_MODELSMANAGEMENTSERVICE']._serialized_start=5649
143
+ _globals['_MODELSMANAGEMENTSERVICE']._serialized_end=7293
142
144
  # @@protoc_insertion_point(module_scope)
frogml/_proto/qwak/models/models_pb2.pyi CHANGED
@@ -260,6 +260,30 @@ class GetModelResponse(google.protobuf.message.Message):
260
260
 
261
261
  global___GetModelResponse = GetModelResponse
262
262
 
263
+ @typing.final
264
+ class ListModelsByModelIdsRequest(google.protobuf.message.Message):
265
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
266
+
267
+ MODEL_IDS_FIELD_NUMBER: builtins.int
268
+ MODEL_UUIDS_FIELD_NUMBER: builtins.int
269
+ @property
270
+ def model_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
271
+ """The model ids"""
272
+
273
+ @property
274
+ def model_uuids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
275
+ """The model uuids"""
276
+
277
+ def __init__(
278
+ self,
279
+ *,
280
+ model_ids: collections.abc.Iterable[builtins.str] | None = ...,
281
+ model_uuids: collections.abc.Iterable[builtins.str] | None = ...,
282
+ ) -> None: ...
283
+ def ClearField(self, field_name: typing.Literal["model_ids", b"model_ids", "model_uuids", b"model_uuids"]) -> None: ...
284
+
285
+ global___ListModelsByModelIdsRequest = ListModelsByModelIdsRequest
286
+
263
287
  @typing.final
264
288
  class DeleteModelRequest(google.protobuf.message.Message):
265
289
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
frogml/_proto/qwak/models/models_pb2_grpc.py CHANGED
@@ -34,6 +34,11 @@ class ModelsManagementServiceStub(object):
34
34
  request_serializer=qwak_dot_models_dot_models__pb2.GetModelRequest.SerializeToString,
35
35
  response_deserializer=qwak_dot_models_dot_models__pb2.GetModelResponse.FromString,
36
36
  )
37
+ self.ListModelsByModelIds = channel.unary_unary(
38
+ '/qwak.models.management.ModelsManagementService/ListModelsByModelIds',
39
+ request_serializer=qwak_dot_models_dot_models__pb2.ListModelsByModelIdsRequest.SerializeToString,
40
+ response_deserializer=qwak_dot_models_dot_models__pb2.ListModelsResponse.FromString,
41
+ )
37
42
  self.DeleteModel = channel.unary_unary(
38
43
  '/qwak.models.management.ModelsManagementService/DeleteModel',
39
44
  request_serializer=qwak_dot_models_dot_models__pb2.DeleteModelRequest.SerializeToString,
@@ -112,6 +117,13 @@ class ModelsManagementServiceServicer(object):
112
117
  context.set_details('Method not implemented!')
113
118
  raise NotImplementedError('Method not implemented!')
114
119
 
120
+ def ListModelsByModelIds(self, request, context):
121
+ """Get models by model IDs and model UUIDs
122
+ """
123
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
124
+ context.set_details('Method not implemented!')
125
+ raise NotImplementedError('Method not implemented!')
126
+
115
127
  def DeleteModel(self, request, context):
116
128
  """Delete model
117
129
  """
@@ -193,6 +205,11 @@ def add_ModelsManagementServiceServicer_to_server(servicer, server):
193
205
  request_deserializer=qwak_dot_models_dot_models__pb2.GetModelRequest.FromString,
194
206
  response_serializer=qwak_dot_models_dot_models__pb2.GetModelResponse.SerializeToString,
195
207
  ),
208
+ 'ListModelsByModelIds': grpc.unary_unary_rpc_method_handler(
209
+ servicer.ListModelsByModelIds,
210
+ request_deserializer=qwak_dot_models_dot_models__pb2.ListModelsByModelIdsRequest.FromString,
211
+ response_serializer=qwak_dot_models_dot_models__pb2.ListModelsResponse.SerializeToString,
212
+ ),
196
213
  'DeleteModel': grpc.unary_unary_rpc_method_handler(
197
214
  servicer.DeleteModel,
198
215
  request_deserializer=qwak_dot_models_dot_models__pb2.DeleteModelRequest.FromString,
@@ -316,6 +333,23 @@ class ModelsManagementService(object):
316
333
  options, channel_credentials,
317
334
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
318
335
 
336
+ @staticmethod
337
+ def ListModelsByModelIds(request,
338
+ target,
339
+ options=(),
340
+ channel_credentials=None,
341
+ call_credentials=None,
342
+ insecure=False,
343
+ compression=None,
344
+ wait_for_ready=None,
345
+ timeout=None,
346
+ metadata=None):
347
+ return grpc.experimental.unary_unary(request, target, '/qwak.models.management.ModelsManagementService/ListModelsByModelIds',
348
+ qwak_dot_models_dot_models__pb2.ListModelsByModelIdsRequest.SerializeToString,
349
+ qwak_dot_models_dot_models__pb2.ListModelsResponse.FromString,
350
+ options, channel_credentials,
351
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
352
+
319
353
  @staticmethod
320
354
  def DeleteModel(request,
321
355
  target,