From bd40d25bc6c50ec5b1e3d050ac4162e0c2439404 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Fri, 10 May 2024 18:08:49 +0800
Subject: [PATCH] add tool resource

---
 api/core/workflow/nodes/llm/knowledge_resource.py | 2 ++
 api/core/workflow/nodes/tool/tool_node.py         | 1 +
 2 files changed, 3 insertions(+)

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'
                     }