Session Management

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

// Create memory instance
const memory = new InMemoryMemory();

// Create session for user 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 }
);

// Create session for user 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 }
);

// Retrieve session-specific history
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"

// User-based: one session per user
const sessionId = `user-${userId}`;

// Conversation-based: multiple sessions per user
const sessionId = `user-${userId}-conv-${conversationId}`;

// Time-based: sessions with timestamps
const sessionId = `user-${userId}-${Date.now()}`;

// Hierarchical: complex organization structure
const sessionId = `org-${orgId}/team-${teamId}/user-${userId}`;

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

const memory = new InMemoryMemory();

// Agent handles request with session isolation
async function handleUserMessage(userId: string, message: string) {
  const sessionId = `user-${userId}`;
  
  // Add user message to session
  await memory.add(
    {
      message: {
        id: crypto.randomUUID(),
        role: 'user',
        content: message,
        timestamp: new Date()
      }
    },
    { sessionId }
  );
  
  // Get session history for agent
  const history = await memory.list({ sessionId });
  
  // Process with agent...
  const response = await agent.run({
    messages: history.map(m => m.message),
    sessionId
  });
  
  return response;
}