Function _patch_comfy_workflow

tools/modly-cli/agent.py:501–543

(workflow: dict[str, Any], *, prompt: str | None, seed: int | None)

def _patch_comfy_workflow(workflow: dict[str, Any], *, prompt: str | None, seed: int | None) -> dict[str, Any]:
    workflow = json.loads(json.dumps(workflow))
    nodes = workflow.get("prompt", workflow)
    if not isinstance(nodes, dict):
        raise ModlyCliError("ComfyUI workflow must be API format (top-level node-id object, or {'prompt': {...}})")
    if "nodes" in workflow and "links" in workflow:
        raise ModlyCliError("ComfyUI workflow is editor format; export it as API format first")

    if prompt is not None:
        patched = False
        for node in nodes.values():
            if not isinstance(node, dict):
                continue
            class_type = str(node.get("class_type", "")).lower()
            inputs = node.get("inputs") if isinstance(node.get("inputs"), dict) else {}
            text = str(inputs.get("text", "")).lower()
            if "cliptextencode" in class_type and "negative" not in text:
                inputs["text"] = prompt
                patched = True
                break
        if not patched:
            for node in nodes.values():
                if isinstance(node, dict) and isinstance(node.get("inputs"), dict):
                    inputs = node["inputs"]
                    for key in ("prompt", "positive", "text"):
                        if key in inputs and isinstance(inputs[key], str):
                            inputs[key] = prompt
                            patched = True
                            break
                if patched:
                    break
        if not patched:
            raise ModlyCliError("Could not find a text/prompt input to patch in ComfyUI workflow")

    if seed is not None:
        for node in nodes.values():
            if not isinstance(node, dict) or not isinstance(node.get("inputs"), dict):
                continue
            inputs = node["inputs"]
            for key in ("seed", "noise_seed"):
                if key in inputs and isinstance(inputs[key], int):
                    inputs[key] = seed
    return nodes


def _run_comfy_workflow(args: argparse.Namespace) -> dict[str, Any]: