frogml-core 0.0.76__py3-none-any.whl → 0.0.78__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 (17) hide show
  1. frogml_core/__init__.py +1 -1
  2. {frogml_core-0.0.76.dist-info → frogml_core-0.0.78.dist-info}/METADATA +1 -1
  3. {frogml_core-0.0.76.dist-info → frogml_core-0.0.78.dist-info}/RECORD +17 -8
  4. {frogml_core-0.0.76.dist-info → frogml_core-0.0.78.dist-info}/WHEEL +1 -1
  5. frogml_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2.py +35 -0
  6. frogml_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2.pyi +17 -0
  7. frogml_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2_grpc.py +4 -0
  8. frogml_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2.py +35 -0
  9. frogml_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2.pyi +17 -0
  10. frogml_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2_grpc.py +4 -0
  11. frogml_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2.py +57 -0
  12. frogml_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2.pyi +79 -0
  13. frogml_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2_grpc.py +4 -0
  14. frogml_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2.py +4 -3
  15. frogml_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2.pyi +7 -2
  16. frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.py +47 -37
  17. frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi +34 -1
frogml_core/__init__.py CHANGED
@@ -1,7 +1,7 @@
1
1
  """Top-level package for frogml."""
2
2
 
3
3
  __author__ = "jfrog"
4
- __version__ = "0.0.76"
4
+ __version__ = "0.0.78"
5
5
 
6
6
  from frogml_core.inner.di_configuration import wire_dependencies
