690619:1128 240 #02

backend/src/modules/ai/services/vram-monitor.service.ts

@@ -138,10 +138,31 @@ export class VramMonitorService {
 
   /**
    * ตรวจสอบว่า VRAM เพียงพอสำหรับความต้องการโหลดโมเดลหรือไม่
+   * ถ้าไม่มีโมเดลโหลดอยู่เลย จะอนุญาตให้โหลดโมเดลใหม่ได้เสมอ (ป้องกัน false positive)
    */
   async hasVramCapacity(requiredMb: number): Promise<boolean> {
     const headroom = await this.getVramHeadroom();
-    return headroom.availableMb >= requiredMb;
+    // ถ้าไม่มีโมเดลโหลดอยู่เลย อนุญาตให้โหลดโมเดลใหม่ได้เสมอ
+    if (headroom.usedMb === 0 && headroom.querySuccess) {
+      this.logger.log(
+        `No models loaded in VRAM, allowing model load (required=${requiredMb}MB)`
+      );
+      return true;
+    }
+    // ถ้า query ล้มเหลว ใช้ optimistic fallback (assume no VRAM used)
+    if (!headroom.querySuccess) {
+      this.logger.log(
+        `VRAM query failed, using optimistic fallback (required=${requiredMb}MB)`
+      );
+      return true;
+    }
+    const hasCapacity = headroom.availableMb >= requiredMb;
+    if (!hasCapacity) {
+      this.logger.warn(
+        `VRAM insufficient: available=${headroom.availableMb}MB, required=${requiredMb}MB`
+      );
+    }
+    return hasCapacity;
   }
 
   /**

backend/src/modules/ai/tests/vram-monitor.service.spec.ts

@@ -100,6 +100,22 @@ describe('VramMonitorService', () => {
       const result = await service.hasVramCapacity(3000); // query available is 2048MB, required 3000MB
       expect(result).toBe(false);
     });
+
+    it('ควรคืน true เมื่อไม่มีโมเดลโหลดอยู่เลย (ป้องกัน false positive)', async () => {
+      mockedAxios.get.mockResolvedValue({
+        data: {
+          models: [], // ไม่มีโมเดลโหลด
+        },
+      });
+      const result = await service.hasVramCapacity(5000); // ต้องการ 5GB แม้ availableMb = 8192MB
+      expect(result).toBe(true);
+    });
+
+    it('ควรคืน true เมื่อ query ล้มเหลว (optimistic fallback)', async () => {
+      mockedAxios.get.mockRejectedValue(new Error('Connection timeout'));
+      const result = await service.hasVramCapacity(5000);
+      expect(result).toBe(true);
+    });
   });
   describe('getVramStatus', () => {
     it('ควรคืน status ที่ถูกต้องเมื่อ Ollama คืน models', async () => {