Resolutions, remove flux.1

2026-07-25 21:19:27 -07:00
parent ca33869007
commit 689cd1244c
3 changed files with 16 additions and 32 deletions
@@ -75,11 +75,15 @@ async def handle(args: Dict[str, Any]) -> List[Dict[str, Any]]:
        if res_set_name and res_set_name in res_cfg:
            res_set = res_cfg[res_set_name]
            if res_preset_name in res_set:
-                res_vals = res_set[res_preset_name]
-                if "Width" in label_map:
-                    payload[label_map["Width"]] = res_vals["width"]
-                if "Height" in label_map:
-                    payload[label_map["Height"]] = res_vals["height"]
+                res_val_str = res_set[res_preset_name]
+                try:
+                    w, h = map(int, res_val_str.split('x'))
+                    if "Width" in label_map:
+                        payload[label_map["Width"]] = w
+                    if "Height" in label_map:
+                        payload[label_map["Height"]] = h
+                except (ValueError, AttributeError):
+                    raise ToolError(f"Invalid resolution format for preset '{res_preset_name}': {res_val_str}. Expected 'WidthxHeight'.")
            else:
                raise ToolError(f"Resolution preset '{res_preset_name}' not found for this model. Available: {', '.join(res_set.keys())}")
        else: