86 lines
2.4 KiB
JavaScript
86 lines
2.4 KiB
JavaScript
const { TaskLoop } = require('../../openmcp-sdk/task-loop');
|
|
const { TaskLoopAdapter } = require('../../openmcp-sdk/service');
|
|
|
|
async function main() {
|
|
// 创建适配器，负责通信和 mcp 连接
|
|
const adapter = new TaskLoopAdapter();
|
|
|
|
// 添加 mcp 服务器
|
|
adapter.addMcp({
|
|
connectionType: 'STDIO',
|
|
commandString: 'uv run mcp run main.py',
|
|
cwd: '~/projects/openmcp-tutorial/crawl4ai-mcp'
|
|
});
|
|
|
|
adapter.addMcp({
|
|
connectionType: 'STDIO',
|
|
commandString: 'node index.js',
|
|
cwd: '~/projects/openmcp-tutorial/my-browser/dist'
|
|
});
|
|
|
|
// 创建事件循环驱动器
|
|
const taskLoop = new TaskLoop({ adapter });
|
|
|
|
// 获取所有工具
|
|
const tools = await taskLoop.getTools();
|
|
|
|
// 配置改次事件循环使用的大模型
|
|
taskLoop.setLlmConfig({
|
|
id: 'deepseek',
|
|
baseUrl: 'https://api.deepseek.com/v1',
|
|
userToken: process.env['DEEPSEEK_API_TOKEN'],
|
|
userModel: 'deepseek-chat'
|
|
});
|
|
|
|
// 创建当前事件循环对应的上下文，并且配置当前上下文的设置
|
|
const storage = {
|
|
messages: [],
|
|
settings: {
|
|
temperature: 0.7,
|
|
// 在本次对话使用所有工具
|
|
enableTools: tools,
|
|
// 系统提示词
|
|
systemPrompt: 'you are a clever bot',
|
|
// 对话上下文的轮数
|
|
contextLength: 20
|
|
}
|
|
};
|
|
|
|
// 本次发出的问题
|
|
const message = 'hello world';
|
|
|
|
// 事件循环结束的句柄
|
|
taskLoop.registerOnDone(() => {
|
|
console.log('taskLoop done');
|
|
});
|
|
|
|
// 事件循环每一次 epoch 开始的句柄
|
|
taskLoop.registerOnError((error) => {
|
|
console.log('taskLoop error', error);
|
|
});
|
|
|
|
// 事件循环出现 error 时的句柄（出现 error 不一定会停止事件循环）
|
|
taskLoop.registerOnEpoch(() => {
|
|
console.log('taskLoop epoch');
|
|
});
|
|
|
|
// 每一次工具调用前
|
|
taskLoop.registerOnToolCall((toolCall) => {
|
|
return toolCall;
|
|
});
|
|
|
|
// 每一次工具调用完后的结果
|
|
taskLoop.registerOnToolCalled((result) => {
|
|
return result;
|
|
});
|
|
|
|
// 开启事件循环
|
|
await taskLoop.start(storage, message);
|
|
|
|
// 打印上下文，最终的回答在 messages.at(-1) 中
|
|
const content = storage.messages.at(-1).content;
|
|
console.log('最终回答：', content);
|
|
}
|
|
|
|
main();
|