fix: cannot access local variable 'icon_dark'

11 months ago · 0e17d1263a
parent 94a13d7d62
commit 0e17d1263a
1 changed files with 2 additions and 0 deletions
--- a/api/core/workflow/nodes/tool/tool_node.py
+++ b/api/core/workflow/nodes/tool/tool_node.py
@ -316,6 +316,7 @@ class ToolNode(BaseNode[ToolNodeData]):
                assert isinstance(message.message, ToolInvokeMessage.LogMessage)
                if message.message.metadata:
                    icon = tool_info.get("icon", "")
+                    icon_dark = None
                    dict_metadata = dict(message.message.metadata)
                    if dict_metadata.get("provider"):
                        manager = PluginInstaller()
@ -327,6 +328,7 @@ class ToolNode(BaseNode[ToolNodeData]):
                                if f"{plugin.plugin_id}/{plugin.name}" == dict_metadata["provider"]
                            )
                            icon = current_plugin.declaration.icon
+                            icon_dark = current_plugin.declaration.icon_dark
                        except StopIteration:
                            pass
                        try: