From ab5dad44f92795e45bed00bbd331e603bb5aa3f7 Mon Sep 17 00:00:00 2001
From: relhamdi <elhamdi.rayan@gmail.com>
Date: Thu, 28 May 2026 11:25:02 +0200
Subject: [PATCH] Add _sub_selected widget to persist sub router value

---
 src/nodes/prompt_router.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/nodes/prompt_router.py b/src/nodes/prompt_router.py
index 9e3c9e2..bb3a193 100644
--- a/src/nodes/prompt_router.py
+++ b/src/nodes/prompt_router.py
@@ -22,6 +22,13 @@ class PromptRouter:
                         "tooltip": "Select the active input.",
                     },
                 ),
+                "_sub_selected": (
+                    "STRING",
+                    {
+                        "default": "--",
+                        "tooltip": "Internal use only.",
+                    },
+                ),
             },
             "optional": {
                 f"input_{i}": ("STRING", {"forceInput": True})
@@ -80,9 +87,8 @@ class PromptRouter:
 
         return mapping
 
-    def process(self, selected, prompt=None, unique_id=None, **kwargs):
+    def process(self, selected, _sub_selected, prompt=None, unique_id=None, **kwargs):
         # Remove internal widgets from kwargs
-        kwargs.pop("_sub_selected", None)
         kwargs.pop("_sub_source", None)
 
         if not selected or selected == "--":