fix(dynamiccpu): allow to send floating for cpu by splitting schemas by memory and cpu

Author: SQuent

app/models/__init__.py

@@ -1,3 +1,15 @@
-from .schemas import DynamicLoadRequest, LoadRequest, ProbeRequest, TerminateRequest
+from .schemas import (
+    DynamicCPULoadRequest,
+    DynamicMemoryLoadRequest,
+    LoadRequest,
+    ProbeRequest,
+    TerminateRequest,
+)
 
-__all__ = ["DynamicLoadRequest", "LoadRequest", "ProbeRequest", "TerminateRequest"]
+__all__ = [
+    "DynamicCPULoadRequest",
+    "DynamicMemoryLoadRequest",
+    "LoadRequest",
+    "ProbeRequest",
+    "TerminateRequest",
+]

app/models/schemas.py

@@ -3,7 +3,16 @@
 from enum import Enum
 
 
-class DynamicLoadRequest(BaseModel):
+class DynamicCPULoadRequest(BaseModel):
+    start_value: float = Field(..., ge=0, description="Starting load")
+    end_value: float = Field(..., ge=0, description="Ending load")
+    duration: int = Field(1, ge=1, description="Duration in secondes")
+    stop_at_end: bool = Field(
+        False, description="Whether to stop load after completion"
+    )
+
+
+class DynamicMemoryLoadRequest(BaseModel):
     start_value: int = Field(..., ge=0, description="Starting load")
     end_value: int = Field(..., ge=0, description="Ending load")
     duration: int = Field(1, ge=1, description="Duration in secondes")

app/routers/load.py

@@ -1,13 +1,17 @@
 from fastapi import APIRouter, HTTPException
-from ..models.schemas import DynamicLoadRequest, LoadRequest
+from ..models.schemas import (
+    DynamicCPULoadRequest,
+    DynamicMemoryLoadRequest,
+    LoadRequest,
+)
 from ..managers.load_manager import LoadManager
 
 router = APIRouter(prefix="/load", tags=["Load Management"])
 load_manager = LoadManager()
 
 
 @router.post("/cpu/dynamic")
-async def start_dynamic_cpu_load(request: DynamicLoadRequest):
+async def start_dynamic_cpu_load(request: DynamicCPULoadRequest):
     """Start a progressive CPU load"""
     try:
         load_manager.dynamic_cpu_load(
@@ -46,7 +50,7 @@ async def stop_cpu_load():
 
 
 @router.post("/memory/dynamic")
-async def start_dynamic_memory_load(request: DynamicLoadRequest):
+async def start_dynamic_memory_load(request: DynamicMemoryLoadRequest):
     """Start a progressive memory load"""
     try:
         load_manager.dynamic_memory_load(