Cleanup

invokeai/app/api/routers/app_info.py

@@ -188,9 +188,5 @@ async def get_invocation_cache_status() -> InvocationCacheStatus:
 @app_router.get("/system-stats", operation_id="get_system_stats", status_code=200, response_model=SystemStats)
 async def get_stats() -> SystemStats:
     """Fetches and returns the system statistics, including CPU, RAM, and GPU stats."""
-    try:
-        stats = get_system_stats()
-        return stats
-    except Exception as e:
-        print(f"Error fetching system stats: {e}")
-        return SystemStats(cpu_usage=0.0, ram_usage=0.0, gpu_usage=[])
+    stats = get_system_stats()
+    return stats

invokeai/frontend/web/src/services/api/endpoints/appInfo.ts

@@ -1,23 +1,10 @@
 import { $openAPISchemaUrl } from 'app/store/nanostores/openAPISchemaUrl';
 import type { OpenAPIV3_1 } from 'openapi-types';
 import type { paths } from 'services/api/schema';
-import type { AppConfig, AppDependencyVersions, AppVersion } from 'services/api/types';
+import type { AppConfig, AppDependencyVersions, AppVersion, SystemStats } from 'services/api/types';
 
 import { api, buildV1Url } from '..';
 
-interface GPUStat {
-  id: number;
-  load: number;
-  memory: number;
-  memory_total: number;
-}
-
-interface SystemStats {
-  cpu_usage: number;
-  ram_usage: number;
-  gpu_usage: GPUStat[];
-}
-
 /**
  * Builds an endpoint URL for the app router
  * @example

invokeai/frontend/web/src/services/api/types.ts

@@ -237,3 +237,17 @@ export type PostUploadAction =
   | RGIPAdapterImagePostUploadAction
   | UpscaleInitialImageAction
   | ReplaceLayerWithImagePostUploadAction;
+
+// System Stats
+export interface GPUStat {
+  id: number;
+  load: number;
+  memory: number;
+  memory_total: number;
+}
+
+export interface SystemStats {
+  cpu_usage: number;
+  ram_usage: number;
+  gpu_usage: GPUStat[];
+}