jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add tool labels (#2178)

Browse source
This commit is contained in:
Yeuoly 2024-01-24 20:14:45 +08:00 • committed by GitHub
commit 7cb75cb2e7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 90 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
api/models/model.py
View file

@ -1003,6 +1003,7 @@ class MessageAgentThought(db.Model):
position = db.Column(db.Integer, nullable=False)
thought = db.Column(db.Text, nullable=True)
tool = db.Column(db.Text, nullable=True)
tool_labels_str = db.Column(db.Text, nullable=False, server_default=db.text("'{}'::text"))
tool_input = db.Column(db.Text, nullable=True)
observation = db.Column(db.Text, nullable=True)
# plugin_id = db.Column(UUID, nullable=True) ## for future design
@ -1030,6 +1031,16 @@ class MessageAgentThought(db.Model):
return json.loads(self.message_files)
else:
return []
@property
def tool_labels(self) -> dict:
try:
if self.tool_labels_str:
return json.loads(self.tool_labels_str)
else:
return {}
except Exception as e:
return {}
class DatasetRetrieverResource(db.Model):
__tablename__ = 'dataset_retriever_resources'