会话管理

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

// 创建内存实例
const memory = new InMemoryMemory();

// 为用户A创建会话
const sessionA = 'user-alice-session-1';
await memory.add(
  {
    message: {
      id: 'msg-1',
      role: 'user',
      content: 'Hello, I\'m Alice',
      timestamp: new Date()
    },
    state: { userId: 'alice' }
  },
  { sessionId: sessionA }
);

// 为用户B创建会话
const sessionB = 'user-bob-session-1';
await memory.add(
  {
    message: {
      id: 'msg-1',
      role: 'user',
      content: 'Hi, I\'m Bob',
      timestamp: new Date()
    },
    state: { userId: 'bob' }
  },
  { sessionId: sessionB }
);

// 获取会话专属历史记录
const aliceHistory = await memory.list({ sessionId: sessionA });
const bobHistory = await memory.list({ sessionId: sessionB });

console.log('Alice:', aliceHistory[0].message.content); // "Hello, I'm Alice"
console.log('Bob:', bobHistory[0].message.content); // "Hi, I'm Bob"

// 基于用户：每个用户一个会话
const sessionId = `user-${userId}`;

// 基于对话：每个用户多个会话
const sessionId = `user-${userId}-conv-${conversationId}`;

// 基于时间：带时间戳的会话
const sessionId = `user-${userId}-${Date.now()}`;

// 层级结构：复杂组织结构
const sessionId = `org-${orgId}/team-${teamId}/user-${userId}`;

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

const memory = new InMemoryMemory();

// Agent处理带会话隔离的请求
async function handleUserMessage(userId: string, message: string) {
  const sessionId = `user-${userId}`;
  
  // 将用户消息添加到会话
  await memory.add(
    {
      message: {
        id: crypto.randomUUID(),
        role: 'user',
        content: message,
        timestamp: new Date()
      }
    },
    { sessionId }
  );
  
  // 获取Agent所需的会话历史
  const history = await memory.list({ sessionId });
  
  // 使用Agent处理...
  const response = await agent.run({
    messages: history.map(m => m.message),
    sessionId
  });
  
  return response;
}