Extract builder nodes

tools/prompt_smoke.py

@@ -2169,6 +2169,63 @@ def smoke_node_insta_registration() -> None:
     _expect(pair.get("options", {}).get("hardcore_cast") == "couple", "Insta/OF Prompt Pair lost options metadata")
 
 
+def smoke_node_builder_registration() -> None:
+    required_nodes = [
+        "SxCPPromptBuilder",
+        "SxCPPromptBuilderFromConfigs",
+    ]
+    for node_name in required_nodes:
+        _expect(node_name in sxcp_nodes.NODE_CLASS_MAPPINGS, f"{node_name} missing from node registry")
+        _expect(node_name in sxcp_nodes.NODE_DISPLAY_NAME_MAPPINGS, f"{node_name} missing from display registry")
+
+    builder_node = sxcp_nodes.NODE_CLASS_MAPPINGS["SxCPPromptBuilder"]
+    builder_inputs = builder_node.INPUT_TYPES().get("required") or {}
+    _expect("category" in builder_inputs, "Prompt Builder lost category input")
+    _expect("tooltip" in builder_inputs["category"][1], "Prompt Builder tooltip injection missing")
+    direct_output = builder_node().build(
+        "woman",
+        "random",
+        1,
+        41,
+        123,
+        "full",
+        "any",
+        "standard",
+        True,
+        0.5,
+        0.0,
+        "random",
+        1,
+        0,
+        -1,
+        -1,
+        Trigger,
+        True,
+    )
+    direct_row = json.loads(direct_output[3])
+    _expect_row_base(direct_row, "node_builder.direct_row")
+    _expect(direct_output[0] == direct_row.get("prompt"), "Prompt Builder prompt output drifted from metadata")
+    _expect(direct_output[4] == direct_row.get("main_category"), "Prompt Builder category output drifted from metadata")
+    _expect_trigger_once("node_builder.direct_prompt", direct_output[0], Trigger)
+
+    config_node = sxcp_nodes.NODE_CLASS_MAPPINGS["SxCPPromptBuilderFromConfigs"]
+    config_inputs = config_node.INPUT_TYPES()
+    _expect("category_config" in (config_inputs.get("optional") or {}), "Prompt Builder From Configs lost category_config input")
+    config_output = config_node().build(
+        1,
+        41,
+        123,
+        category_config=pb.build_category_config_json("woman", "random"),
+        cast_config=pb.build_cast_config_json("solo_woman", 1, 0),
+        generation_profile=pb.build_generation_profile_json(profile="balanced"),
+    )
+    config_row = json.loads(config_output[3])
+    _expect_row_base(config_row, "node_builder.config_row")
+    _expect(config_output[0] == config_row.get("prompt"), "Prompt Builder From Configs prompt output drifted from metadata")
+    _expect(config_output[4] == config_row.get("main_category"), "Prompt Builder From Configs category output drifted from metadata")
+    _expect_text("node_builder.config_caption", config_output[2], 20)
+
+
 def smoke_node_profile_filter_registration() -> None:
     required_nodes = [
         "SxCPGenerationProfile",
@@ -2280,6 +2337,7 @@ SMOKE_CASES: list[tuple[str, Callable[[], None]]] = [
     ("node_hardcore_position_registration", smoke_node_hardcore_position_registration),
     ("node_formatter_registration", smoke_node_formatter_registration),
     ("node_insta_registration", smoke_node_insta_registration),
+    ("node_builder_registration", smoke_node_builder_registration),
     ("node_profile_filter_registration", smoke_node_profile_filter_registration),
 ]