|
@@ -26,8 +26,9 @@ class ResponseTypeDetector:
|
|
continue
|
|
continue
|
|
if msg['role'] not in role_map:
|
|
if msg['role'] not in role_map:
|
|
continue
|
|
continue
|
|
|
|
+ msg_type = msg.get('type', MessageType.TEXT)
|
|
fmt_time = datetime.fromtimestamp(msg['timestamp'] / 1000).strftime('%Y-%m-%d %H:%M:%S')
|
|
fmt_time = datetime.fromtimestamp(msg['timestamp'] / 1000).strftime('%Y-%m-%d %H:%M:%S')
|
|
- messages.append('[{}] [{}] {}'.format(role_map[msg['role']], fmt_time, msg['content']))
|
|
|
|
|
|
+ messages.append('[{}][{}][{}] {}'.format(role_map[msg['role']], fmt_time, msg_type.description, msg['content']))
|
|
return '\n'.join(messages)
|
|
return '\n'.join(messages)
|
|
|
|
|
|
def __init__(self):
|
|
def __init__(self):
|