fix(api-types): introduce QosLevelPascalCase enum and update PodInfo to use it

Author: 0x5457

crates/api-types/src/lib.rs

@@ -10,6 +10,26 @@ use serde::Deserialize;
 use serde::Serialize;
 
 #[derive(Debug, Clone, PartialEq, Copy, Serialize, Deserialize)]
+pub enum QosLevelPascalCase {
+    Low,
+    Medium,
+    High,
+    Critical,
+}
+
+impl From<QosLevel> for QosLevelPascalCase {
+    fn from(qos_level: QosLevel) -> Self {
+        match qos_level {
+            QosLevel::Low => QosLevelPascalCase::Low,
+            QosLevel::Medium => QosLevelPascalCase::Medium,
+            QosLevel::High => QosLevelPascalCase::High,
+            QosLevel::Critical => QosLevelPascalCase::Critical,
+        }
+    }
+}
+
+#[derive(Debug, Clone, PartialEq, Copy, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
 pub enum QosLevel {
     Low,
     Medium,
@@ -105,7 +125,7 @@ pub struct PodInfo {
     /// VRAM limit for the pod in bytes
     pub vram_limit: Option<u64>,
     /// QoS level for the workload
-    pub qos_level: Option<QosLevel>,
+    pub qos_level: Option<QosLevelPascalCase>,
     /// Whether the workload is a compute shard
     pub compute_shard: bool,
     /// Auto-freeze configuration based on QoS level

crates/hypervisor/src/controller/handlers.rs

@@ -102,7 +102,7 @@ where
             .unwrap_or_default(),
         tflops_limit: pod_entry.tflops_limit,
         vram_limit: pod_entry.vram_limit,
-        qos_level: pod_entry.qos_level,
+        qos_level: pod_entry.qos_level.map(|qos| qos.into()),
         compute_shard: pod_entry.compute_shard,
         auto_freeze,
     };