diff --git a/api/core/workflow/nodes/llm/knowledge_resource.py b/api/core/workflow/nodes/llm/knowledge_resource.py index aa1c1bb755..b3e11a96f9 100644 --- a/api/core/workflow/nodes/llm/knowledge_resource.py +++ b/api/core/workflow/nodes/llm/knowledge_resource.py @@ -11,6 +11,7 @@ class KnowledgeResource(BaseModel): title: str url: Optional[str] = None icon: Optional[str] = None + resource_from: Optional[str] = None score: Optional[float] = None metadata: Optional[dict[str, Any]] = None @@ -20,6 +21,7 @@ class KnowledgeResource(BaseModel): 'title': self.title, 'url': self.url, 'icon': self.icon, + 'resource_from': self.resource_from, 'score': self.score, 'metadata': self.metadata } diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py index 6ec16e32f4..9d0a8c7937 100644 --- a/api/core/workflow/nodes/tool/tool_node.py +++ b/api/core/workflow/nodes/tool/tool_node.py @@ -197,6 +197,7 @@ class ToolNode(BaseNode): if message.type == ToolInvokeMessage.MessageType.CHUNK: for chunk in message.message: chunk.icon = icon + chunk.resource_from = node_data.title chunk.metadata = { '_source': 'tool' }