Browse Source

Provide a better error message when attempting to execute the workflow with a missing node. (#3517)

pull/3493/merge
Dr.Lt.Data 6 months ago committed by GitHub
parent
commit
4bc1884478
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 11
      execution.py

11
execution.py

@ -622,8 +622,17 @@ def full_type_name(klass):
def validate_prompt(prompt): def validate_prompt(prompt):
outputs = set() outputs = set()
for x in prompt: for x in prompt:
if 'class_type' not in prompt[x]:
error = {
"type": "invalid_prompt",
"message": f"Cannot execute due to a missing node",
"details": f"Node ID '#{x}'",
"extra_info": {}
}
return (False, error, [], [])
class_ = nodes.NODE_CLASS_MAPPINGS[prompt[x]['class_type']] class_ = nodes.NODE_CLASS_MAPPINGS[prompt[x]['class_type']]
if hasattr(class_, 'OUTPUT_NODE') and class_.OUTPUT_NODE == True: if hasattr(class_, 'OUTPUT_NODE') and class_.OUTPUT_NODE is True:
outputs.add(x) outputs.add(x)
if len(outputs) == 0: if len(outputs) == 0:

Loading…
Cancel
Save