上下文工程

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

// 默认上下文工程 (推荐)
const memory = new InMemoryMemory();

// 禁用自动上下文管理
const manualMemory = new InMemoryMemory({
  enableContextManagement: false  // 禁用自动上下文管理
});

// 带上下文工程的自定义配置
const memoryWithConfig = new InMemoryMemory({
  enableContextManagement: true,   // 启用自动上下文管理 (默认)
  thresholds: {
    preRotThreshold: 150_000,    // 性能退化点
    compactionTrigger: 0.8,      // 在 80% 时压缩 (120K tokens)
    summarizationTrigger: 0.95,   // 在 95% 时摘要 (142.5K tokens)
    recentToKeep: 10,             // 始终保留最后 10 条消息
  }
});

const memory = new InMemoryMemory({
  enableContextManagement: false  // 禁用自动管理
});

// 添加事件而不进行自动上下文管理
await memory.add(event1, { sessionId: 'session-123' });
await memory.add(event2, { sessionId: 'session-123' });
await memory.add(event3, { sessionId: 'session-123' });

// 在需要时手动触发上下文管理
await memory.manageContext('session-123');

import { BaseMemory, IMemoryEvent, StructuredSummary } from '@ag-kit/agents/storage';

// 方法 1: 在配置中传递摘要器
const customMemory = new BaseMemory({
  summarizer: async (events: IMemoryEvent[]): Promise<StructuredSummary> => {
    // 自定义摘要逻辑
    return {
      modifiedFiles: extractFiles(events),
      userGoal: extractUserGoal(events),
      lastStep: getLastStep(events),
      keyDecisions: extractDecisions(events),
      criticalContext: { /* 自定义上下文 */ },
      timestamp: new Date()
    };
  },
  thresholds: {
    preRotThreshold: 100_000,
    compactionTrigger: 0.8,
    summarizationTrigger: 0.95
  }
});

// 方法 2: 扩展 BaseMemory 以进行高级自定义
class CustomMemory extends BaseMemory {
  constructor(config?: any) {
    super({
      ...config
    });
  }

  // 覆盖单个事件的压缩逻辑
  protected async compactEvent(event: IMemoryEvent): Promise<IMemoryEvent> {
    // 自定义压缩逻辑 - 示例: 压缩代码块
    if (event.message.content.includes('```')) {
      event.state.__compaction__ = {
        originalContent: event.message.content,
        compactedAt: new Date().toISOString(),
        tokensSaved: /* 计算节省的 tokens */,
        strategy: 'code_compression'
      };
      event.message.content = '[COMPACTED: 代码讨论]';
      return event
    } else {
      return super.compactEvent(event);
    }
  }

  // 自定义摘要函数
  private async summarizer(events: IMemoryEvent[]): Promise<StructuredSummary> {
    return {
      count: events.length,
      timeRange: {
        start: events[0].message.timestamp,
        end: events[events.length - 1].message.timestamp
      },
      timestamp: new Date()
    };
  }

  // 覆盖存储方法 (可选)
  protected async storeSummary(sessionId: string, summary: StructuredSummary): Promise<void> {
    // 自定义存储逻辑
    console.log(`为 ${sessionId} 存储摘要`);
  }

  protected async clearSummarizedEvents(sessionId: string, recentToKeep: number): Promise<void> {
    // 自定义清理逻辑
    console.log(`为 ${sessionId} 清理事件`);
    await super.clearSummarizedEvents(sessionId, recentToKeep);
  }

  // 添加事件后触发上下文管理
  async add(event: IMemoryEvent): Promise<void> {
    await super.add(event);
    const sessionId = event.state.sessionId || 'default';
    await this.manageContext(sessionId);
  }
}

// 初始问题 (正常操作 - 第 1 层)
await memory.add({
  message: {
    role: 'user',
    content: '我们的生产数据库在 30 秒后超时,影响了 1200 个用户'
  },
  state: {
    issue: 'db_timeout',
    severity: 'critical',
    environment: 'production'
  }
});

// 50 条消息后,上下文达到 120K tokens
// 🟡 系统自动触发第 2 层 (压缩)
// 早期消息被压缩但仍可恢复:

const compactedEvent = {
  message: { content: '[COMPACTED: 数据库超时问题报告]' },
  state: {
    issue: 'db_timeout',
    severity: 'critical',
    __compaction__: {
      originalContent: '我们的生产数据库在 30 秒后超时...',
      tokensSaved: 23,
      compactedAt: '2024-01-15T10:30:00Z'
    }
  }
};

// 200 条消息后,上下文达到 142.5K tokens
// 🔴 系统触发第 3 层 (结构化摘要)
const summary = {
  modifiedFiles: ['database.ts', 'user-service.ts', 'auth-middleware.ts'],
  userGoal: '修复生产环境中的数据库连接超时',
  lastStep: '更新了连接池配置',
  keyDecisions: [
    '将池大小从 10 增加到 50',
    '添加了连接重试逻辑',
    '实现了断路器模式'
  ],
  criticalContext: {
    errorPattern: '30 秒后连接超时',
    environment: 'production',
    affectedUsers: 1200,
    urgency: 'critical'
  }
};

// 最近的消息 (最后 5 条) 保持完整细节以保持上下文连续性