长期记忆

import { Mem0LongTermMemory } from '@ag-kit/agents';

// 初始化Mem0记忆
const memory = new Mem0LongTermMemory({
  apiKey: process.env.MEM0_API_KEY!,
  userId: 'user-123',
  agentId: 'assistant-v1',
  appId: 'my-app'
});

// 测试连接
await memory.ping();

// 从对话中提取并记录记忆
const messages = [
  { role: 'user', content: 'My name is Alice and I love pizza' },
  { role: 'assistant', content: 'Nice to meet you, Alice! I\'ll remember that you love pizza.' }
];

const extractedMemories = await memory.extractAndRecord(messages, {
  userId: 'user-123'
});

console.log('Extracted memories:', extractedMemories);
// [
//   { id: 'mem-1', content: 'User name is Alice', strategy: 'general', ... },
//   { id: 'mem-2', content: 'User loves pizza', strategy: 'preferences', ... }
// ]

// 手动记录特定记忆
await memory.record({
  id: memory.generateMemoryId(),
  strategy: 'preferences',
  role: 'user',
  content: 'User prefers dark mode interface',
  metadata: {
    category: 'ui_preferences',
    confidence: 0.95,
    source: 'explicit_statement'
  },
  createdAt: new Date()
});

// 批量记录多个记忆
await memory.recordBatch([
  {
    id: memory.generateMemoryId(),
    strategy: 'facts',
    content: 'User works as a software engineer',
    metadata: { category: 'professional' },
    createdAt: new Date()
  },
  {
    id: memory.generateMemoryId(),
    strategy: 'preferences',
    content: 'User prefers TypeScript over JavaScript',
    metadata: { category: 'technical' },
    createdAt: new Date()
  }
]);

// 记录单个记忆
await memory.record({
  id: 'mem-001',
  strategy: 'facts',
  role: 'user',
  content: 'User birthday is March 15th',
  metadata: {
    type: 'personal_info',
    verified: true
  },
  createdAt: new Date()
});

// 批量记录记忆
const memories = [
  {
    id: 'mem-002',
    strategy: 'preferences',
    content: 'User likes morning meetings',
    metadata: {},
    createdAt: new Date()
  },
  {
    id: 'mem-003',
    strategy: 'facts',
    content: 'User lives in San Francisco',
    metadata: {},
    createdAt: new Date()
  }
];

await memory.recordBatch(memories);

// 检索所有记忆
const allMemories = await memory.retrieve({});

// 带限制检索
const recentMemories = await memory.retrieve({ limit: 10 });

// 按策略检索
const preferences = await memory.retrieve({
  strategy: 'preferences',
  limit: 20
});

// 分页检索
const page1 = await memory.retrieve({ limit: 10, offset: 0 });
const page2 = await memory.retrieve({ limit: 10, offset: 10 });

// 使用自定义过滤器检索
const verifiedMemories = await memory.retrieve({
  filters: { verified: true },
  limit: 50
});

// 按ID删除
await memory.delete('mem-001');

// 按查询删除（所有匹配的记忆）
await memory.delete({
  strategy: 'temporary',
  filters: { expired: true }
});

// 清除特定策略的所有记忆
await memory.clear('temporary');

// 清除所有记忆（谨慎使用！）
await memory.clear();

// 从对话中提取记忆
const conversation = [
  { role: 'user', content: '我刚搬到纽约并在Google开始工作' },
  { role: 'assistant', content: '恭喜新工作和搬家！' },
  { role: 'user', content: '谢谢！我对这次改变既兴奋又紧张' }
];

const extractedMemories = await memory.extractAndRecord(conversation, {
  userId: 'user-123',
  metadata: {
    conversation_id: 'conv-456',
    extraction_date: new Date().toISOString()
  }
});

// Mem0智能提取：
// - "用户搬到了纽约"
// - "用户在Google工作"
// - "用户对改变既兴奋又紧张"

// 获取与特定记忆相关的记忆
const relatedMemories = await memory.getRelatedMemories('mem-001', 3);
// 返回最多3个相关记忆

// 示例：如果mem-001是"用户喜欢意大利食物"
// 相关记忆可能是：
// - "用户最喜欢的餐厅是Luigi's"
// - "用户对贝类过敏"
// - "用户喜欢烹饪意大利面"

// 触发记忆整合
await memory.consolidate();

// Mem0自动执行：
// - 去重相似记忆
// - 合并相关信息
// - 对老旧未使用的记忆应用衰减
// - 强化频繁访问的记忆

import { Agent, Mem0LongTermMemory } from '@ag-kit/agents';

// 创建长期记忆
const longTermMemory = new Mem0LongTermMemory({
  apiKey: process.env.MEM0_API_KEY!,
  userId: 'user-123',
  agentId: 'assistant-v1'
});

// 创建带长期记忆的Agent
const agent = new Agent({
  name: 'personal-assistant',
  model: provider,
  longTermMemory: longTermMemory,
  /* ... */
});

// Agent自动：
// 1. 从对话中提取记忆
// 2. 检索相关记忆作为上下文
// 3. 根据新信息更新记忆

const response = await agent.run('你还记得关于我的什么？');
// Agent访问长期记忆以提供个性化响应