74 lines
2.1 KiB
JavaScript
74 lines
2.1 KiB
JavaScript
import * as z from 'zod';
|
||
import { tool } from 'langchain';
|
||
import { spawn } from 'child_process';
|
||
|
||
export const winCmdTool = tool(
|
||
async ({ command, cwd, timeout = 30000 }) => {
|
||
const workingDir = cwd || process.cwd();
|
||
|
||
return new Promise((resolve) => {
|
||
let stdout = '';
|
||
let stderr = '';
|
||
let killed = false;
|
||
|
||
const proc = spawn('cmd.exe', ['/c', command], {
|
||
cwd: workingDir,
|
||
shell: false,
|
||
windowsHide: true
|
||
});
|
||
|
||
const timer = setTimeout(() => {
|
||
killed = true;
|
||
proc.kill('SIGKILL');
|
||
}, timeout);
|
||
|
||
proc.stdout.on('data', (data) => {
|
||
stdout += data.toString();
|
||
});
|
||
|
||
proc.stderr.on('data', (data) => {
|
||
stderr += data.toString();
|
||
});
|
||
|
||
proc.on('close', (code) => {
|
||
clearTimeout(timer);
|
||
if (killed) {
|
||
resolve(JSON.stringify({
|
||
success: false,
|
||
error: `Command timed out after ${timeout}ms`,
|
||
command,
|
||
cwd: workingDir
|
||
}));
|
||
return;
|
||
}
|
||
resolve(JSON.stringify({
|
||
success: code === 0,
|
||
exitCode: code,
|
||
stdout: stdout.trim(),
|
||
stderr: stderr.trim(),
|
||
command,
|
||
cwd: workingDir
|
||
}));
|
||
});
|
||
|
||
proc.on('error', (error) => {
|
||
clearTimeout(timer);
|
||
resolve(JSON.stringify({
|
||
success: false,
|
||
error: error.message,
|
||
command,
|
||
cwd: workingDir
|
||
}));
|
||
});
|
||
});
|
||
},
|
||
{
|
||
name: 'win_cmd',
|
||
description: '在 Windows 系统上执行 cmd.exe 命令行命令。适用于运行 Windows 批处理命令、文件系统操作命令（如 dir、copy、del、mkdir 等）。不支持 PowerShell 命令。',
|
||
schema: z.object({
|
||
command: z.string().describe('要执行的 cmd.exe 命令或命令组合，使用 && 连接多个命令，如 "dir /b" 或 "echo hello && dir"'),
|
||
cwd: z.string().optional().describe('执行命令的工作目录，默认为当前工作目录'),
|
||
timeout: z.number().optional().describe('命令执行超时时间(毫秒)，默认30000')
|
||
})
|
||
}
|
||
); |