Add typed action route metadata

prompt_builder.py

@@ -267,6 +267,31 @@ def _action_position_route_metadata(
     )
 
 
+def _action_position_route(
+    *,
+    is_pose_category: bool,
+    subcategory: dict[str, Any],
+    hardcore_position_config: dict[str, Any] | None,
+    item_template_metadata: dict[str, Any] | None,
+    item_text: Any,
+    source_role_graph: Any,
+    source_composition: Any,
+    pose: Any,
+    item_axis_values: dict[str, Any] | None = None,
+) -> row_route_policy.ActionPositionRoute:
+    return row_route_policy.resolve_action_position_route_result(
+        is_pose_category=is_pose_category,
+        subcategory=subcategory,
+        hardcore_position_config=hardcore_position_config,
+        item_template_metadata=item_template_metadata,
+        item_text=item_text,
+        source_role_graph=source_role_graph,
+        source_composition=source_composition,
+        pose=pose,
+        item_axis_values=item_axis_values,
+    )
+
+
 def _oral_acts_for_position(values: list[Any], position: str) -> list[Any]:
     return row_item_policy.oral_acts_for_position(values, position)
 
@@ -2286,7 +2311,7 @@ def _build_custom_row(
     character_expression_text = str(prompt_axes.get("character_expression_text") or "")
     source_composition = str(prompt_axes.get("source_composition") or "")
     composition = str(prompt_axes.get("composition") or "")
-    action_route = _action_position_route_metadata(
+    action_route = _action_position_route(
         is_pose_category=is_pose_category,
         subcategory=subcategory,
         hardcore_position_config=parsed_hardcore_position_config,
@@ -2297,10 +2322,10 @@ def _build_custom_row(
         pose=pose,
         item_axis_values=item_axis_values,
     )
-    position_family = str(action_route.get("position_family") or "")
-    position_keys = list(action_route.get("position_keys") or [])
-    position_key = str(action_route.get("position_key") or "")
-    action_family = str(action_route.get("action_family") or "")
+    position_family = action_route.position_family
+    position_keys = list(action_route.position_keys)
+    position_key = action_route.position_key
+    action_family = action_route.action_family
 
     text_fields = _row_text_fields(category, subcategory, item)