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

Merge pull request #2888 from kirs-hi/fix/robustness-cancel-keyerror-compress-loop

Browse Source 
fix: avoid KeyError on /cancel and infinite loop in image compression
This commit is contained in:
zhayujie
2026-06-12 18:28:10 +08:00
committed by GitHub
parent 80d0a6aeb2 ad64e17a34
commit c9c293f67c
3 changed files with 131 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
channel/chat_channel.py
View File

@@ -519,7 +519,10 @@ class ChatChannel(Channel):
def cancel_session(self, session_id):
with self.lock:
if session_id in self.sessions:
for future in self.futures[session_id]:
# futures[session_id] is only created in consume() when a task is
# dispatched, so it may be absent if cancel happens right after
# produce() but before the first dispatch. Default to [].
for future in self.futures.get(session_id, []):
future.cancel()
cnt = self.sessions[session_id][0].qsize()
if cnt > 0:
@@ -529,7 +532,7 @@ class ChatChannel(Channel):
def cancel_all_session(self):
with self.lock:
for session_id in self.sessions:
for future in self.futures[session_id]:
for future in self.futures.get(session_id, []):
future.cancel()
cnt = self.sessions[session_id][0].qsize()
if cnt > 0: