frogml 1.1.62__py3-none-any.whl → 1.2.19__py3-none-any.whl

frogml/_proto/qwak/audience/v1/audience_api_pb2.pyi

@@ -2,21 +2,18 @@
 @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.message
 import frogml._proto.qwak.audience.v1.audience_pb2
-import sys
-
-if sys.version_info >= (3, 8):
-    import typing as typing_extensions
-else:
-    import typing_extensions
+import typing
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+@typing.final
 class CreateAudienceRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -28,11 +25,12 @@ class CreateAudienceRequest(google.protobuf.message.Message):
         *,
         audience: frogml._proto.qwak.audience.v1.audience_pb2.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"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["audience", b"audience"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["audience", b"audience"]) -> None: ...
 
 global___CreateAudienceRequest = CreateAudienceRequest
 
+@typing.final
 class CreateAudienceResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -43,10 +41,11 @@ class CreateAudienceResponse(google.protobuf.message.Message):
         *,
         audience_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["audience_id", b"audience_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["audience_id", b"audience_id"]) -> None: ...
 
 global___CreateAudienceResponse = CreateAudienceResponse
 
+@typing.final
 class UpdateAudienceRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -61,11 +60,12 @@ class UpdateAudienceRequest(google.protobuf.message.Message):
         id: builtins.str = ...,
         audience: frogml._proto.qwak.audience.v1.audience_pb2.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: ...
+    def HasField(self, field_name: typing.Literal["audience", b"audience"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["audience", b"audience", "id", b"id"]) -> None: ...
 
 global___UpdateAudienceRequest = UpdateAudienceRequest
 
+@typing.final
 class UpdateAudienceResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -75,6 +75,7 @@ class UpdateAudienceResponse(google.protobuf.message.Message):
 
 global___UpdateAudienceResponse = UpdateAudienceResponse
 
+@typing.final
 class GetAudienceRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -85,10 +86,11 @@ class GetAudienceRequest(google.protobuf.message.Message):
         *,
         audience_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["audience_id", b"audience_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["audience_id", b"audience_id"]) -> None: ...
 
 global___GetAudienceRequest = GetAudienceRequest
 
+@typing.final
 class GetAudienceResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -100,11 +102,12 @@ class GetAudienceResponse(google.protobuf.message.Message):
         *,
         audience: frogml._proto.qwak.audience.v1.audience_pb2.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"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["audience", b"audience"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["audience", b"audience"]) -> None: ...
 
 global___GetAudienceResponse = GetAudienceResponse
 
+@typing.final
 class ListAudienceRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -114,6 +117,7 @@ class ListAudienceRequest(google.protobuf.message.Message):
 
 global___ListAudienceRequest = ListAudienceRequest
 
+@typing.final
 class ListAudienceResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -125,10 +129,11 @@ class ListAudienceResponse(google.protobuf.message.Message):
         *,
         audience_entries: collections.abc.Iterable[qwak.audience.v1.audience_pb2.AudienceEntry] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["audience_entries", b"audience_entries"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["audience_entries", b"audience_entries"]) -> None: ...
 
 global___ListAudienceResponse = ListAudienceResponse
 
+@typing.final
 class DeleteAudienceRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -139,10 +144,11 @@ class DeleteAudienceRequest(google.protobuf.message.Message):
         *,
         audience_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["audience_id", b"audience_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["audience_id", b"audience_id"]) -> None: ...
 
 global___DeleteAudienceRequest = DeleteAudienceRequest
 
+@typing.final
 class DeleteAudienceResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -152,6 +158,7 @@ class DeleteAudienceResponse(google.protobuf.message.Message):
 
 global___DeleteAudienceResponse = DeleteAudienceResponse
 
+@typing.final
 class SyncAudiencesRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -162,10 +169,11 @@ class SyncAudiencesRequest(google.protobuf.message.Message):
         *,
         environment_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["environment_id", b"environment_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["environment_id", b"environment_id"]) -> None: ...
 
 global___SyncAudiencesRequest = SyncAudiencesRequest
 
+@typing.final
 class SyncAudiencesResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

frogml/_proto/qwak/audience/v1/audience_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.audience/v1/audience.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -20,11 +21,10 @@ _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['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\024com.qwak.audience.v1B\rAudienceProtoP\001Z\031qwak/audience/v1;audience'
+  _globals['_AUDIENCE_ENVIRONMENTROUTESENTRY']._options = None
+  _globals['_AUDIENCE_ENVIRONMENTROUTESENTRY']._serialized_options = b'8\001'
   _globals['_UNARYOPERATORTYPE']._serialized_start=1356
   _globals['_UNARYOPERATORTYPE']._serialized_end=1642
   _globals['_BINARYOPERATORTYPE']._serialized_start=1644

frogml/_proto/qwak/audience/v1/audience_pb2.pyi

@@ -2,6 +2,7 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import collections.abc
 import google.protobuf.descriptor
@@ -23,7 +24,7 @@ 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
+class _UnaryOperatorTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_UnaryOperatorType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     UNARY_OPERATOR_TYPE_INVALID: _UnaryOperatorType.ValueType  # 0
     UNARY_OPERATOR_TYPE_EXACT_MATCH: _UnaryOperatorType.ValueType  # 1
@@ -49,7 +50,7 @@ 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
+class _BinaryOperatorTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BinaryOperatorType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     BINARY_OPERATOR_TYPE_INVALID: _BinaryOperatorType.ValueType  # 0
     BINARY_OPERATOR_TYPE_RANGE_MATCH: _BinaryOperatorType.ValueType  # 1
@@ -61,6 +62,7 @@ BINARY_OPERATOR_TYPE_INVALID: BinaryOperatorType.ValueType  # 0
 BINARY_OPERATOR_TYPE_RANGE_MATCH: BinaryOperatorType.ValueType  # 1
 global___BinaryOperatorType = BinaryOperatorType
 
+@typing.final
 class AudienceEntry(google.protobuf.message.Message):
     """Describe audience entry with ID its equivalent ID for CRUD operation"""
 
@@ -77,11 +79,12 @@ class AudienceEntry(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["audience", b"audience"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["audience", b"audience", "id", b"id"]) -> None: ...
 
 global___AudienceEntry = AudienceEntry
 
+@typing.final
 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
@@ -90,6 +93,7 @@ class Audience(google.protobuf.message.Message):
 
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
+    @typing.final
     class EnvironmentRoutesEntry(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -104,8 +108,8 @@ class Audience(google.protobuf.message.Message):
             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: ...
+        def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
 
     NAME_FIELD_NUMBER: builtins.int
     DESCRIPTION_FIELD_NUMBER: builtins.int
@@ -120,25 +124,30 @@ class Audience(google.protobuf.message.Message):
     """Audience name"""
     description: builtins.str
     """Audience description"""
+    created_by: builtins.str
+    """Created by"""
+    modified_by: builtins.str
+    """Updated by"""
     @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,
         *,
@@ -152,11 +161,12 @@ class Audience(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["created", b"created", "modified", b"modified"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.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
 
+@typing.final
 class EnvironmentAudienceRoutes(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -168,10 +178,11 @@ class EnvironmentAudienceRoutes(google.protobuf.message.Message):
         *,
         routes: collections.abc.Iterable[global___Route] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["routes", b"routes"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["routes", b"routes"]) -> None: ...
 
 global___EnvironmentAudienceRoutes = EnvironmentAudienceRoutes
 
+@typing.final
 class AudienceRoutesEntry(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -181,9 +192,9 @@ class AudienceRoutesEntry(google.protobuf.message.Message):
     ORDER_FIELD_NUMBER: builtins.int
     audience_id: builtins.str
     audience_name: builtins.str
+    order: builtins.int
     @property
     def routes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Route]: ...
-    order: builtins.int
     def __init__(
         self,
         *,
@@ -192,10 +203,11 @@ class AudienceRoutesEntry(google.protobuf.message.Message):
         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: ...
+    def ClearField(self, field_name: typing.Literal["audience_id", b"audience_id", "audience_name", b"audience_name", "order", b"order", "routes", b"routes"]) -> None: ...
 
 global___AudienceRoutesEntry = AudienceRoutesEntry
 
+@typing.final
 class Condition(google.protobuf.message.Message):
     """Condition which applied on prediction request header"""
 
@@ -207,12 +219,12 @@ class Condition(google.protobuf.message.Message):
     INVERT_FIELD_NUMBER: builtins.int
     key: builtins.str
     """key to extract value for evaluation"""
+    invert: builtins.bool
+    """Condition evaluation should be inverted (For example False will be True)"""
     @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,
         *,
@@ -221,12 +233,13 @@ class Condition(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["binary_condition", b"binary_condition", "condition", b"condition", "unary_condition", b"unary_condition"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.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.Literal["condition", b"condition"]) -> typing.Literal["binary_condition", "unary_condition"] | None: ...
 
 global___Condition = Condition
 
+@typing.final
 class UnaryCondition(google.protobuf.message.Message):
     """Unary condition definition
     For example: unary_operator UnaryOperator.UNARY_OPERATOR_EXACT_MATCH with operand hello,
@@ -245,10 +258,11 @@ class UnaryCondition(google.protobuf.message.Message):
         operator: global___UnaryOperatorType.ValueType = ...,
         operand: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["operand", b"operand", "operator", b"operator"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["operand", b"operand", "operator", b"operator"]) -> None: ...
 
 global___UnaryCondition = UnaryCondition
 
+@typing.final
 class BinaryCondition(google.protobuf.message.Message):
     """Binary condition definition
     For example: unary_operator BinaryOperatorType.BINARY_OPERATOR_TYPE_RANGE_MATCH with first_operand 1,
@@ -270,10 +284,11 @@ class BinaryCondition(google.protobuf.message.Message):
         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: ...
+    def ClearField(self, field_name: typing.Literal["first_operand", b"first_operand", "operator", b"operator", "second_operand", b"second_operand"]) -> None: ...
 
 global___BinaryCondition = BinaryCondition
 
+@typing.final
 class Route(google.protobuf.message.Message):
     """Route configuration to apply on specific variation if all conditions evaluated to true."""
 
@@ -308,6 +323,6 @@ class Route(google.protobuf.message.Message):
         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: ...
+    def ClearField(self, field_name: typing.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/auto_scaling/v1/auto_scaling_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.auto_scaling/v1/auto_scaling.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -19,9 +20,8 @@ _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['DESCRIPTOR']._options = None
+  _globals['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

frogml/_proto/qwak/auto_scaling/v1/auto_scaling_pb2.pyi

@@ -2,6 +2,7 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import collections.abc
 import google.protobuf.descriptor
@@ -22,7 +23,7 @@ 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
+class _MetricTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MetricType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     METRIC_TYPE_NOT_VALID: _MetricType.ValueType  # 0
     METRIC_TYPE_CPU: _MetricType.ValueType  # 1
@@ -47,7 +48,7 @@ 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
+class _AggregationTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AggregationType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     AGGREGATION_TYPE_NOT_VALID: _AggregationType.ValueType  # 0
     AGGREGATION_TYPE_MIN: _AggregationType.ValueType  # 1
@@ -72,6 +73,7 @@ AGGREGATION_TYPE_P95: AggregationType.ValueType  # 7
 AGGREGATION_TYPE_P99: AggregationType.ValueType  # 8
 global___AggregationType = AggregationType
 
+@typing.final
 class AutoScalingConfig(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -99,11 +101,12 @@ class AutoScalingConfig(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["triggers", b"triggers"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.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
 
+@typing.final
 class Triggers(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -111,15 +114,17 @@ class Triggers(google.protobuf.message.Message):
     @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: ...
+    def ClearField(self, field_name: typing.Literal["triggers", b"triggers"]) -> None: ...
 
 global___Triggers = Triggers
 
+@typing.final
 class Trigger(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -127,38 +132,42 @@ class Trigger(google.protobuf.message.Message):
     @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: ...
+    def HasField(self, field_name: typing.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["trigger_type", b"trigger_type"]) -> typing.Literal["prometheus_trigger"] | None: ...
 
 global___Trigger = Trigger
 
+@typing.final
 class PrometheusTrigger(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     QUERY_SPEC_FIELD_NUMBER: builtins.int
     THRESHOLD_FIELD_NUMBER: builtins.int
+    threshold: builtins.int
+    """Value to start scaling for"""
     @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: ...
+    def HasField(self, field_name: typing.Literal["query_spec", b"query_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["query_spec", b"query_spec", "threshold", b"threshold"]) -> None: ...
 
 global___PrometheusTrigger = PrometheusTrigger
 
+@typing.final
 class QuerySpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -182,6 +191,6 @@ class QuerySpec(google.protobuf.message.Message):
         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: ...
+    def ClearField(self, field_name: typing.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_service_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.auto_scaling/v1/auto_scaling_service.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -20,9 +21,8 @@ _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['DESCRIPTOR']._options = None
+  _globals['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

frogml/_proto/qwak/auto_scaling/v1/auto_scaling_service_pb2.pyi

@@ -2,19 +2,16 @@
 @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
+import typing
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+@typing.final
 class AttachAutoScalingRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -28,6 +25,7 @@ class AttachAutoScalingRequest(google.protobuf.message.Message):
     @property
     def auto_scaling_config(self) -> frogml._proto.qwak.auto_scaling.v1.auto_scaling_pb2.AutoScalingConfig:
         """Auto scaling"""
+
     def __init__(
         self,
         *,
@@ -35,11 +33,12 @@ class AttachAutoScalingRequest(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["auto_scaling_config", b"auto_scaling_config"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["auto_scaling_config", b"auto_scaling_config", "model_id", b"model_id", "variation_name", b"variation_name"]) -> None: ...
 
 global___AttachAutoScalingRequest = AttachAutoScalingRequest
 
+@typing.final
 class AttachAutoScalingResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -50,6 +49,6 @@ class AttachAutoScalingResponse(google.protobuf.message.Message):
         *,
         auto_scaling_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["auto_scaling_id", b"auto_scaling_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["auto_scaling_id", b"auto_scaling_id"]) -> None: ...
 
 global___AttachAutoScalingResponse = AttachAutoScalingResponse