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 (#4509)

This commit is contained in:
shaohuzhang1 2025-12-12 15:07:59 +08:00 committed by CaptainB
parent a53b0d5c10
commit d1d64bba00
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -25,6 +25,10 @@ default_pattern_list = [
re.compile("(?<=\\n)(?<!#)###### (?!#).*|(?<=^)(?<!#)###### (?!#).*")
]
end = [".mp4", ".avi", ".mov", ".mkv", ".flv", ".wmv", ".webm", ".mpeg", ".mpg", ".3gp", ".ts", ".rmvb",
".mp3", ".wav", ".flac", ".aac", ".ogg", ".m4a", ".wma", ".opus", ".alac", ".aiff", ".amr",
".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".webp", ".heif", ".raw", ".ico", ".svg", ".pdf"]
class TextSplitHandle(BaseSplitHandle):
def support(self, file, get_buffer):
@ -32,7 +36,8 @@ class TextSplitHandle(BaseSplitHandle):
if file_name.endswith(".md") or file_name.endswith('.txt') or file_name.endswith('.TXT') or file_name.endswith(
'.MD'):
return True
if '.' in file_name:
lower_name = file_name.lower()
if any([True for item in end if lower_name.endswith(item)]):
return False
buffer = get_buffer(file)
result = detect(buffer)