xshat-lite 1.0.0

package/src/modules/chat-helpers.mjs

@@ -0,0 +1,275 @@
+export function sanitizeMessage(message) {
+  return String(message ?? '')
+    .replace(/[\x00-\x08\x0B\x0C\x0E-\x1F\x7F]/g, '')
+    .replace(/\r/g, '');
+}
+
+export function stripProviderWrapper(text) {
+  return String(text ?? '').replace(
+    /^\s*(?:Gemini|AI|Assistant)\s*(?:说|:|：)\s*/i,
+    ''
+  );
+}
+
+function clipText(text, maxChars = 160) {
+  const normalized = sanitizeMessage(stripProviderWrapper(String(text ?? '')))
+    .replace(/\s+/g, ' ')
+    .trim();
+
+  if (normalized.length <= maxChars) {
+    return normalized;
+  }
+
+  return `${normalized.slice(0, Math.max(0, maxChars - 1))}…`;
+}
+
+export function buildCompleteConfig(steps = {}) {
+  const complete = steps.complete || {};
+  const readSelectors = steps.read?.selectors || [];
+
+  return {
+    mode: complete.mode || 'script',
+    script: complete.script || null,
+    selectors: {
+      response: complete.selectors?.response || readSelectors,
+      pending: complete.selectors?.pending || [],
+      done: complete.selectors?.done || [],
+      interimTexts: complete.selectors?.interimTexts || []
+    }
+  };
+}
+
+export function normalizeResult(result) {
+  if (!result || typeof result !== 'object') {
+    return {
+      text: '',
+      pending: false,
+      done: false,
+      error: null,
+      count: 0,
+      transient: false
+    };
+  }
+
+  return {
+    text: typeof result.text === 'string' ? result.text : '',
+    pending: Boolean(result.pending),
+    done: Boolean(result.done),
+    error: result.error ? String(result.error) : null,
+    count: Number.isFinite(result.count) ? result.count : 0,
+    transient: Boolean(result.transient)
+  };
+}
+
+export function summarizeResult(result) {
+  const normalized = normalizeResult(result);
+
+  return {
+    pending: normalized.pending,
+    done: normalized.done,
+    error: normalized.error,
+    count: normalized.count,
+    transient: normalized.transient,
+    preview: normalized.text.replace(/\s+/g, ' ').slice(0, 120)
+  };
+}
+
+export function deriveEffectiveText(result, baseline = {}) {
+  const normalized = normalizeResult(result);
+  const safeBaseline = {
+    count: baseline.count ?? 0,
+    text: baseline.text ?? ''
+  };
+
+  if (normalized.count > safeBaseline.count) {
+    if (safeBaseline.text && normalized.text.startsWith(safeBaseline.text)) {
+      return normalized.text.slice(safeBaseline.text.length);
+    }
+    return normalized.text;
+  }
+
+  if (
+    normalized.count === safeBaseline.count &&
+    safeBaseline.text &&
+    normalized.text.startsWith(safeBaseline.text)
+  ) {
+    return normalized.text.slice(safeBaseline.text.length);
+  }
+
+  if (
+    normalized.count === safeBaseline.count &&
+    safeBaseline.text !== normalized.text
+  ) {
+    return normalized.text;
+  }
+
+  return '';
+}
+
+export function compressConversationContext(messages = [], {
+  compressThreshold = 12,
+  keepRecentMessages = 6,
+  maxSummaryChars = 1200,
+  maxMessageChars = 160
+} = {}) {
+  const safeMessages = Array.isArray(messages) ? messages : [];
+  if (safeMessages.length <= compressThreshold) {
+    return {
+      compressed: false,
+      summary: '',
+      recentMessages: safeMessages
+    };
+  }
+
+  const keepCount = Math.max(2, keepRecentMessages);
+  const recentMessages = safeMessages.slice(-keepCount);
+  const olderMessages = safeMessages.slice(0, Math.max(0, safeMessages.length - keepCount));
+  const lines = [];
+  let totalChars = 0;
+
+  for (const message of olderMessages) {
+    const role = message.role === 'user' ? 'User' : 'Assistant';
+    const line = `${role}: ${clipText(message.content, maxMessageChars)}`;
+    const nextLength = totalChars + line.length + 1;
+
+    if (nextLength > maxSummaryChars) {
+      lines.push('...');
+      break;
+    }
+
+    lines.push(line);
+    totalChars = nextLength;
+  }
+
+  return {
+    compressed: true,
+    summary: lines.join('\n'),
+    recentMessages
+  };
+}
+
+export function buildHistoryPrimer(messages = [], options = {}) {
+  const context = compressConversationContext(messages, options);
+  const historyText = context.recentMessages
+    .map((message) => {
+      const role = message.role === 'user' ? 'User' : 'Assistant';
+      return `${role}: ${stripProviderWrapper(message.content)}`;
+    })
+    .join('\n\n');
+
+  const sections = [
+    '以下是我们之前的对话记录，请记住这些内容作为上下文，不需要对此回复，只需确认你已了解:'
+  ];
+
+  if (context.summary) {
+    sections.push(`较早对话摘要:\n${context.summary}`);
+  }
+
+  if (historyText) {
+    sections.push(`最近对话原文:\n\n${historyText}`);
+  }
+
+  sections.push('请回复"已了解"即可，等待我的新问题。');
+  return sections.join('\n\n');
+}
+
+export function buildAgentPrimer(systemPrompt, messages = [], options = {}) {
+  const context = compressConversationContext(messages, options);
+  const historyText = context.recentMessages
+    .map((message) => {
+      const role = message.role === 'user' ? 'User' : 'Assistant';
+      return `${role}: ${stripProviderWrapper(message.content)}`;
+    })
+    .join('\n\n');
+
+  return [
+    systemPrompt,
+    context.summary ? `以下是较早上下文摘要:\n${context.summary}` : '',
+    historyText ? `以下是最近上下文原文:\n\n${historyText}` : '',
+    '请仅回复符合要求的 JSON。'
+  ]
+    .filter(Boolean)
+    .join('\n\n');
+}
+
+export function wrapAgentTurnMessage(message, { providerId = '' } = {}) {
+  const normalizedProviderId = String(providerId).trim().toLowerCase();
+  const providerHints = [];
+
+  if (normalizedProviderId === 'deepseek' || normalizedProviderId === 'api-deepseek') {
+    providerHints.push(
+      '你当前运行在 DeepSeek 通道。不要输出“根据当前协议我无法访问互联网”“我无法直接浏览网页”“没有搜索工具”。',
+      '如果用户问当前目录、文件是否存在、创建文件、写入内容，这通常是本地任务，不是网页任务，请优先返回文件或 shell 命令。',
+      '不要把路径、域名或文件名混淆。像 app.txt 是文件，baidu.com 更可能是网址，当前目录/项目目录是本地工作区。'
+    );
+  }
+
+  const content = sanitizeMessage(message).trim();
+
+  return [
+    '继续按照既定 Agent 协议工作。',
+    '如果你看不到更早的系统约束，也要继续严格遵守当前这条消息里的协议，不要要求我重新说明格式。',
+    '本轮也只能输出一个 JSON 对象，不要输出任何额外说明。',
+    '如果你输出 reply，请优先先给结果，再补充 2-6 条必要要点；允许在 text 字段中使用简洁 Markdown。',
+    '如果内容适合对比、排行、统计或赛程，请优先用紧凑列表、小表格或“项目: 值”形式，不要生成列很多的大表格。',
+    '面向终端输出时，优先使用: 短结论 + 项目列表；若有对比信息，再用 2-4 列的小表格；每个单元格尽量短，避免横向过宽。',
+    '如果内容包含步骤、建议、优缺点、预测依据、时间线或候选项，请使用编号列表或项目符号列表，不要堆成长段落。',
+    '如果是在做预测、推荐、对比、方案选择或参数说明，优先输出“结论 + 列表”或 2-4 列小表格，让 CLI 里一眼能扫完。',
+    '不要先声明“无法联网”“无法实时”“没有搜索工具”“只是单步 Agent”。先判断是否可以用 open_url、read_agent_page、extract_selector_text、write_file、run_shell 等宿主能力推进。',
+    '如果只需回答，用 {"type":"reply","text":"...","command":null}。',
+    '如果需要宿主执行一步命令，用 {"type":"action","text":"...","command":{"name":"...","args":{}}}。',
+    '如果需要人工处理登录、验证、授权或肉眼判断，用 {"type":"handoff","text":"...","command":null}。',
+    '如果当前能力不足或信息不足，用 {"type":"error","text":"...","command":null}。',
+    '如果你想在网页输入内容，请使用 type_selector，不要使用 input_text / fill_text / fill_input 这些别名。',
+    '如果你想点击网页元素，请使用 click_selector，不要发明 click_element 等别名。',
+    ...providerHints,
+    '',
+    `用户本轮消息如下:\n${content}`
+  ].join('\n');
+}
+
+export function buildAgentFollowupMessage(result, {
+  step = 1,
+  maxSteps = 1,
+  completion = null
+} = {}) {
+  let parsedOutput = result?.output || '';
+  if (typeof parsedOutput === 'string') {
+    try {
+      parsedOutput = JSON.parse(parsedOutput);
+    } catch {
+      parsedOutput = parsedOutput;
+    }
+  }
+
+  const receipt = {
+    ok: Boolean(result?.ok),
+    command: result?.commandName || null,
+    handoff: Boolean(result?.handoff),
+    message: result?.message || '',
+    output: parsedOutput
+  };
+  const lines = [
+    `上一条命令已经执行完毕。这是第 ${step}/${maxSteps} 步的执行结果。`,
+    '请先根据结果判断是否已经完成用户目标。',
+    '如果已经完成，请直接输出 reply，并优先先给结果，再给必要要点；text 字段允许使用简洁 Markdown。',
+    '如果要展示结构化结果，优先小表格、列表或“项目: 值”，避免终端里很宽的大表格。',
+    '终端阅读优先级: 结论一句话 > 2-6 条要点 > 必要时再加短表格，不要先写冗长铺垫。',
+    '如果仍需下一步，只输出一个 action，并保持最小必要动作。',
+    '除非已经确认存在真实阻塞，否则不要直接输出 error，也不要先说自己无法联网、无法持续执行或没有搜索能力。',
+    '如果需要人工处理登录、验证、授权或页面异常，请输出 handoff。',
+    '如果当前信息仍不足，请优先 get_app_state、read_agent_page、extract_selector_text 或 inspect_dom。',
+    '若需要输入网页内容，请使用 type_selector；若需要点击网页元素，请使用 click_selector。',
+    '',
+    `操作回执(JSON):\n${JSON.stringify(receipt, null, 2)}`
+  ];
+
+  if (completion) {
+    lines.push(
+      '',
+      `宿主判断(JSON):\n${JSON.stringify(completion, null, 2)}`
+    );
+  }
+
+  return lines.join('\n');
+}