import { Agent } from "@mastra/core/agent";
import {
  goodsTool,
} from "../tools";
import { createDeepSeek } from '@ai-sdk/deepseek';
import { createOpenAI } from '@ai-sdk/openai';
import { Memory } from "@mastra/memory";
import { LibSQLStore } from "@mastra/libsql";


// const deepseek = createDeepSeek({
//   apiKey: 'sk-8603b08e1125422ca6238c8b4a1a40d8',
// });
const deepseek = createDeepSeek({
  baseURL: 'https://api.xiaomimimo.com/v1',
  apiKey: 'sk-cr54p8d2k20nppdzzwyfir4b4znxoh1py6ccxpkc8qt1ahij'
});

const openai = createOpenAI({
  baseURL: 'https://api.xiaomimimo.com/v1',
  apiKey: 'sk-cr54p8d2k20nppdzzwyfir4b4znxoh1py6ccxpkc8qt1ahij'
});

// 创建一个基本内存实例
const memory = new Memory({
  storage: new LibSQLStore({
    url: "file:./memory.db", // 相对于 `.mastra/output` 目录的相对路径
  }),
  options: {
    lastMessages: 20,
  },
});

export const multiFunctionAgent = new Agent({
  name: "Multi-Function Agent",
  instructions: {
    role: 'system',
    content: `
你是一个具有内存功能的多功能助手，具备以下能力：

1. **商品查询**：使用商品工具查询商品信息，支持分页查询商品列表或根据商品ID获取单个商品详情。

请始终礼貌、乐于助人。如果用户询问超出你能力范围的事情，请礼貌解释你能做什么。
你可以记住之前的对话和用户偏好。
当用户分享关于他们自己的信息时，请确认并记住它以供将来参考。
如果被问及对话中 earlier 提到的事情，请准确地回忆它。
根据用户请求使用适当的工具。如果不确定，请询问澄清。

请使用中文回答用户的问题。
`},
  // model: deepseek('deepseek-chat'),
  // model: openai.chat('mimo-v2-flash'),
  model: deepseek.chat('mimo-v2-flash'),
  memory: memory,
  tools: {
    goodsTool,
  },
});