2025-12-11 15:18:09 +08:00
|
|
|
|
import { Agent } from "@mastra/core/agent";
|
|
|
|
|
|
import {
|
2025-12-12 10:48:12 +08:00
|
|
|
|
goodsTool,
|
2025-12-11 15:18:09 +08:00
|
|
|
|
} from "../tools";
|
|
|
|
|
|
import { createDeepSeek } from '@ai-sdk/deepseek';
|
2025-12-21 23:16:00 +08:00
|
|
|
|
import { createOpenAI } from '@ai-sdk/openai';
|
2025-12-16 11:10:08 +08:00
|
|
|
|
import { Memory } from "@mastra/memory";
|
|
|
|
|
|
import { LibSQLStore } from "@mastra/libsql";
|
2025-12-11 15:18:09 +08:00
|
|
|
|
|
|
|
|
|
|
|
2025-12-21 23:16:00 +08:00
|
|
|
|
// const deepseek = createDeepSeek({
|
|
|
|
|
|
// apiKey: 'sk-8603b08e1125422ca6238c8b4a1a40d8',
|
|
|
|
|
|
// });
|
2025-12-11 15:18:09 +08:00
|
|
|
|
const deepseek = createDeepSeek({
|
2025-12-21 23:16:00 +08:00
|
|
|
|
baseURL: 'https://api.xiaomimimo.com/v1',
|
|
|
|
|
|
apiKey: 'sk-cr54p8d2k20nppdzzwyfir4b4znxoh1py6ccxpkc8qt1ahij'
|
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
|
|
const openai = createOpenAI({
|
|
|
|
|
|
baseURL: 'https://api.xiaomimimo.com/v1',
|
|
|
|
|
|
apiKey: 'sk-cr54p8d2k20nppdzzwyfir4b4znxoh1py6ccxpkc8qt1ahij'
|
2025-12-11 15:18:09 +08:00
|
|
|
|
});
|
|
|
|
|
|
|
2025-12-16 11:10:08 +08:00
|
|
|
|
// 创建一个基本内存实例
|
|
|
|
|
|
const memory = new Memory({
|
|
|
|
|
|
storage: new LibSQLStore({
|
|
|
|
|
|
url: "file:./memory.db", // 相对于 `.mastra/output` 目录的相对路径
|
|
|
|
|
|
}),
|
|
|
|
|
|
options: {
|
|
|
|
|
|
lastMessages: 20,
|
|
|
|
|
|
},
|
|
|
|
|
|
});
|
2025-12-11 15:18:09 +08:00
|
|
|
|
|
|
|
|
|
|
export const multiFunctionAgent = new Agent({
|
|
|
|
|
|
name: "Multi-Function Agent",
|
2025-12-21 23:16:00 +08:00
|
|
|
|
instructions: {
|
|
|
|
|
|
role: 'system',
|
|
|
|
|
|
content: `
|
2025-12-16 11:10:08 +08:00
|
|
|
|
你是一个具有内存功能的多功能助手,具备以下能力:
|
2025-12-11 15:18:09 +08:00
|
|
|
|
|
2026-01-07 11:06:11 +08:00
|
|
|
|
1. **商品查询**:使用商品工具查询商品信息,支持分页查询商品列表或根据商品ID获取单个商品详情。
|
2025-12-11 15:18:09 +08:00
|
|
|
|
|
2025-12-11 17:43:08 +08:00
|
|
|
|
请始终礼貌、乐于助人。如果用户询问超出你能力范围的事情,请礼貌解释你能做什么。
|
2025-12-16 11:10:08 +08:00
|
|
|
|
你可以记住之前的对话和用户偏好。
|
|
|
|
|
|
当用户分享关于他们自己的信息时,请确认并记住它以供将来参考。
|
|
|
|
|
|
如果被问及对话中 earlier 提到的事情,请准确地回忆它。
|
2025-12-11 17:43:08 +08:00
|
|
|
|
根据用户请求使用适当的工具。如果不确定,请询问澄清。
|
|
|
|
|
|
|
|
|
|
|
|
请使用中文回答用户的问题。
|
2025-12-21 23:16:00 +08:00
|
|
|
|
`},
|
|
|
|
|
|
// model: deepseek('deepseek-chat'),
|
|
|
|
|
|
// model: openai.chat('mimo-v2-flash'),
|
|
|
|
|
|
model: deepseek.chat('mimo-v2-flash'),
|
2025-12-16 11:10:08 +08:00
|
|
|
|
memory: memory,
|
2025-12-11 15:18:09 +08:00
|
|
|
|
tools: {
|
2025-12-12 10:48:12 +08:00
|
|
|
|
goodsTool,
|
2025-12-11 15:18:09 +08:00
|
|
|
|
},
|
|
|
|
|
|
});
|