From 8bea245e96e6b4c82784b7ba83385f9f33facee8 Mon Sep 17 00:00:00 2001 From: dacert Date: Sun, 19 Jul 2026 19:43:40 +0100 Subject: [PATCH] Fix create links --- src/comfy_graph_bind/api.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/comfy_graph_bind/api.py b/src/comfy_graph_bind/api.py index 251fdaf..3e4d0ac 100644 --- a/src/comfy_graph_bind/api.py +++ b/src/comfy_graph_bind/api.py @@ -2,7 +2,6 @@ from __future__ import annotations -import contextlib from collections.abc import Sequence from dataclasses import dataclass, field @@ -22,7 +21,7 @@ from .scene import GraphScene from .start_node import StartNodeItem, StartOutputConfig from .view import GraphView -from .workflow_parser import WorkflowGraph, WorkflowNode, parse_workflow_dict +from .workflow_parser import WorkflowGraph, WorkflowNode, _parse_node_id, parse_workflow_dict from .workflow_parser import is_link as _is_api_link @@ -163,9 +162,11 @@ def _create_existing_links(self, graph: WorkflowGraph) -> None: for name, value in node.raw_inputs.items(): if not _is_api_link(value): continue - with contextlib.suppress(ValueError, TypeError): - src_id = int(value[0]) + src_id = _parse_node_id(str(value[0])) + try: src_idx = int(value[1]) + except (ValueError, TypeError): + continue if src_id not in self._node_items: continue