Skip to content

Commit 9249c17

Browse files
committed
refactor: Support image_list, document_list while using swagger api
--story=1017867 --user=刘瑞斌 对话API支持上传文件、语音、图像和视频,用于实现maxkb的文件解析及多模态对话#2228 https://www.tapd.cn/57709429/s/1654842
1 parent 83cd69e commit 9249c17

File tree

2 files changed

+14
-8
lines changed

2 files changed

+14
-8
lines changed

apps/application/serializers/chat_message_serializers.py

+13-7
Original file line numberDiff line numberDiff line change
@@ -222,13 +222,19 @@ def chat(self, instance: Dict, with_valid=True):
222222
client_type = self.data.get('client_type')
223223
chat_id = self.generate_chat(chat_id, application_id, message, client_id)
224224
return ChatMessageSerializer(
225-
data={'chat_id': chat_id, 'message': message,
226-
're_chat': re_chat,
227-
'stream': stream,
228-
'application_id': application_id,
229-
'client_id': client_id,
230-
'client_type': client_type, 'form_data': instance.get('form_data', {})}).chat(
231-
base_to_response=OpenaiToResponse())
225+
data={
226+
'chat_id': chat_id, 'message': message,
227+
're_chat': re_chat,
228+
'stream': stream,
229+
'application_id': application_id,
230+
'client_id': client_id,
231+
'client_type': client_type,
232+
'form_data': instance.get('form_data', {}),
233+
'image_list': instance.get('image_list', []),
234+
'document_list': instance.get('document_list', []),
235+
'audio_list': instance.get('audio_list', []),
236+
}
237+
).chat(base_to_response=OpenaiToResponse())
232238

233239

234240
class ChatMessageSerializer(serializers.Serializer):

apps/dataset/serializers/file_serializers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ class FileSerializer(serializers.Serializer):
6262
def upload(self, with_valid=True):
6363
if with_valid:
6464
self.is_valid(raise_exception=True)
65-
meta = self.data.get('meta')
65+
meta = self.data.get('meta', {'debug': True})
6666
file_id = meta.get('file_id', uuid.uuid1())
6767
file = File(id=file_id, file_name=self.data.get('file').name, meta=meta)
6868
file.save(self.data.get('file').read())

0 commit comments

Comments
 (0)