AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 02:55:54 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 2 deletions
Show Stats 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()