kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-20 13:47:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(wechatmp): flush cached segments while task still running

Browse Source 
Previously the passive reply only drained the cache after the agent task fully finished, so for long multi-turn tasks the user could not retrieve already-cached intermediate segments. Now return cached segments as soon as they are available, even while the task is still running; the next user message fetches the rest.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
6vision
2026-05-30 15:48:27 +08:00
parent 458b1a1d88
commit 5aca54c083
1 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
channel/wechatmp/passive_reply.py
View File

@@ -103,14 +103,21 @@ class Query:
task_running = True task_running = True
waiting_until = request_time + 4 waiting_until = request_time + 4
while time.time() < waiting_until: while time.time() < waiting_until:
if from_user in channel.running: if from_user not in channel.running:
time.sleep(0.1)
else:
task_running = False task_running = False
break break
# Task still running, but if it has already produced cached
# segments (e.g. multi-turn thinking output), return them now
# instead of forcing the user to wait for the whole task. The
# remaining segments are fetched by the user's next message.
if channel.cache_dict.get(from_user):
break
time.sleep(0.1)
reply_text = "" reply_text = ""
if task_running: # Only fall back to retry / "thinking" hint when the task is still
# running AND there is nothing cached to send yet.
if task_running and not channel.cache_dict.get(from_user):
if request_cnt < 3: if request_cnt < 3:
# waiting for timeout (the POST request will be closed by Wechat official server) # waiting for timeout (the POST request will be closed by Wechat official server)
time.sleep(2) time.sleep(2)