feat: add tool labels (#2178)

2024-01-24 20:14:45 +08:00
parent 0940084fd2
commit 7cb75cb2e7
8 changed files with 90 additions and 3 deletions
--- a/api/fields/conversation_fields.py
+++ b/api/fields/conversation_fields.py
@@ -49,10 +49,11 @@ agent_thought_fields = {
    'position': fields.Integer,
    'thought': fields.String,
    'tool': fields.String,
+    'tool_labels': fields.Raw,
    'tool_input': fields.String,
    'created_at': TimestampField,
    'observation': fields.String,
-    'files': fields.List(fields.String)
+    'files': fields.List(fields.String),
 }

 message_detail_fields = {