从 800 行到 1000 行：我的 Agent 实现了分层记忆，接入了 MCP 能力，还暴露了 WebUI

tools/
├── index.js        # 工具注册中心
├── filesystem.js   # 文件系统（readFile, writeFile, editFile...）
├── shell.js        # Shell 命令（runCommand, searchContent）
├── plan.js         # 计划管理（createPlan, updatePlanStep...）
├── lint.js         # 代码检查（lintFile, lintFiles）
└── utils.js        # 通用工具（getCurrentTime, think）

// tools/index.js
const toolModules = [filesystem, shell, plan, utils, lint];

// 自动收集所有工具
const toolRegistry = {};
for (const module of toolModules) {
  for (const [name, tool] of Object.entries(module)) {
    if (tool?.schema && tool?.execute) {
      toolRegistry[tool.schema.function.name] = tool;
    }
  }
}

// 之前：整体覆盖（危险、费 token）
writeFile({ filePath: "app.js", content: "整个文件的所有内容..." })

// 现在：搜索替换（精确、安全）
editFile({
  filePath: "app.js",
  oldText: "const name = 'old'",
  newText: "const name = 'new'"
})

// 检查是否有多处匹配
const matches = content.split(oldText).length - 1;
if (matches > 1) {
return`替换失败: 找到 ${matches} 处匹配，请提供更多上下文使其唯一。`;
}

// 找不到时给出有用的错误提示
if (!content.includes(oldText)) {
const firstLine = oldText.split("\n")[0].trim();
if (content.includes(firstLine)) {
    return`找到了相似内容，但不完全匹配。请检查空格、缩进。`;
  }
}

// 根据文件类型选择检查方式
switch (ext) {
case".js":
    return`node --check "${filePath}"`;  // JavaScript
case".ts":
    return`npx tsc --noEmit "${filePath}"`;  // TypeScript
case".py":
    return`python3 -m py_compile "${filePath}"`;  // Python
case".json":
    return`node -e "JSON.parse(require('fs').readFileSync('${filePath}'))"`;
}

getMessages() {
  return this.messages;  // 全部发送
}

# AGENT.md - 项目上下文

## 技术栈
- Node.js (ESM 模块)
- DeepSeek API (LLM)

## 编码规范
- 使用 ES6+ 语法
- 注释使用中文

// 配置
constCONFIG = {
SUMMARIZE_THRESHOLD: 10,  // 超过 10 轮触发摘要
KEEP_RECENT_TURNS: 5,     // 保留最近 5 轮完整消息
};

// 增量摘要流程
asynccheckAndSummarize() {
// 1. 判断是否需要摘要
if (userMessageCount <= SUMMARIZE_THRESHOLD) return;

// 2. 只摘要 [summarizedUpTo, startIndex] 范围的新消息
const messagesToSummarize = messages.slice(this.summarizedUpTo, startIndex);

// 3. 生成摘要
const newSummary = awaitthis.memory.generateSummary(messagesToSummarize);

// 4. 合并到已有摘要
this.summary = this.summary ? `${this.summary}\n\n${newSummary}` : newSummary;

// 5. 更新 checkpoint
this.summarizedUpTo = startIndex;
}

buildContextMessages({ systemPrompt, summary, recentMessages }) {
const messages = [];

// 1. 系统提示词 + AGENT.md（长期记忆）
let fullSystemPrompt = systemPrompt;
if (agentMd) {
    fullSystemPrompt = `<project_context>\n${agentMd}\n</project_context>\n\n${systemPrompt}`;
  }
  messages.push({ role: "system", content: fullSystemPrompt });

// 2. 会话摘要（中期记忆）
if (summary) {
    messages.push({
      role: "system",
      content: `<conversation_summary>\n${summary}\n</conversation_summary>`,
    });
  }

// 3. 最近的消息（短期记忆）
  messages.push(...recentMessages);

return messages;
}

agent/
├── core/
│   └── agent.js      # Agent 核心逻辑
├── index.js          # CLI 入口
└── server.js         # Web 服务入口

class AgentextendsEventEmitter {
asyncchat(userMessage) {
    this.emit('assistant_start');
    
    forawait (const chunk of stream) {
      this.emit('assistant_chunk', { content: chunk });
    }
    
    if (toolCalls.length > 0) {
      this.emit('tool_call', { name, args, id });
      const result = awaitexecuteTool(name, args);
      this.emit('tool_result', { id, result });
    }
    
    this.emit('assistant_end');
  }
}

// CLI
agent.on('assistant_chunk', ({ content }) => {
  process.stdout.write(content);
});

// Web Server (WebSocket)
agent.on('assistant_chunk', ({ content }) => {
  ws.send(JSON.stringify({ type: 'assistant_chunk', payload: { content } }));
});

web/
├── src/
│   ├── App.jsx
│   ├── components/
│   │   ├── ChatPanel.jsx   # 对话区域
│   │   ├── MessageItem.jsx # 消息气泡
│   │   └── Sidebar.jsx     # 侧边栏（会话列表、工具列表）
│   └── hooks/
│       └── useWebSocket.js # WebSocket 封装
├── vite.config.js
└── tailwind.config.js

// useWebSocket.js
const { sendMessage, lastMessage, connectionStatus } = useWebSocket('ws://localhost:3001/ws');

// 处理流式消息
case 'assistant_chunk':
  setMessages(prev => {
    const newMessages = [...prev];
    const lastMsg = newMessages[newMessages.length - 1];
    if (lastMsg?.role === 'assistant') {
      lastMsg.content += payload.content;  // 逐字追加
    }
    return newMessages;
  });

// ~/.agent/mcp.json
{
"servers":{
    "filesystem":{
      "command":"npx",
      "args":["-y","@anthropic/mcp-server-filesystem","/path/to/dir"]
    }
}
}

// mcp-client.js
classMCPServerextendsEventEmitter {
asyncconnect() {
    // 启动子进程
    this.process = spawn(command, args, { stdio: ["pipe", "pipe", "pipe"] });
    
    // JSON-RPC 通信
    awaitthis.sendRequest("initialize", { capabilities: {} });
    
    // 获取工具列表
    const { tools } = awaitthis.sendRequest("tools/list");
    this.tools = tools;
  }

asynccallTool(name, arguments) {
    returnawaitthis.sendRequest("tools/call", { name, arguments });
  }
}