qwak-sdk 0.5.89__tar.gz → 0.5.90__tar.gz

{qwak_sdk-0.5.89 → qwak_sdk-0.5.90}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: qwak-sdk
-Version: 0.5.89
+Version: 0.5.90
 Summary: Qwak SDK and CLI for qwak models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{qwak_sdk-0.5.89 → qwak_sdk-0.5.90}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "qwak-sdk"
-version = "0.5.89"
+version = "0.5.90"
 description = "Qwak SDK and CLI for qwak models"
 authors = ["Qwak <info@qwak.com>"]
 readme = "README.md"

{qwak_sdk-0.5.89 → qwak_sdk-0.5.90}/qwak_sdk/__init__.py

@@ -1,6 +1,6 @@
 # fmt: off
 __author__ = '''Qwak.ai'''
-__version__ = '0.5.89'
+__version__ = '0.5.90'
 
 from qwak.inner import wire_dependencies
 

{qwak_sdk-0.5.89 → qwak_sdk-0.5.90}/qwak_sdk/commands/auto_scalling/_logic/config/config.py

@@ -12,9 +12,11 @@ from _qwak_proto.qwak.auto_scaling.v1.auto_scaling_pb2 import (
     AGGREGATION_TYPE_P99,
     AGGREGATION_TYPE_SUM,
     METRIC_TYPE_CPU,
+    METRIC_TYPE_ERROR_RATE,
     METRIC_TYPE_GPU,
     METRIC_TYPE_LATENCY,
     METRIC_TYPE_MEMORY,
+    METRIC_TYPE_THROUGHPUT,
     AutoScalingConfig,
     PrometheusTrigger,
     QuerySpec,
@@ -22,12 +24,35 @@ from _qwak_proto.qwak.auto_scaling.v1.auto_scaling_pb2 import (
     Triggers,
 )
 
+from qwak_sdk.exceptions import QwakCommandException
+
 
 class MetricType(Enum):
     cpu = METRIC_TYPE_CPU
     gpu = METRIC_TYPE_GPU
     latency = METRIC_TYPE_LATENCY
     memory = METRIC_TYPE_MEMORY
+    throughput = METRIC_TYPE_THROUGHPUT
+    error_rate = METRIC_TYPE_ERROR_RATE
+
+    @classmethod
+    def get_metric_type(cls, metric_type: str) -> "MetricType":
+        """
+        Converts a string representation of a metric type to its corresponding `MetricType` enum.
+        The method performs a case-insensitive lookup. If the provided metric type does not exist
+        in the `MetricType` enum, a `QwakCommandException` is raised.
+
+        :param metric_type: The name of the metric type to retrieve.
+        :return: The corresponding `MetricType` enum.
+        :raises: QwakCommandException if the provided `metric_type` does not match any member
+        """
+
+        try:
+            return cls[metric_type.lower()]
+        except KeyError as exception:
+            raise QwakCommandException(
+                f"The metric type {metric_type} doesn't exist"
+            ) from exception
 
 
 class AggregationType(Enum):
@@ -40,6 +65,24 @@ class AggregationType(Enum):
     p95 = AGGREGATION_TYPE_P95
     p99 = AGGREGATION_TYPE_P99
 
+    @classmethod
+    def get_aggregation_type(cls, aggregation_type: str) -> "AggregationType":
+        """
+        Converts a string representation of an aggregation type to its corresponding `AggregationType` enum.
+        The method performs a case-insensitive lookup. If the provided aggregation type does not exist
+        in the `AggregationType` enum, a `QwakCommandException` is raised.
+
+        :param aggregation_type: The name of the aggregation type to retrieve
+        :return: The corresponding `AggregationType` enum.
+        :raises: QwakCommandException if the provided `aggregation_type` does not match
+        """
+        try:
+            return cls[aggregation_type.lower()]
+        except KeyError as exception:
+            raise QwakCommandException(
+                f"The aggregation type {aggregation_type} doesn't exist"
+            ) from exception
+
 
 @dataclass
 class AutoScaling:
@@ -68,16 +111,18 @@ class AutoScaling:
     def to_autoscaling_api(self) -> AutoScalingConfig:
         triggers = Triggers()
         for trigger in self.triggers.prometheus_trigger:
+            metric_type: MetricType = MetricType.get_metric_type(
+                trigger.query_spec.metric_type
+            )
+            aggregation_type: AggregationType = AggregationType.get_aggregation_type(
+                trigger.query_spec.aggregation_type
+            )
             triggers.triggers.append(
                 Trigger(
                     prometheus_trigger=PrometheusTrigger(
                         query_spec=QuerySpec(
-                            metric_type=MetricType[
-                                trigger.query_spec.metric_type.lower()
-                            ].value,
-                            aggregation_type=AggregationType[
-                                trigger.query_spec.aggregation_type.lower()
-                            ].value,
+                            metric_type=metric_type.value,
+                            aggregation_type=aggregation_type.value,
                             time_period=trigger.query_spec.time_period,
                             error_code=trigger.query_spec.error_code,
                         ),