superagent-ai-agent 0.1.0

package/src/query.js

@@ -0,0 +1,239 @@
+/**
+ * src/query.js — SDK 交互层
+ *
+ * 封装与 @anthropic-ai/claude-agent-sdk 的所有交互细节：
+ * - 底层 query 调用（runQuery）
+ * - 流式消费 + 状态收集（consumeQuery）
+ * - 内容审查降级包装器（withFallback）
+ * - Plan JSON 解析（parsePlan）
+ * - 辅助工具函数
+ *
+ * 对上层（agent.js）只暴露 3 个接口：withFallback、phaseModel、parsePlan
+ */
+
+import { readFileSync, readdirSync, statSync } from 'node:fs';
+import path from 'node:path';
+import { query } from '@anthropic-ai/claude-agent-sdk';
+import { webMcpServer } from './web-tool.js';
+import {
+  MODEL_TEXT, MODEL_VISION, MODEL_FALLBACK,
+  EXECUTOR_TOOLS, INSPECTION_RE,
+  PERSONA_PATH, PROJECT_SKILLS_DIR,
+} from './config.js';
+
+// ═══════════════════════════════════════════════
+// 辅助函数
+// ═══════════════════════════════════════════════
+
+/**
+ * 根据是否有图片选择模型
+ * 有图片 → MODEL_VISION（多模态），无图片 → MODEL_TEXT（纯文本）
+ */
+export function phaseModel(images) {
+  return images && images.length ? MODEL_VISION : MODEL_TEXT;
+}
+
+/** 读取 agent-persona.md（每次调用都重读，改完即生效） */
+function loadPersona() {
+  try {
+    return readFileSync(PERSONA_PATH, 'utf8').trim();
+  } catch (e) {
+    if (e.code !== 'ENOENT') console.warn('[persona] 读取失败:', e.message);
+    return '';
+  }
+}
+
+/** 扫描 .claude/skills/ 目录，提取已注册的 skill 名称列表 */
+function listProjectSkills() {
+  try {
+    const entries = readdirSync(PROJECT_SKILLS_DIR, { withFileTypes: true });
+    const names = [];
+    for (const e of entries) {
+      // symlink 处理：readdirSync 的 Dirent 走 lstat，需 statSync 跟随
+      let full;
+      try { full = statSync(path.join(PROJECT_SKILLS_DIR, e.name)); } catch { continue; }
+      if (!full.isDirectory()) continue;
+      try {
+        const content = readFileSync(path.join(PROJECT_SKILLS_DIR, e.name, 'SKILL.md'), 'utf8');
+        const m = content.match(/^---[\s\S]*?\nname:\s*([^\n]+)/);
+        names.push((m?.[1]?.trim()) || e.name);
+      } catch { /* SKILL.md 缺失，跳过 */ }
+    }
+    return names;
+  } catch (e) {
+    if (e.code !== 'ENOENT') console.warn('[skills] 扫描失败:', e.message);
+    return [];
+  }
+}
+
+/** 构建 prompt：纯文本直接返回字符串，带图片则包装为异步可迭代 SDK 用户消息 */
+function buildPrompt(text, images) {
+  if (!images || images.length === 0) return text;
+  const content = [
+    ...images.map((img) => ({
+      type: 'image',
+      source: { type: 'base64', media_type: img.mediaType, data: img.base64 },
+    })),
+    { type: 'text', text: text || '' },
+  ];
+  return (async function* () {
+    yield { type: 'user', message: { role: 'user', content }, parent_tool_use_id: null };
+  })();
+}
+
+/** 从各种错误结构中提取可读文本 */
+function pickErrorText(msg, err) {
+  return [err?.message, err?.stack, msg?.message, msg?.error?.message,
+    typeof msg?.error === 'string' ? msg.error : null,
+    msg?.data, typeof msg?.data === 'string' ? msg.data : null]
+    .filter(Boolean).join('\n');
+}
+
+// ═══════════════════════════════════════════════
+// SDK 调用封装
+// ═══════════════════════════════════════════════
+
+/**
+ * 底层 SDK query 调用
+ * 将业务参数转换为 claude-agent-sdk 的 options 格式
+ */
+function runQuery({ prompt, images, model, sessionId, abortController, tools = EXECUTOR_TOOLS, roleAppend = '' }) {
+  const persona = loadPersona();
+  const projectSkills = listProjectSkills();
+  const append = [persona, roleAppend].filter(Boolean).join('\n');
+  return query({
+    prompt: buildPrompt(prompt, images),
+    options: {
+      permissionMode: 'bypassPermissions',
+      allowDangerouslySkipPermissions: true,
+      tools,
+      mcpServers: { web: webMcpServer },
+      cwd: process.cwd(),
+      settingSources: ['project'],
+      skills: projectSkills,
+      // 关闭扩展思考:模型不再产出 thinking 块,UI 也就没有 💭 思考卡片
+      thinking: { type: 'disabled' },
+      env: {
+        ...process.env,
+        ANTHROPIC_API_KEY: process.env.ANTHROPIC_AUTH_TOKEN,
+        ANTHROPIC_AUTH_TOKEN: undefined,
+        API_TIMEOUT_MS: '300000000',
+        ANTHROPIC_BASE_URL: process.env.ANTHROPIC_BASE_URL,
+        ANTHROPIC_MODEL: model,
+      },
+      systemPrompt: {
+        type: 'preset',
+        preset: 'claude_code',
+        ...(append ? { append } : {}),
+      },
+      ...(sessionId ? { resume: sessionId } : {}),
+      ...(abortController ? { abortController } : {}),
+    },
+  });
+}
+
+/**
+ * 流式消费一次 query，透传所有 msg 给上层
+ *
+ * 返回的 generator 附带 .state 属性，包含：
+ * - text: 累积的 assistant 文本
+ * - sessionId: SDK 返回的会话 ID
+ * - inspectionHit: 是否命中内容审查
+ * - yieldedContent: 是否已产出有效内容
+ * - failures: 失败的 tool_result 错误文本数组
+ */
+function consumeQuery({ prompt, images, model, sessionId, abortController, tools, roleAppend }) {
+  const state = {
+    text: '',
+    sessionId: sessionId ?? null,
+    inspectionHit: false,
+    yieldedContent: false,
+    failures: [],
+  };
+  const it = runQuery({ prompt, images, model, sessionId, abortController, tools, roleAppend });
+  const gen = (async function* () {
+    for await (const msg of it) {
+      // 命中审查的 error 吞掉，交给上层 withFallback 切兜底模型
+      if (msg?.type === 'error' && INSPECTION_RE.test(pickErrorText(msg))) {
+        state.inspectionHit = true;
+        continue;
+      }
+      if (msg?.type !== 'error' && msg?.type !== 'result') state.yieldedContent = true;
+      if (msg?.type === 'assistant') {
+        for (const b of msg.message?.content ?? []) {
+          if (b.type === 'text' && b.text) state.text += b.text;
+        }
+      }
+      if (msg?.type === 'user') {
+        for (const b of msg.message?.content ?? []) {
+          if (b.type === 'tool_result' && b.is_error) {
+            const t = typeof b.content === 'string'
+              ? b.content
+              : (b.content ?? []).map((c) => c.text ?? JSON.stringify(c)).join('\n');
+            state.failures.push(t);
+          }
+        }
+      }
+      if (msg?.type === 'result' && msg.session_id) state.sessionId = msg.session_id;
+      yield msg;
+    }
+  })();
+  gen.state = state;
+  return gen;
+}
+
+// ═══════════════════════════════════════════════
+// 公开接口
+// ═══════════════════════════════════════════════
+
+/**
+ * 统一的内容审查降级包装器
+ *
+ * 先用 primaryModel 跑，若命中审查且无有效输出，自动切 MODEL_FALLBACK 重跑。
+ * 返回最终的 state 对象（text / sessionId / failures 等）。
+ */
+export async function* withFallback({ prompt, images, model, sessionId, abortController, tools, roleAppend }) {
+  const gen = consumeQuery({ prompt, images, model, sessionId, abortController, tools, roleAppend });
+  for await (const msg of gen) yield msg;
+  if (gen.state.inspectionHit && !gen.state.yieldedContent) {
+    console.warn(`[agent] ${model} 命中内容审查,切 ${MODEL_FALLBACK}`);
+    const fb = consumeQuery({ prompt, images, model: MODEL_FALLBACK, sessionId: gen.state.sessionId ?? sessionId, abortController, tools, roleAppend });
+    for await (const msg of fb) yield msg;
+    return fb.state;
+  }
+  return gen.state;
+}
+
+// ═══════════════════════════════════════════════
+// Plan JSON 解析
+// ═══════════════════════════════════════════════
+
+/** 校验 plan 对象结构是否合法 */
+function validatePlan(obj) {
+  if (obj && (typeof obj.needPlan === 'boolean' || Array.isArray(obj.steps))) return obj;
+  return null;
+}
+
+/**
+ * 从 Planner 输出的文本中提取 plan JSON
+ *
+ * 支持 ```plan / ```json 代码块包裹，或裸 JSON。
+ * 严格解析失败后会做宽松重试（去尾逗号、修中文标点）。
+ */
+export function parsePlan(text) {
+  if (!text) return null;
+  const block = text.match(/```(?:plan|json)?\s*(\{[\s\S]*?\})\s*```/);
+  const raw = block ? block[1] : text;
+  const m = raw.match(/\{[\s\S]*\}/);
+  if (!m) return null;
+  let json = m[0];
+  // 先严格解析
+  try { return validatePlan(JSON.parse(json)); } catch { /* strict failed */ }
+  // 宽松重试：去尾逗号、修中文标点
+  json = json
+    .replace(/,\s*([}\]])/g, '$1')
+    .replace(/[\u201c\u201d]/g, '"')
+    .replace(/\uFF1A/g, ':');
+  try { return validatePlan(JSON.parse(json)); } catch { /* lenient failed */ }
+  return null;
+}

