From 8311254009b511fadcae16d63dab48c8203abd56 Mon Sep 17 00:00:00 2001
From: wl-zhao <zhaowl20@mails.tsinghua.edu.cn>
Date: Thu, 17 Oct 2024 15:17:30 +0800
Subject: [PATCH] update error message when missing nodes

---
 dependency_checker.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/dependency_checker.py b/dependency_checker.py
index 199fb23..05039b2 100644
--- a/dependency_checker.py
+++ b/dependency_checker.py
@@ -99,7 +99,9 @@ def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes an
     
     file_mapping_dict = {}
     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"):
             custom_nodes.append(node_cls.RELATIVE_PYTHON_MODULE)