fix: add used vram into system event body (#1012)

janhq · Aug 12, 2024 · 84e1e19 · 84e1e19
1 parent 80b0350
commit 84e1e19
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/cortex-js/src/domain/models/resource.interface.ts b/cortex-js/src/domain/models/resource.interface.ts
@@ -17,5 +17,5 @@ export interface UsedMemInfo {
 
 export interface GpuInfo {
   name: string | undefined;
-  vram: number | null;
+  vram: UsedMemInfo;
 }
diff --git a/cortex-js/src/infrastructure/services/resources-manager/resources-manager.service.ts b/cortex-js/src/infrastructure/services/resources-manager/resources-manager.service.ts
@@ -25,7 +25,10 @@ export class ResourcesManagerService {
       },
       gpus: (await si.graphics()).controllers.map((gpu) => ({
         name: gpu.name,
-        vram: gpu.vram,
+        vram: {
+          total: gpu.vram ?? 0,
+          used: gpu.memoryUsed ?? 0,
+        }
       })),
     };
   }