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

Add typed category route metadata

Browse Source
This commit is contained in:
Ethanfel
2026-06-27 10:39:45 +02:00
parent 00139d0cd9
commit 2c978c7eab
5 changed files with 146 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/prompt_smoke.py
+23
View File
@@ -1027,6 +1027,16 @@ def smoke_row_category_route_policy() -> None:
men_count=1,
hardcore_position_config=hard_config,
)
route_result = row_category_route.select_category_item_route_result(
category_choice="custom_random",
subcategory_choice="Hardcore sexual poses / Oral sex",
seed_config=seed_cfg,
seed=2301,
row_number=1,
women_count=1,
men_count=1,
hardcore_position_config=hard_config,
)
delegated = pb._select_category_item_route(
category_choice="custom_random",
subcategory_choice="Hardcore sexual poses / Oral sex",
@@ -1037,7 +1047,20 @@ def smoke_row_category_route_policy() -> None:
men_count=1,
hardcore_position_config=hard_config,
)
typed_delegated = pb._select_category_item_route_result(
category_choice="custom_random",
subcategory_choice="Hardcore sexual poses / Oral sex",
seed_config=seed_cfg,
seed=2301,
row_number=1,
women_count=1,
men_count=1,
hardcore_position_config=hard_config,
)
_expect(delegated == route, "Prompt builder category/item route should delegate to row_category_route")
_expect(route_result.as_dict() == route, "Typed category/item route should match legacy dict route")
_expect(typed_delegated == route_result, "Prompt builder typed category/item route should delegate")
_expect(route_result.content_axis == "pose", "Typed category/item route lost content seed axis")
_expect(route["category"]["slug"] == "hardcore_sexual_poses", "Row category route selected wrong hardcore category")
_expect(route["subcategory"]["slug"] == "oral_sex", "Row category route selected wrong hardcore subcategory")
_expect(route["content_axis"] == "pose", "Hardcore pose category should use pose seed axis")