dsh-data-cleaning-agent 0.7.0 → 0.8.0

package/lib/client.js

@@ -347,6 +347,78 @@ window.__ModuleLoader__.load({
   font-size: 12px;
 }
 .dcAgentPromptFile input { max-width: 100%; }
+.dcAgentPromptImageDrop {
+  display: grid;
+  gap: 10px;
+  padding: 12px;
+  border: 1px dashed light-dark(#b9c5d8, #43516a);
+  border-radius: 12px;
+  background: light-dark(#fbfcfe, #11171f);
+  outline: none;
+}
+.dcAgentPromptImageDrop.is-active {
+  border-color: #2869e6;
+  background: light-dark(#edf4ff, #172841);
+}
+.dcAgentPromptImageDrop > p {
+  margin: 0;
+  color: light-dark(#697386, #aab5c7);
+  font-size: 11px;
+  line-height: 17px;
+}
+.dcAgentPromptImagePreview {
+  display: flex;
+  align-items: center;
+  gap: 10px;
+  min-width: 0;
+}
+.dcAgentPromptImageThumb {
+  width: 64px;
+  height: 64px;
+  flex: 0 0 auto;
+  overflow: hidden;
+  padding: 0;
+  border: 1px solid light-dark(#d9dee8, #394352);
+  border-radius: 10px;
+  background: light-dark(#f3f6fa, #202733);
+  cursor: zoom-in;
+}
+.dcAgentPromptImageThumb img { width: 100%; height: 100%; object-fit: cover; display: block; }
+.dcAgentPromptImageMeta { min-width: 0; flex: 1 1 auto; }
+.dcAgentPromptImageMeta b,
+.dcAgentPromptImageMeta small { display: block; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
+.dcAgentPromptImageMeta small { margin-top: 3px; color: light-dark(#697386, #aab5c7); }
+.dcAgentPromptImageRemove {
+  flex: 0 0 auto;
+  padding: 5px 9px;
+  border: 1px solid light-dark(#d9dee8, #394352);
+  border-radius: 8px;
+  background: transparent;
+  color: inherit;
+  font: inherit;
+  font-size: 11px;
+  cursor: pointer;
+}
+.dcAgentImageLightbox {
+  position: fixed;
+  inset: 0;
+  z-index: 1400;
+  display: grid;
+  place-items: center;
+  padding: 32px;
+  border: 0;
+  background: rgba(8, 12, 20, .76);
+  cursor: zoom-out;
+  pointer-events: auto;
+}
+.dcAgentImageLightbox img {
+  display: block;
+  max-width: min(1100px, 92vw);
+  max-height: 88vh;
+  object-fit: contain;
+  border-radius: 12px;
+  box-shadow: 0 24px 80px rgba(0, 0, 0, .4);
+}
 .dcAgentPromptGroup { margin-top: 12px; }
 .dcAgentPromptGroup > b { display: block; margin-bottom: 7px; font-size: 12px; }
 .dcAgentPromptChoices { display: flex; flex-wrap: wrap; gap: 6px; }
@@ -982,7 +1054,10 @@ window.__ModuleLoader__.load({
       const executionDraft = value.startsWith('请执行已在「数据清洗补全工作台」确认的企业数据任务。')
         && value.includes('安全任务凭证：dcq-')
         && value.includes('data_cleaning_qcc_run');
-      return wizardDraft || executionDraft;
+      const imageDraft = value.startsWith('请识别我刚刚在向导中安全暂存的企业名单图片')
+        && value.includes('安全图片凭证：dci-')
+        && value.includes('data_cleaning_extract_image_companies');
+      return wizardDraft || executionDraft || imageDraft;
     }
 
     function clearCleaningDraft(ctx, sessionId, onlyDefault = false) {
@@ -1119,7 +1194,7 @@ window.__ModuleLoader__.load({
         : [];
       const entryCount = Number.isFinite(config.entryCount) ? config.entryCount : entries.length;
       const source = mode === 'image'
-        ? `已附加图片${config.fileName ? `「${config.fileName}」` : ''}`
+        ? `图片${config.fileName ? `「${config.fileName}」` : ''}，已识别并人工核验 ${entryCount} 条主体标识`
         : mode === 'excel'
           ? `本地表格${config.fileName ? `「${config.fileName}」` : ''}，已解析 ${entryCount} 条主体标识，完整数据已载入数据清洗补全工作台`
           : `手工录入，共 ${entryCount} 条主体标识`;
@@ -1127,9 +1202,6 @@ window.__ModuleLoader__.load({
         '请执行一项企业名单数据清洗补全任务。',
         `输入来源：${source}。`,
       ];
-      if (mode === 'image') {
-        lines.push('请先使用当前已连接且可用的企查查智能文档解析 MCP 能力，从图片中提取企业全称或统一社会信用代码；若该能力未连接或不可用，请先提示我改用文本或 Excel。');
-      }
       if (entries.length) {
         lines.push(`${mode === 'excel' ? '主体预览' : '待处理主体'}：\n${entries.map((entry, index) => `${index + 1}. ${entry}`).join('\n')}`);
       }
@@ -1523,8 +1595,8 @@ window.__ModuleLoader__.load({
     }
 
     /**
-     * 图片接入兼容 Bridge。仅在运行时同时探测到 createDraftImages / input.shell.addImages
-     * 时使用；失败会释放浏览器临时对象，不假设 alpha 实验 API 稳定。
+     * 图片接入兼容 Bridge。返回官方 ComposerAttachment，供向导复用其 previewUrl
+     * 显示 64px 缩略图；失败会释放浏览器临时对象，不假设 alpha 实验 API 稳定。
      */
     function attachPromptImages(ctx, sessionId, files) {
       const conversation = typeof ctx.get === 'function' ? ctx.get('conversation') : ctx.conversation;
@@ -1540,7 +1612,17 @@ window.__ModuleLoader__.load({
         conversation.releaseDraftImages?.(images);
         throw new Error('图片未能加入当前对话，请检查格式或数量限制。');
       }
-      return images.length;
+      return images;
+    }
+
+    function removePromptImage(ctx, sessionId, attachment) {
+      const conversation = typeof ctx.get === 'function' ? ctx.get('conversation') : ctx.conversation;
+      const id = attachment?.id;
+      if (!conversation || !id) return false;
+      const shell = conversation.input?.shell?.(sessionId);
+      shell?.removeImage?.(id);
+      conversation.releaseDraftImage?.(id);
+      return true;
     }
 
     async function sendQccAgentCommand(ctx, sessionId, prompt) {
@@ -1873,8 +1955,30 @@ window.__ModuleLoader__.load({
       }).filter(Boolean).slice(0, 50);
     }
 
+    async function imageFileBase64(file) {
+      if (!file || typeof file.arrayBuffer !== 'function') throw new Error('图片文件不可读。');
+      const bytes = new Uint8Array(await file.arrayBuffer());
+      let binary = '';
+      for (let offset = 0; offset < bytes.length; offset += 0x8000) {
+        binary += String.fromCharCode(...bytes.subarray(offset, offset + 0x8000));
+      }
+      return btoa(binary);
+    }
+
+    async function stagePromptImage(file) {
+      if (!['image/png', 'image/jpeg', 'image/webp'].includes(String(file?.type ?? ''))) {
+        throw new Error('仅支持 PNG、JPEG 或 WebP 图片。');
+      }
+      if (Number(file?.size ?? 0) > 8 * 1024 * 1024) throw new Error('图片不能超过 8 MiB。');
+      return api('/data-cleaning/api/images/commands', {
+        fileName: file.name,
+        mimeType: file.type,
+        content: await imageFileBase64(file),
+      });
+    }
+
     function PromptGenerator(props) {
-      const { sessionId, inputActions, attachImages } = props;
+      const { sessionId, inputActions, attachImages, removeImage } = props;
       const enabled = useCleaningSession(sessionId);
       const [open, setOpen] = react.useState(false);
       const [wizardStep, setWizardStep] = react.useState(1);
@@ -1884,7 +1988,11 @@ window.__ModuleLoader__.load({
       const [entryCount, setEntryCount] = react.useState(0);
       const [spreadsheetFileName, setSpreadsheetFileName] = react.useState('');
       const [imageFileName, setImageFileName] = react.useState('');
-      const [imageAttached, setImageAttached] = react.useState(false);
+      const [imageSizeBytes, setImageSizeBytes] = react.useState(0);
+      const [imageAttachment, setImageAttachment] = react.useState(null);
+      const [imageCommand, setImageCommand] = react.useState(null);
+      const [imageDropActive, setImageDropActive] = react.useState(false);
+      const [imageLightbox, setImageLightbox] = react.useState(false);
       const [cleaningKeys, setCleaningKeys] = react.useState(DEFAULT_CLEANING_KEYS);
       const [enrichmentKeys, setEnrichmentKeys] = react.useState(DEFAULT_ENRICHMENT_KEYS);
       const [anchorKeys, setAnchorKeys] = react.useState(['company_name', 'credit_no']);
@@ -1897,38 +2005,127 @@ window.__ModuleLoader__.load({
       const [busy, setBusy] = react.useState(false);
       const [error, setError] = react.useState(null);
 
-      if (!enabled) return null;
+      react.useEffect(() => {
+        const commandId = imageCommand?.commandId;
+        if (!commandId || ['completed', 'failed'].includes(imageCommand?.state)) return undefined;
+        let disposed = false;
+        let timer = null;
+        const poll = async () => {
+          try {
+            const response = await api(`/data-cleaning/api/images/commands/${encodeURIComponent(commandId)}`);
+            if (disposed) return;
+            const command = response?.command;
+            if (!command) return;
+            setImageCommand((current) => current?.commandId === commandId ? { ...current, ...command } : current);
+            if (command.state === 'completed') {
+              const recognized = Array.isArray(command.result?.entries) ? command.result.entries : [];
+              setEntries(recognized);
+              setEntryCount(recognized.length);
+              requestWorkbenchDataset(entriesToDataset(recognized), sessionId, {
+                type: 'image', fileName: imageFileName, sizeBytes: imageSizeBytes,
+              });
+              setError(null);
+              setWizardStep(2);
+              setOpen(true);
+              return;
+            }
+            if (command.state === 'failed') {
+              setError(command.error?.message || '图片识别失败，请换用清晰图片或文本/Excel 名单。');
+              setOpen(true);
+              return;
+            }
+            timer = setTimeout(poll, 1000);
+          } catch (pollError) {
+            if (!disposed) {
+              setError(pollError instanceof Error ? pollError.message : String(pollError));
+              setOpen(true);
+            }
+          }
+        };
+        timer = setTimeout(poll, 800);
+        return () => {
+          disposed = true;
+          if (timer !== null) clearTimeout(timer);
+        };
+      }, [imageCommand?.commandId, imageCommand?.state, imageFileName, imageSizeBytes, sessionId]);
 
+      const discardCurrentImage = () => {
+        if (imageAttachment && typeof removeImage === 'function') removeImage(sessionId, imageAttachment);
+        if (imageCommand?.commandId && imageCommand.state !== 'running') {
+          requestJson(`/data-cleaning/api/images/commands/${encodeURIComponent(imageCommand.commandId)}`, 'DELETE').catch(() => {});
+        }
+        setImageAttachment(null);
+        setImageCommand(null);
+        setImageFileName('');
+        setImageSizeBytes(0);
+        setEntries([]);
+        setEntryCount(0);
+        setImageLightbox(false);
+      };
       const setSourceMode = (nextMode) => {
+        if (mode === 'image' && nextMode !== 'image') {
+          if (imageCommand?.state === 'running') {
+            setError('图片正在识别，请等待完成后再切换数据来源。');
+            return;
+          }
+          discardCurrentImage();
+        }
         setMode(nextMode);
         setError(null);
       };
       const toggle = (setter, values, key) => setter(values.includes(key)
         ? values.filter((item) => item !== key)
         : [...values, key]);
+      const handleImageFile = async (file) => {
+        if (!file || busy) return;
+        setBusy(true);
+        setError(null);
+        let attached = null;
+        try {
+          if (typeof attachImages !== 'function') throw new Error('当前 DSH 没有可用的图片接入 Bridge。');
+          const images = await attachImages(sessionId, [file]);
+          attached = Array.isArray(images) ? images[0] : null;
+          if (!attached?.id) throw new Error('图片未能加入当前对话。');
+          const staged = await stagePromptImage(file);
+          if (imageAttachment && typeof removeImage === 'function') removeImage(sessionId, imageAttachment);
+          if (imageCommand?.commandId) {
+            requestJson(`/data-cleaning/api/images/commands/${encodeURIComponent(imageCommand.commandId)}`, 'DELETE').catch(() => {});
+          }
+          setImageAttachment(attached);
+          setImageFileName(file.name);
+          setImageSizeBytes(Number(file.size ?? 0));
+          setImageCommand(staged.command);
+          setEntries([]);
+          setEntryCount(0);
+        } catch (fileError) {
+          if (attached && typeof removeImage === 'function') removeImage(sessionId, attached);
+          throw fileError;
+        } finally {
+          setBusy(false);
+        }
+      };
       const handleFile = async (event) => {
         const file = event.target?.files?.[0];
         if (!file || busy) return;
+        if (mode === 'image') {
+          try { await handleImageFile(file); }
+          catch (fileError) { setError(fileError instanceof Error ? fileError.message : String(fileError)); }
+          finally { if (event.target) event.target.value = ''; }
+          return;
+        }
         setBusy(true);
         setError(null);
         try {
-          if (mode === 'image') {
-            if (typeof attachImages !== 'function') throw new Error('当前 DSH 没有可用的图片接入 Bridge。');
-            await attachImages(sessionId, [file]);
-            setImageFileName(file.name);
-            setImageAttached(true);
-          } else {
-            const result = await parseFile(file);
-            if (!result || result.ok === false) throw new Error(result?.message || result?.error || '表格解析失败');
-            const extracted = extractPromptEntries(result);
-            if (!extracted.length) throw new Error('未识别到企业名称或统一社会信用代码列，请检查表头。');
-            setEntries(extracted);
-            setEntryCount(Number.isFinite(result.rowCount) ? result.rowCount : (Array.isArray(result.rows) ? result.rows.length : extracted.length));
-            setSpreadsheetFileName(file.name);
-            requestWorkbenchDataset(result, sessionId, {
-              type: result.fmt || 'xlsx', fileName: file.name, sizeBytes: file.size || 0,
-            });
-          }
+          const result = await parseFile(file);
+          if (!result || result.ok === false) throw new Error(result?.message || result?.error || '表格解析失败');
+          const extracted = extractPromptEntries(result);
+          if (!extracted.length) throw new Error('未识别到企业名称或统一社会信用代码列，请检查表头。');
+          setEntries(extracted);
+          setEntryCount(Number.isFinite(result.rowCount) ? result.rowCount : (Array.isArray(result.rows) ? result.rows.length : extracted.length));
+          setSpreadsheetFileName(file.name);
+          requestWorkbenchDataset(result, sessionId, {
+            type: result.fmt || 'xlsx', fileName: file.name, sizeBytes: file.size || 0,
+          });
         } catch (fileError) {
           setError(fileError instanceof Error ? fileError.message : String(fileError));
         } finally {
@@ -1943,7 +2140,9 @@ window.__ModuleLoader__.load({
         const values = selectedEntries();
         if (mode === 'text' && !values.length) return '请先录入企业名称或统一社会信用代码，每行一条。';
         if (mode === 'excel' && !values.length) return '请先上传并解析 Excel / CSV / JSON 文件。';
-        if (mode === 'image' && !imageAttached) return '请先选择并附加一张包含企业名单的图片。';
+        if (mode === 'image' && !imageCommand) return '请先选择、粘贴或拖入一张包含企业名单的图片。';
+        if (mode === 'image' && imageCommand?.state !== 'completed') return '请先把图片识别指令回填到对话框并发送，等待名单识别完成。';
+        if (mode === 'image' && !values.length) return '图片中未提取到可核验的企业名称或统一社会信用代码。';
         return null;
       };
       const promptConfig = () => ({
@@ -1984,6 +2183,22 @@ window.__ModuleLoader__.load({
 
       const nextWizard = () => {
         if (wizardStep === 1) {
+          if (mode === 'image' && imageCommand?.state === 'prepared') {
+            if (!inputActions || typeof inputActions.setDraft !== 'function') {
+              setError('当前会话输入机尚未就绪，请稍后重试。');
+              return;
+            }
+            // Host 已持久暂存图片；发送识别命令前释放 Composer 附件，避免不支持
+            // 视觉输入的文本模型拒绝本轮。识别工具只按 commandId 读取 Host 暂存副本。
+            if (imageAttachment && typeof removeImage === 'function') {
+              removeImage(sessionId, imageAttachment);
+              setImageAttachment(null);
+            }
+            inputActions.setDraft(imageCommand.prompt);
+            setError(null);
+            setOpen(false);
+            return;
+          }
           const sourceError = validateSource();
           if (sourceError) { setError(sourceError); return; }
         }
@@ -1993,6 +2208,47 @@ window.__ModuleLoader__.load({
         generatePrompt();
       };
 
+      const acceptImageFiles = async (files) => {
+        const file = [...(files || [])].find((candidate) => String(candidate?.type ?? '').startsWith('image/'));
+        if (!file) { setError('剪贴板或拖入内容中没有可用图片。'); return; }
+        try { await handleImageFile(file); }
+        catch (imageError) { setError(imageError instanceof Error ? imageError.message : String(imageError)); }
+      };
+      const removeCurrentImage = async () => {
+        if (busy) return;
+        discardCurrentImage();
+        setError(null);
+      };
+
+      // 在数据清洗会话中从 window 捕获阶段接管 Composer 图片粘贴，早于
+      // document 级视觉兼容插件。向导未打开时会自动进入图片模式；只消费包含
+      // 图片且目标为原生 Composer 的事件，不影响文字粘贴、右侧工作台或其他会话。
+      react.useEffect(() => {
+        if (!enabled) return undefined;
+        const handleWindowPaste = (event) => {
+          const files = event.clipboardData?.files;
+          const hasImage = [...(files || [])].some((file) => String(file?.type ?? '').startsWith('image/'));
+          if (!hasImage) return;
+          const target = event.target;
+          const inComposer = !open && Boolean(target?.closest?.('textarea, [contenteditable="true"], [role="textbox"]'));
+          const inImageStep = open && wizardStep === 1 && mode === 'image';
+          if (!inComposer && !inImageStep) return;
+          event.preventDefault();
+          event.stopImmediatePropagation?.();
+          if (!open) {
+            setMode('image');
+            setWizardStep(1);
+            setOpen(true);
+            setError(null);
+          }
+          acceptImageFiles(files);
+        };
+        window.addEventListener?.('paste', handleWindowPaste, true);
+        return () => window.removeEventListener?.('paste', handleWindowPaste, true);
+      }, [enabled, open, wizardStep, mode, busy, sessionId]);
+
+      if (!enabled) return null;
+
       const choice = (options, values, setter) => options.map(([key, label]) => h('label', {
         key,
         className: `dcAgentPromptChoice${values.includes(key) ? ' is-selected' : ''}`,
@@ -2067,6 +2323,55 @@ window.__ModuleLoader__.load({
                 placeholder: '企查查科技股份有限公司\n9132…\n某某信息技术（上海）有限公司',
                 onChange: (event) => setRawText(event.target.value),
               }),
+            ) : mode === 'image' ? h('div', {
+              className: `dcAgentPromptImageDrop${imageDropActive ? ' is-active' : ''}`,
+              tabIndex: 0,
+              role: 'group',
+              'aria-label': '粘贴、拖入或选择企业名单图片',
+              onPaste: (event) => {
+                const files = event.clipboardData?.files;
+                if (files?.length) {
+                  event.preventDefault();
+                  acceptImageFiles(files);
+                }
+              },
+              onDragOver: (event) => { event.preventDefault(); setImageDropActive(true); },
+              onDragLeave: () => setImageDropActive(false),
+              onDrop: (event) => {
+                event.preventDefault();
+                setImageDropActive(false);
+                acceptImageFiles(event.dataTransfer?.files);
+              },
+            },
+              h('p', null, '可直接在此粘贴图片，或拖入 / 选择 PNG、JPEG、WebP。图片会在对话框显示缩略图，点击可放大。'),
+              imageAttachment ? h('div', { className: 'dcAgentPromptImagePreview' },
+                h('button', {
+                  type: 'button', className: 'dcAgentPromptImageThumb',
+                  'aria-label': `放大查看 ${imageFileName}`,
+                  onClick: () => setImageLightbox(true),
+                }, imageAttachment.previewUrl ? h('img', { src: imageAttachment.previewUrl, alt: imageFileName }) : '图片'),
+                h('div', { className: 'dcAgentPromptImageMeta' },
+                  h('b', null, imageFileName),
+                  h('small', null, imageCommand?.state === 'completed'
+                    ? `已识别 ${entryCount} 条，下一步逐条核验`
+                    : imageCommand?.state === 'running'
+                      ? '识别中…'
+                      : imageCommand?.state === 'failed'
+                        ? '识别失败，可移除后重试'
+                        : '已暂存，下一步生成可编辑识别指令'),
+                ),
+                h('button', { type: 'button', className: 'dcAgentPromptImageRemove', disabled: busy || imageCommand?.state === 'running', onClick: removeCurrentImage }, '移除'),
+              ) : h('label', { className: 'dcAgentPromptFile' },
+                h('span', null, '选择包含企业名单的图片'),
+                h('input', {
+                  type: 'file', accept: 'image/png,image/jpeg,image/webp',
+                  disabled: busy, onChange: handleFile,
+                }),
+              ),
+              imageCommand?.state === 'completed' && entries.length ? h('div', { className: 'dcAgentPromptChoices' },
+                entries.slice(0, 12).map((entry, index) => h('span', { key: `${entry}-${index}`, className: 'dcAgentPromptChoice is-selected' }, `${index + 1}. ${entry}`)),
+                entries.length > 12 ? h('span', { className: 'dcAgentPromptChoice' }, `另 ${entries.length - 12} 条`) : null,
+              ) : null,
             ) : h('label', { className: 'dcAgentPromptFile' },
               h('span', null, mode === 'image' ? '选择包含企业名单的图片' : '选择 CSV / XLSX / JSON 名单'),
               h('input', {
@@ -2118,9 +2423,20 @@ window.__ModuleLoader__.load({
           error ? h('p', { className: 'dcAgentPromptError', role: 'alert' }, error) : null,
           h('div', { className: 'dcAgentPromptActions' },
             h('button', { type: 'button', className: 'dcAgentPromptAction', disabled: wizardStep === 1, onClick: () => setWizardStep(Math.max(1, wizardStep - 1)) }, '上一步'),
-            h('button', { type: 'button', className: 'dcAgentPromptAction is-primary', disabled: busy, onClick: nextWizard }, busy ? '处理中…' : wizardStep === 4 ? '回填到对话框' : '下一步'),
+            h('button', { type: 'button', className: 'dcAgentPromptAction is-primary', disabled: busy || imageCommand?.state === 'running', onClick: nextWizard }, busy
+              ? '处理中…'
+              : wizardStep === 1 && mode === 'image' && imageCommand?.state === 'prepared'
+                ? '回填图片识别指令'
+                : wizardStep === 1 && mode === 'image' && imageCommand?.state === 'running'
+                  ? '图片识别中…'
+                  : wizardStep === 4 ? '回填到对话框' : '下一步'),
           ),
         ) : null,
+        imageLightbox && imageAttachment?.previewUrl ? h('button', {
+          type: 'button', className: 'dcAgentImageLightbox',
+          'aria-label': '关闭图片预览',
+          onClick: () => setImageLightbox(false),
+        }, h('img', { src: imageAttachment.previewUrl, alt: imageFileName })) : null,
       );
     }
 
@@ -2143,11 +2459,12 @@ window.__ModuleLoader__.load({
       );
     }
 
-    /** M3 · 三工具 tool.call.toolview 卡片元数据（wire 名 → 展示文案）。 */
+    /** 模型工具 tool.call.toolview 卡片元数据（wire 名 → 展示文案）。 */
     const TOOL_VIEW_META = {
       data_clean_rows: { icon: '🧹', label: '数据清洗', hint: '去重 · 剔除非法金额 · 缺失补 0' },
       data_complete_rows: { icon: '🧩', label: '数据补全', hint: '名称 / 金额 / 手机号归一' },
       data_profile: { icon: '🩺', label: '数据体检', hint: '缺失率 · 去重值 · 金额分布' },
+      data_cleaning_extract_image_companies: { icon: '🖼️', label: '图片名单识别', hint: 'OCR · 企业名称 / 信用代码提取 · 人工核验' },
     };
     const TOOL_VIEW_STATE = { running: '运行中', stopped: '已停止', error: '失败', ok: '完成' };
 
@@ -3419,7 +3736,10 @@ window.__ModuleLoader__.load({
           name: 'conversation.input.overlay',
           id: 'data-cleaning-agent-prompt-generator',
           order: 110,
-          inject: () => ({ attachImages: (sessionId, files) => attachPromptImages(ctx, sessionId, files) }),
+          inject: () => ({
+            attachImages: (sessionId, files) => attachPromptImages(ctx, sessionId, files),
+            removeImage: (sessionId, attachment) => removePromptImage(ctx, sessionId, attachment),
+          }),
         }, PromptGenerator));
 
         // 会话头部恢复入口：用户关闭右栏后可随时重新打开当前任务工作台。
@@ -3429,9 +3749,9 @@ window.__ModuleLoader__.load({
           order: 110,
         }, WorkbenchHeaderEntry));
 
-        // M3：三个工具的 tool.call.toolview 富化卡片（keyed by wire name，替代裸 JSON 摘要）。
+        // 模型工具的 tool.call.toolview 富化卡片（keyed by wire name，替代裸 JSON 摘要）。
         // 说明：分三条独立 inject（而非 generator）——测试 shim 对 inject 回调仅执行一次并 push 其返回值。
-        const toolviewKeys = ['data_clean_rows', 'data_complete_rows', 'data_profile'];
+        const toolviewKeys = ['data_clean_rows', 'data_complete_rows', 'data_profile', 'data_cleaning_extract_image_companies'];
         for (const key of toolviewKeys) {
           ctx.slots.inject('tool.call.toolview', () => ctx.slots.register({
             name: 'tool.call.toolview',