diff --git a/api/chat_routes.py b/api/chat_routes.py
index a78e031..3373e1a 100644
--- a/api/chat_routes.py
+++ b/api/chat_routes.py
@@ -1373,12 +1373,20 @@ _FIGURE_ANSWER_KEYWORDS = frozenset([
def _replace_table_image_placeholders(answer: str, images: List[Dict]) -> str:
"""
- 将回答中 markdown 表格内的 [图片] 占位符替换为实际图片语法。
+ 将回答中 markdown 表格内的图片占位符替换为实际图片语法。
- LLM 看到的上下文里表格单元格是 [图片](MinerU 解析产物),它也原样输出。
- 此函数按顺序将每个 [图片] 映射到 images 列表中对应图片的 URL,
+ LLM 看到的上下文里表格单元格图片以 [图片] 或 [N张图片](MinerU 解析产物)表示,
+ LLM 通常会原样输出,有时还会加上  格式。
+
+ 此函数按顺序将占位符映射到 images 列表中的 URL,
生成  语法,前端 markdown-it 可直接渲染为 标签。
+ 处理的占位符格式:
+ - [图片] →  (单张)
+ -  →  (LLM 加了 ! 和假 URL)
+ - [3张图片] →   
+ -  → 同上 (LLM 加了前缀和假 URL)
+
Args:
answer: LLM 生成的回答文本
images: select_images + _filter_images_by_answer 后的最终图片列表
@@ -1386,28 +1394,35 @@ def _replace_table_image_placeholders(answer: str, images: List[Dict]) -> str:
Returns:
替换后的回答文本
"""
- if not images or '[图片]' not in answer:
+ if not images or '图片' not in answer:
return answer
- # 按顺序替换:第 N 个 [图片] → 第 N 张图片
img_iter = iter(images)
replaced_count = 0
def _replacer(match):
nonlocal replaced_count
- try:
- img = next(img_iter)
- url = img.get('url', '')
- img_id = img.get('id', 'img')
- replaced_count += 1
- return f''
- except StopIteration:
- # 图片用完了,保留剩余占位符
- return match.group(0)
+ n_str = match.group(1) # 数字部分,None 表示单张
+ n = int(n_str) if n_str else 1
- result = re.sub(r'\[图片\]', _replacer, answer)
+ parts = []
+ for _ in range(n):
+ try:
+ img = next(img_iter)
+ url = img.get('url', '')
+ img_id = img.get('id', 'img')
+ parts.append(f'')
+ replaced_count += 1
+ except StopIteration:
+ parts.append('[图片]') # 图片用完了
+ return ' '.join(parts)
+
+ # !? — 可选前导 !(LLM 有时加)
+ # \[(\d+)?张?图片\] — [图片] 或 [3张图片](张 可选)
+ # (?:\(图片URL\))? — 可选的 LLM 从上下文复制的假 URL
+ result = re.sub(r'!?\[(\d+)?张?图片\](?:\(图片URL\))?', _replacer, answer)
if replaced_count > 0:
- logger.info(f"[图片占位符替换] 替换了 {replaced_count}/{len(images)} 个 [图片] 为 markdown 图片语法")
+ logger.info(f"[图片占位符替换] 替换了 {replaced_count}/{len(images)} 个图片为 markdown 图片语法")
return result
diff --git a/knowledge/manager.py b/knowledge/manager.py
index 66b2f2f..0fd8fe6 100644
--- a/knowledge/manager.py
+++ b/knowledge/manager.py
@@ -39,6 +39,7 @@ from pathlib import Path
import logging
import chromadb
+from bs4 import BeautifulSoup
# 从 base.py 导入基础类和常量
from .base import (
@@ -550,8 +551,39 @@ class KnowledgeBaseManager(
curr_html = getattr(current, 'table_html', '') or ''
next_html = getattr(next_chunk, 'table_html', '') or ''
if curr_html and next_html:
- # 合并两个表格的 HTML
- current.table_html = curr_html + '\n' + next_html
+ # 正确合并两个表格的 HTML:
+ # 将第二个表格的