frogml-core 0.0.116__py3-none-any.whl → 0.0.117__py3-none-any.whl

_proto/qwak/audience/v1/audience_api_pb2_grpc.py

@@ -0,0 +1,240 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml._proto.qwak.audience.v1 import audience_api_pb2 as qwak_dot_audience_dot_v1_dot_audience__api__pb2
+
+
+class AudienceAPIStub(object):
+    """Audience API
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateAudience = channel.unary_unary(
+                '/qwak.audience.v1.AudienceAPI/CreateAudience',
+                request_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.CreateAudienceRequest.SerializeToString,
+                response_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.CreateAudienceResponse.FromString,
+                )
+        self.UpdateAudience = channel.unary_unary(
+                '/qwak.audience.v1.AudienceAPI/UpdateAudience',
+                request_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.UpdateAudienceRequest.SerializeToString,
+                response_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.UpdateAudienceResponse.FromString,
+                )
+        self.GetAudience = channel.unary_unary(
+                '/qwak.audience.v1.AudienceAPI/GetAudience',
+                request_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.GetAudienceRequest.SerializeToString,
+                response_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.GetAudienceResponse.FromString,
+                )
+        self.ListAudience = channel.unary_unary(
+                '/qwak.audience.v1.AudienceAPI/ListAudience',
+                request_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.ListAudienceRequest.SerializeToString,
+                response_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.ListAudienceResponse.FromString,
+                )
+        self.DeleteAudience = channel.unary_unary(
+                '/qwak.audience.v1.AudienceAPI/DeleteAudience',
+                request_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.DeleteAudienceRequest.SerializeToString,
+                response_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.DeleteAudienceResponse.FromString,
+                )
+        self.SyncAudiences = channel.unary_unary(
+                '/qwak.audience.v1.AudienceAPI/SyncAudiences',
+                request_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.SyncAudiencesRequest.SerializeToString,
+                response_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.SyncAudiencesResponse.FromString,
+                )
+
+
+class AudienceAPIServicer(object):
+    """Audience API
+    """
+
+    def CreateAudience(self, request, context):
+        """Create audience
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def UpdateAudience(self, request, context):
+        """Update audience by ID
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetAudience(self, request, context):
+        """Get audience by ID
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ListAudience(self, request, context):
+        """List audiences by model ID
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteAudience(self, request, context):
+        """Delete audience by ID
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def SyncAudiences(self, request, context):
+        """Sync DB <-> Proxy configuration for a specific environment
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_AudienceAPIServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'CreateAudience': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateAudience,
+                    request_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.CreateAudienceRequest.FromString,
+                    response_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.CreateAudienceResponse.SerializeToString,
+            ),
+            'UpdateAudience': grpc.unary_unary_rpc_method_handler(
+                    servicer.UpdateAudience,
+                    request_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.UpdateAudienceRequest.FromString,
+                    response_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.UpdateAudienceResponse.SerializeToString,
+            ),
+            'GetAudience': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetAudience,
+                    request_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.GetAudienceRequest.FromString,
+                    response_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.GetAudienceResponse.SerializeToString,
+            ),
+            'ListAudience': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListAudience,
+                    request_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.ListAudienceRequest.FromString,
+                    response_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.ListAudienceResponse.SerializeToString,
+            ),
+            'DeleteAudience': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteAudience,
+                    request_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.DeleteAudienceRequest.FromString,
+                    response_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.DeleteAudienceResponse.SerializeToString,
+            ),
+            'SyncAudiences': grpc.unary_unary_rpc_method_handler(
+                    servicer.SyncAudiences,
+                    request_deserializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.SyncAudiencesRequest.FromString,
+                    response_serializer=qwak_dot_audience_dot_v1_dot_audience__api__pb2.SyncAudiencesResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.audience.v1.AudienceAPI', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class AudienceAPI(object):
+    """Audience API
+    """
+
+    @staticmethod
+    def CreateAudience(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.audience.v1.AudienceAPI/CreateAudience',
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.CreateAudienceRequest.SerializeToString,
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.CreateAudienceResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def UpdateAudience(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.audience.v1.AudienceAPI/UpdateAudience',
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.UpdateAudienceRequest.SerializeToString,
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.UpdateAudienceResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetAudience(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.audience.v1.AudienceAPI/GetAudience',
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.GetAudienceRequest.SerializeToString,
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.GetAudienceResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ListAudience(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.audience.v1.AudienceAPI/ListAudience',
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.ListAudienceRequest.SerializeToString,
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.ListAudienceResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def DeleteAudience(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.audience.v1.AudienceAPI/DeleteAudience',
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.DeleteAudienceRequest.SerializeToString,
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.DeleteAudienceResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def SyncAudiences(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.audience.v1.AudienceAPI/SyncAudiences',
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.SyncAudiencesRequest.SerializeToString,
+            qwak_dot_audience_dot_v1_dot_audience__api__pb2.SyncAudiencesResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_proto/qwak/audience/v1/audience_pb2.py

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.audience/v1/audience.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fqwak/audience/v1/audience.proto\x12\x10qwak.audience.v1\x1a\x1fgoogle/protobuf/timestamp.proto\"I\n\rAudienceEntry\x12\n\n\x02id\x18\x01 \x01(\t\x12,\n\x08\x61udience\x18\x02 \x01(\x0b\x32\x1a.qwak.audience.v1.Audience\"\xc1\x03\n\x08\x41udience\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12/\n\nconditions\x18\x03 \x03(\x0b\x32\x1b.qwak.audience.v1.Condition\x12\'\n\x06routes\x18\x04 \x03(\x0b\x32\x17.qwak.audience.v1.Route\x12+\n\x07\x63reated\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x06 \x01(\t\x12,\n\x08modified\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x13\n\x0bmodified_by\x18\x08 \x01(\t\x12M\n\x12\x65nvironment_routes\x18\t \x03(\x0b\x32\x31.qwak.audience.v1.Audience.EnvironmentRoutesEntry\x1a\x65\n\x16\x45nvironmentRoutesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12:\n\x05value\x18\x02 \x01(\x0b\x32+.qwak.audience.v1.EnvironmentAudienceRoutes:\x02\x38\x01\"D\n\x19\x45nvironmentAudienceRoutes\x12\'\n\x06routes\x18\x01 \x03(\x0b\x32\x17.qwak.audience.v1.Route\"y\n\x13\x41udienceRoutesEntry\x12\x13\n\x0b\x61udience_id\x18\x01 \x01(\t\x12\x15\n\raudience_name\x18\x02 \x01(\t\x12\'\n\x06routes\x18\x04 \x03(\x0b\x32\x17.qwak.audience.v1.Route\x12\r\n\x05order\x18\x05 \x01(\x05\"\xb1\x01\n\tCondition\x12\x0b\n\x03key\x18\x01 \x01(\t\x12=\n\x10\x62inary_condition\x18\x02 \x01(\x0b\x32!.qwak.audience.v1.BinaryConditionH\x00\x12;\n\x0funary_condition\x18\x03 \x01(\x0b\x32 .qwak.audience.v1.UnaryConditionH\x00\x12\x0e\n\x06invert\x18\x04 \x01(\x08\x42\x0b\n\tcondition\"X\n\x0eUnaryCondition\x12\x35\n\x08operator\x18\x01 \x01(\x0e\x32#.qwak.audience.v1.UnaryOperatorType\x12\x0f\n\x07operand\x18\x02 \x01(\t\"x\n\x0f\x42inaryCondition\x12\x36\n\x08operator\x18\x01 \x01(\x0e\x32$.qwak.audience.v1.BinaryOperatorType\x12\x15\n\rfirst_operand\x18\x02 \x01(\t\x12\x16\n\x0esecond_operand\x18\x03 \x01(\t\"\x9a\x01\n\x05Route\x12\x16\n\x0evariation_name\x18\x01 \x01(\t\x12\x0e\n\x06weight\x18\x02 \x01(\r\x12\x0e\n\x06shadow\x18\x03 \x01(\x08\x12\x10\n\x08model_id\x18\x04 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\x12\x13\n\x0b\x61udience_id\x18\x06 \x01(\t\x12\x1a\n\x12request_timeout_ms\x18\x07 \x01(\x05*\x9e\x02\n\x11UnaryOperatorType\x12\x1f\n\x1bUNARY_OPERATOR_TYPE_INVALID\x10\x00\x12#\n\x1fUNARY_OPERATOR_TYPE_EXACT_MATCH\x10\x01\x12(\n$UNARY_OPERATOR_TYPE_SAFE_REGEX_MATCH\x10\x02\x12%\n!UNARY_OPERATOR_TYPE_PRESENT_MATCH\x10\x03\x12$\n UNARY_OPERATOR_TYPE_PREFIX_MATCH\x10\x04\x12$\n UNARY_OPERATOR_TYPE_SUFFIX_MATCH\x10\x05\x12&\n\"UNARY_OPERATOR_TYPE_CONTAINS_MATCH\x10\x06*\\\n\x12\x42inaryOperatorType\x12 \n\x1c\x42INARY_OPERATOR_TYPE_INVALID\x10\x00\x12$\n BINARY_OPERATOR_TYPE_RANGE_MATCH\x10\x01\x42\x42\n\x14\x63om.qwak.audience.v1B\rAudienceProtoP\x01Z\x19qwak/audience/v1;audienceb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.audience.v1.audience_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\024com.qwak.audience.v1B\rAudienceProtoP\001Z\031qwak/audience/v1;audience'
+  _AUDIENCE_ENVIRONMENTROUTESENTRY._options = None
+  _AUDIENCE_ENVIRONMENTROUTESENTRY._serialized_options = b'8\001'
+  _globals['_UNARYOPERATORTYPE']._serialized_start=1356
+  _globals['_UNARYOPERATORTYPE']._serialized_end=1642
+  _globals['_BINARYOPERATORTYPE']._serialized_start=1644
+  _globals['_BINARYOPERATORTYPE']._serialized_end=1736
+  _globals['_AUDIENCEENTRY']._serialized_start=86
+  _globals['_AUDIENCEENTRY']._serialized_end=159
+  _globals['_AUDIENCE']._serialized_start=162
+  _globals['_AUDIENCE']._serialized_end=611
+  _globals['_AUDIENCE_ENVIRONMENTROUTESENTRY']._serialized_start=510
+  _globals['_AUDIENCE_ENVIRONMENTROUTESENTRY']._serialized_end=611
+  _globals['_ENVIRONMENTAUDIENCEROUTES']._serialized_start=613
+  _globals['_ENVIRONMENTAUDIENCEROUTES']._serialized_end=681
+  _globals['_AUDIENCEROUTESENTRY']._serialized_start=683
+  _globals['_AUDIENCEROUTESENTRY']._serialized_end=804
+  _globals['_CONDITION']._serialized_start=807
+  _globals['_CONDITION']._serialized_end=984
+  _globals['_UNARYCONDITION']._serialized_start=986
+  _globals['_UNARYCONDITION']._serialized_end=1074
+  _globals['_BINARYCONDITION']._serialized_start=1076
+  _globals['_BINARYCONDITION']._serialized_end=1196
+  _globals['_ROUTE']._serialized_start=1199
+  _globals['_ROUTE']._serialized_end=1353
+# @@protoc_insertion_point(module_scope)

_proto/qwak/auto_scaling/v1/auto_scaling_pb2.py

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.auto_scaling/v1/auto_scaling.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'qwak/auto_scaling/v1/auto_scaling.proto\x12\x14qwak.auto_scaling.v1\"\xaf\x01\n\x11\x41utoScalingConfig\x12\x19\n\x11min_replica_count\x18\x01 \x01(\x05\x12\x19\n\x11max_replica_count\x18\x02 \x01(\x05\x12\x18\n\x10polling_interval\x18\x03 \x01(\x05\x12\x18\n\x10\x63ool_down_period\x18\x04 \x01(\x05\x12\x30\n\x08triggers\x18\x05 \x01(\x0b\x32\x1e.qwak.auto_scaling.v1.Triggers\";\n\x08Triggers\x12/\n\x08triggers\x18\x01 \x03(\x0b\x32\x1d.qwak.auto_scaling.v1.Trigger\"`\n\x07Trigger\x12\x45\n\x12prometheus_trigger\x18\x01 \x01(\x0b\x32\'.qwak.auto_scaling.v1.PrometheusTriggerH\x00\x42\x0e\n\x0ctrigger_type\"[\n\x11PrometheusTrigger\x12\x33\n\nquery_spec\x18\x01 \x01(\x0b\x32\x1f.qwak.auto_scaling.v1.QuerySpec\x12\x11\n\tthreshold\x18\x02 \x01(\x05\"\xac\x01\n\tQuerySpec\x12\x35\n\x0bmetric_type\x18\x01 \x01(\x0e\x32 .qwak.auto_scaling.v1.MetricType\x12?\n\x10\x61ggregation_type\x18\x02 \x01(\x0e\x32%.qwak.auto_scaling.v1.AggregationType\x12\x13\n\x0btime_period\x18\x03 \x01(\x05\x12\x12\n\nerror_code\x18\x04 \x01(\t*\xba\x01\n\nMetricType\x12\x19\n\x15METRIC_TYPE_NOT_VALID\x10\x00\x12\x13\n\x0fMETRIC_TYPE_CPU\x10\x01\x12\x16\n\x12METRIC_TYPE_MEMORY\x10\x02\x12\x17\n\x13METRIC_TYPE_LATENCY\x10\x03\x12\x13\n\x0fMETRIC_TYPE_GPU\x10\x04\x12\x1a\n\x16METRIC_TYPE_ERROR_RATE\x10\x05\x12\x1a\n\x16METRIC_TYPE_THROUGHPUT\x10\x06*\x85\x02\n\x0f\x41ggregationType\x12\x1e\n\x1a\x41GGREGATION_TYPE_NOT_VALID\x10\x00\x12\x18\n\x14\x41GGREGATION_TYPE_MIN\x10\x01\x12\x18\n\x14\x41GGREGATION_TYPE_MAX\x10\x02\x12\x1c\n\x18\x41GGREGATION_TYPE_AVERAGE\x10\x03\x12\x18\n\x14\x41GGREGATION_TYPE_SUM\x10\x04\x12\x18\n\x14\x41GGREGATION_TYPE_P50\x10\x05\x12\x18\n\x14\x41GGREGATION_TYPE_P90\x10\x06\x12\x18\n\x14\x41GGREGATION_TYPE_P95\x10\x07\x12\x18\n\x14\x41GGREGATION_TYPE_P99\x10\x08\x42K\n%com.qwak.ai.management.deployment.apiP\x01Z qwak/auto_scaling/v1;autoscalingb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.auto_scaling.v1.auto_scaling_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.management.deployment.apiP\001Z qwak/auto_scaling/v1;autoscaling'
+  _globals['_METRICTYPE']._serialized_start=671
+  _globals['_METRICTYPE']._serialized_end=857
+  _globals['_AGGREGATIONTYPE']._serialized_start=860
+  _globals['_AGGREGATIONTYPE']._serialized_end=1121
+  _globals['_AUTOSCALINGCONFIG']._serialized_start=66
+  _globals['_AUTOSCALINGCONFIG']._serialized_end=241
+  _globals['_TRIGGERS']._serialized_start=243
+  _globals['_TRIGGERS']._serialized_end=302
+  _globals['_TRIGGER']._serialized_start=304
+  _globals['_TRIGGER']._serialized_end=400
+  _globals['_PROMETHEUSTRIGGER']._serialized_start=402
+  _globals['_PROMETHEUSTRIGGER']._serialized_end=493
+  _globals['_QUERYSPEC']._serialized_start=496
+  _globals['_QUERYSPEC']._serialized_end=668
+# @@protoc_insertion_point(module_scope)

_proto/qwak/auto_scaling/v1/auto_scaling_service_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.auto_scaling/v1/auto_scaling_service.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from frogml._proto.qwak.auto_scaling.v1 import auto_scaling_pb2 as qwak_dot_auto__scaling_dot_v1_dot_auto__scaling__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n/qwak/auto_scaling/v1/auto_scaling_service.proto\x12\x14qwak.auto_scaling.v1\x1a\'qwak/auto_scaling/v1/auto_scaling.proto\"\x8a\x01\n\x18\x41ttachAutoScalingRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x16\n\x0evariation_name\x18\x02 \x01(\t\x12\x44\n\x13\x61uto_scaling_config\x18\x03 \x01(\x0b\x32\'.qwak.auto_scaling.v1.AutoScalingConfig\"4\n\x19\x41ttachAutoScalingResponse\x12\x17\n\x0f\x61uto_scaling_id\x18\x01 \x01(\t2\x8a\x01\n\x12\x41utoScalingService\x12t\n\x11\x41ttachAutoScaling\x12..qwak.auto_scaling.v1.AttachAutoScalingRequest\x1a/.qwak.auto_scaling.v1.AttachAutoScalingResponseBK\n%com.qwak.ai.management.deployment.apiP\x01Z qwak/auto_scaling/v1;autoscalingb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.auto_scaling.v1.auto_scaling_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.management.deployment.apiP\001Z qwak/auto_scaling/v1;autoscaling'
+  _globals['_ATTACHAUTOSCALINGREQUEST']._serialized_start=115
+  _globals['_ATTACHAUTOSCALINGREQUEST']._serialized_end=253
+  _globals['_ATTACHAUTOSCALINGRESPONSE']._serialized_start=255
+  _globals['_ATTACHAUTOSCALINGRESPONSE']._serialized_end=307
+  _globals['_AUTOSCALINGSERVICE']._serialized_start=310
+  _globals['_AUTOSCALINGSERVICE']._serialized_end=448
+# @@protoc_insertion_point(module_scope)

_proto/qwak/auto_scaling/v1/auto_scaling_service_pb2.pyi

@@ -0,0 +1,55 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import frogml._proto.qwak.auto_scaling.v1.auto_scaling_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class AttachAutoScalingRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    VARIATION_NAME_FIELD_NUMBER: builtins.int
+    AUTO_SCALING_CONFIG_FIELD_NUMBER: builtins.int
+    model_id: builtins.str
+    """The ID of the model to scale"""
+    variation_name: builtins.str
+    """Deployment variation name"""
+    @property
+    def auto_scaling_config(self) -> frogml._proto.qwak.auto_scaling.v1.auto_scaling_pb2.AutoScalingConfig:
+        """Auto scaling"""
+    def __init__(
+        self,
+        *,
+        model_id: builtins.str = ...,
+        variation_name: builtins.str = ...,
+        auto_scaling_config: frogml._proto.qwak.auto_scaling.v1.auto_scaling_pb2.AutoScalingConfig | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["auto_scaling_config", b"auto_scaling_config"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["auto_scaling_config", b"auto_scaling_config", "model_id", b"model_id", "variation_name", b"variation_name"]) -> None: ...
+
+global___AttachAutoScalingRequest = AttachAutoScalingRequest
+
+class AttachAutoScalingResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    AUTO_SCALING_ID_FIELD_NUMBER: builtins.int
+    auto_scaling_id: builtins.str
+    def __init__(
+        self,
+        *,
+        auto_scaling_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["auto_scaling_id", b"auto_scaling_id"]) -> None: ...
+
+global___AttachAutoScalingResponse = AttachAutoScalingResponse

_proto/qwak/auto_scaling/v1/auto_scaling_service_pb2_grpc.py

@@ -0,0 +1,67 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml._proto.qwak.auto_scaling.v1 import auto_scaling_service_pb2 as qwak_dot_auto__scaling_dot_v1_dot_auto__scaling__service__pb2
+
+
+class AutoScalingServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.AttachAutoScaling = channel.unary_unary(
+                '/qwak.auto_scaling.v1.AutoScalingService/AttachAutoScaling',
+                request_serializer=qwak_dot_auto__scaling_dot_v1_dot_auto__scaling__service__pb2.AttachAutoScalingRequest.SerializeToString,
+                response_deserializer=qwak_dot_auto__scaling_dot_v1_dot_auto__scaling__service__pb2.AttachAutoScalingResponse.FromString,
+                )
+
+
+class AutoScalingServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def AttachAutoScaling(self, request, context):
+        """Attach auto scaling
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_AutoScalingServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'AttachAutoScaling': grpc.unary_unary_rpc_method_handler(
+                    servicer.AttachAutoScaling,
+                    request_deserializer=qwak_dot_auto__scaling_dot_v1_dot_auto__scaling__service__pb2.AttachAutoScalingRequest.FromString,
+                    response_serializer=qwak_dot_auto__scaling_dot_v1_dot_auto__scaling__service__pb2.AttachAutoScalingResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.auto_scaling.v1.AutoScalingService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class AutoScalingService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def AttachAutoScaling(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.auto_scaling.v1.AutoScalingService/AttachAutoScaling',
+            qwak_dot_auto__scaling_dot_v1_dot_auto__scaling__service__pb2.AttachAutoScalingRequest.SerializeToString,
+            qwak_dot_auto__scaling_dot_v1_dot_auto__scaling__service__pb2.AttachAutoScalingResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_proto/qwak/automation/v1/action_pb2.py

@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.automation/v1/action.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
+from frogml._proto.qwak.automation.v1 import common_pb2 as qwak_dot_automation_dot_v1_dot_common__pb2
+from frogml._proto.qwak.automation.v1 import auto_scaling_pb2 as qwak_dot_automation_dot_v1_dot_auto__scaling__pb2
+from frogml._proto.qwak.batch_job.v1 import batch_job_service_pb2 as qwak_dot_batch__job_dot_v1_dot_batch__job__service__pb2
+from frogml._proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fqwak/automation/v1/action.proto\x12\x12qwak.automation.v1\x1a\x1egoogle/protobuf/wrappers.proto\x1a\x1fqwak/automation/v1/common.proto\x1a%qwak/automation/v1/auto_scaling.proto\x1a)qwak/batch_job/v1/batch_job_service.proto\x1a/qwak/user_application/common/v0/resources.proto\"\x99\x01\n\x06\x41\x63tion\x12@\n\x0c\x62uild_deploy\x18\x01 \x01(\x0b\x32(.qwak.automation.v1.BuildAndDeployActionH\x00\x12\x43\n\x0f\x62\x61tch_execution\x18\x02 \x01(\x0b\x32(.qwak.automation.v1.BatchExecutionActionH\x00\x42\x08\n\x06\x61\x63tion\"Q\n\x14\x42\x61tchExecutionAction\x12\x39\n\x11\x65xecute_batch_job\x18\x01 \x01(\x0b\x32\x1e.qwak.batchjob.BatchJobRequest\"\xcd\x01\n\x14\x42uildAndDeployAction\x12\x31\n\nbuild_spec\x18\x01 \x01(\x0b\x32\x1d.qwak.automation.v1.BuildSpec\x12\x45\n\x14\x64\x65ployment_condition\x18\x02 \x01(\x0b\x32\'.qwak.automation.v1.DeploymentCondition\x12;\n\x0f\x64\x65ployment_spec\x18\x03 \x01(\x0b\x32\".qwak.automation.v1.DeploymentSpec\"\xb7\x01\n\x13\x44\x65ploymentCondition\x12@\n\x0c\x62uild_metric\x18\x01 \x01(\x0b\x32(.qwak.automation.v1.BuildMetricConditionH\x00\x12Q\n\x15\x64\x65ployed_build_metric\x18\x02 \x01(\x0b\x32\x30.qwak.automation.v1.DeployedBuildMetricConditionH\x00\x42\x0b\n\tcondition\"\x89\x01\n\x14\x42uildMetricCondition\x12\x13\n\x0bmetric_name\x18\x01 \x01(\t\x12I\n\x13threshold_direction\x18\x02 \x01(\x0e\x32,.qwak.automation.v1.MetricThresholdDirection\x12\x11\n\tthreshold\x18\x03 \x01(\t\"\x91\x01\n\x1c\x44\x65ployedBuildMetricCondition\x12\x13\n\x0bmetric_name\x18\x01 \x01(\t\x12I\n\x13threshold_direction\x18\x02 \x01(\x0e\x32,.qwak.automation.v1.MetricThresholdDirection\x12\x11\n\tvariation\x18\x03 \x01(\t\"\xdf\x01\n\x0e\x44\x65ploymentSpec\x12;\n\x0f\x64\x65ployment_size\x18\x01 \x01(\x0b\x32\".qwak.automation.v1.DeploymentSize\x12G\n\x10\x61\x64vanced_options\x18\x02 \x01(\x0b\x32-.qwak.automation.v1.AdvancedDeploymentOptions\x12\x1f\n\x17selected_variation_name\x18\x03 \x01(\t\x12\x14\n\x0c\x65nvironments\x18\x04 \x03(\t\x12\x10\n\x08\x65nv_vars\x18\x05 \x03(\t\"\xc4\x02\n\x19\x41\x64vancedDeploymentOptions\x12%\n\x1dnumber_of_http_server_workers\x18\x01 \x01(\x05\x12\x1f\n\x17http_request_timeout_ms\x18\x02 \x01(\x05\x12\x13\n\x0b\x64\x61\x65mon_mode\x18\x03 \x01(\x08\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x04 \x01(\t\x12\x42\n\x13\x61uto_scaling_config\x18\x05 \x01(\x0b\x32%.qwak.automation.v1.AutoScalingConfig\x12\x16\n\x0emax_batch_size\x18\x06 \x01(\x05\x12(\n deployment_process_timeout_limit\x18\x07 \x01(\x05\x12\'\n\x1fservice_account_key_secret_name\x18\x08 \x01(\t\"\xad\x02\n\x0e\x44\x65ploymentSize\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0f\n\x03\x63pu\x18\x02 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x03 \x01(\x05\x42\x02\x18\x01\x12\x38\n\x0cmemory_units\x18\x04 \x01(\x0e\x32\x1e.qwak.automation.v1.MemoryUnitB\x02\x18\x01\x12;\n\rgpu_resources\x18\x05 \x01(\x0b\x32 .qwak.automation.v1.GpuResourcesB\x02\x18\x01\x12`\n\x1c\x63lient_pod_compute_resources\x18\x06 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResources\"\xc0\x04\n\tBuildSpec\x12<\n\x10git_model_source\x18\x01 \x01(\x0b\x32\".qwak.automation.v1.GitModelSource\x12\x41\n\nparameters\x18\x02 \x03(\x0b\x32-.qwak.automation.v1.BuildSpec.ParametersEntry\x12\x0c\n\x04tags\x18\x03 \x03(\t\x12\x12\n\nbase_image\x18\x04 \x01(\t\x12\x18\n\x10\x61ssumed_iam_role\x18\x05 \x01(\t\x12/\n\x08resource\x18\x06 \x01(\x0b\x32\x1d.qwak.automation.v1.Resources\x12\x10\n\x08main_dir\x18\x07 \x01(\t\x12\x1c\n\x14\x64\x65pendency_file_path\x18\x08 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\t \x03(\t\x12\x16\n\x0egpu_compatible\x18\n \x01(\x08\x12.\n\npush_image\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.BoolValue\x12;\n\x0fpurchase_option\x18\x0c \x01(\x0e\x32\".qwak.automation.v1.PurchaseOption\x12\'\n\x1fservice_account_key_secret_name\x18\r \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x0e \x01(\x05\x1a\x31\n\x0fParametersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xf9\x01\n\tResources\x12=\n\rcpu_resources\x18\x01 \x01(\x0b\x32 .qwak.automation.v1.CpuResourcesB\x02\x18\x01H\x00\x12=\n\rgpu_resources\x18\x02 \x01(\x0b\x32 .qwak.automation.v1.GpuResourcesB\x02\x18\x01H\x00\x12\x62\n\x1c\x63lient_pod_compute_resources\x18\x03 \x01(\x0b\x32:.qwak.user_application.common.v0.ClientPodComputeResourcesH\x00\x42\n\n\x08resource\"h\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12\x34\n\x0cmemory_units\x18\x03 \x01(\x0e\x32\x1e.qwak.automation.v1.MemoryUnit\"Q\n\x0cGpuResources\x12-\n\x08gpu_type\x18\x01 \x01(\x0e\x32\x1b.qwak.automation.v1.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"w\n\x0eGitModelSource\x12\x0f\n\x07git_uri\x18\x01 \x01(\t\x12#\n\x1bgit_credentials_secret_name\x18\x02 \x01(\t\x12\x12\n\ngit_branch\x18\x03 \x01(\t\x12\x1b\n\x13git_ssh_secret_name\x18\x04 \x01(\t*w\n\x07GpuType\x12\x0b\n\x07INVALID\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06*+\n\nMemoryUnit\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*)\n\x0ePurchaseOption\x12\r\n\tON_DEMAND\x10\x00\x12\x08\n\x04SPOT\x10\x01\x42!\n\x1d\x63om.qwak.ai.automation.api.v1P\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.automation.v1.action_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\035com.qwak.ai.automation.api.v1P\001'
+  _DEPLOYMENTSIZE.fields_by_name['cpu']._options = None
+  _DEPLOYMENTSIZE.fields_by_name['cpu']._serialized_options = b'\030\001'
+  _DEPLOYMENTSIZE.fields_by_name['memory_amount']._options = None
+  _DEPLOYMENTSIZE.fields_by_name['memory_amount']._serialized_options = b'\030\001'
+  _DEPLOYMENTSIZE.fields_by_name['memory_units']._options = None
+  _DEPLOYMENTSIZE.fields_by_name['memory_units']._serialized_options = b'\030\001'
+  _DEPLOYMENTSIZE.fields_by_name['gpu_resources']._options = None
+  _DEPLOYMENTSIZE.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
+  _BUILDSPEC_PARAMETERSENTRY._options = None
+  _BUILDSPEC_PARAMETERSENTRY._serialized_options = b'8\001'
+  _RESOURCES.fields_by_name['cpu_resources']._options = None
+  _RESOURCES.fields_by_name['cpu_resources']._serialized_options = b'\030\001'
+  _RESOURCES.fields_by_name['gpu_resources']._options = None
+  _RESOURCES.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
+  _globals['_GPUTYPE']._serialized_start=3170
+  _globals['_GPUTYPE']._serialized_end=3289
+  _globals['_MEMORYUNIT']._serialized_start=3291
+  _globals['_MEMORYUNIT']._serialized_end=3334
+  _globals['_PURCHASEOPTION']._serialized_start=3336
+  _globals['_PURCHASEOPTION']._serialized_end=3377
+  _globals['_ACTION']._serialized_start=252
+  _globals['_ACTION']._serialized_end=405
+  _globals['_BATCHEXECUTIONACTION']._serialized_start=407
+  _globals['_BATCHEXECUTIONACTION']._serialized_end=488
+  _globals['_BUILDANDDEPLOYACTION']._serialized_start=491
+  _globals['_BUILDANDDEPLOYACTION']._serialized_end=696
+  _globals['_DEPLOYMENTCONDITION']._serialized_start=699
+  _globals['_DEPLOYMENTCONDITION']._serialized_end=882
+  _globals['_BUILDMETRICCONDITION']._serialized_start=885
+  _globals['_BUILDMETRICCONDITION']._serialized_end=1022
+  _globals['_DEPLOYEDBUILDMETRICCONDITION']._serialized_start=1025
+  _globals['_DEPLOYEDBUILDMETRICCONDITION']._serialized_end=1170
+  _globals['_DEPLOYMENTSPEC']._serialized_start=1173
+  _globals['_DEPLOYMENTSPEC']._serialized_end=1396
+  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_start=1399
+  _globals['_ADVANCEDDEPLOYMENTOPTIONS']._serialized_end=1723
+  _globals['_DEPLOYMENTSIZE']._serialized_start=1726
+  _globals['_DEPLOYMENTSIZE']._serialized_end=2027
+  _globals['_BUILDSPEC']._serialized_start=2030
+  _globals['_BUILDSPEC']._serialized_end=2606
+  _globals['_BUILDSPEC_PARAMETERSENTRY']._serialized_start=2557
+  _globals['_BUILDSPEC_PARAMETERSENTRY']._serialized_end=2606
+  _globals['_RESOURCES']._serialized_start=2609
+  _globals['_RESOURCES']._serialized_end=2858
+  _globals['_CPURESOURCES']._serialized_start=2860
+  _globals['_CPURESOURCES']._serialized_end=2964
+  _globals['_GPURESOURCES']._serialized_start=2966
+  _globals['_GPURESOURCES']._serialized_end=3047
+  _globals['_GITMODELSOURCE']._serialized_start=3049
+  _globals['_GITMODELSOURCE']._serialized_end=3168
+# @@protoc_insertion_point(module_scope)