From 1ecf581ba176e60852ed025a7b5e8b0b3e8aef01 Mon Sep 17 00:00:00 2001 From: Novice Date: Wed, 7 Jan 2026 15:05:18 +0800 Subject: [PATCH] chore: add model name in detail --- api/core/workflow/nodes/llm/entities.py | 1 + api/core/workflow/nodes/llm/node.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/api/core/workflow/nodes/llm/entities.py b/api/core/workflow/nodes/llm/entities.py index f441efa8e2..0d61e45003 100644 --- a/api/core/workflow/nodes/llm/entities.py +++ b/api/core/workflow/nodes/llm/entities.py @@ -123,6 +123,7 @@ class LLMTraceSegment(BaseModel): # Common metadata for both model and tool segments provider: str | None = Field(default=None, description="Model or tool provider identifier") + name: str | None = Field(default=None, description="Name of the model or tool") icon: str | None = Field(default=None, description="Icon for the provider") icon_dark: str | None = Field(default=None, description="Dark theme icon for the provider") error: str | None = Field(default=None, description="Error message if segment failed") diff --git a/api/core/workflow/nodes/llm/node.py b/api/core/workflow/nodes/llm/node.py index a8a9c9063c..f1f07addd3 100644 --- a/api/core/workflow/nodes/llm/node.py +++ b/api/core/workflow/nodes/llm/node.py @@ -1720,6 +1720,7 @@ class LLMNode(Node[LLMNodeData]): # Generate model provider icon URL provider = self._node_data.model.provider + model_name = self._node_data.model.name model_icon = self._generate_model_provider_icon_url(provider) model_icon_dark = self._generate_model_provider_icon_url(provider, dark=True) @@ -1734,6 +1735,7 @@ class LLMNode(Node[LLMNodeData]): tool_calls=list(buffers.pending_tool_calls), ), provider=provider, + name=model_name, icon=model_icon, icon_dark=model_icon_dark, error=error, @@ -1848,6 +1850,7 @@ class LLMNode(Node[LLMNodeData]): output=result_str, ), provider=tool_provider, + name=tool_name, icon=tool_icon, icon_dark=tool_icon_dark, error=str(tool_error) if tool_error else None,