15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: The image uploaded from the workflow knowledge base zip file cannot be parsed (#4505)

This commit is contained in:
shaohuzhang1 2025-12-12 14:01:29 +08:00 committed by GitHub
parent 6e53d819ca
commit 621fbd3bd6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/common/handle/impl/text/text_split_handle.py
View File

@ -28,11 +28,13 @@ default_pattern_list = [
class TextSplitHandle(BaseSplitHandle):
def support(self, file, get_buffer):
buffer = get_buffer(file)
file_name: str = file.name.lower()
if file_name.endswith(".md") or file_name.endswith('.txt') or file_name.endswith('.TXT') or file_name.endswith(
'.MD'):
return True
if file_name.index('.') > 0:
return False
buffer = get_buffer(file)
result = detect(buffer)
if result['encoding'] is not None and result['confidence'] is not None and result['encoding'] != 'ascii' and \
result['confidence'] > 0.5:

6
apps/common/handle/impl/text/zip_split_handle.py
View File

@ -216,12 +216,10 @@ class ZipSplitHandle(BaseSplitHandle):
real_name = get_file_name(zf.name)
except Exception:
real_name = zf.name
# 为 split_handle 提供可重复读取的 file-like 对象
zf.name = real_name
get_buffer = FileBufferHandle().get_buffer
for split_handle in split_handles:
# 准备一个简单的 get_buffer 回调，返回当前 raw
get_buffer = FileBufferHandle().get_buffer
if split_handle.support(zf, get_buffer):
row = get_buffer(zf)
md_text = split_handle.get_content(io.BytesIO(row), save_image)
@ -229,7 +227,7 @@ class ZipSplitHandle(BaseSplitHandle):
break
for file_content in file_content_list:
_image_list, content = get_image_list_by_content(file_content.get('name'), file_content.get("content"),
files)
files)
content_parts.append(content)
for image in _image_list:
image_list.append(image)