From 32c9eaff454fd83e1423d0951015f5d864ff8af0 Mon Sep 17 00:00:00 2001 From: Typer_Body Date: Tue, 5 May 2026 16:16:33 +0800 Subject: [PATCH] =?UTF-8?q?=E8=BF=98=E6=98=AF=E4=BF=AE=E4=B8=8D=E5=A5=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/langbot/pkg/workflow/registry.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/src/langbot/pkg/workflow/registry.py b/src/langbot/pkg/workflow/registry.py index 65738912..560d25f0 100644 --- a/src/langbot/pkg/workflow/registry.py +++ b/src/langbot/pkg/workflow/registry.py @@ -69,6 +69,18 @@ class NodeTypeRegistry: for registered_type, node_class in self._nodes.items(): if node_class.type_name == node_type: return node_class + + # Lazy-process pending registrations so execution paths that didn't + # explicitly warm the registry can still resolve newly imported nodes. + if get_pending_registrations(): + self.process_pending_registrations() + + if node_type in self._nodes: + return self._nodes[node_type] + + for registered_type, node_class in self._nodes.items(): + if node_class.type_name == node_type: + return node_class return None