add tool resource

feat/add-resource-from-tools
jyong 2 years ago
parent 919c45b639
commit bd40d25bc6

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

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

Loading…
Cancel
Save