YueHuihui
/
RAGflow
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #162 from zstar1003/dev 

refactor(api): 优化图片插入的 Markdown 格式，增加最大宽度限制
This commit is contained in:
zstar 2025-06-12 12:20:41 +08:00 committed by GitHub
parent 9852b32715 a6b55ba533
commit d888c4a8d1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/db/services/dialog_service.py
View File

@ -270,8 +270,8 @@ def chat(dialog, messages, stream=True, **kwargs):
processed_image_urls.add(img_url) processed_image_urls.add(img_url)
inserted_images[idx] = img_url inserted_images[idx] = img_url
# 插入图片，不加任何括号包裹引用标记 # 插入图片，并限制最大宽度
return f"{match.group(0)}\n\n![img_url]({img_url})" return f'{match.group(0)}\n\n<img src="{img_url}" alt="{img_url}" style="max-width:800px;">'
# 用正则替换插图 # 用正则替换插图
answer = re.sub(r"##(\d+)\$\$", insert_image_markdown, answer) answer = re.sub(r"##(\d+)\$\$", insert_image_markdown, answer)

2
api/db/services/write_service.py
View File

@ -92,7 +92,7 @@ def write_dialog(question, kb_ids, tenant_id, similarity_threshold, keyword_simi
if img_url not in image_urls: if img_url not in image_urls:
image_urls.add(img_url) image_urls.add(img_url)
image_markdowns.append(f"\n![{img_url}]({img_url})") image_markdowns.append(f'\n<img src="{img_url}" alt="{img_url}" style="max-width:500px; display:block; margin:auto;">')
if image_markdowns: if image_markdowns:
final_answer += "".join(image_markdowns) final_answer += "".join(image_markdowns)