jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix event/stream ping (#3553)

Browse source
This commit is contained in:
liuzhenghua 2024-04-17 18:28:24 +08:00 • committed by GitHub
commit e02ee3bb2e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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()