jhzhang/rag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: detect table-embedded images as figure intent in post-filter

Browse Source
This commit is contained in:
lacerate551
2026-06-20 22:57:43 +08:00
parent 5bc86a7c1f
commit b5bcb3c941
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/chat_routes.py
View File

@@ -1285,6 +1285,9 @@ def _filter_images_by_answer(selected_images: List[Dict], answer: str, query: st
_has_figure_in_answer = any(kw in answer for kw in _FIGURE_ANSWER_KEYWORDS) _has_figure_in_answer = any(kw in answer for kw in _FIGURE_ANSWER_KEYWORDS)
if not _has_figure_in_answer: if not _has_figure_in_answer:
_has_figure_in_answer = bool(re.search(r'[图表]\s*\d+', answer)) _has_figure_in_answer = bool(re.search(r'[图表]\s*\d+', answer))
# 表格内嵌图片检测：回答中出现 [图片] 占位符或 markdown 图片语法
if not _has_figure_in_answer:
_has_figure_in_answer = bool(re.search(r'\[图片\]|!\[', answer))
_has_figure_in_query = any(kw in query for kw in _FIGURE_QUERY_KEYWORDS) _has_figure_in_query = any(kw in query for kw in _FIGURE_QUERY_KEYWORDS)
if not _has_figure_in_query: if not _has_figure_in_query:
_has_figure_in_query = bool(re.search(r'[图表]\s*\d+', query)) _has_figure_in_query = bool(re.search(r'[图表]\s*\d+', query))