171535833a
* Made an install script and auto updates env for mac * GPU sensors and initial training working for MAC. Still WIP. * Switch dataloader to single threaded until I can work around some mac pickeling issues. * Get quantization working on mac * Fix mac exclusive imports so they don't break other builds. * Add mac instructions to the UI
67 lines
2.3 KiB
TypeScript
67 lines
2.3 KiB
TypeScript
import { NextResponse } from 'next/server';
|
|
import si from 'systeminformation';
|
|
import { createRequire } from 'module';
|
|
import os from 'os';
|
|
import { CpuInfo } from '@/types';
|
|
|
|
const isMac = os.platform() === 'darwin';
|
|
|
|
export async function GET() {
|
|
try {
|
|
const cpuInfoRaw = await si.cpu();
|
|
let cpuInfo: CpuInfo;
|
|
|
|
if (isMac) {
|
|
try {
|
|
const nativeRequire = createRequire(import.meta.url);
|
|
const ms = nativeRequire('macstats') as any;
|
|
const ramData = ms.getRAMUsageSync();
|
|
const cpuData = ms.getCpuDataSync();
|
|
|
|
cpuInfo = {
|
|
name: `${cpuInfoRaw.manufacturer} ${cpuInfoRaw.brand}`,
|
|
cores: cpuInfoRaw.cores,
|
|
temperature: cpuData.temperature || 0,
|
|
totalMemory: ramData.total / (1024 * 1024),
|
|
availableMemory: ramData.free / (1024 * 1024),
|
|
freeMemory: ramData.free / (1024 * 1024),
|
|
currentLoad: (await si.currentLoad()).currentLoad || 0,
|
|
};
|
|
} catch {
|
|
// Fallback to systeminformation if macstats fails
|
|
const memoryData = await si.mem();
|
|
cpuInfo = {
|
|
name: `${cpuInfoRaw.manufacturer} ${cpuInfoRaw.brand}`,
|
|
cores: cpuInfoRaw.cores,
|
|
temperature: (await si.cpuTemperature()).main || 0,
|
|
totalMemory: memoryData.total / (1024 * 1024),
|
|
availableMemory: memoryData.available / (1024 * 1024),
|
|
freeMemory: memoryData.free / (1024 * 1024),
|
|
currentLoad: (await si.currentLoad()).currentLoad || 0,
|
|
};
|
|
}
|
|
} else {
|
|
const memoryData = await si.mem();
|
|
cpuInfo = {
|
|
name: `${cpuInfoRaw.manufacturer} ${cpuInfoRaw.brand}`,
|
|
cores: cpuInfoRaw.cores,
|
|
temperature: (await si.cpuTemperature()).main || 0,
|
|
totalMemory: memoryData.total / (1024 * 1024),
|
|
availableMemory: memoryData.available / (1024 * 1024),
|
|
freeMemory: memoryData.free / (1024 * 1024),
|
|
currentLoad: (await si.currentLoad()).currentLoad || 0,
|
|
};
|
|
}
|
|
|
|
return NextResponse.json(cpuInfo);
|
|
} catch (error) {
|
|
console.error('Error fetching CPU stats:', error);
|
|
return NextResponse.json(
|
|
{
|
|
error: `Failed to fetch CPU stats: ${error instanceof Error ? error.message : String(error)}`,
|
|
},
|
|
{ status: 500 },
|
|
);
|
|
}
|
|
}
|