nodes/bentoml-comfy-pack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: get node identifier

Browse Source 
Signed-off-by: Frost Ming <me@frostming.com>
This commit is contained in:
Frost Ming
2024-11-07 16:49:35 +08:00
parent 17031261cb
commit 5feae11a55
1 changed files with 4 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/comfyui_idl/utils.py
+4 -11
View File
@@ -45,7 +45,7 @@ def _normalize_to_identifier(s: str) -> str:
return s.lower()
def _get_input_name(node, dep_map) -> str:
def _get_node_identifier(node, dep_map=None) -> str:
"""
Get the input name from the node
"""
@@ -53,20 +53,13 @@ def _get_input_name(node, dep_map) -> str:
if title.isidentifier():
return title
nid = node["id"]
if (nid, 0) in dep_map:
if dep_map and (nid, 0) in dep_map:
_, input_name = dep_map[(nid, 0)]
return _normalize_to_identifier(input_name)
return _normalize_to_identifier(title)
def _get_output_name(node) -> str:
title = node["_meta"]["title"]
if title.isidentifier():
return title
return _normalize_to_identifier(title)
def _parse_workflow(workflow: dict) -> tuple[dict, dict]:
"""
Parse the workflow template and return the input and output definition
@@ -83,12 +76,12 @@ def _parse_workflow(workflow: dict) -> tuple[dict, dict]:
for id, node in workflow.items():
node["id"] = id
if node["class_type"].startswith("BentoInput"):
name = _get_input_name(node, dep_map)
name = _get_node_identifier(node, dep_map)
if name in inputs:
name = f"{name}_{id}"
inputs[name] = node
elif node["class_type"].startswith("BentoOutput"):
name = _get_output_name(node)
name = _get_node_identifier(node)
if name in inputs:
name = f"{name}_{id}"
outputs[name] = node