frogml-core 0.0.1__py3-none-any.whl

_frogml_proto/qwak/audience/v1/audience_pb2.pyi

@@ -0,0 +1,313 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _UnaryOperatorType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _UnaryOperatorTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_UnaryOperatorType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    UNARY_OPERATOR_TYPE_INVALID: _UnaryOperatorType.ValueType  # 0
+    UNARY_OPERATOR_TYPE_EXACT_MATCH: _UnaryOperatorType.ValueType  # 1
+    UNARY_OPERATOR_TYPE_SAFE_REGEX_MATCH: _UnaryOperatorType.ValueType  # 2
+    UNARY_OPERATOR_TYPE_PRESENT_MATCH: _UnaryOperatorType.ValueType  # 3
+    UNARY_OPERATOR_TYPE_PREFIX_MATCH: _UnaryOperatorType.ValueType  # 4
+    UNARY_OPERATOR_TYPE_SUFFIX_MATCH: _UnaryOperatorType.ValueType  # 5
+    UNARY_OPERATOR_TYPE_CONTAINS_MATCH: _UnaryOperatorType.ValueType  # 6
+
+class UnaryOperatorType(_UnaryOperatorType, metaclass=_UnaryOperatorTypeEnumTypeWrapper):
+    """Unary operators available types"""
+
+UNARY_OPERATOR_TYPE_INVALID: UnaryOperatorType.ValueType  # 0
+UNARY_OPERATOR_TYPE_EXACT_MATCH: UnaryOperatorType.ValueType  # 1
+UNARY_OPERATOR_TYPE_SAFE_REGEX_MATCH: UnaryOperatorType.ValueType  # 2
+UNARY_OPERATOR_TYPE_PRESENT_MATCH: UnaryOperatorType.ValueType  # 3
+UNARY_OPERATOR_TYPE_PREFIX_MATCH: UnaryOperatorType.ValueType  # 4
+UNARY_OPERATOR_TYPE_SUFFIX_MATCH: UnaryOperatorType.ValueType  # 5
+UNARY_OPERATOR_TYPE_CONTAINS_MATCH: UnaryOperatorType.ValueType  # 6
+global___UnaryOperatorType = UnaryOperatorType
+
+class _BinaryOperatorType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _BinaryOperatorTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BinaryOperatorType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    BINARY_OPERATOR_TYPE_INVALID: _BinaryOperatorType.ValueType  # 0
+    BINARY_OPERATOR_TYPE_RANGE_MATCH: _BinaryOperatorType.ValueType  # 1
+
+class BinaryOperatorType(_BinaryOperatorType, metaclass=_BinaryOperatorTypeEnumTypeWrapper):
+    """Binary operators available types"""
+
+BINARY_OPERATOR_TYPE_INVALID: BinaryOperatorType.ValueType  # 0
+BINARY_OPERATOR_TYPE_RANGE_MATCH: BinaryOperatorType.ValueType  # 1
+global___BinaryOperatorType = BinaryOperatorType
+
+class AudienceEntry(google.protobuf.message.Message):
+    """Describe audience entry with ID its equivalent ID for CRUD operation"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ID_FIELD_NUMBER: builtins.int
+    AUDIENCE_FIELD_NUMBER: builtins.int
+    id: builtins.str
+    @property
+    def audience(self) -> global___Audience: ...
+    def __init__(
+        self,
+        *,
+        id: builtins.str = ...,
+        audience: global___Audience | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["audience", b"audience"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["audience", b"audience", "id", b"id"]) -> None: ...
+
+global___AudienceEntry = AudienceEntry
+
+class Audience(google.protobuf.message.Message):
+    """Specifies Condition based route configuration, If all conditions evaluated to true prediction request will route to routes config.
+    For example: Condition check if prediction request include an header key, value which value is exactly "hello" than route
+                 configuration will take effect.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class EnvironmentRoutesEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        @property
+        def value(self) -> global___EnvironmentAudienceRoutes: ...
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: global___EnvironmentAudienceRoutes | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    NAME_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    CONDITIONS_FIELD_NUMBER: builtins.int
+    ROUTES_FIELD_NUMBER: builtins.int
+    CREATED_FIELD_NUMBER: builtins.int
+    CREATED_BY_FIELD_NUMBER: builtins.int
+    MODIFIED_FIELD_NUMBER: builtins.int
+    MODIFIED_BY_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ROUTES_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Audience name"""
+    description: builtins.str
+    """Audience description"""
+    @property
+    def conditions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Condition]:
+        """All conditions should evaluated to true"""
+    @property
+    def routes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Route]:
+        """If all evaluated to true the prediction will the routes"""
+    @property
+    def created(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Create date"""
+    created_by: builtins.str
+    """Created by"""
+    @property
+    def modified(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Update date"""
+    modified_by: builtins.str
+    """Updated by"""
+    @property
+    def environment_routes(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___EnvironmentAudienceRoutes]:
+        """Routes by environment id"""
+    def __init__(
+        self,
+        *,
+        name: builtins.str = ...,
+        description: builtins.str = ...,
+        conditions: collections.abc.Iterable[global___Condition] | None = ...,
+        routes: collections.abc.Iterable[global___Route] | None = ...,
+        created: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        created_by: builtins.str = ...,
+        modified: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        modified_by: builtins.str = ...,
+        environment_routes: collections.abc.Mapping[builtins.str, global___EnvironmentAudienceRoutes] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["created", b"created", "modified", b"modified"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["conditions", b"conditions", "created", b"created", "created_by", b"created_by", "description", b"description", "environment_routes", b"environment_routes", "modified", b"modified", "modified_by", b"modified_by", "name", b"name", "routes", b"routes"]) -> None: ...
+
+global___Audience = Audience
+
+class EnvironmentAudienceRoutes(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ROUTES_FIELD_NUMBER: builtins.int
+    @property
+    def routes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Route]: ...
+    def __init__(
+        self,
+        *,
+        routes: collections.abc.Iterable[global___Route] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["routes", b"routes"]) -> None: ...
+
+global___EnvironmentAudienceRoutes = EnvironmentAudienceRoutes
+
+class AudienceRoutesEntry(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    AUDIENCE_ID_FIELD_NUMBER: builtins.int
+    AUDIENCE_NAME_FIELD_NUMBER: builtins.int
+    ROUTES_FIELD_NUMBER: builtins.int
+    ORDER_FIELD_NUMBER: builtins.int
+    audience_id: builtins.str
+    audience_name: builtins.str
+    @property
+    def routes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Route]: ...
+    order: builtins.int
+    def __init__(
+        self,
+        *,
+        audience_id: builtins.str = ...,
+        audience_name: builtins.str = ...,
+        routes: collections.abc.Iterable[global___Route] | None = ...,
+        order: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["audience_id", b"audience_id", "audience_name", b"audience_name", "order", b"order", "routes", b"routes"]) -> None: ...
+
+global___AudienceRoutesEntry = AudienceRoutesEntry
+
+class Condition(google.protobuf.message.Message):
+    """Condition which applied on prediction request header"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    KEY_FIELD_NUMBER: builtins.int
+    BINARY_CONDITION_FIELD_NUMBER: builtins.int
+    UNARY_CONDITION_FIELD_NUMBER: builtins.int
+    INVERT_FIELD_NUMBER: builtins.int
+    key: builtins.str
+    """key to extract value for evaluation"""
+    @property
+    def binary_condition(self) -> global___BinaryCondition: ...
+    @property
+    def unary_condition(self) -> global___UnaryCondition: ...
+    invert: builtins.bool
+    """Condition evaluation should be inverted (For example False will be True)"""
+    def __init__(
+        self,
+        *,
+        key: builtins.str = ...,
+        binary_condition: global___BinaryCondition | None = ...,
+        unary_condition: global___UnaryCondition | None = ...,
+        invert: builtins.bool = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["binary_condition", b"binary_condition", "condition", b"condition", "unary_condition", b"unary_condition"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["binary_condition", b"binary_condition", "condition", b"condition", "invert", b"invert", "key", b"key", "unary_condition", b"unary_condition"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["condition", b"condition"]) -> typing_extensions.Literal["binary_condition", "unary_condition"] | None: ...
+
+global___Condition = Condition
+
+class UnaryCondition(google.protobuf.message.Message):
+    """Unary condition definition
+    For example: unary_operator UnaryOperator.UNARY_OPERATOR_EXACT_MATCH with operand hello,
+                 will check that header value will be exactly hello.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    OPERATOR_FIELD_NUMBER: builtins.int
+    OPERAND_FIELD_NUMBER: builtins.int
+    operator: global___UnaryOperatorType.ValueType
+    operand: builtins.str
+    def __init__(
+        self,
+        *,
+        operator: global___UnaryOperatorType.ValueType = ...,
+        operand: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["operand", b"operand", "operator", b"operator"]) -> None: ...
+
+global___UnaryCondition = UnaryCondition
+
+class BinaryCondition(google.protobuf.message.Message):
+    """Binary condition definition
+    For example: unary_operator BinaryOperatorType.BINARY_OPERATOR_TYPE_RANGE_MATCH with first_operand 1,
+                 and second operand 3 will check that header value will be in range of 1-3 included)
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    OPERATOR_FIELD_NUMBER: builtins.int
+    FIRST_OPERAND_FIELD_NUMBER: builtins.int
+    SECOND_OPERAND_FIELD_NUMBER: builtins.int
+    operator: global___BinaryOperatorType.ValueType
+    first_operand: builtins.str
+    second_operand: builtins.str
+    def __init__(
+        self,
+        *,
+        operator: global___BinaryOperatorType.ValueType = ...,
+        first_operand: builtins.str = ...,
+        second_operand: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["first_operand", b"first_operand", "operator", b"operator", "second_operand", b"second_operand"]) -> None: ...
+
+global___BinaryCondition = BinaryCondition
+
+class Route(google.protobuf.message.Message):
+    """Route configuration to apply on specific variation if all conditions evaluated to true."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VARIATION_NAME_FIELD_NUMBER: builtins.int
+    WEIGHT_FIELD_NUMBER: builtins.int
+    SHADOW_FIELD_NUMBER: builtins.int
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    AUDIENCE_ID_FIELD_NUMBER: builtins.int
+    REQUEST_TIMEOUT_MS_FIELD_NUMBER: builtins.int
+    variation_name: builtins.str
+    """Deployment variation name"""
+    weight: builtins.int
+    """Weight of variation on splitting between other variations (Not percentage)"""
+    shadow: builtins.bool
+    """Shadow all traffic to this variation (All response from this variation will be ignored)"""
+    model_id: builtins.str
+    environment_id: builtins.str
+    audience_id: builtins.str
+    request_timeout_ms: builtins.int
+    """Timeout of the request in ms"""
+    def __init__(
+        self,
+        *,
+        variation_name: builtins.str = ...,
+        weight: builtins.int = ...,
+        shadow: builtins.bool = ...,
+        model_id: builtins.str = ...,
+        environment_id: builtins.str = ...,
+        audience_id: builtins.str = ...,
+        request_timeout_ms: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["audience_id", b"audience_id", "environment_id", b"environment_id", "model_id", b"model_id", "request_timeout_ms", b"request_timeout_ms", "shadow", b"shadow", "variation_name", b"variation_name", "weight", b"weight"]) -> None: ...
+
+global___Route = Route

_frogml_proto/qwak/audience/v1/audience_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

_frogml_proto/qwak/auto_scaling/v1/auto_scaling_pb2.py

@@ -0,0 +1,100 @@
+# -*- 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.internal import enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@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')
+
+_METRICTYPE = DESCRIPTOR.enum_types_by_name['MetricType']
+MetricType = enum_type_wrapper.EnumTypeWrapper(_METRICTYPE)
+_AGGREGATIONTYPE = DESCRIPTOR.enum_types_by_name['AggregationType']
+AggregationType = enum_type_wrapper.EnumTypeWrapper(_AGGREGATIONTYPE)
+METRIC_TYPE_NOT_VALID = 0
+METRIC_TYPE_CPU = 1
+METRIC_TYPE_MEMORY = 2
+METRIC_TYPE_LATENCY = 3
+METRIC_TYPE_GPU = 4
+METRIC_TYPE_ERROR_RATE = 5
+METRIC_TYPE_THROUGHPUT = 6
+AGGREGATION_TYPE_NOT_VALID = 0
+AGGREGATION_TYPE_MIN = 1
+AGGREGATION_TYPE_MAX = 2
+AGGREGATION_TYPE_AVERAGE = 3
+AGGREGATION_TYPE_SUM = 4
+AGGREGATION_TYPE_P50 = 5
+AGGREGATION_TYPE_P90 = 6
+AGGREGATION_TYPE_P95 = 7
+AGGREGATION_TYPE_P99 = 8
+
+
+_AUTOSCALINGCONFIG = DESCRIPTOR.message_types_by_name['AutoScalingConfig']
+_TRIGGERS = DESCRIPTOR.message_types_by_name['Triggers']
+_TRIGGER = DESCRIPTOR.message_types_by_name['Trigger']
+_PROMETHEUSTRIGGER = DESCRIPTOR.message_types_by_name['PrometheusTrigger']
+_QUERYSPEC = DESCRIPTOR.message_types_by_name['QuerySpec']
+AutoScalingConfig = _reflection.GeneratedProtocolMessageType('AutoScalingConfig', (_message.Message,), {
+  'DESCRIPTOR' : _AUTOSCALINGCONFIG,
+  '__module__' : 'qwak.auto_scaling.v1.auto_scaling_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.auto_scaling.v1.AutoScalingConfig)
+  })
+_sym_db.RegisterMessage(AutoScalingConfig)
+
+Triggers = _reflection.GeneratedProtocolMessageType('Triggers', (_message.Message,), {
+  'DESCRIPTOR' : _TRIGGERS,
+  '__module__' : 'qwak.auto_scaling.v1.auto_scaling_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.auto_scaling.v1.Triggers)
+  })
+_sym_db.RegisterMessage(Triggers)
+
+Trigger = _reflection.GeneratedProtocolMessageType('Trigger', (_message.Message,), {
+  'DESCRIPTOR' : _TRIGGER,
+  '__module__' : 'qwak.auto_scaling.v1.auto_scaling_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.auto_scaling.v1.Trigger)
+  })
+_sym_db.RegisterMessage(Trigger)
+
+PrometheusTrigger = _reflection.GeneratedProtocolMessageType('PrometheusTrigger', (_message.Message,), {
+  'DESCRIPTOR' : _PROMETHEUSTRIGGER,
+  '__module__' : 'qwak.auto_scaling.v1.auto_scaling_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.auto_scaling.v1.PrometheusTrigger)
+  })
+_sym_db.RegisterMessage(PrometheusTrigger)
+
+QuerySpec = _reflection.GeneratedProtocolMessageType('QuerySpec', (_message.Message,), {
+  'DESCRIPTOR' : _QUERYSPEC,
+  '__module__' : 'qwak.auto_scaling.v1.auto_scaling_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.auto_scaling.v1.QuerySpec)
+  })
+_sym_db.RegisterMessage(QuerySpec)
+
+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'
+  _METRICTYPE._serialized_start=671
+  _METRICTYPE._serialized_end=857
+  _AGGREGATIONTYPE._serialized_start=860
+  _AGGREGATIONTYPE._serialized_end=1121
+  _AUTOSCALINGCONFIG._serialized_start=66
+  _AUTOSCALINGCONFIG._serialized_end=241
+  _TRIGGERS._serialized_start=243
+  _TRIGGERS._serialized_end=302
+  _TRIGGER._serialized_start=304
+  _TRIGGER._serialized_end=400
+  _PROMETHEUSTRIGGER._serialized_start=402
+  _PROMETHEUSTRIGGER._serialized_end=493
+  _QUERYSPEC._serialized_start=496
+  _QUERYSPEC._serialized_end=668
+# @@protoc_insertion_point(module_scope)

_frogml_proto/qwak/auto_scaling/v1/auto_scaling_pb2.pyi

@@ -0,0 +1,187 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _MetricType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _MetricTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MetricType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    METRIC_TYPE_NOT_VALID: _MetricType.ValueType  # 0
+    METRIC_TYPE_CPU: _MetricType.ValueType  # 1
+    METRIC_TYPE_MEMORY: _MetricType.ValueType  # 2
+    METRIC_TYPE_LATENCY: _MetricType.ValueType  # 3
+    METRIC_TYPE_GPU: _MetricType.ValueType  # 4
+    METRIC_TYPE_ERROR_RATE: _MetricType.ValueType  # 5
+    METRIC_TYPE_THROUGHPUT: _MetricType.ValueType  # 6
+
+class MetricType(_MetricType, metaclass=_MetricTypeEnumTypeWrapper): ...
+
+METRIC_TYPE_NOT_VALID: MetricType.ValueType  # 0
+METRIC_TYPE_CPU: MetricType.ValueType  # 1
+METRIC_TYPE_MEMORY: MetricType.ValueType  # 2
+METRIC_TYPE_LATENCY: MetricType.ValueType  # 3
+METRIC_TYPE_GPU: MetricType.ValueType  # 4
+METRIC_TYPE_ERROR_RATE: MetricType.ValueType  # 5
+METRIC_TYPE_THROUGHPUT: MetricType.ValueType  # 6
+global___MetricType = MetricType
+
+class _AggregationType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _AggregationTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AggregationType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    AGGREGATION_TYPE_NOT_VALID: _AggregationType.ValueType  # 0
+    AGGREGATION_TYPE_MIN: _AggregationType.ValueType  # 1
+    AGGREGATION_TYPE_MAX: _AggregationType.ValueType  # 2
+    AGGREGATION_TYPE_AVERAGE: _AggregationType.ValueType  # 3
+    AGGREGATION_TYPE_SUM: _AggregationType.ValueType  # 4
+    AGGREGATION_TYPE_P50: _AggregationType.ValueType  # 5
+    AGGREGATION_TYPE_P90: _AggregationType.ValueType  # 6
+    AGGREGATION_TYPE_P95: _AggregationType.ValueType  # 7
+    AGGREGATION_TYPE_P99: _AggregationType.ValueType  # 8
+
+class AggregationType(_AggregationType, metaclass=_AggregationTypeEnumTypeWrapper): ...
+
+AGGREGATION_TYPE_NOT_VALID: AggregationType.ValueType  # 0
+AGGREGATION_TYPE_MIN: AggregationType.ValueType  # 1
+AGGREGATION_TYPE_MAX: AggregationType.ValueType  # 2
+AGGREGATION_TYPE_AVERAGE: AggregationType.ValueType  # 3
+AGGREGATION_TYPE_SUM: AggregationType.ValueType  # 4
+AGGREGATION_TYPE_P50: AggregationType.ValueType  # 5
+AGGREGATION_TYPE_P90: AggregationType.ValueType  # 6
+AGGREGATION_TYPE_P95: AggregationType.ValueType  # 7
+AGGREGATION_TYPE_P99: AggregationType.ValueType  # 8
+global___AggregationType = AggregationType
+
+class AutoScalingConfig(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MIN_REPLICA_COUNT_FIELD_NUMBER: builtins.int
+    MAX_REPLICA_COUNT_FIELD_NUMBER: builtins.int
+    POLLING_INTERVAL_FIELD_NUMBER: builtins.int
+    COOL_DOWN_PERIOD_FIELD_NUMBER: builtins.int
+    TRIGGERS_FIELD_NUMBER: builtins.int
+    min_replica_count: builtins.int
+    """The minimum number of replicas will scale the resource down to"""
+    max_replica_count: builtins.int
+    """The maximum number of replicas of the target resource"""
+    polling_interval: builtins.int
+    """This is the interval to check each trigger on. By default it's every 30 seconds."""
+    cool_down_period: builtins.int
+    """The period to wait after the last trigger reported active before scaling the resource back to 0. By default it’s 5 minutes (300 seconds)."""
+    @property
+    def triggers(self) -> global___Triggers: ...
+    def __init__(
+        self,
+        *,
+        min_replica_count: builtins.int = ...,
+        max_replica_count: builtins.int = ...,
+        polling_interval: builtins.int = ...,
+        cool_down_period: builtins.int = ...,
+        triggers: global___Triggers | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["triggers", b"triggers"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cool_down_period", b"cool_down_period", "max_replica_count", b"max_replica_count", "min_replica_count", b"min_replica_count", "polling_interval", b"polling_interval", "triggers", b"triggers"]) -> None: ...
+
+global___AutoScalingConfig = AutoScalingConfig
+
+class Triggers(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TRIGGERS_FIELD_NUMBER: builtins.int
+    @property
+    def triggers(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Trigger]:
+        """List of triggers to activate scaling(up/down) of the target resource based on"""
+    def __init__(
+        self,
+        *,
+        triggers: collections.abc.Iterable[global___Trigger] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["triggers", b"triggers"]) -> None: ...
+
+global___Triggers = Triggers
+
+class Trigger(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PROMETHEUS_TRIGGER_FIELD_NUMBER: builtins.int
+    @property
+    def prometheus_trigger(self) -> global___PrometheusTrigger:
+        """Prometheus trigger type"""
+    def __init__(
+        self,
+        *,
+        prometheus_trigger: global___PrometheusTrigger | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["trigger_type", b"trigger_type"]) -> typing_extensions.Literal["prometheus_trigger"] | None: ...
+
+global___Trigger = Trigger
+
+class PrometheusTrigger(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    QUERY_SPEC_FIELD_NUMBER: builtins.int
+    THRESHOLD_FIELD_NUMBER: builtins.int
+    @property
+    def query_spec(self) -> global___QuerySpec:
+        """Query specification"""
+    threshold: builtins.int
+    """Value to start scaling for"""
+    def __init__(
+        self,
+        *,
+        query_spec: global___QuerySpec | None = ...,
+        threshold: builtins.int = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["query_spec", b"query_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["query_spec", b"query_spec", "threshold", b"threshold"]) -> None: ...
+
+global___PrometheusTrigger = PrometheusTrigger
+
+class QuerySpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    METRIC_TYPE_FIELD_NUMBER: builtins.int
+    AGGREGATION_TYPE_FIELD_NUMBER: builtins.int
+    TIME_PERIOD_FIELD_NUMBER: builtins.int
+    ERROR_CODE_FIELD_NUMBER: builtins.int
+    metric_type: global___MetricType.ValueType
+    """The type of the metric"""
+    aggregation_type: global___AggregationType.ValueType
+    """The type of the aggregation"""
+    time_period: builtins.int
+    """The period to run the query based on"""
+    error_code: builtins.str
+    """The error code to scale by in case of MetricType.METRIC_TYPE_ERROR_RATE - default is ".*" """
+    def __init__(
+        self,
+        *,
+        metric_type: global___MetricType.ValueType = ...,
+        aggregation_type: global___AggregationType.ValueType = ...,
+        time_period: builtins.int = ...,
+        error_code: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["aggregation_type", b"aggregation_type", "error_code", b"error_code", "metric_type", b"metric_type", "time_period", b"time_period"]) -> None: ...
+
+global___QuerySpec = QuerySpec

_frogml_proto/qwak/auto_scaling/v1/auto_scaling_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

_frogml_proto/qwak/auto_scaling/v1/auto_scaling_service_pb2.py

@@ -0,0 +1,49 @@
+# -*- 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 message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@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')
+
+
+
+_ATTACHAUTOSCALINGREQUEST = DESCRIPTOR.message_types_by_name['AttachAutoScalingRequest']
+_ATTACHAUTOSCALINGRESPONSE = DESCRIPTOR.message_types_by_name['AttachAutoScalingResponse']
+AttachAutoScalingRequest = _reflection.GeneratedProtocolMessageType('AttachAutoScalingRequest', (_message.Message,), {
+  'DESCRIPTOR' : _ATTACHAUTOSCALINGREQUEST,
+  '__module__' : 'qwak.auto_scaling.v1.auto_scaling_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.auto_scaling.v1.AttachAutoScalingRequest)
+  })
+_sym_db.RegisterMessage(AttachAutoScalingRequest)
+
+AttachAutoScalingResponse = _reflection.GeneratedProtocolMessageType('AttachAutoScalingResponse', (_message.Message,), {
+  'DESCRIPTOR' : _ATTACHAUTOSCALINGRESPONSE,
+  '__module__' : 'qwak.auto_scaling.v1.auto_scaling_service_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.auto_scaling.v1.AttachAutoScalingResponse)
+  })
+_sym_db.RegisterMessage(AttachAutoScalingResponse)
+
+_AUTOSCALINGSERVICE = DESCRIPTOR.services_by_name['AutoScalingService']
+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'
+  _ATTACHAUTOSCALINGREQUEST._serialized_start=115
+  _ATTACHAUTOSCALINGREQUEST._serialized_end=253
+  _ATTACHAUTOSCALINGRESPONSE._serialized_start=255
+  _ATTACHAUTOSCALINGRESPONSE._serialized_end=307
+  _AUTOSCALINGSERVICE._serialized_start=310
+  _AUTOSCALINGSERVICE._serialized_end=448
+# @@protoc_insertion_point(module_scope)