@mrpatronz/nexusflow 0.2.13 → 0.2.17

package/src/utils/system-scanner.test.ts

@@ -1,89 +1,89 @@
-import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { execa } from 'execa';
-import * as os from 'node:os';
-import { scanSystemSpecs } from './system-scanner.js';
-
-vi.mock('execa');
-vi.mock('node:os', async () => {
-  const actual = await vi.importActual<typeof import('node:os')>('node:os');
-  return {
-    ...actual,
-    totalmem: vi.fn(),
-    platform: vi.fn(),
-  };
-});
-
-describe('scanSystemSpecs', () => {
-  beforeEach(() => {
-    vi.clearAllMocks();
-  });
-
-  it('should recommend 7b on windows with 16GB RAM even if no GPU detected', async () => {
-    vi.mocked(os.totalmem).mockReturnValue(16 * 1024 * 1024 * 1024);
-    vi.mocked(os.platform).mockReturnValue('win32');
-    vi.mocked(execa).mockResolvedValue({ stdout: '' } as any);
-
-    const specs = await scanSystemSpecs();
-
-    expect(specs.totalRamGb).toBe(16);
-    expect(specs.gpuName).toBe('Unknown/Integrated');
-    expect(specs.hasHardwareAcceleration).toBe(false);
-    expect(specs.recommendedModel).toBe('qwen2.5-coder:7b');
-  });
-
-  it('should recommend 1.5b on windows with 8GB RAM and no GPU', async () => {
-    vi.mocked(os.totalmem).mockReturnValue(8 * 1024 * 1024 * 1024);
-    vi.mocked(os.platform).mockReturnValue('win32');
-    vi.mocked(execa).mockResolvedValue({ stdout: '' } as any);
-
-    const specs = await scanSystemSpecs();
-
-    expect(specs.totalRamGb).toBe(8);
-    expect(specs.recommendedModel).toBe('qwen2.5-coder:1.5b');
-  });
-
-  it('should detect Nvidia GPU on Windows and recommend 7b with 12GB RAM', async () => {
-    vi.mocked(os.totalmem).mockReturnValue(12 * 1024 * 1024 * 1024);
-    vi.mocked(os.platform).mockReturnValue('win32');
-    vi.mocked(execa).mockResolvedValue({
-      stdout: 'NVIDIA GeForce RTX 4070 Laptop GPU\r\n'
-    } as any);
-
-    const specs = await scanSystemSpecs();
-
-    expect(specs.totalRamGb).toBe(12);
-    expect(specs.gpuName).toBe('NVIDIA GeForce RTX 4070 Laptop GPU');
-    expect(specs.hasHardwareAcceleration).toBe(true);
-    expect(specs.recommendedModel).toBe('qwen2.5-coder:7b');
-  });
-
-  it('should detect Apple Silicon on Darwin and recommend 7b with 12GB RAM', async () => {
-    vi.mocked(os.totalmem).mockReturnValue(12 * 1024 * 1024 * 1024);
-    vi.mocked(os.platform).mockReturnValue('darwin');
-    vi.mocked(execa).mockResolvedValue({
-      stdout: 'Apple M3 Max\n'
-    } as any);
-
-    const specs = await scanSystemSpecs();
-
-    expect(specs.totalRamGb).toBe(12);
-    expect(specs.gpuName).toBe('Apple Silicon (Integrated)');
-    expect(specs.hasHardwareAcceleration).toBe(true);
-    expect(specs.recommendedModel).toBe('qwen2.5-coder:7b');
-  });
-
-  it('should detect Linux GPU and recommend 7b with 12GB RAM', async () => {
-    vi.mocked(os.totalmem).mockReturnValue(12 * 1024 * 1024 * 1024);
-    vi.mocked(os.platform).mockReturnValue('linux');
-    vi.mocked(execa).mockResolvedValue({
-      stdout: '01:00.0 VGA compatible controller: NVIDIA Corporation AD104 [GeForce RTX 4070 Ti] (rev a1)\n'
-    } as any);
-
-    const specs = await scanSystemSpecs();
-
-    expect(specs.totalRamGb).toBe(12);
-    expect(specs.gpuName).toBe('NVIDIA Corporation AD104 [GeForce RTX 4070 Ti] (rev a1)');
-    expect(specs.hasHardwareAcceleration).toBe(true);
-    expect(specs.recommendedModel).toBe('qwen2.5-coder:7b');
-  });
-});
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { execa } from 'execa';
+import * as os from 'node:os';
+import { scanSystemSpecs } from './system-scanner.js';
+
+vi.mock('execa');
+vi.mock('node:os', async () => {
+  const actual = await vi.importActual<typeof import('node:os')>('node:os');
+  return {
+    ...actual,
+    totalmem: vi.fn(),
+    platform: vi.fn(),
+  };
+});
+
+describe('scanSystemSpecs', () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it('should recommend 7b on windows with 16GB RAM even if no GPU detected', async () => {
+    vi.mocked(os.totalmem).mockReturnValue(16 * 1024 * 1024 * 1024);
+    vi.mocked(os.platform).mockReturnValue('win32');
+    vi.mocked(execa).mockResolvedValue({ stdout: '' } as any);
+
+    const specs = await scanSystemSpecs();
+
+    expect(specs.totalRamGb).toBe(16);
+    expect(specs.gpuName).toBe('Unknown/Integrated');
+    expect(specs.hasHardwareAcceleration).toBe(false);
+    expect(specs.recommendedModel).toBe('qwen2.5-coder:7b');
+  });
+
+  it('should recommend 1.5b on windows with 8GB RAM and no GPU', async () => {
+    vi.mocked(os.totalmem).mockReturnValue(8 * 1024 * 1024 * 1024);
+    vi.mocked(os.platform).mockReturnValue('win32');
+    vi.mocked(execa).mockResolvedValue({ stdout: '' } as any);
+
+    const specs = await scanSystemSpecs();
+
+    expect(specs.totalRamGb).toBe(8);
+    expect(specs.recommendedModel).toBe('qwen2.5-coder:1.5b');
+  });
+
+  it('should detect Nvidia GPU on Windows and recommend 7b with 12GB RAM', async () => {
+    vi.mocked(os.totalmem).mockReturnValue(12 * 1024 * 1024 * 1024);
+    vi.mocked(os.platform).mockReturnValue('win32');
+    vi.mocked(execa).mockResolvedValue({
+      stdout: 'NVIDIA GeForce RTX 4070 Laptop GPU\r\n'
+    } as any);
+
+    const specs = await scanSystemSpecs();
+
+    expect(specs.totalRamGb).toBe(12);
+    expect(specs.gpuName).toBe('NVIDIA GeForce RTX 4070 Laptop GPU');
+    expect(specs.hasHardwareAcceleration).toBe(true);
+    expect(specs.recommendedModel).toBe('qwen2.5-coder:7b');
+  });
+
+  it('should detect Apple Silicon on Darwin and recommend 7b with 12GB RAM', async () => {
+    vi.mocked(os.totalmem).mockReturnValue(12 * 1024 * 1024 * 1024);
+    vi.mocked(os.platform).mockReturnValue('darwin');
+    vi.mocked(execa).mockResolvedValue({
+      stdout: 'Apple M3 Max\n'
+    } as any);
+
+    const specs = await scanSystemSpecs();
+
+    expect(specs.totalRamGb).toBe(12);
+    expect(specs.gpuName).toBe('Apple Silicon (Integrated)');
+    expect(specs.hasHardwareAcceleration).toBe(true);
+    expect(specs.recommendedModel).toBe('qwen2.5-coder:7b');
+  });
+
+  it('should detect Linux GPU and recommend 7b with 12GB RAM', async () => {
+    vi.mocked(os.totalmem).mockReturnValue(12 * 1024 * 1024 * 1024);
+    vi.mocked(os.platform).mockReturnValue('linux');
+    vi.mocked(execa).mockResolvedValue({
+      stdout: '01:00.0 VGA compatible controller: NVIDIA Corporation AD104 [GeForce RTX 4070 Ti] (rev a1)\n'
+    } as any);
+
+    const specs = await scanSystemSpecs();
+
+    expect(specs.totalRamGb).toBe(12);
+    expect(specs.gpuName).toBe('NVIDIA Corporation AD104 [GeForce RTX 4070 Ti] (rev a1)');
+    expect(specs.hasHardwareAcceleration).toBe(true);
+    expect(specs.recommendedModel).toBe('qwen2.5-coder:7b');
+  });
+});

