hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-11 03:46:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix event/stream ping (#3553)

Browse Source
This commit is contained in:
liuzhenghua
2024-04-17 18:28:24 +08:00
committed by GitHub
parent 394ceee141
commit e02ee3bb2e
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/app/apps/base_app_generate_response_converter.py
View File

@@ -26,7 +26,10 @@ class AppGenerateResponseConverter(ABC):
else: else:
def _generate(): def _generate():
for chunk in cls.convert_stream_full_response(response): 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() return _generate()
else: else:
@@ -35,7 +38,10 @@ class AppGenerateResponseConverter(ABC):
else: else:
def _generate(): def _generate():
for chunk in cls.convert_stream_simple_response(response): 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() return _generate()