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: handle exceptions in get_file_base64 method

This commit is contained in:
wxg0103 2025-09-19 11:12:57 +08:00
parent 0693ae99ca
commit 5bea2b2566
1 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
apps/application/flow/step_node/image_to_video_step_node/impl/base_image_to_video_node.py
View File

@ -71,17 +71,22 @@ class BaseImageToVideoNode(IImageToVideoNode):
'history_message': history_message, 'question': question}, {})
def get_file_base64(self, image_url):
if isinstance(image_url, list):
image_url = image_url[0].get('file_id')
if isinstance(image_url, str) and not image_url.startswith('http'):
file = QuerySet(File).filter(id=image_url).first()
file_bytes = file.get_bytes()
# 如果我不知道content_type 可以用 magic 库去检测
file_type = file.file_name.split(".")[-1].lower()
content_type = mime_types.get(file_type, 'application/octet-stream')
encoded_bytes = base64.b64encode(file_bytes)
return f'data:{content_type};base64,{encoded_bytes.decode()}'
return image_url
try :
if isinstance(image_url, list):
image_url = image_url[0].get('file_id')
if isinstance(image_url, str) and not image_url.startswith('http'):
file = QuerySet(File).filter(id=image_url).first()
file_bytes = file.get_bytes()
# 如果我不知道content_type 可以用 magic 库去检测
file_type = file.file_name.split(".")[-1].lower()
content_type = mime_types.get(file_type, 'application/octet-stream')
encoded_bytes = base64.b64encode(file_bytes)
return f'data:{content_type};base64,{encoded_bytes.decode()}'
return image_url
except Exception as e:
raise ValueError(
gettext("Failed to obtain the image"))
def generate_history_ai_message(self, chat_record):
for val in chat_record.details.values():