import { LettaClient, Agent } from '@letta-ai/sdk';

// 初始化 Letta 客户端
const client = new LettaClient({
  apiKey: process.env.LETTA_API_KEY,
  baseUrl: process.env.LETTA_SERVER_URL
});

// 创建后台代理
const agent = new Agent(client, {
  name: 'claude-subconscious',
  description: '后台编码助手，观察并学习编码模式',
  tools: ['read_file', 'write_file', 'run_command'],
  memory: {
    type: 'persistent',  // 持久化记忆
    maxSize: 10000       // 最大记忆条目数
  }
});

class SubconsciousAgent {
  // 代理核心工作流程
  async lifecycle() {
    // 1. 初始化：加载配置和工具集
    await this.loadConfig();
    await this.registerTools();

    // 2. 进入事件循环
    while (this.isActive) {
      // 监听 -> 感知 -> 思考 -> 行动 -> 记忆
      const event = await this.observe();
      const context = this.recall(event);
      const insight = await this.think(event, context);
      if (insight.confidence > 0.8) {
        await this.whisper(insight);
      }
      this.remember(event, insight);
    }
  }
}

# 步骤 1: 克隆仓库
git clone https://github.com/letta-ai/claude-subconscious.git
cd claude-subconscious

# 步骤 2: 安装依赖
npm install

# 步骤 3: 配置环境变量
cp .env.example .env
# 编辑 .env，填入：
#   LETTA_API_KEY=your_api_key
#   LETTA_SERVER_URL=https://api.letta.com
#   LETTA_MODE=observe   (observe=silent, assist=proactive)

# 步骤 4: 启用插件
claude plugin enable .

# 步骤 5: 验证安装
claude plugin list # 应显示 subconscious (active)

import { Agent, EventTypes } from '@letta-ai/sdk';

class CodeReviewListener {
  constructor(private agent: Agent) {}

  // 监听文件保存事件
  @on(EventTypes.FILE_SAVE)
  async onFileSave(event: FileSaveEvent) {
    const changes = await this.diff(event.filePath);

    // 分析代码变更
    const analysis = await this.agent.think({
      prompt: `分析以下代码变更，检测潜在问题：
        文件: ${event.filePath}
        变更: ${changes}`,
      context: this.agent.recall(event.filePath)
    });

    // 高风险时立即提醒
    if (analysis.riskLevel === 'high') {
      await this.agent.whisper({
        message: analysis.warning,
        priority: 'high'
      });
    }

    // 记录到记忆系统
    this.agent.remember({
      type: 'code_change',
      file: event.filePath,
      summary: analysis.summary
    });
  }
}

class TestSuggestAgent extends Agent {
  """自动建议测试的专用代理"""

  name = 'test-suggester';
  tools = ['read_file', 'run_command'];

  async onFileSave(event: FileSaveEvent) {
    // 只处理源代码文件
    if (!event.filePath.match(/\.(ts|js|py)$/)) return;

    // 检查是否已有对应测试文件
    const testPath = this.getTestPath(event.filePath);
    const hasTest = await this.tools.read_file(testPath);

    if (!hasTest) {
      const code = await this.tools.read_file(event.filePath);
      const testCode = await this.generateTest(code);
      await this.whisper({
        message: `检测到 ${event.filePath} 没有测试。建议添加：`,
        suggestion: testCode
      });
    }
  }
}

class OptimizedAgent extends Agent {
  private eventBuffer: Event[] = [];
  private analysisCache = new Map<string, Analysis>();

  async onFileSave(event: FileSaveEvent) {
    // 1. 事件合并：积累 5 秒内的变更
    this.eventBuffer.push(event);
    await this.debounce(5000);

    // 2. 缓存检查：相似代码复用分析
    const cacheKey = this.hashChanges(event);
    if (this.analysisCache.has(cacheKey)) {
      return this.analysisCache.get(cacheKey);
    }

    // 3. 模型路由：简单判断用 Haiku
    const model = event.isComplex
      ? 'claude-sonnet-4-20250514'
      : 'claude-haiku-4-20250414';

    // 4. 批量分析
    const analysis = await this.batchAnalyze(
      this.eventBuffer, { model }
    );

    this.analysisCache.set(cacheKey, analysis);
    this.eventBuffer = [];
    return analysis;
  }
}

class CodeReviewAgent extends Agent {
  """代码审查代理 — 请补充完整实现"""

  name = 'code-reviewer';

  // TODO 1: 注册事件监听器
  async onFileSave(event: FileSaveEvent) {
    pass
  }

  // TODO 2: 分析代码质量
  async analyzeQuality(code: string): Promise {
    pass
  }

  // TODO 3: 生成改进建议
  async generateSuggestions(analysis: Analysis): Promise {
    pass
  }

  // TODO 4: 更新记忆
  updateMemory(result: ReviewResult) {
    pass
  }
}