outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add tool resource

Browse Source
This commit is contained in:
jyong 2024-05-10 18:08:49 +08:00
parent 919c45b639
commit bd40d25bc6
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/workflow/nodes/llm/knowledge_resource.py
View File

@ -11,6 +11,7 @@ class KnowledgeResource(BaseModel):
title: str title: str
url: Optional[str] = None url: Optional[str] = None
icon: Optional[str] = None icon: Optional[str] = None
resource_from: Optional[str] = None
score: Optional[float] = None score: Optional[float] = None
metadata: Optional[dict[str, Any]] = None metadata: Optional[dict[str, Any]] = None
@ -20,6 +21,7 @@ class KnowledgeResource(BaseModel):
'title': self.title, 'title': self.title,
'url': self.url, 'url': self.url,
'icon': self.icon, 'icon': self.icon,
'resource_from': self.resource_from,
'score': self.score, 'score': self.score,
'metadata': self.metadata 'metadata': self.metadata
} }

1
api/core/workflow/nodes/tool/tool_node.py
View File

@ -197,6 +197,7 @@ class ToolNode(BaseNode):
if message.type == ToolInvokeMessage.MessageType.CHUNK: if message.type == ToolInvokeMessage.MessageType.CHUNK:
for chunk in message.message: for chunk in message.message:
chunk.icon = icon chunk.icon = icon
chunk.resource_from = node_data.title
chunk.metadata = { chunk.metadata = {
'_source': 'tool' '_source': 'tool'
} }