update error message

2024-10-17 16:22:49 +08:00
parent f78f07961e
commit f2926099fe
1 changed files with 3 additions and 1 deletions
@@ -55,7 +55,9 @@ def schema_validator(prompt):
        "outputs": {}
    }
    for node_id, node_info in prompt.items():
-        node_class_type = node_info["class_type"]
+        node_class_type = node_info.get("class_type")
+        if node_class_type is None:
+            raise NotImplementedError(f"Missing nodes founded, please first install the missing nodes using ComfyUI Manager")
        node_cls = NODE_CLASS_MAPPINGS[node_class_type]
        if hasattr(node_cls, "RELATIVE_PYTHON_MODULE") and node_cls.RELATIVE_PYTHON_MODULE == "custom_nodes.ComfyUI-ShellAgent-Plugin":
            schema = {}