15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-25 17:22:55 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: raise error
Some checks failed
sync2gitee / repo-sync (push) Has been cancelled
Details
Typos Check / Spell Check with Typos (push) Has been cancelled
Details

This commit is contained in:
zhangzhanwei 2025-09-29 18:05:43 +08:00 committed by zhanweizhang7
parent f943cc52bb
commit f834569f91
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/chat/serializers/chat.py
View File

@ -187,11 +187,13 @@ class PromptGenerateSerializer(serializers.Serializer):
def process():
model = get_model_instance_by_model_workspace_id(model_id=model_id, workspace_id=workspace_id,**application.model_params_setting)
for r in model.stream([SystemMessage(content=system_content),
try:
for r in model.stream([SystemMessage(content=system_content),
*[HumanMessage(content=m.get('content')) if m.get('role') == 'user' else AIMessage(
content=m.get('content')) for m in messages]]):
yield 'data: ' + json.dumps({'content': r.content}) + '\n\n'
yield 'data: ' + json.dumps({'content': r.content}) + '\n\n'
except Exception as e:
yield 'data: ' + json.dumps({'error': str(e)}) + '\n\n'
return to_stream_response_simple(process())