zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: limit chapter title length to 256 characters in pdf_split_handle.py

Browse Source 
--bug=1054363 --user=刘瑞斌 【知识库】导入PDF文档，分段标题长度超长时，没有自动截断 https://www.tapd.cn/57709429/s/1681044
This commit is contained in:
CaptainB 2025-04-07 10:53:11 +08:00 committed by 刘瑞斌
parent 675adeeb63
commit 560890f717
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/common/handle/impl/pdf_split_handle.py
View File

@ -173,14 +173,15 @@ class PdfSplitHandle(BaseSplitHandle):
# Null characters are not allowed. # Null characters are not allowed.
chapter_text = chapter_text.replace('\0', '') chapter_text = chapter_text.replace('\0', '')
# 限制标题长度
real_chapter_title = chapter_title[:256]
# 限制章节内容长度 # 限制章节内容长度
if 0 < limit < len(chapter_text): if 0 < limit < len(chapter_text):
split_text = PdfSplitHandle.split_text(chapter_text, limit) split_text = PdfSplitHandle.split_text(chapter_text, limit)
for text in split_text: for text in split_text:
chapters.append({"title": chapter_title, "content": text}) chapters.append({"title": real_chapter_title, "content": text})
else: else:
chapters.append({"title": chapter_title, "content": chapter_text if chapter_text else chapter_title}) chapters.append({"title": real_chapter_title, "content": chapter_text if chapter_text else real_chapter_title})
# 保存章节内容和章节标题 # 保存章节内容和章节标题
return chapters return chapters