package/src/server.js

@@ -0,0 +1,303 @@
+import express from 'express';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { runAgent } from './agent.js';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const PORT = Number(process.env.PORT ?? 3000);
+
+for (const key of ['ANTHROPIC_BASE_URL', 'ANTHROPIC_AUTH_TOKEN']) {
+  if (!process.env[key]) {
+    console.error(`[fatal] 缺少环境变量 ${key}，请在 .env 里配置后再启动`);
+    process.exit(1);
+  }
+}
+
+const app = express();
+// 图片走 base64 走 JSON,放宽到 30MB(单图建议 ≤5MB,base64 后约 7MB)
+app.use(express.json({ limit: '30mb' }));
+app.use(express.static(path.join(__dirname, 'public'), {
+  // HTML 入口强制 no-cache,确保前端代码改动后浏览器每次都拿最新版
+  // (JS/CSS 资源可继续走默认缓存,因为内联在 HTML 里随 HTML 一起更新)
+  setHeaders: (res, filePath) => {
+    if (filePath.endsWith('.html')) {
+      res.setHeader('Cache-Control', 'no-cache, no-store, must-revalidate');
+    }
+  },
+}));
+
+// 把 marked 当 vendor 静态资源对外暴露
+app.get('/vendor/marked.umd.js', (_req, res) => {
+  res.sendFile(path.resolve(__dirname, '../node_modules/marked/lib/marked.umd.js'));
+});
+
+// agent-config.json:Web UI 显示的 agent/user 名称等。每次请求重读,文件改完立刻生效
+const CONFIG_PATH = path.resolve(process.cwd(), 'agent-config.json');
+const DEFAULT_CONFIG = { agentName: '个人助理', userName: '你' };
+app.get('/api/config', async (_req, res) => {
+  try {
+    const fs = await import('node:fs/promises');
+    const text = await fs.readFile(CONFIG_PATH, 'utf8');
+    res.json({ ...DEFAULT_CONFIG, ...JSON.parse(text) });
+  } catch {
+    res.json(DEFAULT_CONFIG);
+  }
+});
+
+// Server 启动时间戳，用于前端自愈：发现变化就 reload
+const SERVER_START_TIME = Date.now();
+app.get('/api/version', (_req, res) => {
+  res.setHeader('Cache-Control', 'no-cache, no-store, must-revalidate');
+  res.json({ startTime: SERVER_START_TIME });
+});
+
+// ---------- 重启后自动自测 ----------
+// Luna 改完代码重启前会写 .self-test-pending.json,server 启动时读到就暴露给前端,
+// 前端 init 时拉取,有 pending 就自动发一条测试消息验证工具是否工作,完成后上报清除
+const SELF_TEST_FILE = path.resolve(process.cwd(), '.self-test-pending.json');
+app.get('/api/self-test', async (_req, res) => {
+  try {
+    const fs = await import('node:fs/promises');
+    const raw = await fs.readFile(SELF_TEST_FILE, 'utf8');
+    res.json(JSON.parse(raw));
+  } catch (e) {
+    if (e?.code === 'ENOENT') return res.json({ pending: false });
+    res.status(500).json({ error: String(e?.message ?? e) });
+  }
+});
+app.post('/api/self-test/done', async (_req, res) => {
+  try {
+    const fs = await import('node:fs/promises');
+    await fs.unlink(SELF_TEST_FILE);
+    res.json({ ok: true });
+  } catch (e) {
+    if (e?.code === 'ENOENT') return res.json({ ok: true });
+    res.status(500).json({ error: String(e?.message ?? e) });
+  }
+});
+
+// 头像图片路由:如果项目根有 luna-avatar.png 之类的图片就 serve 出去
+app.get('/avatar', async (_req, res) => {
+  try {
+    const fs = await import('node:fs/promises');
+    const cfg = JSON.parse(await fs.readFile(CONFIG_PATH, 'utf8'));
+    if (!cfg.avatar) return res.status(404).end();
+    const filePath = path.resolve(process.cwd(), cfg.avatar);
+    const buf = await fs.readFile(filePath);
+    const ext = (cfg.avatar.split('.').pop() || 'png').toLowerCase();
+    const mime = { png: 'image/png', jpg: 'image/jpeg', jpeg: 'image/jpeg', webp: 'image/webp', gif: 'image/gif', svg: 'image/svg+xml' }[ext] || 'application/octet-stream';
+    res.setHeader('Content-Type', mime);
+    res.setHeader('Cache-Control', 'public, max-age=300');
+    res.send(buf);
+  } catch (e) {
+    res.status(404).end();
+  }
+});
+
+// 动态 SVG favicon:取 agentName 第一个字,珊瑚红渐变背景
+function escapeXml(s) {
+  return s.replace(/[&<>"']/g, (c) => ({ '&': '&amp;', '<': '&lt;', '>': '&gt;', '"': '&quot;', "'": '&apos;' }[c]));
+}
+function firstChar(s) {
+  return [...String(s || '')][0] || '';
+}
+// 某些浏览器无视 <link rel="icon"> 仍然请求 /favicon.ico，给它返回 204 静默掉
+app.get('/favicon.ico', (_req, res) => res.status(204).end());
+
+app.get('/favicon.svg', async (_req, res) => {
+  let agentName = DEFAULT_CONFIG.agentName;
+  try {
+    const fs = await import('node:fs/promises');
+    const cfg = JSON.parse(await fs.readFile(CONFIG_PATH, 'utf8'));
+    if (cfg.agentName) agentName = cfg.agentName;
+  } catch { /* 用默认 */ }
+  const ch = escapeXml(firstChar(agentName) || '助');
+  const svg = `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 64 64">
+  <defs>
+    <linearGradient id="g" x1="0" y1="0" x2="1" y2="1">
+      <stop offset="0%" stop-color="#9b87d4"/>
+      <stop offset="100%" stop-color="#f0c2dc"/>
+    </linearGradient>
+  </defs>
+  <rect width="64" height="64" rx="14" fill="url(#g)"/>
+  <text x="32" y="44" text-anchor="middle"
+        font-family="-apple-system,'PingFang SC','Hiragino Sans GB','Microsoft YaHei',system-ui,sans-serif"
+        font-size="36" font-weight="700" fill="#fff">${ch}</text>
+</svg>`;
+  res.setHeader('Content-Type', 'image/svg+xml');
+  res.setHeader('Cache-Control', 'no-cache, no-store, must-revalidate');
+  res.send(svg);
+});
+
+// ---------- 对话历史归档：./sessions/{sessionId}.json ----------
+const SESSIONS_DIR = path.resolve(process.cwd(), 'sessions');
+// sessionId 只允许字母数字、横线、下划线，防路径穿越
+const SESSION_ID_RE = /^[A-Za-z0-9_-]{1,128}$/;
+
+async function ensureSessionsDir() {
+  const fs = await import('node:fs/promises');
+  await fs.mkdir(SESSIONS_DIR, { recursive: true });
+}
+
+app.get('/api/sessions', async (_req, res) => {
+  try {
+    await ensureSessionsDir();
+    const fs = await import('node:fs/promises');
+    const files = (await fs.readdir(SESSIONS_DIR)).filter((f) => f.endsWith('.json'));
+    const items = [];
+    for (const f of files) {
+      try {
+        const raw = await fs.readFile(path.join(SESSIONS_DIR, f), 'utf8');
+        const obj = JSON.parse(raw);
+        items.push({
+          sessionId: obj.sessionId,
+          title: obj.title || '(无标题)',
+          createdAt: obj.createdAt,
+          updatedAt: obj.updatedAt,
+        });
+      } catch { /* 跳过坏文件 */ }
+    }
+    // 按 updatedAt 倒序
+    items.sort((a, b) => (b.updatedAt ?? 0) - (a.updatedAt ?? 0));
+    res.json(items);
+  } catch (e) {
+    res.status(500).json({ error: String(e?.message ?? e) });
+  }
+});
+
+app.get('/api/sessions/:id', async (req, res) => {
+  const { id } = req.params;
+  if (!SESSION_ID_RE.test(id)) return res.status(400).json({ error: 'bad id' });
+  try {
+    const fs = await import('node:fs/promises');
+    const raw = await fs.readFile(path.join(SESSIONS_DIR, `${id}.json`), 'utf8');
+    res.type('application/json').send(raw);
+  } catch (e) {
+    if (e?.code === 'ENOENT') return res.status(404).json({ error: 'not found' });
+    res.status(500).json({ error: String(e?.message ?? e) });
+  }
+});
+
+app.post('/api/sessions', async (req, res) => {
+  const { sessionId, title, history, createdAt } = req.body ?? {};
+  if (!sessionId || !SESSION_ID_RE.test(sessionId)) {
+    return res.status(400).json({ error: 'bad sessionId' });
+  }
+  if (!Array.isArray(history) || history.length === 0) {
+    return res.status(400).json({ error: 'history 不能为空' });
+  }
+  try {
+    await ensureSessionsDir();
+    const fs = await import('node:fs/promises');
+    const now = Date.now();
+    const filePath = path.join(SESSIONS_DIR, `${sessionId}.json`);
+    // 同一 sessionId 已存在 → 保留旧 createdAt
+    let prevCreatedAt = null;
+    try {
+      const old = JSON.parse(await fs.readFile(filePath, 'utf8'));
+      if (typeof old?.createdAt === 'number') prevCreatedAt = old.createdAt;
+    } catch { /* 文件不存在或损坏，忽略 */ }
+    const payload = {
+      sessionId,
+      title: (typeof title === 'string' && title.trim()) || '(无标题)',
+      createdAt: prevCreatedAt ?? (typeof createdAt === 'number' ? createdAt : now),
+      updatedAt: now,
+      history,
+    };
+    await fs.writeFile(filePath, JSON.stringify(payload, null, 2), 'utf8');
+    res.json({ ok: true, sessionId });
+  } catch (e) {
+    res.status(500).json({ error: String(e?.message ?? e) });
+  }
+});
+
+app.delete('/api/sessions/:id', async (req, res) => {
+  const { id } = req.params;
+  if (!SESSION_ID_RE.test(id)) return res.status(400).json({ error: 'bad id' });
+  try {
+    const fs = await import('node:fs/promises');
+    await fs.unlink(path.join(SESSIONS_DIR, `${id}.json`));
+    res.json({ ok: true });
+  } catch (e) {
+    if (e?.code === 'ENOENT') return res.status(404).json({ error: 'not found' });
+    res.status(500).json({ error: String(e?.message ?? e) });
+  }
+});
+
+const ALLOWED_IMAGE_TYPES = new Set(['image/jpeg', 'image/png', 'image/gif', 'image/webp']);
+const MAX_IMAGES = 8;
+
+app.post('/api/chat', async (req, res) => {
+  const { prompt, sessionId, images } = req.body ?? {};
+  const text = typeof prompt === 'string' ? prompt : '';
+  const imgs = Array.isArray(images) ? images : [];
+
+  if (!text.trim() && imgs.length === 0) {
+    res.status(400).json({ error: 'prompt 和 images 至少要有一个' });
+    return;
+  }
+  if (imgs.length > MAX_IMAGES) {
+    res.status(400).json({ error: `图片最多 ${MAX_IMAGES} 张` });
+    return;
+  }
+  for (const img of imgs) {
+    if (!img || !ALLOWED_IMAGE_TYPES.has(img.mediaType) || typeof img.base64 !== 'string') {
+      res.status(400).json({ error: `图片格式不合法,支持 ${[...ALLOWED_IMAGE_TYPES].join('/')}` });
+      return;
+    }
+  }
+
+  res.setHeader('Content-Type', 'text/event-stream');
+  res.setHeader('Cache-Control', 'no-cache');
+  res.setHeader('Connection', 'keep-alive');
+  res.flushHeaders?.();
+
+  const send = (event, data) => {
+    res.write(`event: ${event}\n`);
+    res.write(`data: ${JSON.stringify(data)}\n\n`);
+  };
+
+  const abortController = new AbortController();
+  // 注意:req.on('close') 是「请求体读完」就触发(express.json 一读完就 fire),不能用
+  // 真正的「客户端断开」用 res.on('close'),并用 writableEnded 排除"我们自己 end 掉"的正常完成
+  res.on('close', () => {
+    if (!res.writableEnded && !abortController.signal.aborted) {
+      console.log('[chat] 客户端断开 → 中止 agent');
+      abortController.abort();
+    }
+  });
+
+  try {
+    for await (const msg of runAgent({ prompt: text, sessionId, images: imgs, abortController })) {
+      // plan-execute 阶段事件单独走专用 SSE event,前端渲染成独立卡片
+      if (msg?.type === 'plan') send('plan', msg);
+      else if (msg?.type === 'reflection') send('reflection', msg);
+      else send('message', msg);
+      if (msg?.type === 'result') {
+        send('done', { sessionId: msg.session_id });
+      }
+    }
+    try { res.end(); } catch {}
+  } catch (err) {
+    if (abortController.signal.aborted) {
+      console.log('[chat] 已中止');
+      try { res.end(); } catch {}
+      return;
+    }
+    console.error('[chat] agent 异常:', err);
+    try { send('error', { message: err?.message ?? String(err) }); res.end(); } catch {}
+  }
+});
+
+app
+  .listen(PORT, () => {
+    console.log(`Listening on http://localhost:${PORT}`);
+  })
+  .on('error', (err) => {
+    if (err.code === 'EADDRINUSE') {
+      console.error(`[fatal] 端口 ${PORT} 被占用，换一个：PORT=${PORT + 1} npm start`);
+    } else {
+      console.error('[fatal]', err);
+    }
+    process.exit(1);
+  });