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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: qwak-sdk
-Version: 0.5.89
+Version: 0.5.91
 Summary: Qwak SDK and CLI for qwak models
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model
@@ -34,7 +34,7 @@ Requires-Dist: pyarrow (>=6.0.0) ; extra == "batch" or extra == "feature-store"
 Requires-Dist: pyathena (>=2.2.0,!=2.18.0) ; extra == "feature-store"
 Requires-Dist: pyspark (==3.4.2) ; extra == "feature-store"
 Requires-Dist: python-json-logger (>=3.2.1,<4.0.0)
-Requires-Dist: qwak-core (==0.4.220)
+Requires-Dist: qwak-core (==0.4.247)
 Requires-Dist: qwak-inference (>=0.1.20,<0.2.0)
 Requires-Dist: rich (>=13.0.0)
 Requires-Dist: tabulate (>=0.8.0)
@@ -46,3 +46,5 @@ Description-Content-Type: text/markdown
 
 Qwak is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.
 
+### Qwak SDK
+

{qwak_sdk-0.5.89 → qwak_sdk-0.5.91}/README.md

@@ -1,3 +1,5 @@
 # Qwak SDK
 
 Qwak is an end-to-end production ML platform designed to allow data scientists to build, deploy, and monitor their models in production with minimal engineering friction.
+
+### Qwak SDK

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "qwak-sdk"
-version = "0.5.89"
+version = "0.5.91"
 description = "Qwak SDK and CLI for qwak models"
 authors = ["Qwak <info@qwak.com>"]
 readme = "README.md"
@@ -29,7 +29,7 @@ qwak = "qwak_sdk.main:qwak_cli"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<3.12"
-qwak-core = "0.4.220"
+qwak-core = "0.4.247"
 qwak-inference = "^0.1.20"
 croniter = "1.4.1"
 tabulate = ">=0.8.0"

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

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

{qwak_sdk-0.5.89 → qwak_sdk-0.5.91}/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,
                         ),