7
7
  from frogml_core.model.model_version_tracking import ( # noqa: F401,E501
{frogml_core-0.0.76.dist-info → frogml_core-0.0.78.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: frogml-core
3
- Version: 0.0.76
3
+ Version: 0.0.78
4
4
  Summary: frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform
5
5
  License: Apache-2.0
6
6
  Keywords: mlops,ml,deployment,serving,model
{frogml_core-0.0.76.dist-info → frogml_core-0.0.78.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- frogml_core/__init__.py,sha256=JFOVXLDbP0Z1qhJzQj1rahRHurUp_xGYNagocwb-9EI,777
1
+ frogml_core/__init__.py,sha256=Uql7AUqVfJsciUbDjWMoJoILzqPYZkBjdkdyY8skAsM,777
2
2
  frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
3
3
  frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
4
4
  frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -495,9 +495,15 @@ frogml_proto/qwak/administration/runtime_configuration/v0/external/elasticsearch
495
495
  frogml_proto/qwak/administration/runtime_configuration/v0/external/kafka_config_pb2.py,sha256=kNYjS4Rzbrk2u1TrJT1WoCAz2TRHUc5NrMhpl2OxacA,5188
496
496
  frogml_proto/qwak/administration/runtime_configuration/v0/external/kafka_config_pb2.pyi,sha256=clCZQ9IKh9Sof_IiefMaaPP99nLNH9whytth3diiR0o,3493
497
497
  frogml_proto/qwak/administration/runtime_configuration/v0/external/kafka_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
498
+ frogml_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2.py,sha256=UOWLCW8L8RnzfEajUPgu5euBihnPp4NXL3ndM6bA6VE,2164
499
+ frogml_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2.pyi,sha256=RuZooE_GiRTfbA4_PwWEPvv3M8CWILkDRzcpWEg_Ono,427
500
+ frogml_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
498
501
  frogml_proto/qwak/administration/runtime_configuration/v0/external/redis_config_pb2.py,sha256=pTJrJgDoC0HQiEPDiz6rF9UVXB4BG3K-RZjYdSQiF8U,2544
499
502
  frogml_proto/qwak/administration/runtime_configuration/v0/external/redis_config_pb2.pyi,sha256=eZjYhz_rcAmnQFdU4KeTGZEeePP6GB11PV8SOi4kz9g,1540
500
503
  frogml_proto/qwak/administration/runtime_configuration/v0/external/redis_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
504
+ frogml_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2.py,sha256=hpgqHlnpofBC2QJi_7Vlk9r-a55gVQklJvbDF7TIMtU,2249
505
+ frogml_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2.pyi,sha256=GrXC12VIePiETq-2YxEyugeYrtq5mEtpw93WlnlgURM,442
506
+ frogml_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
501
507
  frogml_proto/qwak/administration/runtime_configuration/v0/feature_store_config_pb2.py,sha256=sxrA7pP71wnnCTh_fN4nurxW9zHizKrPUw9isH7sEE8,4217
502
508
  frogml_proto/qwak/administration/runtime_configuration/v0/feature_store_config_pb2.pyi,sha256=A5OXe8GQmk3Rw60CpOvBqbiO56rPQv46-HdJCKfhQAk,3008
503
509
  frogml_proto/qwak/administration/runtime_configuration/v0/feature_store_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -516,8 +522,11 @@ frogml_proto/qwak/administration/runtime_configuration/v0/network_config_pb2_grp
516
522
  frogml_proto/qwak/administration/runtime_configuration/v0/object_storage_config_pb2.py,sha256=p39Mb6VPtI0aYnsiILAjPcKPYWWdGkPpCNOyMgMOqmY,3626
517
523
  frogml_proto/qwak/administration/runtime_configuration/v0/object_storage_config_pb2.pyi,sha256=I5lypi84OoePvdpz7FtuzVq8J2PRyU--mqV680WlNzc,2562
518
524
  frogml_proto/qwak/administration/runtime_configuration/v0/object_storage_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
519
- frogml_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2.py,sha256=F5uo2c_cPvaaKmJ6aovak41pevl6VBk2tELXXMzjg4g,4924
520
- frogml_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2.pyi,sha256=QLTmYD45iyIZFPj4ozOVR7foK6To8UC0RY_TPCy8Mp4,4979
525
+ frogml_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2.py,sha256=NmAJTK4VMJAB-_mTEtUZGGtGEk7isXnd-4nxsNM5bXA,5049
526
+ frogml_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2.pyi,sha256=8fJqSJJX4QVJdBZpYiGCGaotrPVqX_zEJnnxIRwvFG0,4403
527
+ frogml_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
528
+ frogml_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2.py,sha256=OAdD6pbS8HDbKBPEzCqhwb8HTdiSh5UqcnmYoMvSWE0,5348
529
+ frogml_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2.pyi,sha256=dnQAsKFrNsqpZ8idbAUNq8FKfpS4w0eQmBLZ1Y3owFg,5589
521
530
  frogml_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
522
531
  frogml_proto/qwak/administration/runtime_configuration/v0/sql_engine_config_pb2.py,sha256=4qWJev_E2ZzU3jaSTjZ9I-pcAYNIYwJ-UoE3_tgkD7A,3767
523
532
  frogml_proto/qwak/administration/runtime_configuration/v0/sql_engine_config_pb2.pyi,sha256=86lYyaMcsIOIxqx3J0q9DKA6HdVQzFpzq_guhJcgV9s,2588
@@ -900,8 +909,8 @@ frogml_proto/qwak/kube_deployment_captain/alert_pb2_grpc.py,sha256=1oboBPFxaTEXt
900
909
  frogml_proto/qwak/kube_deployment_captain/alerting_pb2.py,sha256=2WquRLoQXoQMVRn4q0eK4ouA0bbOFUeXBYqqhuUER4s,3717
901
910
  frogml_proto/qwak/kube_deployment_captain/alerting_pb2.pyi,sha256=AGXtot0uXW-mGd2xuixCCJB6eTt0UuF3b2oAhb4IsNc,3759
902
911
  frogml_proto/qwak/kube_deployment_captain/alerting_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
903
- frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.py,sha256=ZsfBxb7kK6WFaArChEkd9V8PFJ_C62s51S4dZute6OA,28172
904
- frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi,sha256=KLyFPYDjltXubE2x-TjNgbN6Vcp4_6VCbtqD8653ZEM,37284
912
+ frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.py,sha256=btbJlJfan4bao4ktbvviClA6H8To5H9MFsECk8G7B-4,29136
913
+ frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi,sha256=4u4U_F76gCFfsLb9ivpZKtLuzR7zssncYrlHNIj3CK0,39265
905
914
  frogml_proto/qwak/kube_deployment_captain/batch_job_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
906
915
  frogml_proto/qwak/kube_deployment_captain/deployment_pb2.py,sha256=3u38KAg__KYW1n-YlLDQCJxTDXj7eyQSbILEoX7fsz8,14050
907
916
  frogml_proto/qwak/kube_deployment_captain/deployment_pb2.pyi,sha256=Sz11iTrL3BlUEl1jXNf0Q1SDe1SX_MjjyB9bMrtsOVg,24090
@@ -1087,6 +1096,6 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
1087
1096
  frogml_services_mock/services_mock.py,sha256=xfbYvl05gFUq2oT82cT-6K2CEJ6d9kFX_tk77QZHTFI,20787
1088
1097
  frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1089
1098
  frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
1090
- frogml_core-0.0.76.dist-info/METADATA,sha256=6XSGnrnc493daGZM95FWRrfuJM7npUDHZ7fwc-APiC8,2004
1091
- frogml_core-0.0.76.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
1092
- frogml_core-0.0.76.dist-info/RECORD,,
1099
+ frogml_core-0.0.78.dist-info/METADATA,sha256=6S_8ftAvq2OeIvAM9chAPCpSGRYiqX9G7-eEXRayQq0,2004
1100
+ frogml_core-0.0.78.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
1101
+ frogml_core-0.0.78.dist-info/RECORD,,
{frogml_core-0.0.76.dist-info → frogml_core-0.0.78.dist-info}/WHEEL RENAMED
@@ -1,4 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: poetry-core 2.1.1
2
+ Generator: poetry-core 2.1.3
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
frogml_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2.py ADDED
@@ -0,0 +1,35 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: frogml_proto.qwak.administration/runtime_configuration/v0/external/prometheus_config.proto
4
+ """Generated protocol buffer code."""
5
+ from google.protobuf import descriptor as _descriptor
6
+ from google.protobuf import descriptor_pool as _descriptor_pool
7
+ from google.protobuf import message as _message
8
+ from google.protobuf import reflection as _reflection
9
+ from google.protobuf import symbol_database as _symbol_database
10
+ # @@protoc_insertion_point(imports)
11
+
12
+ _sym_db = _symbol_database.Default()
13
+
14
+
15
+
16
+
17
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nMqwak/administration/runtime_configuration/v0/external/prometheus_config.proto\x12Gqwak.administration.runtime_configuration.v0.external.prometheus_config\"\x19\n\x17PrometheusConfigurationB\xf9\x01\n@com.jfrog.ml.runtime_configuration.v0.external.prometheus_configP\x01Z\xb2\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0/external;runtime_configuration_v0_externalb\x06proto3')
18
+
19
+
20
+
21
+ _PROMETHEUSCONFIGURATION = DESCRIPTOR.message_types_by_name['PrometheusConfiguration']
22
+ PrometheusConfiguration = _reflection.GeneratedProtocolMessageType('PrometheusConfiguration', (_message.Message,), {
23
+ 'DESCRIPTOR' : _PROMETHEUSCONFIGURATION,
24
+ '__module__' : 'qwak.administration.runtime_configuration.v0.external.prometheus_config_pb2'
25
+ # @@protoc_insertion_point(class_scope:qwak.administration.runtime_configuration.v0.external.prometheus_config.PrometheusConfiguration)
26
+ })
27
+ _sym_db.RegisterMessage(PrometheusConfiguration)
28
+
29
+ if _descriptor._USE_C_DESCRIPTORS == False:
30
+
31
+ DESCRIPTOR._options = None
32
+ DESCRIPTOR._serialized_options = b'\n@com.jfrog.ml.runtime_configuration.v0.external.prometheus_configP\001Z\262\001github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0/external;runtime_configuration_v0_external'
33
+ _PROMETHEUSCONFIGURATION._serialized_start=154
34
+ _PROMETHEUSCONFIGURATION._serialized_end=179
35
+ # @@protoc_insertion_point(module_scope)
frogml_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_pb2.pyi ADDED
@@ -0,0 +1,17 @@
1
+ """
2
+ @generated by mypy-protobuf. Do not edit manually!
3
+ isort:skip_file
4
+ """
5
+ import google.protobuf.descriptor
6
+ import google.protobuf.message
7
+
8
+ DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
9
+
10
+ class PrometheusConfiguration(google.protobuf.message.Message):
11
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
12
+
13
+ def __init__(
14
+ self,
15
+ ) -> None: ...
16
+
17
+ global___PrometheusConfiguration = PrometheusConfiguration
frogml_proto/qwak/administration/runtime_configuration/v0/external/prometheus_config_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/administration/runtime_configuration/v0/external/victoriametrics_config_pb2.py ADDED
@@ -0,0 +1,35 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: frogml_proto.qwak.administration/runtime_configuration/v0/external/victoriametrics_config.proto
4
+ """Generated protocol buffer code."""
5
+ from google.protobuf import descriptor as _descriptor
6
+ from google.protobuf import descriptor_pool as _descriptor_pool
7
+ from google.protobuf import message as _message
8
+ from google.protobuf import reflection as _reflection
9
+ from google.protobuf import symbol_database as _symbol_database
10
+ # @@protoc_insertion_point(imports)
11
+
12
+ _sym_db = _symbol_database.Default()
13
+
14
+
15
+
16
+
17
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nRqwak/administration/runtime_configuration/v0/external/victoriametrics_config.proto\x12Lqwak.administration.runtime_configuration.v0.external.victoriametrics_config\"\x1e\n\x1cVictoriaMetricsConfigurationB\xfe\x01\nEcom.jfrog.ml.runtime_configuration.v0.external.victoriametrics_configP\x01Z\xb2\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0/external;runtime_configuration_v0_externalb\x06proto3')
18
+
19
+
20
+
21
+ _VICTORIAMETRICSCONFIGURATION = DESCRIPTOR.message_types_by_name['VictoriaMetricsConfiguration']
22
+ VictoriaMetricsConfiguration = _reflection.GeneratedProtocolMessageType('VictoriaMetricsConfiguration', (_message.Message,), {
23
+ 'DESCRIPTOR' : _VICTORIAMETRICSCONFIGURATION,
24
+ '__module__' : 'qwak.administration.runtime_configuration.v0.external.victoriametrics_config_pb2'
25
+ # @@protoc_insertion_point(class_scope:qwak.administration.runtime_configuration.v0.external.victoriametrics_config.VictoriaMetricsConfiguration)
26
+ })
27
+ _sym_db.RegisterMessage(VictoriaMetricsConfiguration)
28
+
29
+ if _descriptor._USE_C_DESCRIPTORS == False:
30
+
31
+ DESCRIPTOR._options = None
32
+ DESCRIPTOR._serialized_options = b'\nEcom.jfrog.ml.runtime_configuration.v0.external.victoriametrics_configP\001Z\262\001github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0/external;runtime_configuration_v0_external'
33
+ _VICTORIAMETRICSCONFIGURATION._serialized_start=164
34
+ _VICTORIAMETRICSCONFIGURATION._serialized_end=194
35
+ # @@protoc_insertion_point(module_scope)
frogml_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_pb2.pyi ADDED
@@ -0,0 +1,17 @@
1
+ """
2
+ @generated by mypy-protobuf. Do not edit manually!
3
+ isort:skip_file
4
+ """
5
+ import google.protobuf.descriptor
6
+ import google.protobuf.message
7
+
8
+ DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
9
+
10
+ class VictoriaMetricsConfiguration(google.protobuf.message.Message):
11
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
12
+
13
+ def __init__(
14
+ self,
15
+ ) -> None: ...
16
+
17
+ global___VictoriaMetricsConfiguration = VictoriaMetricsConfiguration
frogml_proto/qwak/administration/runtime_configuration/v0/external/victoriametrics_config_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/administration/runtime_configuration/v0/observability_config_pb2.py ADDED
@@ -0,0 +1,57 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: frogml_proto.qwak.administration/runtime_configuration/v0/observability_config.proto
4
+ """Generated protocol buffer code."""
5
+ from google.protobuf import descriptor as _descriptor
6
+ from google.protobuf import descriptor_pool as _descriptor_pool
7
+ from google.protobuf import message as _message
8
+ from google.protobuf import reflection as _reflection
9
+ from google.protobuf import symbol_database as _symbol_database
10
+ # @@protoc_insertion_point(imports)
11
+
12
+ _sym_db = _symbol_database.Default()
13
+
14
+
15
+ from frogml_proto.qwak.administration.runtime_configuration.v0.external import prometheus_config_pb2 as qwak_dot_administration_dot_runtime__configuration_dot_v0_dot_external_dot_prometheus__config__pb2
16
+ from frogml_proto.qwak.administration.runtime_configuration.v0.external import victoriametrics_config_pb2 as qwak_dot_administration_dot_runtime__configuration_dot_v0_dot_external_dot_victoriametrics__config__pb2
17
+
18
+
19
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nGqwak/administration/runtime_configuration/v0/observability_config.proto\x12\x41qwak.administration.runtime_configuration.v0.observability_config\x1aMqwak/administration/runtime_configuration/v0/external/prometheus_config.proto\x1aRqwak/administration/runtime_configuration/v0/external/victoriametrics_config.proto\"\x83\x02\n\x1aObservabilityConfiguration\x12n\n\nmonitoring\x18\x01 \x01(\x0b\x32Z.qwak.administration.runtime_configuration.v0.observability_config.MonitoringConfiguration\x12u\n\x08\x61lerting\x18\x02 \x01(\x0b\x32\x63.qwak.administration.runtime_configuration.v0.observability_config.AlertingIntegrationConfiguration\"\xa2\x02\n\x17MonitoringConfiguration\x12v\n\nprometheus\x18\x01 \x01(\x0b\x32`.qwak.administration.runtime_configuration.v0.external.prometheus_config.PrometheusConfigurationH\x00\x12\x86\x01\n\x10victoria_metrics\x18\x02 \x01(\x0b\x32j.qwak.administration.runtime_configuration.v0.external.victoriametrics_config.VictoriaMetricsConfigurationH\x00\x42\x06\n\x04type\"\xab\x02\n AlertingIntegrationConfiguration\x12v\n\nprometheus\x18\x01 \x01(\x0b\x32`.qwak.administration.runtime_configuration.v0.external.prometheus_config.PrometheusConfigurationH\x00\x12\x86\x01\n\x10victoria_metrics\x18\x02 \x01(\x0b\x32j.qwak.administration.runtime_configuration.v0.external.victoriametrics_config.VictoriaMetricsConfigurationH\x00\x42\x06\n\x04typeB\xe1\x01\n:com.jfrog.ml.runtime_configuration.v0.observability_configP\x01Z\xa0\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0;runtime_configuration_v0b\x06proto3')
20
+
21
+
22
+
23
+ _OBSERVABILITYCONFIGURATION = DESCRIPTOR.message_types_by_name['ObservabilityConfiguration']
24
+ _MONITORINGCONFIGURATION = DESCRIPTOR.message_types_by_name['MonitoringConfiguration']
25
+ _ALERTINGINTEGRATIONCONFIGURATION = DESCRIPTOR.message_types_by_name['AlertingIntegrationConfiguration']
26
+ ObservabilityConfiguration = _reflection.GeneratedProtocolMessageType('ObservabilityConfiguration', (_message.Message,), {
27
+ 'DESCRIPTOR' : _OBSERVABILITYCONFIGURATION,
28
+ '__module__' : 'qwak.administration.runtime_configuration.v0.observability_config_pb2'
29
+ # @@protoc_insertion_point(class_scope:qwak.administration.runtime_configuration.v0.observability_config.ObservabilityConfiguration)
30
+ })
31
+ _sym_db.RegisterMessage(ObservabilityConfiguration)
32
+
33
+ MonitoringConfiguration = _reflection.GeneratedProtocolMessageType('MonitoringConfiguration', (_message.Message,), {
34
+ 'DESCRIPTOR' : _MONITORINGCONFIGURATION,
35
+ '__module__' : 'qwak.administration.runtime_configuration.v0.observability_config_pb2'
36
+ # @@protoc_insertion_point(class_scope:qwak.administration.runtime_configuration.v0.observability_config.MonitoringConfiguration)
37
+ })
38
+ _sym_db.RegisterMessage(MonitoringConfiguration)
39
+
40
+ AlertingIntegrationConfiguration = _reflection.GeneratedProtocolMessageType('AlertingIntegrationConfiguration', (_message.Message,), {
41
+ 'DESCRIPTOR' : _ALERTINGINTEGRATIONCONFIGURATION,
42
+ '__module__' : 'qwak.administration.runtime_configuration.v0.observability_config_pb2'
43
+ # @@protoc_insertion_point(class_scope:qwak.administration.runtime_configuration.v0.observability_config.AlertingIntegrationConfiguration)
44
+ })
45
+ _sym_db.RegisterMessage(AlertingIntegrationConfiguration)
46
+
47
+ if _descriptor._USE_C_DESCRIPTORS == False:
48
+
49
+ DESCRIPTOR._options = None
50
+ DESCRIPTOR._serialized_options = b'\n:com.jfrog.ml.runtime_configuration.v0.observability_configP\001Z\240\001github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0;runtime_configuration_v0'
51
+ _OBSERVABILITYCONFIGURATION._serialized_start=306
52
+ _OBSERVABILITYCONFIGURATION._serialized_end=565
53
+ _MONITORINGCONFIGURATION._serialized_start=568
54
+ _MONITORINGCONFIGURATION._serialized_end=858
55
+ _ALERTINGINTEGRATIONCONFIGURATION._serialized_start=861
56
+ _ALERTINGINTEGRATIONCONFIGURATION._serialized_end=1160
57
+ # @@protoc_insertion_point(module_scope)
frogml_proto/qwak/administration/runtime_configuration/v0/observability_config_pb2.pyi ADDED
@@ -0,0 +1,79 @@
1
+ """
2
+ @generated by mypy-protobuf. Do not edit manually!
3
+ isort:skip_file
4
+ """
5
+ import builtins
6
+ import google.protobuf.descriptor
7
+ import google.protobuf.message
8
+ import frogml_proto.qwak.administration.runtime_configuration.v0.external.prometheus_config_pb2
9
+ import frogml_proto.qwak.administration.runtime_configuration.v0.external.victoriametrics_config_pb2
10
+ import sys
11
+
12
+ if sys.version_info >= (3, 8):
13
+ import typing as typing_extensions
14
+ else:
15
+ import typing_extensions
16
+
17
+ DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
18
+
19
+ class ObservabilityConfiguration(google.protobuf.message.Message):
20
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
21
+
22
+ MONITORING_FIELD_NUMBER: builtins.int
23
+ ALERTING_FIELD_NUMBER: builtins.int
24
+ @property
25
+ def monitoring(self) -> global___MonitoringConfiguration: ...
26
+ @property
27
+ def alerting(self) -> global___AlertingIntegrationConfiguration: ...
28
+ def __init__(
29
+ self,
30
+ *,
31
+ monitoring: global___MonitoringConfiguration | None = ...,
32
+ alerting: global___AlertingIntegrationConfiguration | None = ...,
33
+ ) -> None: ...
34
+ def HasField(self, field_name: typing_extensions.Literal["alerting", b"alerting", "monitoring", b"monitoring"]) -> builtins.bool: ...
35
+ def ClearField(self, field_name: typing_extensions.Literal["alerting", b"alerting", "monitoring", b"monitoring"]) -> None: ...
36
+
37
+ global___ObservabilityConfiguration = ObservabilityConfiguration
38
+
39
+ class MonitoringConfiguration(google.protobuf.message.Message):
40
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
41
+
42
+ PROMETHEUS_FIELD_NUMBER: builtins.int
43
+ VICTORIA_METRICS_FIELD_NUMBER: builtins.int
44
+ @property
45
+ def prometheus(self) -> frogml_proto.qwak.administration.runtime_configuration.v0.external.prometheus_config_pb2.PrometheusConfiguration: ...
46
+ @property
47
+ def victoria_metrics(self) -> frogml_proto.qwak.administration.runtime_configuration.v0.external.victoriametrics_config_pb2.VictoriaMetricsConfiguration: ...
48
+ def __init__(
49
+ self,
50
+ *,
51
+ prometheus: frogml_proto.qwak.administration.runtime_configuration.v0.external.prometheus_config_pb2.PrometheusConfiguration | None = ...,
52
+ victoria_metrics: frogml_proto.qwak.administration.runtime_configuration.v0.external.victoriametrics_config_pb2.VictoriaMetricsConfiguration | None = ...,
53
+ ) -> None: ...
54
+ def HasField(self, field_name: typing_extensions.Literal["prometheus", b"prometheus", "type", b"type", "victoria_metrics", b"victoria_metrics"]) -> builtins.bool: ...
55
+ def ClearField(self, field_name: typing_extensions.Literal["prometheus", b"prometheus", "type", b"type", "victoria_metrics", b"victoria_metrics"]) -> None: ...
56
+ def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["prometheus", "victoria_metrics"] | None: ...
57
+
58
+ global___MonitoringConfiguration = MonitoringConfiguration
59
+
60
+ class AlertingIntegrationConfiguration(google.protobuf.message.Message):
61
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
62
+
63
+ PROMETHEUS_FIELD_NUMBER: builtins.int
64
+ VICTORIA_METRICS_FIELD_NUMBER: builtins.int
65
+ @property
66
+ def prometheus(self) -> frogml_proto.qwak.administration.runtime_configuration.v0.external.prometheus_config_pb2.PrometheusConfiguration: ...
67
+ @property
68
+ def victoria_metrics(self) -> frogml_proto.qwak.administration.runtime_configuration.v0.external.victoriametrics_config_pb2.VictoriaMetricsConfiguration: ...
69
+ def __init__(
70
+ self,
71
+ *,
72
+ prometheus: frogml_proto.qwak.administration.runtime_configuration.v0.external.prometheus_config_pb2.PrometheusConfiguration | None = ...,
73
+ victoria_metrics: frogml_proto.qwak.administration.runtime_configuration.v0.external.victoriametrics_config_pb2.VictoriaMetricsConfiguration | None = ...,
74
+ ) -> None: ...
75
+ def HasField(self, field_name: typing_extensions.Literal["prometheus", b"prometheus", "type", b"type", "victoria_metrics", b"victoria_metrics"]) -> builtins.bool: ...
76
+ def ClearField(self, field_name: typing_extensions.Literal["prometheus", b"prometheus", "type", b"type", "victoria_metrics", b"victoria_metrics"]) -> None: ...
77
+ def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["prometheus", "victoria_metrics"] | None: ...
78
+
79
+ global___AlertingIntegrationConfiguration = AlertingIntegrationConfiguration
frogml_proto/qwak/administration/runtime_configuration/v0/observability_config_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/administration/runtime_configuration/v0/runtime_config_pb2.py CHANGED
@@ -19,9 +19,10 @@ from frogml_proto.qwak.administration.runtime_configuration.v0 import logs_stora
19
19
  from frogml_proto.qwak.administration.runtime_configuration.v0 import object_storage_config_pb2 as qwak_dot_administration_dot_runtime__configuration_dot_v0_dot_object__storage__config__pb2
20
20
  from frogml_proto.qwak.administration.runtime_configuration.v0 import hosting_config_pb2 as qwak_dot_administration_dot_runtime__configuration_dot_v0_dot_hosting__config__pb2
21
21
  from frogml_proto.qwak.administration.runtime_configuration.v0 import model_analytics_storage_config_pb2 as qwak_dot_administration_dot_runtime__configuration_dot_v0_dot_model__analytics__storage__config__pb2
22
+ from frogml_proto.qwak.administration.runtime_configuration.v0 import observability_config_pb2 as qwak_dot_administration_dot_runtime__configuration_dot_v0_dot_observability__config__pb2
22
23
 
23
24
 
24
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nAqwak/administration/runtime_configuration/v0/runtime_config.proto\x12,qwak.administration.runtime_configuration.v0\x1aGqwak/administration/runtime_configuration/v0/feature_store_config.proto\x1a\x44qwak/administration/runtime_configuration/v0/sql_engine_config.proto\x1a\x46qwak/administration/runtime_configuration/v0/data_catalog_config.proto\x1a\x46qwak/administration/runtime_configuration/v0/logs_storage_config.proto\x1aHqwak/administration/runtime_configuration/v0/object_storage_config.proto\x1a\x41qwak/administration/runtime_configuration/v0/hosting_config.proto\x1aQqwak/administration/runtime_configuration/v0/model_analytics_storage_config.proto\"\xae\x07\n\x14RuntimeConfiguration\x12p\n\x15hosting_configuration\x18\x01 \x01(\x0b\x32Q.qwak.administration.runtime_configuration.v0.hosting_config.HostingConfiguration\x12\x81\x01\n\x1b\x66\x65\x61ture_store_configuration\x18\x02 \x01(\x0b\x32\\.qwak.administration.runtime_configuration.v0.feature_store_config.FeatureStoreConfiguration\x12\x84\x01\n\x1cobject_storage_configuration\x18\x03 \x01(\x0b\x32^.qwak.administration.runtime_configuration.v0.object_storage_config.ObjectStorageConfiguration\x12~\n\x1alogs_storage_configuration\x18\x04 \x01(\x0b\x32Z.qwak.administration.runtime_configuration.v0.logs_storage_config.LogsStorageConfiguration\x12~\n\x1a\x64\x61ta_catalog_configuration\x18\x05 \x01(\x0b\x32Z.qwak.administration.runtime_configuration.v0.data_catalog_config.DataCatalogConfiguration\x12x\n\x18sql_engine_configuration\x18\x06 \x01(\x0b\x32V.qwak.administration.runtime_configuration.v0.sql_engine_config.SqlEngineConfiguration\x12\x9e\x01\n%model_analytics_storage_configuration\x18\x07 \x01(\x0b\x32o.qwak.administration.runtime_configuration.v0.model_analytics_storage_config.ModelAnalyticsStorageConfigurationB\xcc\x01\n%com.jfrog.ml.runtime_configuration.v0P\x01Z\xa0\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0;runtime_configuration_v0b\x06proto3')
25
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nAqwak/administration/runtime_configuration/v0/runtime_config.proto\x12,qwak.administration.runtime_configuration.v0\x1aGqwak/administration/runtime_configuration/v0/feature_store_config.proto\x1a\x44qwak/administration/runtime_configuration/v0/sql_engine_config.proto\x1a\x46qwak/administration/runtime_configuration/v0/data_catalog_config.proto\x1a\x46qwak/administration/runtime_configuration/v0/logs_storage_config.proto\x1aHqwak/administration/runtime_configuration/v0/object_storage_config.proto\x1a\x41qwak/administration/runtime_configuration/v0/hosting_config.proto\x1aQqwak/administration/runtime_configuration/v0/model_analytics_storage_config.proto\x1aGqwak/administration/runtime_configuration/v0/observability_config.proto\"\xb3\x08\n\x14RuntimeConfiguration\x12p\n\x15hosting_configuration\x18\x01 \x01(\x0b\x32Q.qwak.administration.runtime_configuration.v0.hosting_config.HostingConfiguration\x12\x81\x01\n\x1b\x66\x65\x61ture_store_configuration\x18\x02 \x01(\x0b\x32\\.qwak.administration.runtime_configuration.v0.feature_store_config.FeatureStoreConfiguration\x12\x84\x01\n\x1cobject_storage_configuration\x18\x03 \x01(\x0b\x32^.qwak.administration.runtime_configuration.v0.object_storage_config.ObjectStorageConfiguration\x12~\n\x1alogs_storage_configuration\x18\x04 \x01(\x0b\x32Z.qwak.administration.runtime_configuration.v0.logs_storage_config.LogsStorageConfiguration\x12~\n\x1a\x64\x61ta_catalog_configuration\x18\x05 \x01(\x0b\x32Z.qwak.administration.runtime_configuration.v0.data_catalog_config.DataCatalogConfiguration\x12x\n\x18sql_engine_configuration\x18\x06 \x01(\x0b\x32V.qwak.administration.runtime_configuration.v0.sql_engine_config.SqlEngineConfiguration\x12\x9e\x01\n%model_analytics_storage_configuration\x18\x07 \x01(\x0b\x32o.qwak.administration.runtime_configuration.v0.model_analytics_storage_config.ModelAnalyticsStorageConfiguration\x12\x82\x01\n\x1bobservability_configuration\x18\x08 \x01(\x0b\x32].qwak.administration.runtime_configuration.v0.observability_config.ObservabilityConfigurationB\xcc\x01\n%com.jfrog.ml.runtime_configuration.v0P\x01Z\xa0\x01github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0;runtime_configuration_v0b\x06proto3')
25
26
 
26
27
 
27
28
 
@@ -37,6 +38,6 @@ if _descriptor._USE_C_DESCRIPTORS == False:
37
38
 
38
39
  DESCRIPTOR._options = None
39
40
  DESCRIPTOR._serialized_options = b'\n%com.jfrog.ml.runtime_configuration.v0P\001Z\240\001github.com/qwak-ai/qwak-platform/services/core/java/user-management/user-management-api/pb/qwak/administration/runtime_configuration/v0;runtime_configuration_v0'
40
- _RUNTIMECONFIGURATION._serialized_start=627
41
- _RUNTIMECONFIGURATION._serialized_end=1569
41
+ _RUNTIMECONFIGURATION._serialized_start=700
42
+ _RUNTIMECONFIGURATION._serialized_end=1775
42
43
  # @@protoc_insertion_point(module_scope)
frogml_proto/qwak/administration/runtime_configuration/v0/runtime_config_pb2.pyi CHANGED
@@ -11,6 +11,7 @@ import frogml_proto.qwak.administration.runtime_configuration.v0.hosting_config_
11
11
  import frogml_proto.qwak.administration.runtime_configuration.v0.logs_storage_config_pb2
12
12
  import frogml_proto.qwak.administration.runtime_configuration.v0.model_analytics_storage_config_pb2
13
13
  import frogml_proto.qwak.administration.runtime_configuration.v0.object_storage_config_pb2
14
+ import frogml_proto.qwak.administration.runtime_configuration.v0.observability_config_pb2
14
15
  import frogml_proto.qwak.administration.runtime_configuration.v0.sql_engine_config_pb2
15
16
  import sys
16
17
 
@@ -31,6 +32,7 @@ class RuntimeConfiguration(google.protobuf.message.Message):
31
32
  DATA_CATALOG_CONFIGURATION_FIELD_NUMBER: builtins.int
32
33
  SQL_ENGINE_CONFIGURATION_FIELD_NUMBER: builtins.int
33
34
  MODEL_ANALYTICS_STORAGE_CONFIGURATION_FIELD_NUMBER: builtins.int
35
+ OBSERVABILITY_CONFIGURATION_FIELD_NUMBER: builtins.int
34
36
  @property
35
37
  def hosting_configuration(self) -> frogml_proto.qwak.administration.runtime_configuration.v0.hosting_config_pb2.HostingConfiguration: ...
36
38
  @property
@@ -45,6 +47,8 @@ class RuntimeConfiguration(google.protobuf.message.Message):
45
47
  def sql_engine_configuration(self) -> frogml_proto.qwak.administration.runtime_configuration.v0.sql_engine_config_pb2.SqlEngineConfiguration: ...
46
48
  @property
47
49
  def model_analytics_storage_configuration(self) -> frogml_proto.qwak.administration.runtime_configuration.v0.model_analytics_storage_config_pb2.ModelAnalyticsStorageConfiguration: ...
50
+ @property
51
+ def observability_configuration(self) -> frogml_proto.qwak.administration.runtime_configuration.v0.observability_config_pb2.ObservabilityConfiguration: ...
48
52
  def __init__(
49
53
  self,
50
54
  *,
@@ -55,8 +59,9 @@ class RuntimeConfiguration(google.protobuf.message.Message):
55
59
  data_catalog_configuration: frogml_proto.qwak.administration.runtime_configuration.v0.data_catalog_config_pb2.DataCatalogConfiguration | None = ...,
56
60
  sql_engine_configuration: frogml_proto.qwak.administration.runtime_configuration.v0.sql_engine_config_pb2.SqlEngineConfiguration | None = ...,
57
61
  model_analytics_storage_configuration: frogml_proto.qwak.administration.runtime_configuration.v0.model_analytics_storage_config_pb2.ModelAnalyticsStorageConfiguration | None = ...,
62
+ observability_configuration: frogml_proto.qwak.administration.runtime_configuration.v0.observability_config_pb2.ObservabilityConfiguration | None = ...,
58
63
  ) -> None: ...
59
- def HasField(self, field_name: typing_extensions.Literal["data_catalog_configuration", b"data_catalog_configuration", "feature_store_configuration", b"feature_store_configuration", "hosting_configuration", b"hosting_configuration", "logs_storage_configuration", b"logs_storage_configuration", "model_analytics_storage_configuration", b"model_analytics_storage_configuration", "object_storage_configuration", b"object_storage_configuration", "sql_engine_configuration", b"sql_engine_configuration"]) -> builtins.bool: ...
60
- def ClearField(self, field_name: typing_extensions.Literal["data_catalog_configuration", b"data_catalog_configuration", "feature_store_configuration", b"feature_store_configuration", "hosting_configuration", b"hosting_configuration", "logs_storage_configuration", b"logs_storage_configuration", "model_analytics_storage_configuration", b"model_analytics_storage_configuration", "object_storage_configuration", b"object_storage_configuration", "sql_engine_configuration", b"sql_engine_configuration"]) -> None: ...
64
+ def HasField(self, field_name: typing_extensions.Literal["data_catalog_configuration", b"data_catalog_configuration", "feature_store_configuration", b"feature_store_configuration", "hosting_configuration", b"hosting_configuration", "logs_storage_configuration", b"logs_storage_configuration", "model_analytics_storage_configuration", b"model_analytics_storage_configuration", "object_storage_configuration", b"object_storage_configuration", "observability_configuration", b"observability_configuration", "sql_engine_configuration", b"sql_engine_configuration"]) -> builtins.bool: ...
65
+ def ClearField(self, field_name: typing_extensions.Literal["data_catalog_configuration", b"data_catalog_configuration", "feature_store_configuration", b"feature_store_configuration", "hosting_configuration", b"hosting_configuration", "logs_storage_configuration", b"logs_storage_configuration", "model_analytics_storage_configuration", b"model_analytics_storage_configuration", "object_storage_configuration", b"object_storage_configuration", "observability_configuration", b"observability_configuration", "sql_engine_configuration", b"sql_engine_configuration"]) -> None: ...
61
66
 
62
67
  global___RuntimeConfiguration = RuntimeConfiguration
frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.py CHANGED
@@ -15,7 +15,7 @@ _sym_db = _symbol_database.Default()
15
15
 
16
16
 
17
17
 
18
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,qwak/kube_deployment_captain/batch_job.proto\x12\x1cqwak.kube.deployment.captain\"\xd2\x03\n\x1cListInferenceJobFilesRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x16\n\x0e\x64irectory_path\x18\x02 \x01(\t\x12\x18\n\x0ctoken_secret\x18\x03 \x01(\tB\x02\x18\x01\x12\x19\n\rsecret_secret\x18\x04 \x01(\tB\x02\x18\x01\x12\x1a\n\x12secret_service_url\x18\x05 \x01(\t\x12\x14\n\x08role_arn\x18\x06 \x01(\tB\x02\x18\x01\x12G\n\x0f\x61ws_credentials\x18\x07 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x08 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12P\n\x14grouped_task_details\x18\t \x01(\x0b\x32\x30.qwak.kube.deployment.captain.GroupedTaskDetailsH\x01\x12\x16\n\x0e\x65nvironment_id\x18\n \x01(\tB\x1a\n\x18\x63loud_client_credentialsB\x0b\n\ttask_flow\"c\n\x12GroupedTaskDetails\x12\x17\n\x0f\x63oncurrentTasks\x18\x01 \x01(\x05\x12\x18\n\x10\x64\x65stination_path\x18\x02 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x03 \x01(\t\"\xb4\x02\n)CleanStorageObjectFromCloudStorageRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x15\n\robject_prefix\x18\x02 \x03(\t\x12\x1a\n\x12secret_service_url\x18\x03 \x01(\t\x12G\n\x0f\x61ws_credentials\x18\x04 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x05 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12\x16\n\x0e\x65nvironment_id\x18\x06 \x01(\tB\x1a\n\x18\x63loud_client_credentials\"O\n\x0e\x41wsCredentials\x12\x10\n\x08role_arn\x18\x01 \x01(\t\x12\x14\n\x0ctoken_secret\x18\x02 \x01(\t\x12\x15\n\rsecret_secret\x18\x04 \x01(\t\"9\n\x0eGcpCredentials\x12\'\n\x1fservice_account_json_key_secret\x18\x01 \x01(\t\"\xa6\x01\n\x1dListInferenceJobFilesResponse\x12\x12\n\nfile_names\x18\x01 \x03(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x03 \x01(\t\x12H\n\x11list_file_details\x18\x04 \x03(\x0b\x32-.qwak.kube.deployment.captain.ListFileDetails\"W\n\x0fListFileDetails\x12\x15\n\ris_point_file\x18\x01 \x01(\x08\x12\x17\n\x0fpoint_file_path\x18\x02 \x01(\t\x12\x14\n\x0cobject_paths\x18\x03 \x03(\t\"\xf0\x01\n\"CreateInferenceTaskExecutorRequest\x12\x63\n\x1btask_executor_configuration\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.TaskExecutorConfigurationMessage\x12\x65\n\x1cinference_task_configuration\x18\x02 \x01(\x0b\x32?.qwak.kube.deployment.captain.InferenceTaskConfigurationMessage\"\x89\x01\n\x1aPrepareInferenceJobRequest\x12k\n#inference_job_configuration_message\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.InferenceJobConfigurationMessage\"F\n\x1bPrepareInferenceJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xcd\x04\n TaskExecutorConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\x12G\n\x10model_identifier\x18\x05 \x01(\x0b\x32-.qwak.kube.deployment.captain.ModelIdentifier\x12\x11\n\timage_url\x18\x06 \x01(\t\x12\x15\n\rbackoff_limit\x18\x07 \x01(\x05\x12\x0f\n\x03\x63pu\x18\x08 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\t \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x0c \x01(\t\x12\x41\n\x08job_size\x18\r \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x17\n\x0fpurchase_option\x18\x0e \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0f \x01(\t\x12%\n\x1d\x63ustom_service_account_secret\x18\x10 \x01(\t\x12#\n\x1bjfrog_token_api_secret_name\x18\x11 \x01(\t\x12\x16\n\x0ejfrog_base_url\x18\x12 \x01(\t\"q\n InferenceJobConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x03 \x01(\t\"`\n\x0fModelIdentifier\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x15\n\tbranch_id\x18\x03 \x01(\tB\x02\x18\x01\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\x8c\x04\n!InferenceTaskConfigurationMessage\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x02 \x01(\t\x12\x10\n\x08\x66ilepath\x18\x03 \x01(\t\x12\x18\n\x10\x64\x65stination_path\x18\x04 \x01(\t\x12\x44\n\x0finput_file_type\x18\x05 \x01(\x0e\x32+.qwak.kube.deployment.captain.InputFileType\x12\x46\n\x10output_file_type\x18\x06 \x01(\x0e\x32,.qwak.kube.deployment.captain.OutputFileType\x12\x14\n\x0ctoken_secret\x18\x07 \x01(\t\x12\x15\n\rsecret_secret\x18\x08 \x01(\t\x12\x43\n\nparameters\x18\t \x03(\x0b\x32/.qwak.kube.deployment.captain.BatchJobParameter\x12\x43\n\x0e\x63loud_provider\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.CloudProvider\x12\'\n\x1fservice_account_json_key_secret\x18\x0b \x01(\t\x12\x1a\n\x12is_point_file_path\x18\x0c \x01(\x08\"/\n\x11\x42\x61tchJobParameter\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"?\n#CleanInferenceTasksExecutorsRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\"O\n$CleanInferenceTasksExecutorsResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"X\n!CleanInferenceTaskExecutorRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\"M\n\"CleanInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"N\n#CreateInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xf7\x02\n\x1eStartInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x11\n\timage_url\x18\x04 \x01(\t\x12\x0f\n\x03\x63pu\x18\x05 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x06 \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\x07 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x11\n\texecutors\x18\x08 \x01(\x05\x12\x0f\n\x07timeout\x18\t \x01(\x05\x12\x41\n\x08job_size\x18\n \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x12\n\nmodel_uuid\x18\x0b \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0c \x01(\t\"!\n\x1fStartInferenceJobWarmupResponse\"p\n\x1f\x43\x61ncelInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\"\n CancelInferenceJobWarmupResponse\"\xd5\x01\n\x11\x42\x61tchJobResources\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0b\n\x03\x63pu\x18\x02 \x01(\x02\x12\x15\n\rmemory_amount\x18\x03 \x01(\x05\x12\x41\n\x0cmemory_units\x18\x04 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApi\x12\x41\n\rgpu_resources\x18\x05 \x01(\x0b\x32*.qwak.kube.deployment.captain.GpuResources\"[\n\x0cGpuResources\x12\x37\n\x08gpu_type\x18\x01 \x01(\x0e\x32%.qwak.kube.deployment.captain.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05*:\n\rMemoryUnitApi\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x81\x01\n\rInputFileType\x12\x1d\n\x19UNDEFINED_INPUT_FILE_TYPE\x10\x00\x12\x17\n\x13\x43SV_INPUT_FILE_TYPE\x10\x01\x12\x1b\n\x17\x46\x45\x41THER_INPUT_FILE_TYPE\x10\x02\x12\x1b\n\x17PARQUET_INPUT_FILE_TYPE\x10\x03*\x86\x01\n\x0eOutputFileType\x12\x1e\n\x1aUNDEFINED_OUTPUT_FILE_TYPE\x10\x00\x12\x18\n\x14\x43SV_OUTPUT_FILE_TYPE\x10\x01\x12\x1c\n\x18\x46\x45\x41THER_OUTPUT_FILE_TYPE\x10\x02\x12\x1c\n\x18PARQUET_OUTPUT_FILE_TYPE\x10\x03*\x90\x02\n\x07GpuType\x12\x0f\n\x0bINVALID_GPU\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*=\n\rCloudProvider\x12\x1a\n\x16UNKNOWN_CLOUD_PROVIDER\x10\x00\x12\x07\n\x03\x41WS\x10\x01\x12\x07\n\x03GCP\x10\x02\x42+\n\'com.qwak.ai.kube.deployment.captain.apiP\x01\x62\x06proto3')
18
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,qwak/kube_deployment_captain/batch_job.proto\x12\x1cqwak.kube.deployment.captain\"\xd2\x03\n\x1cListInferenceJobFilesRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x16\n\x0e\x64irectory_path\x18\x02 \x01(\t\x12\x18\n\x0ctoken_secret\x18\x03 \x01(\tB\x02\x18\x01\x12\x19\n\rsecret_secret\x18\x04 \x01(\tB\x02\x18\x01\x12\x1a\n\x12secret_service_url\x18\x05 \x01(\t\x12\x14\n\x08role_arn\x18\x06 \x01(\tB\x02\x18\x01\x12G\n\x0f\x61ws_credentials\x18\x07 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x08 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12P\n\x14grouped_task_details\x18\t \x01(\x0b\x32\x30.qwak.kube.deployment.captain.GroupedTaskDetailsH\x01\x12\x16\n\x0e\x65nvironment_id\x18\n \x01(\tB\x1a\n\x18\x63loud_client_credentialsB\x0b\n\ttask_flow\"c\n\x12GroupedTaskDetails\x12\x17\n\x0f\x63oncurrentTasks\x18\x01 \x01(\x05\x12\x18\n\x10\x64\x65stination_path\x18\x02 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x03 \x01(\t\"\xb4\x02\n)CleanStorageObjectFromCloudStorageRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x15\n\robject_prefix\x18\x02 \x03(\t\x12\x1a\n\x12secret_service_url\x18\x03 \x01(\t\x12G\n\x0f\x61ws_credentials\x18\x04 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x05 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12\x16\n\x0e\x65nvironment_id\x18\x06 \x01(\tB\x1a\n\x18\x63loud_client_credentials\"O\n\x0e\x41wsCredentials\x12\x10\n\x08role_arn\x18\x01 \x01(\t\x12\x14\n\x0ctoken_secret\x18\x02 \x01(\t\x12\x15\n\rsecret_secret\x18\x04 \x01(\t\"9\n\x0eGcpCredentials\x12\'\n\x1fservice_account_json_key_secret\x18\x01 \x01(\t\"\xa6\x01\n\x1dListInferenceJobFilesResponse\x12\x12\n\nfile_names\x18\x01 \x03(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x03 \x01(\t\x12H\n\x11list_file_details\x18\x04 \x03(\x0b\x32-.qwak.kube.deployment.captain.ListFileDetails\"W\n\x0fListFileDetails\x12\x15\n\ris_point_file\x18\x01 \x01(\x08\x12\x17\n\x0fpoint_file_path\x18\x02 \x01(\t\x12\x14\n\x0cobject_paths\x18\x03 \x03(\t\"\xf0\x01\n\"CreateInferenceTaskExecutorRequest\x12\x63\n\x1btask_executor_configuration\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.TaskExecutorConfigurationMessage\x12\x65\n\x1cinference_task_configuration\x18\x02 \x01(\x0b\x32?.qwak.kube.deployment.captain.InferenceTaskConfigurationMessage\"\x89\x01\n\x1aPrepareInferenceJobRequest\x12k\n#inference_job_configuration_message\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.InferenceJobConfigurationMessage\"F\n\x1bPrepareInferenceJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xcd\x04\n TaskExecutorConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\x12G\n\x10model_identifier\x18\x05 \x01(\x0b\x32-.qwak.kube.deployment.captain.ModelIdentifier\x12\x11\n\timage_url\x18\x06 \x01(\t\x12\x15\n\rbackoff_limit\x18\x07 \x01(\x05\x12\x0f\n\x03\x63pu\x18\x08 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\t \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x0c \x01(\t\x12\x41\n\x08job_size\x18\r \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x17\n\x0fpurchase_option\x18\x0e \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0f \x01(\t\x12%\n\x1d\x63ustom_service_account_secret\x18\x10 \x01(\t\x12#\n\x1bjfrog_token_api_secret_name\x18\x11 \x01(\t\x12\x16\n\x0ejfrog_base_url\x18\x12 \x01(\t\"q\n InferenceJobConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x03 \x01(\t\"`\n\x0fModelIdentifier\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x15\n\tbranch_id\x18\x03 \x01(\tB\x02\x18\x01\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\x85\x01\n\x1dListFilesPreStepConfiguration\x12\x13\n\x0bsource_path\x18\x01 \x01(\t\x12\x1c\n\x14list_files_file_name\x18\x02 \x01(\t\x12\x1d\n\x15total_number_of_tasks\x18\x03 \x01(\x05\x12\x12\n\ntask_index\x18\x04 \x01(\x05\"\xf4\x04\n!InferenceTaskConfigurationMessage\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x02 \x01(\t\x12\x10\n\x08\x66ilepath\x18\x03 \x01(\t\x12\x18\n\x10\x64\x65stination_path\x18\x04 \x01(\t\x12\x44\n\x0finput_file_type\x18\x05 \x01(\x0e\x32+.qwak.kube.deployment.captain.InputFileType\x12\x46\n\x10output_file_type\x18\x06 \x01(\x0e\x32,.qwak.kube.deployment.captain.OutputFileType\x12\x14\n\x0ctoken_secret\x18\x07 \x01(\t\x12\x15\n\rsecret_secret\x18\x08 \x01(\t\x12\x43\n\nparameters\x18\t \x03(\x0b\x32/.qwak.kube.deployment.captain.BatchJobParameter\x12\x43\n\x0e\x63loud_provider\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.CloudProvider\x12\'\n\x1fservice_account_json_key_secret\x18\x0b \x01(\t\x12\x1a\n\x12is_point_file_path\x18\x0c \x01(\x08\x12\x66\n!list_files_pre_step_configuration\x18\r \x01(\x0b\x32;.qwak.kube.deployment.captain.ListFilesPreStepConfiguration\"/\n\x11\x42\x61tchJobParameter\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"?\n#CleanInferenceTasksExecutorsRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\"O\n$CleanInferenceTasksExecutorsResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"X\n!CleanInferenceTaskExecutorRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\"M\n\"CleanInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"N\n#CreateInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xf7\x02\n\x1eStartInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x11\n\timage_url\x18\x04 \x01(\t\x12\x0f\n\x03\x63pu\x18\x05 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x06 \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\x07 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x11\n\texecutors\x18\x08 \x01(\x05\x12\x0f\n\x07timeout\x18\t \x01(\x05\x12\x41\n\x08job_size\x18\n \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x12\n\nmodel_uuid\x18\x0b \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0c \x01(\t\"!\n\x1fStartInferenceJobWarmupResponse\"p\n\x1f\x43\x61ncelInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\"\n CancelInferenceJobWarmupResponse\"\xd5\x01\n\x11\x42\x61tchJobResources\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0b\n\x03\x63pu\x18\x02 \x01(\x02\x12\x15\n\rmemory_amount\x18\x03 \x01(\x05\x12\x41\n\x0cmemory_units\x18\x04 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApi\x12\x41\n\rgpu_resources\x18\x05 \x01(\x0b\x32*.qwak.kube.deployment.captain.GpuResources\"[\n\x0cGpuResources\x12\x37\n\x08gpu_type\x18\x01 \x01(\x0e\x32%.qwak.kube.deployment.captain.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05*:\n\rMemoryUnitApi\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x81\x01\n\rInputFileType\x12\x1d\n\x19UNDEFINED_INPUT_FILE_TYPE\x10\x00\x12\x17\n\x13\x43SV_INPUT_FILE_TYPE\x10\x01\x12\x1b\n\x17\x46\x45\x41THER_INPUT_FILE_TYPE\x10\x02\x12\x1b\n\x17PARQUET_INPUT_FILE_TYPE\x10\x03*\x86\x01\n\x0eOutputFileType\x12\x1e\n\x1aUNDEFINED_OUTPUT_FILE_TYPE\x10\x00\x12\x18\n\x14\x43SV_OUTPUT_FILE_TYPE\x10\x01\x12\x1c\n\x18\x46\x45\x41THER_OUTPUT_FILE_TYPE\x10\x02\x12\x1c\n\x18PARQUET_OUTPUT_FILE_TYPE\x10\x03*\x90\x02\n\x07GpuType\x12\x0f\n\x0bINVALID_GPU\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*=\n\rCloudProvider\x12\x1a\n\x16UNKNOWN_CLOUD_PROVIDER\x10\x00\x12\x07\n\x03\x41WS\x10\x01\x12\x07\n\x03GCP\x10\x02\x42+\n\'com.qwak.ai.kube.deployment.captain.apiP\x01\x62\x06proto3')
19
19
 
20
20
  _MEMORYUNITAPI = DESCRIPTOR.enum_types_by_name['MemoryUnitApi']
21
21
  MemoryUnitApi = enum_type_wrapper.EnumTypeWrapper(_MEMORYUNITAPI)
@@ -69,6 +69,7 @@ _PREPAREINFERENCEJOBRESPONSE = DESCRIPTOR.message_types_by_name['PrepareInferenc
69
69
  _TASKEXECUTORCONFIGURATIONMESSAGE = DESCRIPTOR.message_types_by_name['TaskExecutorConfigurationMessage']
70
70
  _INFERENCEJOBCONFIGURATIONMESSAGE = DESCRIPTOR.message_types_by_name['InferenceJobConfigurationMessage']
71
71
  _MODELIDENTIFIER = DESCRIPTOR.message_types_by_name['ModelIdentifier']
72
+ _LISTFILESPRESTEPCONFIGURATION = DESCRIPTOR.message_types_by_name['ListFilesPreStepConfiguration']
72
73
  _INFERENCETASKCONFIGURATIONMESSAGE = DESCRIPTOR.message_types_by_name['InferenceTaskConfigurationMessage']
73
74
  _BATCHJOBPARAMETER = DESCRIPTOR.message_types_by_name['BatchJobParameter']
74
75
  _CLEANINFERENCETASKSEXECUTORSREQUEST = DESCRIPTOR.message_types_by_name['CleanInferenceTasksExecutorsRequest']
@@ -173,6 +174,13 @@ ModelIdentifier = _reflection.GeneratedProtocolMessageType('ModelIdentifier', (_
173
174
  })
174
175
  _sym_db.RegisterMessage(ModelIdentifier)
175
176
 
177
+ ListFilesPreStepConfiguration = _reflection.GeneratedProtocolMessageType('ListFilesPreStepConfiguration', (_message.Message,), {
178
+ 'DESCRIPTOR' : _LISTFILESPRESTEPCONFIGURATION,
179
+ '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
180
+ # @@protoc_insertion_point(class_scope:qwak.kube.deployment.captain.ListFilesPreStepConfiguration)
181
+ })
182
+ _sym_db.RegisterMessage(ListFilesPreStepConfiguration)
183
+
176
184
  InferenceTaskConfigurationMessage = _reflection.GeneratedProtocolMessageType('InferenceTaskConfigurationMessage', (_message.Message,), {
177
185
  'DESCRIPTOR' : _INFERENCETASKCONFIGURATIONMESSAGE,
178
186
  '__module__' : 'qwak.kube_deployment_captain.batch_job_pb2'
@@ -292,16 +300,16 @@ if _descriptor._USE_C_DESCRIPTORS == False:
292
300
  _STARTINFERENCEJOBWARMUPREQUEST.fields_by_name['memory_units']._serialized_options = b'\030\001'
293
301
  _CANCELINFERENCEJOBWARMUPREQUEST.fields_by_name['branch_id']._options = None
294
302
  _CANCELINFERENCEJOBWARMUPREQUEST.fields_by_name['branch_id']._serialized_options = b'\030\001'
295
- _MEMORYUNITAPI._serialized_start=4460
296
- _MEMORYUNITAPI._serialized_end=4518
297
- _INPUTFILETYPE._serialized_start=4521
298
- _INPUTFILETYPE._serialized_end=4650
299
- _OUTPUTFILETYPE._serialized_start=4653
300
- _OUTPUTFILETYPE._serialized_end=4787
301
- _GPUTYPE._serialized_start=4790
302
- _GPUTYPE._serialized_end=5062
303
- _CLOUDPROVIDER._serialized_start=5064
304
- _CLOUDPROVIDER._serialized_end=5125
303
+ _MEMORYUNITAPI._serialized_start=4700
304
+ _MEMORYUNITAPI._serialized_end=4758
305
+ _INPUTFILETYPE._serialized_start=4761
306
+ _INPUTFILETYPE._serialized_end=4890
307
+ _OUTPUTFILETYPE._serialized_start=4893
308
+ _OUTPUTFILETYPE._serialized_end=5027
309
+ _GPUTYPE._serialized_start=5030
310
+ _GPUTYPE._serialized_end=5302
311
+ _CLOUDPROVIDER._serialized_start=5304
312
+ _CLOUDPROVIDER._serialized_end=5365
305
313
  _LISTINFERENCEJOBFILESREQUEST._serialized_start=79
306
314
  _LISTINFERENCEJOBFILESREQUEST._serialized_end=545
307
315
  _GROUPEDTASKDETAILS._serialized_start=547
@@ -328,30 +336,32 @@ if _descriptor._USE_C_DESCRIPTORS == False:
328
336
  _INFERENCEJOBCONFIGURATIONMESSAGE._serialized_end=2517
329
337
  _MODELIDENTIFIER._serialized_start=2519
330
338
  _MODELIDENTIFIER._serialized_end=2615
331
- _INFERENCETASKCONFIGURATIONMESSAGE._serialized_start=2618
332
- _INFERENCETASKCONFIGURATIONMESSAGE._serialized_end=3142
333
- _BATCHJOBPARAMETER._serialized_start=3144
334
- _BATCHJOBPARAMETER._serialized_end=3191
335
- _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_start=3193
336
- _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_end=3256
337
- _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_start=3258
338
- _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_end=3337
339
- _CLEANINFERENCETASKEXECUTORREQUEST._serialized_start=3339
340
- _CLEANINFERENCETASKEXECUTORREQUEST._serialized_end=3427
341
- _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_start=3429
342
- _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_end=3506
343
- _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_start=3508
344
- _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_end=3586
345
- _STARTINFERENCEJOBWARMUPREQUEST._serialized_start=3589
346
- _STARTINFERENCEJOBWARMUPREQUEST._serialized_end=3964
347
- _STARTINFERENCEJOBWARMUPRESPONSE._serialized_start=3966
348
- _STARTINFERENCEJOBWARMUPRESPONSE._serialized_end=3999
349
- _CANCELINFERENCEJOBWARMUPREQUEST._serialized_start=4001
350
- _CANCELINFERENCEJOBWARMUPREQUEST._serialized_end=4113
351
- _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_start=4115
352
- _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_end=4149
353
- _BATCHJOBRESOURCES._serialized_start=4152
354
- _BATCHJOBRESOURCES._serialized_end=4365
355
- _GPURESOURCES._serialized_start=4367
356
- _GPURESOURCES._serialized_end=4458
339
+ _LISTFILESPRESTEPCONFIGURATION._serialized_start=2618
340
+ _LISTFILESPRESTEPCONFIGURATION._serialized_end=2751
341
+ _INFERENCETASKCONFIGURATIONMESSAGE._serialized_start=2754
342
+ _INFERENCETASKCONFIGURATIONMESSAGE._serialized_end=3382
343
+ _BATCHJOBPARAMETER._serialized_start=3384
344
+ _BATCHJOBPARAMETER._serialized_end=3431
345
+ _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_start=3433
346
+ _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_end=3496
347
+ _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_start=3498
348
+ _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_end=3577
349
+ _CLEANINFERENCETASKEXECUTORREQUEST._serialized_start=3579
350
+ _CLEANINFERENCETASKEXECUTORREQUEST._serialized_end=3667
351
+ _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_start=3669
352
+ _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_end=3746
353
+ _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_start=3748
354
+ _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_end=3826
355
+ _STARTINFERENCEJOBWARMUPREQUEST._serialized_start=3829
356
+ _STARTINFERENCEJOBWARMUPREQUEST._serialized_end=4204
357
+ _STARTINFERENCEJOBWARMUPRESPONSE._serialized_start=4206
358
+ _STARTINFERENCEJOBWARMUPRESPONSE._serialized_end=4239
359
+ _CANCELINFERENCEJOBWARMUPREQUEST._serialized_start=4241
360
+ _CANCELINFERENCEJOBWARMUPREQUEST._serialized_end=4353
361
+ _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_start=4355
362
+ _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_end=4389
363
+ _BATCHJOBRESOURCES._serialized_start=4392
364
+ _BATCHJOBRESOURCES._serialized_end=4605
365
+ _GPURESOURCES._serialized_start=4607
366
+ _GPURESOURCES._serialized_end=4698
357
367
  # @@protoc_insertion_point(module_scope)
frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi CHANGED
@@ -524,6 +524,33 @@ class ModelIdentifier(google.protobuf.message.Message):
524
524
 
525
525
  global___ModelIdentifier = ModelIdentifier
526
526
 
527
+ class ListFilesPreStepConfiguration(google.protobuf.message.Message):
528
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
529
+
530
+ SOURCE_PATH_FIELD_NUMBER: builtins.int
531
+ LIST_FILES_FILE_NAME_FIELD_NUMBER: builtins.int
532
+ TOTAL_NUMBER_OF_TASKS_FIELD_NUMBER: builtins.int
533
+ TASK_INDEX_FIELD_NUMBER: builtins.int
534
+ source_path: builtins.str
535
+ """Directory path in source bucket of data files to run inference on"""
536
+ list_files_file_name: builtins.str
537
+ """Name for the file to save the list of files in the source path"""
538
+ total_number_of_tasks: builtins.int
539
+ """Total number of inference tasks, used to divide the files list into similarly sized groups"""
540
+ task_index: builtins.int
541
+ """Index of the specific task, used to determine on which files group to run inference on"""
542
+ def __init__(
543
+ self,
544
+ *,
545
+ source_path: builtins.str = ...,
546
+ list_files_file_name: builtins.str = ...,
547
+ total_number_of_tasks: builtins.int = ...,
548
+ task_index: builtins.int = ...,
549
+ ) -> None: ...
550
+ def ClearField(self, field_name: typing_extensions.Literal["list_files_file_name", b"list_files_file_name", "source_path", b"source_path", "task_index", b"task_index", "total_number_of_tasks", b"total_number_of_tasks"]) -> None: ...
551
+
552
+ global___ListFilesPreStepConfiguration = ListFilesPreStepConfiguration
553
+
527
554
  class InferenceTaskConfigurationMessage(google.protobuf.message.Message):
528
555
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
529
556
 
@@ -539,6 +566,7 @@ class InferenceTaskConfigurationMessage(google.protobuf.message.Message):
539
566
  CLOUD_PROVIDER_FIELD_NUMBER: builtins.int
540
567
  SERVICE_ACCOUNT_JSON_KEY_SECRET_FIELD_NUMBER: builtins.int
541
568
  IS_POINT_FILE_PATH_FIELD_NUMBER: builtins.int
569
+ LIST_FILES_PRE_STEP_CONFIGURATION_FIELD_NUMBER: builtins.int
542
570
  source_bucket: builtins.str
543
571
  """The source bucket in cloud storage"""
544
572
  destination_bucket: builtins.str
@@ -564,6 +592,9 @@ class InferenceTaskConfigurationMessage(google.protobuf.message.Message):
564
592
  """The service account json key secret"""
565
593
  is_point_file_path: builtins.bool
566
594
  """Is the file contains points to inference files"""
595
+ @property
596
+ def list_files_pre_step_configuration(self) -> global___ListFilesPreStepConfiguration:
597
+ """Optional pre step configuration, used to list files in the source directory and run inference on subset of files"""
567
598
  def __init__(
568
599
  self,
569
600
  *,
@@ -579,8 +610,10 @@ class InferenceTaskConfigurationMessage(google.protobuf.message.Message):
579
610
  cloud_provider: global___CloudProvider.ValueType = ...,
580
611
  service_account_json_key_secret: builtins.str = ...,
581
612
  is_point_file_path: builtins.bool = ...,
613
+ list_files_pre_step_configuration: global___ListFilesPreStepConfiguration | None = ...,
582
614
  ) -> None: ...
583
- def ClearField(self, field_name: typing_extensions.Literal["cloud_provider", b"cloud_provider", "destination_bucket", b"destination_bucket", "destination_path", b"destination_path", "filepath", b"filepath", "input_file_type", b"input_file_type", "is_point_file_path", b"is_point_file_path", "output_file_type", b"output_file_type", "parameters", b"parameters", "secret_secret", b"secret_secret", "service_account_json_key_secret", b"service_account_json_key_secret", "source_bucket", b"source_bucket", "token_secret", b"token_secret"]) -> None: ...
615
+ def HasField(self, field_name: typing_extensions.Literal["list_files_pre_step_configuration", b"list_files_pre_step_configuration"]) -> builtins.bool: ...
616
+ def ClearField(self, field_name: typing_extensions.Literal["cloud_provider", b"cloud_provider", "destination_bucket", b"destination_bucket", "destination_path", b"destination_path", "filepath", b"filepath", "input_file_type", b"input_file_type", "is_point_file_path", b"is_point_file_path", "list_files_pre_step_configuration", b"list_files_pre_step_configuration", "output_file_type", b"output_file_type", "parameters", b"parameters", "secret_secret", b"secret_secret", "service_account_json_key_secret", b"service_account_json_key_secret", "source_bucket", b"source_bucket", "token_secret", b"token_secret"]) -> None: ...
584
617
 
585
618
  global___InferenceTaskConfigurationMessage = InferenceTaskConfigurationMessage
586
619