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/message_fields.py
+++ b/api/fields/message_fields.py
@@ -36,6 +36,7 @@ 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,