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

feat: decode base64 file chunks before uploading

This commit is contained in:
CaptainB 2025-12-09 10:38:26 +08:00
parent 6d24d66b9d
commit 8e7d8d6615
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
apps/application/flow/step_node/tool_lib_node/impl/base_tool_lib_node.py
View File

@ -8,6 +8,7 @@
"""
import ast
import base64
import io
import json
import mimetypes
@ -207,9 +208,13 @@ class BaseToolLibNodeNode(IToolLibNode):
{**all_params, **self.workflow_params.get('data_source'),
'download_item': item},
function_name='download')
file = bytes_to_uploaded_file(ast.literal_eval(result.get('file_bytes')), result.get('name'))
file_bytes = result.get('file_bytes', [])
chunks = []
for chunk in file_bytes:
chunks.append(base64.b64decode(chunk))
file = bytes_to_uploaded_file(b''.join(chunks), result.get('name'))
file_url = self.upload_knowledge_file(file)
download_file_list.append({'file_id': file_url, 'name': result.get('name')})
download_file_list.append({'file_id': file_url.split('/')[-1], 'name': result.get('name')})
result = download_file_list
else:
result = function_executor.exec_code(tool_lib.code, all_params)