zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: The workflow cannot function properly after the conversation flow is disconnected (#2428)

Browse Source
This commit is contained in:
shaohuzhang1 2025-02-27 14:21:47 +08:00 committed by GitHub
parent c433c03fc0
commit afaf3d6e26
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/application/flow/workflow_manage.py
View File

@ -382,6 +382,8 @@ class WorkflowManage:
break break
yield chunk yield chunk
finally: finally:
while self.is_run():
pass
details = self.get_runtime_details() details = self.get_runtime_details()
message_tokens = sum([row.get('message_tokens') for row in details.values() if message_tokens = sum([row.get('message_tokens') for row in details.values() if
'message_tokens' in row and row.get('message_tokens') is not None]) 'message_tokens' in row and row.get('message_tokens') is not None])

4
apps/common/middleware/gzip.py
View File

@ -44,10 +44,10 @@ class GZipMiddleware(MiddlewareMixin):
if response.is_async: if response.is_async:
# pull to lexical scope to capture fixed reference in case # pull to lexical scope to capture fixed reference in case
# streaming_content is set again later. # streaming_content is set again later.
orignal_iterator = response.streaming_content original_iterator = response.streaming_content
async def gzip_wrapper(): async def gzip_wrapper():
async for chunk in orignal_iterator: async for chunk in original_iterator:
yield compress_string( yield compress_string(
chunk, chunk,
max_random_bytes=self.max_random_bytes, max_random_bytes=self.max_random_bytes,