package/src/utils/system-scanner.ts

@@ -1,96 +1,96 @@
-/**
- * @module utils/system-scanner
- * Detects local system specifications (RAM, GPU) and recommends the optimal local LLM size.
- */
-
-import { execa } from 'execa';
-import * as os from 'node:os';
-
-export interface SystemSpecs {
-  totalRamGb: number;
-  gpuName: string;
-  hasHardwareAcceleration: boolean;
-  recommendedModel: string;
-}
-
-/**
- * Probes the local operating system to detect RAM and GPU specifications,
- * returning a recommended local LLM model name.
- */
-export async function scanSystemSpecs(): Promise<SystemSpecs> {
-  const totalRamGb = Math.round(os.totalmem() / (1024 * 1024 * 1024));
-  let gpuName = 'Unknown/Integrated';
-  let hasHardwareAcceleration = false;
-
-  const platform = os.platform();
-
-  try {
-    if (platform === 'win32') {
-      const { stdout } = await execa('powershell', ['-NoProfile', '-Command', 'Get-CimInstance Win32_VideoController | Select-Object -ExpandProperty Name'], { reject: false });
-      if (stdout) {
-        const lines = stdout.split('\n')
-          .map(l => l.trim())
-          .filter(l => l && l.toLowerCase() !== 'name');
-        
-        if (lines.length > 0) {
-          gpuName = lines.join(', ');
-          const lowerGpu = gpuName.toLowerCase();
-          if (
-            lowerGpu.includes('nvidia') || 
-            lowerGpu.includes('radeon') || 
-            lowerGpu.includes('geforce') || 
-            lowerGpu.includes('rtx') || 
-            lowerGpu.includes('gtx')
-          ) {
-            hasHardwareAcceleration = true;
-          }
-        }
-      }
-    } else if (platform === 'darwin') {
-      const { stdout } = await execa('sysctl', ['-n', 'machdep.cpu.brand_string'], { reject: false });
-      if (stdout && stdout.toLowerCase().includes('apple')) {
-        gpuName = 'Apple Silicon (Integrated)';
-        hasHardwareAcceleration = true;
-      } else {
-        const { stdout: spOut } = await execa('system_profiler', ['SPDisplaysDataType'], { reject: false });
-        if (spOut) {
-          gpuName = 'Intel Mac GPU';
-          const lowerGpu = spOut.toLowerCase();
-          if (lowerGpu.includes('amd') || lowerGpu.includes('radeon') || lowerGpu.includes('nvidia')) {
-            hasHardwareAcceleration = true;
-          }
-        }
-      }
-    } else if (platform === 'linux') {
-      const { stdout } = await execa('lspci', [], { reject: false });
-      if (stdout) {
-        const gpuLines = stdout.split('\n').filter(line => line.toLowerCase().includes('vga') || line.toLowerCase().includes('3d'));
-        if (gpuLines.length > 0) {
-          gpuName = gpuLines.map(line => line.split(':').pop()?.trim()).filter(Boolean).join(', ');
-          const lowerGpu = gpuName.toLowerCase();
-          if (lowerGpu.includes('nvidia') || lowerGpu.includes('amd') || lowerGpu.includes('radeon')) {
-            hasHardwareAcceleration = true;
-          }
-        }
-      }
-    }
-  } catch {
-    // Ignore execution failures, fall back to default values
-  }
-
-  // Recommendation logic:
-  // - RAM >= 12GB AND has dedicated GPU/Apple Silicon -> Recommend 7B
-  // - RAM >= 16GB regardless of GPU -> Recommend 7B (can run fine on CPU)
-  // - Otherwise -> Recommend 1.5B (fast and safe)
-  let recommendedModel = 'qwen2.5-coder:1.5b';
-  if ((totalRamGb >= 12 && hasHardwareAcceleration) || totalRamGb >= 16) {
-    recommendedModel = 'qwen2.5-coder:7b';
-  }
-
-  return {
-    totalRamGb,
-    gpuName,
-    hasHardwareAcceleration,
-    recommendedModel,
-  };
-}
+/**
+ * @module utils/system-scanner
+ * Detects local system specifications (RAM, GPU) and recommends the optimal local LLM size.
+ */
+
+import { execa } from 'execa';
+import * as os from 'node:os';
+
+export interface SystemSpecs {
+  totalRamGb: number;
+  gpuName: string;
+  hasHardwareAcceleration: boolean;
+  recommendedModel: string;
+}
+
+/**
+ * Probes the local operating system to detect RAM and GPU specifications,
+ * returning a recommended local LLM model name.
+ */
+export async function scanSystemSpecs(): Promise<SystemSpecs> {
+  const totalRamGb = Math.round(os.totalmem() / (1024 * 1024 * 1024));
+  let gpuName = 'Unknown/Integrated';
+  let hasHardwareAcceleration = false;
+
+  const platform = os.platform();
+
+  try {
+    if (platform === 'win32') {
+      const { stdout } = await execa('powershell', ['-NoProfile', '-Command', 'Get-CimInstance Win32_VideoController | Select-Object -ExpandProperty Name'], { reject: false });
+      if (stdout) {
+        const lines = stdout.split('\n')
+          .map(l => l.trim())
+          .filter(l => l && l.toLowerCase() !== 'name');
+        
+        if (lines.length > 0) {
+          gpuName = lines.join(', ');
+          const lowerGpu = gpuName.toLowerCase();
+          if (
+            lowerGpu.includes('nvidia') || 
+            lowerGpu.includes('radeon') || 
+            lowerGpu.includes('geforce') || 
+            lowerGpu.includes('rtx') || 
+            lowerGpu.includes('gtx')
+          ) {
+            hasHardwareAcceleration = true;
+          }
+        }
+      }
+    } else if (platform === 'darwin') {
+      const { stdout } = await execa('sysctl', ['-n', 'machdep.cpu.brand_string'], { reject: false });
+      if (stdout && stdout.toLowerCase().includes('apple')) {
+        gpuName = 'Apple Silicon (Integrated)';
+        hasHardwareAcceleration = true;
+      } else {
+        const { stdout: spOut } = await execa('system_profiler', ['SPDisplaysDataType'], { reject: false });
+        if (spOut) {
+          gpuName = 'Intel Mac GPU';
+          const lowerGpu = spOut.toLowerCase();
+          if (lowerGpu.includes('amd') || lowerGpu.includes('radeon') || lowerGpu.includes('nvidia')) {
+            hasHardwareAcceleration = true;
+          }
+        }
+      }
+    } else if (platform === 'linux') {
+      const { stdout } = await execa('lspci', [], { reject: false });
+      if (stdout) {
+        const gpuLines = stdout.split('\n').filter(line => line.toLowerCase().includes('vga') || line.toLowerCase().includes('3d'));
+        if (gpuLines.length > 0) {
+          gpuName = gpuLines.map(line => line.split(':').pop()?.trim()).filter(Boolean).join(', ');
+          const lowerGpu = gpuName.toLowerCase();
+          if (lowerGpu.includes('nvidia') || lowerGpu.includes('amd') || lowerGpu.includes('radeon')) {
+            hasHardwareAcceleration = true;
+          }
+        }
+      }
+    }
+  } catch {
+    // Ignore execution failures, fall back to default values
+  }
+
+  // Recommendation logic:
+  // - RAM >= 12GB AND has dedicated GPU/Apple Silicon -> Recommend 7B
+  // - RAM >= 16GB regardless of GPU -> Recommend 7B (can run fine on CPU)
+  // - Otherwise -> Recommend 1.5B (fast and safe)
+  let recommendedModel = 'qwen2.5-coder:1.5b';
+  if ((totalRamGb >= 12 && hasHardwareAcceleration) || totalRamGb >= 16) {
+    recommendedModel = 'qwen2.5-coder:7b';
+  }
+
+  return {
+    totalRamGb,
+    gpuName,
+    hasHardwareAcceleration,
+    recommendedModel,
+  };
+}