add tool resource

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

View File

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

View File

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