From f834569f91995baad6b3afaeadce6ee6af2645ae Mon Sep 17 00:00:00 2001 From: zhangzhanwei Date: Mon, 29 Sep 2025 18:05:43 +0800 Subject: [PATCH] feat: raise error --- apps/chat/serializers/chat.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/apps/chat/serializers/chat.py b/apps/chat/serializers/chat.py index 369b0065b..6ac58f3b6 100644 --- a/apps/chat/serializers/chat.py +++ b/apps/chat/serializers/chat.py @@ -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())