15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 10:12:51 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

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

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
yield chunk
finally:
while self.is_run():
pass
details = self.get_runtime_details()
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])

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

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