Add GPU type information to health check response

- Updated the health check endpoint to include the type of GPU available (CUDA or MPS).
- Modified the HealthResponse model to accommodate the new gpu_type field, enhancing the response with detailed GPU information.
- This change improves the clarity of system capabilities for users and developers.
This commit is contained in:
Jamie Pine
2026-01-29 03:12:11 -08:00
parent 123e8215e4
commit 229841e05e
2 changed files with 8 additions and 0 deletions
+1
View File
@@ -118,6 +118,7 @@ class HealthResponse(BaseModel):
model_downloaded: Optional[bool] = None # Whether model is cached/downloaded
model_size: Optional[str] = None # Current model size if loaded
gpu_available: bool
gpu_type: Optional[str] = None # GPU type (CUDA, MPS, or None)
vram_used_mb: Optional[float] = None