113 lines
3.7 KiB
JavaScript
113 lines
3.7 KiB
JavaScript
import * as z from "zod"
|
||
import { tool } from "langchain"
|
||
import os from 'os'
|
||
import { promisify } from 'util'
|
||
import { exec } from 'child_process'
|
||
|
||
const execPromise = promisify(exec)
|
||
|
||
/**
|
||
* 获取系统硬件配置信息工具
|
||
* @returns {string} - 返回系统硬件配置信息的 JSON 字符串
|
||
*/
|
||
export const getHardwareInfoTool = tool(
|
||
async (input) => {
|
||
try {
|
||
const hardwareInfo = {
|
||
cpu: {
|
||
cores: os.cpus().length,
|
||
model: os.cpus()[0].model,
|
||
speed: os.cpus()[0].speed
|
||
},
|
||
memory: {
|
||
total: `${(os.totalmem() / (1024 ** 3)).toFixed(2)} GB`,
|
||
free: `${(os.freemem() / (1024 ** 3)).toFixed(2)} GB`
|
||
},
|
||
disks: [],
|
||
gpu: []
|
||
}
|
||
|
||
// 获取硬盘信息（Windows）
|
||
if (os.platform() === 'win32') {
|
||
try {
|
||
const { stdout } = await execPromise('powershell -Command "Get-WmiObject Win32_DiskDrive | Select-Object Model,InterfaceType,Size | ConvertTo-Json"')
|
||
const diskData = JSON.parse(stdout)
|
||
const diskArray = Array.isArray(diskData) ? diskData : [diskData]
|
||
hardwareInfo.disks = diskArray.map(disk => ({
|
||
model: disk.Model || 'N/A',
|
||
interfaceType: disk.InterfaceType || 'N/A',
|
||
size: disk.Size ? `${(parseInt(disk.Size) / (1024 * 1024 * 1024)).toFixed(2)} GB` : 'N/A'
|
||
})).filter(item => item)
|
||
} catch (error) {
|
||
console.error('Error getting disk info:', error)
|
||
}
|
||
|
||
// 获取显卡信息（Windows）
|
||
try {
|
||
const { stdout } = await execPromise('wmic path win32_videocontroller get name')
|
||
hardwareInfo.gpu = stdout.trim().split('\n').slice(1).filter(line => line.trim())
|
||
} catch (error) {
|
||
console.error('Error getting GPU info:', error)
|
||
}
|
||
}
|
||
// 获取硬盘和显卡信息（Linux）
|
||
else if (os.platform() === 'linux') {
|
||
try {
|
||
const { stdout } = await execPromise('lsblk -o NAME,MODEL,SIZE,TYPE | grep disk')
|
||
const diskLines = stdout.trim().split('\n')
|
||
hardwareInfo.disks = diskLines.map(line => {
|
||
const parts = line.trim().split(/\s+/)
|
||
if (parts.length >= 3) {
|
||
return {
|
||
name: parts[0],
|
||
model: parts.slice(1, -1).join(' '),
|
||
size: parts[parts.length - 1]
|
||
}
|
||
}
|
||
return null
|
||
}).filter(item => item)
|
||
} catch (error) {
|
||
console.error('Error getting disk info:', error)
|
||
}
|
||
|
||
try {
|
||
const { stdout } = await execPromise('lspci | grep VGA')
|
||
hardwareInfo.gpu = stdout.trim().split('\n').map(line => line.trim())
|
||
} catch (error) {
|
||
console.error('Error getting GPU info:', error)
|
||
}
|
||
}
|
||
// 获取硬盘和显卡信息（macOS）
|
||
else if (os.platform() === 'darwin') {
|
||
try {
|
||
const { stdout } = await execPromise('diskutil list')
|
||
hardwareInfo.disks = [{
|
||
info: stdout.trim()
|
||
}]
|
||
} catch (error) {
|
||
console.error('Error getting disk info:', error)
|
||
}
|
||
|
||
try {
|
||
const { stdout } = await execPromise('system_profiler SPDisplaysDataType')
|
||
hardwareInfo.gpu = [{
|
||
info: stdout.trim()
|
||
}]
|
||
} catch (error) {
|
||
console.error('Error getting GPU info:', error)
|
||
}
|
||
}
|
||
|
||
return JSON.stringify(hardwareInfo, null, 2)
|
||
} catch (error) {
|
||
console.error('Error getting hardware info:', error)
|
||
return JSON.stringify({ error: error.message }, null, 2)
|
||
}
|
||
},
|
||
{
|
||
name: "get_hardware_info",
|
||
description: `获取系统硬件配置信息，包括CPU、内存、硬盘、显卡等`,
|
||
schema: z.object({})
|
||
}
|
||
)
|