降低自主AI代理计算成本的智能多模型路由架构设计

标准代理流程（高成本）：
[ Claude Code 顶级模型 ] ---> 完整读取 1500 行源代码 ---> 消耗 12000 Tokens ---> 成本高昂

优化后的路由流程（Spotify 架构）：
[ Claude Code 顶级模型 ] ---> 被 Shunt 钩子拦截 ---> [ Gemini 2.5 Flash ] ---> 生成结构化摘要 (500 Tokens) ---> 低成本

                         开发者 Prompt 指令
                                │
                                ▼
  ┌───────────────────────────────────────────────────────────┐
  │                       Claude Code                         │
  │                    (顶级前沿推理模型)                       │
  └─────────────────────────────┬─────────────────────────────┘
                                │ 拦截文件读取工具调用
                                ▼
                     ┌─────────────────────┐
                     │   动态阈值判断:     │
                     │ 文件行数 > 350 行?  │
                     └──────────┬──────────┘
                                │
                ┌───────────────┴───────────────┐
              是│                               │否
                ▼                               ▼
  ┌───────────────────────────┐   ┌───────────────────────────┐
  │    Bulk-Reader 模式       │   │    直接读取文件内容       │
  │   (Gemini 2.5 Flash)      │   │  (载入 Claude 上下文)     │
  └─────────────┬─────────────┘   └───────────────────────────┘
                │ 生成代码摘要
                ▼
  ┌───────────────────────────┐
  │  精简 AST 结构化摘要      │
  │  返回给 Claude 快速决策   │
  └───────────────────────────┘

import fs from 'fs';
import readline from 'readline';
import { OpenAI } from 'openai';

// 初始化 n1n.ai 统一多模型 API 客户端
const client = new OpenAI({
  baseURL: 'https://api.n1n.ai/v1',
  apiKey: process.env.N1N_API_KEY || '',
});

interface ReadFileOptions {
  filePath: string;
}

/**
 * 拦截 Agent 的文件读取工具调用，按代码行数动态路由
 */
export async function interceptedReadFile(options: ReadFileOptions): Promise<string> {
  const { filePath } = options;

  if (!fs.existsSync(filePath)) {
    throw new Error(`文件不存在: ${filePath}`);
  }

  const lineCount = await calculateLineCount(filePath);
  const LINE_THRESHOLD = 350; // 设置拦截阈值为 350 行

  if (lineCount > LINE_THRESHOLD) {
    console.log(`[Shunt 拦截器] 文件超过 ${LINE_THRESHOLD} 行（当前 ${lineCount} 行）。已启动 Bulk-Reader 降级模式。`);
    return await runBulkReaderMode(filePath);
  }

  // 小文件直接返回原内容给主模型
  return fs.readFileSync(filePath, 'utf-8');
}

/**
 * 使用轻量级模型生成代码结构摘要
 */
async function runBulkReaderMode(filePath: string): Promise<string> {
  const fileContent = fs.readFileSync(filePath, 'utf-8');

  const completion = await client.chat.completions.create({
    model: 'google/gemini-2.5-flash', // 通过 n1n.ai 聚合平台调用低成本模型
    messages: [
      {
        role: 'system',
        content: `你是一个代码结构化提取分析器。请仅提取并输出以下内容：
1. 核心导出接口、类及函数签名；
2. 全局变量与关键状态变更逻辑；
3. 核心函数的业务逻辑概括。
严禁输出具体的实现细节与琐碎代码，将回复字数严格控制在 400 字以内。`,
      },
      {
        role: 'user',
        content: `分析以下源代码文件：

${fileContent}`,
      },
    ],
    temperature: 0.1,
  });

  const summary = completion.choices[0].message.content;
  return `[系统提示：该文件已由 Bulk-Reader 子代理精简，以下为结构化摘要]

${summary}`;
}

function calculateLineCount(filePath: string): Promise<number> {
  return new Promise((resolve) => {
    let count = 0;
    const rl = readline.createInterface({
      input: fs.createReadStream(filePath),
      crlfDelay: Infinity,
    });
    rl.on('line', () => count++);
    rl.on('close', () => resolve(count));
  });
}