dsh-code-ui 1.6.2 → 1.6.3

package/lib/client.js

@@ -1367,7 +1367,6 @@ window.__ModuleLoader__.load({
     }
     function consumePendingRefs() {
       const pend = pendingRefs()
-      if (pend.length === 0) return
       const ids = pend.map((r) => r.id)
       for (let i = 0; i < pend.length; i++) {
         // 冻结发送时刻的编号到记录上：AI 经 get_referenced_snippets 读到的编号 = 用户在 tag 上看到的编号
@@ -1376,37 +1375,10 @@ window.__ModuleLoader__.load({
         pend[i].seq = i + 1
       }
       emit()
-      callHost('use-references', { ids }).then(() => {})
-    }
-    function buildSendPayload(draft) {
-      // 引用文本不粘贴进消息（明面上不展示代码正文）：引用统一存 host 端，AI 通过 get_referenced_snippets
-      // 工具读取；消息里附「结构化指针」——人可读、AI 可读，消息区引用卡也按它解析还原（单一事实源）。
-      // 只取未使用引用（used 的已随旧消息消耗，不再重复附带）。整文件引用：给出绝对路径。
-      const text = draft || ''
-      if (text.indexOf('[code UI] 已标记') >= 0) return text // 幂等：已带指针不重复附加
-      // 编号 = 待用引用创建序（1 起），与输入框 tag、AI 工具输出三处同号：用户说「引用N」AI 即知所指
-      // 已被行内 chip 携带的引用不再自动附带（发送时由 codec.serialize 在 chip 位置展开）。
-      // chip 集合以草稿 occurrences 实时为准（用户删 chip 后自动回到自动附带），chipRefs 为兜底
-      const chipIds = store.draftChipIds || store.chipRefs
-      const pend = numberedPendingRefs().filter((x) => !chipIds || !chipIds.has(x.r.id))
-      const codeRefs = pend.filter((x) => x.r.kind === 'cite' && x.r.lineStart)
-      const fileRefs = pend.filter((x) => x.r.kind === 'cite' && !x.r.lineStart)
-      const trRefs = pend.filter((x) => x.r.kind === 'translate')
-      const items = []
-      for (const x of codeRefs) items.push('- ' + x.n + ': 代码引用 ' + baseName(x.r.path) + ':' + x.r.lineStart + '-' + x.r.lineEnd + '（' + x.r.path + '）')
-      for (const x of fileRefs) items.push('- ' + x.n + ': 文件引用 ' + baseName(x.r.path) + '（整文件，绝对路径：' + x.r.path + '）')
-      for (const x of trRefs) items.push(x.r.lineStart
-        ? '- ' + x.n + ': 翻译引用 ' + baseName(x.r.path) + ':' + x.r.lineStart + '-' + x.r.lineEnd + '（' + x.r.path + '）'
-        : '- ' + x.n + ': 翻译引用 ' + baseName(x.r.path) + '（' + x.r.path + '）')
-      const parts = []
-      if (codeRefs.length > 0) parts.push(codeRefs.length + ' 段代码引用')
-      if (fileRefs.length > 0) parts.push(fileRefs.length + ' 个文件引用')
-      if (trRefs.length > 0) parts.push(trRefs.length + ' 条翻译请求')
-      if (parts.length === 0) return text
-      let ptr = '[code UI] 已标记 ' + parts.join('、') + '（代码正文不进消息，请调用 get_referenced_snippets 工具读取；条目编号与用户输入框上的引用 tag 序号一致，用户说「引用N」即指该条）：'
-      if (items.length > 0) ptr += '\n' + items.join('\n')
-      const sep = text.trim() ? '\n\n' : ''
-      return text + sep + ptr
+      // 每次发送都通知宿主（含空列表=清除该会话的「最近消息引用批次」）：
+      // 批次经宿主 systemPrompt 动态 section 进模型上下文——消息正文不再附加指针文本
+      // （用户要求：对话区不展示引用指针，但 AI 要知道并调用 get_referenced_snippets 读取）。
+      callHost('use-references', { ids: ids, sessionId: store.inputSessionId || '' }).then(() => {})
     }
 
     /* —— 消息区引用卡：解析消息里的结构化指针，在用户气泡内追加 Trae 式引用行 —— */
@@ -1587,18 +1559,10 @@ window.__ModuleLoader__.load({
         patchedShells.add(shell)
         const origSubmit = shell.submit
         shell.submit = function () {
-          let hadRefs = false
-          try {
-            const hasRefs = pendingRefs().length > 0
-            hadRefs = hasRefs
-            const draft = store.inputDraft || ''
-            if (hasRefs && !selfSending && String(draft).indexOf('[code UI]') < 0) {
-              shell.setDraft(buildSendPayload(draft))
-            }
-          } catch (e) {}
-          const result = origSubmit.apply(shell, arguments)
-          try { if (hadRefs && !selfSending) consumePendingRefs() } catch (e) {}
-          return result
+          // 引用不再附加到消息正文：消耗只做两件事——本地标记 used + 通知宿主更新
+          // 「最近消息引用批次」（宿主经 systemPrompt section 透给模型，对话区不展示）。
+          try { if (!selfSending) consumePendingRefs() } catch (e) {}
+          return origSubmit.apply(shell, arguments)
         }
       } catch (e) {}
     }
@@ -1638,21 +1602,9 @@ window.__ModuleLoader__.load({
         const orig = inputActions.submit
         inputActions.__cdexPatched = true
         inputActions.submit = function () {
-          let attached = false
-          let hadRefs = false
-          try {
-            const hasRefs = pendingRefs().length > 0
-            hadRefs = hasRefs
-            const draft = store.inputDraft || ''
-            if (hasRefs && !selfSending && draft.indexOf('[code UI]') < 0) {
-              inputActions.setDraft(buildSendPayload(draft))
-              attached = true
-            }
-          } catch (e) {}
-          const result = orig.apply(inputActions, arguments)
-          // 引用随本条消息消耗：chip 展开（attached=false）与自动附带（attached=true）两条路径都要消耗
-          try { if (hadRefs && !selfSending) consumePendingRefs() } catch (e) {}
-          return result
+          // 引用不进消息正文（指针经宿主 systemPrompt section 只给模型）；这里只消耗。
+          try { if (!selfSending) consumePendingRefs() } catch (e) {}
+          return orig.apply(inputActions, arguments)
         }
         return () => {
           try { inputActions.submit = orig; delete inputActions.__cdexPatched } catch (e) {}
@@ -1690,10 +1642,10 @@ window.__ModuleLoader__.load({
         }
         const text = taEl.value
         if (!text.trim() && refs.length === 0) return
-        const payload = buildSendPayload(text)
+        // 引用不进消息正文（指针经宿主 systemPrompt section 只给模型），原文直发
         selfSending = true
         try {
-          st.inputActions.setDraft(payload)
+          st.inputActions.setDraft(text)
           st.inputActions.submit()
         } finally { selfSending = false }
         // 引用随本条消息消耗（chip 清空；记录留在【引用/备注】已使用区）
@@ -2523,10 +2475,13 @@ window.__ModuleLoader__.load({
                 const r = parseChipRef(refJson)
                 return r ? chipShortLabel(r) : String(refJson)
               },
+              // 发送展开为「空」：chip 不进消息正文（用户要求对话区不展示引用指针）。
+              // AI 侧靠宿主 systemPrompt 动态 section 获知本条消息的引用并调
+              // get_referenced_snippets 读取——编号与 chip tag 一致。无效数据仍透出提示。
               serialize: async (refJson) => {
                 const r = parseChipRef(refJson)
                 if (!r) return '（code UI 引用数据无效: ' + refJson + '）'
-                return formatRefModelLine(r)
+                return ''
               },
             },
           }), 'code-ui: @ 引用源原型'))

package/lib/index.js

@@ -17,6 +17,37 @@ const references = [];
 const notes = [];
 let idSeq = 1;
 
+// 会话「最近一条消息」的引用批次（sessionId → [{ seq, line }]）：发送时由 use-references
+// 写入（空列表 = 清除），systemPrompt 动态 section 在模型上下文组装时按会话读出——
+// 引用指针只进模型上下文（AI 可见、对话区不展示），不再附加到消息正文。
+const lastRefBatches = new Map();
+
+// 引用条目 → 模型侧指针行（与客户端 formatRefModelLine 同格式，编号由批次位置给出）
+function refModelLine(r) {
+  if (r.kind === "translate") {
+    return r.lineStart
+      ? "- 翻译引用 " + baseName(r.path) + ":" + r.lineStart + "-" + r.lineEnd + "（" + r.path + "）"
+      : "- 翻译引用 " + baseName(r.path) + "（" + r.path + "）";
+  }
+  return r.lineStart
+    ? "- 代码引用 " + baseName(r.path) + ":" + r.lineStart + "-" + r.lineEnd + "（" + r.path + "）"
+    : "- 文件引用 " + baseName(r.path) + "（整文件，绝对路径：" + r.path + "）";
+}
+// systemPrompt 动态 section 文本：按组装上下文里的 agent → session 取批次；无批次返回空串。
+function refSectionText(ac) {
+  try {
+    const agent = ac && (ac.agent || ac.scope);
+    const header = agent && agent.session && agent.session.header;
+    const sid = header && header.id ? String(header.id) : "";
+    const batch = sid ? lastRefBatches.get(sid) : undefined;
+    if (!batch || batch.length === 0) return "";
+    return [
+      "用户刚发送的消息中通过 code UI 编辑器标记了以下引用（编号与用户在编辑器引用 tag 上看到的序号一致）。代码正文不进消息正文：请调用 get_referenced_snippets 工具按编号读取（kind: cite=代码引用 / translate=翻译请求 / all=全部；用户说「引用N」即指编号 N）。",
+      ...batch.map((b) => b.line),
+    ].join("\n");
+  } catch (e) { return ""; }
+}
+
 function now() {
   return new Date().toISOString();
 }
@@ -1025,10 +1056,22 @@ function rpcHandler(ctx) {
         // 引用一次性消耗：把本条消息附带的引用标记为 used（不删除——AI 在消息发出后才经
         // get_referenced_snippets 读取，删除就读不到了；used 记录保留供用户在【引用/备注】查找）。
         const ids = Array.isArray(args.ids) ? args.ids : [];
+        const sid = typeof args.sessionId === "string" ? args.sessionId : "";
         let n = 0;
         for (const r of references) {
           if (!r.used && ids.indexOf(r.id) >= 0) { r.used = true; r.usedAt = now(); n++; }
         }
+        // 记录/清除该会话「最近一条消息」的引用批次：模型上下文组装时经 systemPrompt
+        // 动态 section 透出（消息正文不再附加指针文本）。
+        if (sid) {
+          const batch = [];
+          for (let i = 0; i < ids.length; i++) {
+            const r = references.find((x) => x.id === ids[i]);
+            if (r) batch.push({ seq: i + 1, line: refModelLine(r) });
+          }
+          if (batch.length > 0) lastRefBatches.set(sid, batch);
+          else lastRefBatches.delete(sid);
+        }
         return ok({ ok: true, used: n });
       }
 
@@ -1370,6 +1413,18 @@ export function apply(ctx) {
     registerSnippetsTool(c.tools);
   });
 
+  // 模型上下文隐藏 section：引用指针只进 system prompt（AI 可见），不再附加到消息正文。
+  // text 回调在每次模型上下文组装时按组装 agent → session 取「最近一条消息的引用批次」。
+  // 顺序取 2950（工具说明之后、交付引用之前）；无 systemPrompt 服务的宿主静默跳过。
+  ctx.inject(["systemPrompt"], (c) => {
+    try {
+      const sp = c.systemPrompt;
+      if (!sp || typeof sp.section !== "function") return;
+      const dispose = sp.section({ name: "code-ui-references", order: 2950, text: refSectionText });
+      ctx.effect(() => () => { Promise.resolve(dispose()).catch(() => {}); }, "dsh-code-ui: ref prompt section");
+    } catch (e) {}
+  });
+
   // 注册 Typert Remote 服务到宿主 root Context（api-gateway 只扫描它所在 Context 的
   // reflect.props；root 与 gateway 同层，0.1.5-rc.2 / 0.2.0-rc.2 一致）。服务生命周期
   // 挂到本插件 fiber：卸载时自动从 root 摘除，gateway 经 internal/service 事件失效缓存。

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-code-ui",
-  "version": "1.6.2",
+  "version": "1.6.3",
   "description": "Cursor 风格的 AI 代码编辑器工作台插件（DSH）：文件树、多标签页编辑、语法高亮、引用/翻译行内标签给 AI、右侧停靠/悬浮窗口双布局",
   "type": "module",
   "main": "lib/index.js",