add tool resource
This commit is contained in:
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…
Reference in New Issue
Block a user