Updated system status response.

main/schemas.py

@@ -1,7 +1,7 @@
 import json
 from pathlib import Path
 from pydantic import BaseModel, Field, create_model, ConfigDict
-from typing import List, Optional, Dict, Union
+from typing import List, Optional, Dict, Any
 from time import time
 
 class QueryExpansionRequest(BaseModel):
@@ -107,12 +107,23 @@ class ModelStatusInfo(BaseModel):
     embedding_model: EmbeddingModelStatus
 
 class SystemStatusResponse(BaseModel):
-    """Pydantic model for system status response"""
-    cpu: Optional[Dict[str, Union[float, str]]] = None
-    memory: Optional[Dict[str, Union[float, str]]] = None
-    gpu: Optional[Dict[str, Union[bool, str, float]]] = None
+    """Pydantic model for comprehensive system status response"""
+
+    os: Dict[str, Any]
+    cpu: Dict[str, Any]
+    memory: Dict[str, Any]
+    gpus: List[Dict[str, Any]]
+    cuda_available: bool
+    cuda_version: Optional[str] = None
+    gpu_count: int
+    gpu_total: Optional[Dict[str, Any]] = None
+    gpu_error: Optional[str] = None
+
+    # Storage information - unchanged
     storage: Optional[Dict[str, str]] = None
-    model: ModelStatusInfo
+
+    # Model information - might need type adjustment
+    model: Optional[ModelStatusInfo] = None
 
 class ValidationResponse(BaseModel):
     config_validation: Dict[str, bool]