From e02ee3bb2e1a576fafc708036e75d621d0fcf32a Mon Sep 17 00:00:00 2001 From: liuzhenghua <1090179900@qq.com> Date: Wed, 17 Apr 2024 18:28:24 +0800 Subject: [PATCH] fix event/stream ping (#3553) --- .../app/apps/base_app_generate_response_converter.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/api/core/app/apps/base_app_generate_response_converter.py b/api/core/app/apps/base_app_generate_response_converter.py index 7202822975..bacd1a5477 100644 --- a/api/core/app/apps/base_app_generate_response_converter.py +++ b/api/core/app/apps/base_app_generate_response_converter.py @@ -26,7 +26,10 @@ class AppGenerateResponseConverter(ABC): else: def _generate(): for chunk in cls.convert_stream_full_response(response): - yield f'data: {chunk}\n\n' + if chunk == 'ping': + yield f'event: {chunk}\n\n' + else: + yield f'data: {chunk}\n\n' return _generate() else: @@ -35,7 +38,10 @@ class AppGenerateResponseConverter(ABC): else: def _generate(): for chunk in cls.convert_stream_simple_response(response): - yield f'data: {chunk}\n\n' + if chunk == 'ping': + yield f'event: {chunk}\n\n' + else: + yield f'data: {chunk}\n\n' return _generate()