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

const longTermMemory = new Mem0LongTermMemory({
  apiKey: 'your-mem0-api-key',
  host: 'https://api.mem0.ai',           // 可选的自定义主机
  organizationName: 'my-org',            // 可选的组织
  projectName: 'my-project',             // 可选的项目
  userId: 'user-123',                    // 默认用户 ID
  agentId: 'assistant-v1',               // 默认 Agent ID
  appId: 'my-app'                        // 默认应用 ID
});

// 记录单个记忆
await longTermMemory.record({
  id: 'mem-1',
  strategy: 'user_preferences',
  role: 'user',
  content: '用户偏好暗色模式和简洁的 UI',
  metadata: {
    category: 'ui_preferences',
    confidence: 0.9,
    source: 'explicit_statement'
  },
  createdAt: new Date()
});

// 批量记录多个记忆
const memories = [
  { id: 'mem-2', strategy: 'facts', content: '用户是软件工程师', metadata: {} },
  { id: 'mem-3', strategy: 'preferences', content: '用户喜欢 TypeScript', metadata: {} }
];
await longTermMemory.recordBatch(memories);

// 从对话中自动提取
const messages = [
  { role: 'user', content: '我是一名数据科学家,热爱 Python' },
  { role: 'assistant', content: '太好了!Python 非常适合数据科学。' }
];

const extractedMemories = await longTermMemory.extractAndRecord(messages, {
  userId: 'user-123',
  sessionId: 'session-456',
  metadata: { conversation_type: 'onboarding' }
});

console.log('提取的记忆:', extractedMemories);

// 带过滤器的语义搜索
const searchResults = await longTermMemory.semanticSearch('编程语言', {
  threshold: 0.7,
  limit: 10,
  strategy: ['preferences', 'skills'],
  filters: { category: 'technical' }
});

// 复杂查询
const queryResults = await longTermMemory.retrieve({
  strategy: ['user_preferences', 'facts'],
  limit: 20,
  filters: { confidence: { $gte: 0.8 } }
});

// 更新现有记忆
await longTermMemory.update('mem-1', {
  content: '用户强烈偏好暗色模式和干净、简洁的 UI 设计',
  metadata: { confidence: 0.95, last_confirmed: new Date() }
});

// 使用图关系获取相关记忆
const relatedMemories = await longTermMemory.getRelatedMemories('mem-1', 3);

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

// 按策略清除记忆
await longTermMemory.clear('temporary');