qwak-sdk 0.5.89__py3-none-any.whl → 0.5.91__py3-none-any.whl

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/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,
                         ),

{qwak_sdk-0.5.89.dist-info → qwak_sdk-0.5.91.dist-info}/METADATA

@@ -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.dist-info → qwak_sdk-0.5.91.dist-info}/RECORD

@@ -1,4 +1,4 @@
-qwak_sdk/__init__.py,sha256=q3XWCXwhGmFGETMTEoIiiggcGW4O5OkYgl-K8JDOlPM,135
+qwak_sdk/__init__.py,sha256=x6-9NBYL7J0HwjIkwNAp8QNsoknm4YVTPW4pNDTzCqs,135
 qwak_sdk/cli.py,sha256=frMefF0AmLI1g5A9aADNueUkMDjRHikKlmXIA2i4kYA,4510
 qwak_sdk/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -55,7 +55,7 @@ qwak_sdk/commands/audience/update/ui.py,sha256=BRZvhZPM8d3Xcj64X_gWpe8e42XjXCvah
 qwak_sdk/commands/auto_scalling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/auto_scalling/_logic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/auto_scalling/_logic/config/__init__.py,sha256=XtEX4zx1KhshBzM-pOs9G7fUlrk6Fl15ejmvOetdf40,49
-qwak_sdk/commands/auto_scalling/_logic/config/config.py,sha256=ofTq19SJSxy-IjZNsYqNhyxm5QLDi-3siUlbqYMzXsU,3294
+qwak_sdk/commands/auto_scalling/_logic/config/config.py,sha256=IYF-goAE3niI3c-UoduOBE-SdBf_sZ12mywXOyaM51w,5250
 qwak_sdk/commands/auto_scalling/_logic/config/parser.py,sha256=PPo5cgE02X0MpqELLBHIPohUafvIUndZbzamPR0Sxg0,696
 qwak_sdk/commands/auto_scalling/attach/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 qwak_sdk/commands/auto_scalling/attach/_logic.py,sha256=8V4DUIsCHbUaBzQls62MnAZOoOW4BSpVQFKVLofXAJs,1359
@@ -319,7 +319,7 @@ qwak_sdk/tools/colors.py,sha256=7pui_GGjC4uZKYFsIyXaJjYsjLxJVHb4OrfTgr93hqo,287
 qwak_sdk/tools/files.py,sha256=AyKJTOy7NhvP3SrqwIw_lxYNCOy1CvLgMmSJpWZ0OKM,2257
 qwak_sdk/tools/log_handling.py,sha256=Aa1EmxUPCX8YWiZRutUvnqPv6K_z1zoGMwIWsEv24mM,6327
 qwak_sdk/tools/utils.py,sha256=SHmU4r_m2ABZyFYMC03P17GvltPbYbmB39hvalIZEtI,1168
-qwak_sdk-0.5.89.dist-info/METADATA,sha256=5MR1Pkv-W7ijD-R81sjmE0M7PpGIGgMvolht_CvcnUA,2316
-qwak_sdk-0.5.89.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-qwak_sdk-0.5.89.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
-qwak_sdk-0.5.89.dist-info/RECORD,,
+qwak_sdk-0.5.91.dist-info/METADATA,sha256=9HxbbVIyV4OqXwPxY-idhNu0FXEA9UQmN3GIp2nBl9Y,2330
+qwak_sdk-0.5.91.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+qwak_sdk-0.5.91.dist-info/entry_points.txt,sha256=vSl0ELYDyj640oMM57u0AjBP87wtLYxCcGOendhEx80,47
+qwak_sdk-0.5.91.dist-info/RECORD,,