refactor: Support image_list, document_list while using swagger api

--story=1017867 --user=刘瑞斌 对话API支持上传文件、语音、图像和视频,用于实现maxkb的文件解析及多模态对话#2228 https://www.tapd.cn/57709429/s/1654842
This commit is contained in:
CaptainB 2025-02-17 16:05:26 +08:00 committed by 刘瑞斌
parent 83cd69e5b7
commit 9249c1756f
2 changed files with 14 additions and 8 deletions

View File

@ -222,13 +222,19 @@ class OpenAIChatSerializer(serializers.Serializer):
client_type = self.data.get('client_type')
chat_id = self.generate_chat(chat_id, application_id, message, client_id)
return ChatMessageSerializer(
data={'chat_id': chat_id, 'message': message,
're_chat': re_chat,
'stream': stream,
'application_id': application_id,
'client_id': client_id,
'client_type': client_type, 'form_data': instance.get('form_data', {})}).chat(
base_to_response=OpenaiToResponse())
data={
'chat_id': chat_id, 'message': message,
're_chat': re_chat,
'stream': stream,
'application_id': application_id,
'client_id': client_id,
'client_type': client_type,
'form_data': instance.get('form_data', {}),
'image_list': instance.get('image_list', []),
'document_list': instance.get('document_list', []),
'audio_list': instance.get('audio_list', []),
}
).chat(base_to_response=OpenaiToResponse())
class ChatMessageSerializer(serializers.Serializer):

View File

@ -62,7 +62,7 @@ class FileSerializer(serializers.Serializer):
def upload(self, with_valid=True):
if with_valid:
self.is_valid(raise_exception=True)
meta = self.data.get('meta')
meta = self.data.get('meta', {'debug': True})
file_id = meta.get('file_id', uuid.uuid1())
file = File(id=file_id, file_name=self.data.get('file').name, meta=meta)
file.save(self.data.get('file').read())