Ethanfel/ComfyUI-Ethanfel-Prompt-Builder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Krea2 POV prompt restore wiring

Browse Source
This commit is contained in:
Ethanfel
2026-06-30 23:00:08 +02:00
parent 665a23a7b2
commit 85c577024b
3 changed files with 80 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/prompt_smoke.py
+69 -5
View File
@@ -11353,6 +11353,15 @@ def smoke_node_hardcore_position_registration() -> None:
restore_node = sxcp_nodes.NODE_CLASS_MAPPINGS["SxCPKrea2POVPromptRestore"]
restore_inputs = restore_node.INPUT_TYPES().get("required") or {}
restore_optional = restore_node.INPUT_TYPES().get("optional") or {}
_expect(
"hardcore_position_config" not in restore_inputs,
"Krea2 POV Prompt Restore should not require an incoming position config connection",
)
_expect(
"hardcore_position_config" in restore_optional,
"Krea2 POV Prompt Restore should expose incoming position config as an optional connection",
)
for key in (
"restore_clothing_detail",
"restore_face_expression_detail",
@@ -11361,13 +11370,68 @@ def smoke_node_hardcore_position_registration() -> None:
"relax_non_pose_axis_conflicts",
):
_expect(key in restore_inputs, f"Krea2 POV Prompt Restore lost input {key}")
standalone_restore_config, standalone_restore_summary = restore_node().build(
True,
True,
False,
False,
True,
"",
)
parsed_standalone_restore = json.loads(standalone_restore_config)
_expect(
parsed_standalone_restore.get("restore_prompt_axes") == [
"clothing_detail",
"face_detail",
"expression_detail",
"mouth_detail",
"reaction_detail",
],
"Krea2 POV Prompt Restore should emit restore axes without an incoming connection",
)
_expect(
"restore_axes=clothing_detail,face_detail" in standalone_restore_summary,
"Krea2 POV Prompt Restore standalone summary lost restored axes",
)
pre_restore_config, _pre_restore_summary = restore_node().build(
True,
True,
False,
False,
True,
"",
)
pre_restored_top_config, _pre_restored_top_keys, _pre_restored_top_positions, _, _, _ = oral_filter().build(
"replace",
pre_restore_config,
include_blowjob_top_down_vertical_shaft=True,
)
parsed_pre_restored_top = json.loads(pre_restored_top_config)
_expect(
parsed_pre_restored_top.get("krea2_variant_keys") == ["pov_blowjob_top_down_vertical_shaft"],
"Restore-before-filter should keep the selected top-view atlas variant metadata",
)
_expect(
parsed_pre_restored_top.get("restore_prompt_axes") == [
"clothing_detail",
"face_detail",
"expression_detail",
"mouth_detail",
"reaction_detail",
],
"Restore-before-filter should preserve restore axes through replace-mode pose filtering",
)
_expect(
parsed_pre_restored_top.get("relax_non_pose_axis_conflicts") is True,
"Restore-before-filter should preserve non-pose conflict relaxation through replace-mode pose filtering",
)
restored_config, restored_summary = restore_node().build(
True,
True,
False,
False,
True,
doggy_config,
True,
True,
False,
False,
True,
)
parsed_restored_config = json.loads(restored_config)
_expect(