DemonGatanjieuandClaude Opus 5.5
c3433a9241
Ways: "Use it this time" in the Ways view, for one task. It joins the task's picks as a new kind, the AI is told, and add_nodes builds that way for its use whichever one the AI asks for. "Only" lets in the nodes a picked way builds. Examples: text to picture, two passes, picture to picture and repainting an area, on ComfyUI's own nodes. One card, first on the Workflows page, opens them in a tab of their own. The files name no model or picture; a checkpoint and a picture that are here are filled in when one is read. list_workflows gives them to an AI when no saved workflow fits, and they open and run by name like saved ones. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
92 lines
5.6 KiB
Python
92 lines
5.6 KiB
Python
"""The examples Anomalous ships (api/example_workflows.py): ComfyUI's own nodes only, no model or picture
|
|
of any machine in the files, filled in from what is here, and reachable by an AI when no saved workflow fits."""
|
|
import json
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
from unittest import mock
|
|
|
|
PLUGIN = Path(__file__).resolve().parents[1]
|
|
sys.path[:0] = [str(PLUGIN), str(PLUGIN.parents[1]), str(PLUGIN / "tests")]
|
|
|
|
from api import example_workflows as examples, mcp_fill, mcp_tools # noqa: E402
|
|
|
|
OWN_NODES = {"CheckpointLoaderSimple", "CLIPTextEncode", "EmptyLatentImage", "KSampler", "VAEDecode", "VAEEncode",
|
|
"SaveImage", "LoadImage", "LatentUpscaleBy", "SetLatentNoiseMask"}
|
|
|
|
|
|
class ExampleTests(unittest.TestCase):
|
|
def test_the_files_use_own_nodes_and_name_nothing_of_a_machine(self):
|
|
self.assertEqual(set(examples.EXAMPLES), {"text_to_picture", "two_pass", "picture_to_picture", "repaint"})
|
|
for key, example in examples.EXAMPLES.items():
|
|
nodes = example["workflow"]["nodes"]
|
|
self.assertLessEqual({node["type"] for node in nodes}, OWN_NODES, key)
|
|
for node in nodes:
|
|
if node["type"] in ("CheckpointLoaderSimple", "LoadImage"):
|
|
self.assertEqual(node["widgets_values"][0], "", key)
|
|
self.assertTrue(example["title"]["zh"] and example["title"]["en"] and example["about"]["zh"] and example["about"]["en"] and example["ai"], key)
|
|
text = json.dumps(example)
|
|
for mark in (".safetensors", ".ckpt", ".png", ".jpg", ".webp", "widgets_values_named"):
|
|
self.assertNotIn(mark, text, key)
|
|
|
|
def test_every_link_joins_nodes_that_are_there(self):
|
|
for key, example in examples.EXAMPLES.items():
|
|
ids = {node["id"] for node in example["workflow"]["nodes"]}
|
|
for link in example["workflow"]["links"]:
|
|
self.assertLessEqual({link[1], link[3]}, ids, key)
|
|
|
|
def test_read_fills_in_what_is_here(self):
|
|
with tempfile.TemporaryDirectory() as folder:
|
|
Path(folder, "a.png").write_bytes(b"x")
|
|
with mock.patch.object(examples, "_checkpoint", return_value="mine.safetensors"), \
|
|
mock.patch.object(examples.folder_paths, "get_input_directory", return_value=folder, create=True):
|
|
data = examples.read("repaint")
|
|
values = {node["type"]: node["widgets_values"][0] for node in data["workflow"]["nodes"] if node["type"] in ("CheckpointLoaderSimple", "LoadImage")}
|
|
self.assertEqual(values, {"CheckpointLoaderSimple": "mine.safetensors", "LoadImage": "a.png"})
|
|
self.assertFalse(data["lacks_model"])
|
|
self.assertEqual(next(node for node in examples.EXAMPLES["repaint"]["workflow"]["nodes"] if node["type"] == "LoadImage")["widgets_values"][0], "", "the shipped one stays blank")
|
|
self.assertIsNone(examples.read("nothing"))
|
|
|
|
def test_read_says_when_there_is_no_model(self):
|
|
with mock.patch.object(examples, "_checkpoint", return_value=""), mock.patch.object(examples, "_picture", return_value=""):
|
|
self.assertTrue(examples.read("text_to_picture")["lacks_model"])
|
|
|
|
def test_a_checkpoint_the_settings_suit_comes_first(self):
|
|
from api import model_advice
|
|
names = ["flux.safetensors", "xl.safetensors"]
|
|
with mock.patch.object(examples.folder_paths, "get_filename_list", return_value=names, create=True), \
|
|
mock.patch.object(examples.folder_paths, "get_full_path", side_effect=lambda _kind, name: name, create=True), \
|
|
mock.patch.object(model_advice, "base_of", side_effect=lambda path: "SDXL 1.0" if path.startswith("xl") else "Flux.1 D"):
|
|
self.assertEqual(examples._checkpoint(), "xl.safetensors")
|
|
with mock.patch.object(examples.folder_paths, "get_filename_list", return_value=[], create=True):
|
|
self.assertEqual(examples._checkpoint(), "")
|
|
|
|
def test_an_ai_gets_them_when_no_saved_workflow_fits(self):
|
|
with mock.patch.object(mcp_tools, "_list_recipes", return_value=[]), mock.patch.object(mcp_tools, "plans", return_value={}), \
|
|
mock.patch.object(mcp_tools, "get_recipes_dir", return_value=""):
|
|
listed = mcp_tools.list_workflows()
|
|
found = mcp_tools.list_workflows("repaint")
|
|
self.assertEqual(listed["total_matches"], 4)
|
|
self.assertTrue(all(item["example"] for item in listed["workflows"]))
|
|
self.assertEqual([item["name"] for item in found["workflows"]], ["Example: repaint an area"])
|
|
|
|
def test_saved_workflows_come_instead_of_them(self):
|
|
recipe = {"filename": "recipe_1.json", "data": {"name": "Mine"}}
|
|
with mock.patch.object(mcp_tools, "_list_recipes", return_value=[recipe]), mock.patch.object(mcp_tools, "plans", return_value={}), \
|
|
mock.patch.object(mcp_tools, "get_recipes_dir", return_value=""):
|
|
self.assertEqual([item["name"] for item in mcp_tools.list_workflows()["workflows"]], ["Mine"])
|
|
|
|
def test_they_open_by_name_but_a_saved_one_of_that_name_wins(self):
|
|
name = "Example: two passes"
|
|
with mock.patch("api.recipe_store._list_recipes", return_value=[]), mock.patch("api.recipe_store.get_recipes_dir", return_value=""):
|
|
self.assertEqual(mcp_fill.saved_workflow(name), ("example:two_pass", name, None))
|
|
recipe = {"filename": "recipe_1.json", "data": {"name": name}}
|
|
with mock.patch("api.recipe_store._list_recipes", return_value=[recipe]), mock.patch("api.recipe_store.get_recipes_dir", return_value=""):
|
|
self.assertEqual(mcp_fill.saved_workflow(name)[0], "recipe_1.json")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|