From ee68cf222c901f52df6c030eeff097b2be894f26 Mon Sep 17 00:00:00 2001 From: Concedo <39025047+LostRuins@users.noreply.github.com> Date: Tue, 8 Sep 2026 18:15:18 +0800 Subject: [PATCH] fix genkey image gen status leak --- embd_res/kcpp_sdui.embd | 8 ++++---- koboldcpp.py | 8 ++++++-- 2 files changed, 10 insertions(+), 6 deletions(-) diff --git a/embd_res/kcpp_sdui.embd b/embd_res/kcpp_sdui.embd index 4a8609d4a..76be75b43 100644 --- a/embd_res/kcpp_sdui.embd +++ b/embd_res/kcpp_sdui.embd @@ -6,7 +6,7 @@ Stable UI for KoboldCpp diff --git a/koboldcpp.py b/koboldcpp.py index 6d8545533..0c50d2f62 100644 --- a/koboldcpp.py +++ b/koboldcpp.py @@ -6833,9 +6833,13 @@ Change Mode
parsed_dict = urllib.parse.parse_qs(parsed_url.query) genkey = parsed_dict.get('genkey', [''])[0] skip_current_image = parse_query_bool(parsed_dict, 'skip_current_image') - # with no auth, reveal status without preview image + # Only expose progress and previews for the requested active generation. auth = bool(genkey and genkey==currgenimgkey) - info = a1111_progress_response(auth and not skip_current_image) + info = build_a1111_progress_response('idle') + if auth: + active_info = a1111_progress_response(not skip_current_image) + if genkey == currgenimgkey: + info = active_info response_body = json.dumps(info).encode() elif clean_path=='/history' or clean_path=='/api/history' or clean_path.startswith('/api/history/') or clean_path.startswith('/history/'): #emulate comfyui modelNameToReturn = friendlysdmodelname