mirror/cube-studio
2
0
Fork 0
mirror of https://github.com/tencentmusic/cube-studio.git synced 2025-02-17 14:40:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复智能对话响应问题

Browse Source
This commit is contained in:
xiaoman.gao@extremevision.com.cn 2024-04-19 11:02:52 +08:00
parent b50c32404a
commit b03fdd7495
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
myapp/views/view_chat.py
View File

@ -213,7 +213,7 @@ class Chat_View_Base():
description= _('接口类型，并不一定是openai，只需要符合http请求响应格式即可'),
widget=Select2Widget(),
default='openai',
choices=[[x, x] for x in ["chatgpt3.5",'chatgpt4','chatglm','belle', 'llama','AIGC','autogpt',_('召回列表')]],
choices=[[x, x] for x in ["openai",'AIGC','chatbi','autogpt',_('召回列表')]],
validators=[]
),
"service_config": StringField(
@ -1167,7 +1167,11 @@ AI:
message = event.data
finish = False
if message != '[DONE]':
message = json.loads(event.data)['choices'][0].get('delta', {}).get('content', '')
choices = json.loads(event.data)['choices']
if choices:
message = choices[0].get('delta', {}).get('content', '')
else:
message=''
print(message, flush=True, end='')
# print(message)
if message == '[DONE]':