|
@@ -7,6 +7,7 @@ from logging_service import logger
|
|
|
import time
|
|
|
|
|
|
import configs
|
|
|
+from message import MessageType
|
|
|
|
|
|
|
|
|
class HistoryDialogueService:
|
|
@@ -36,10 +37,12 @@ class HistoryDialogueService:
|
|
|
else:
|
|
|
logger.warning("Unknown sender in dialogue history: {}".format(sender))
|
|
|
continue
|
|
|
+ msg_type = record.get('type', MessageType.TEXT.value)
|
|
|
ret.append({
|
|
|
'role': role,
|
|
|
'content': record.get('content', ''),
|
|
|
- 'timestamp': record.get('sendtime', 0)
|
|
|
+ 'timestamp': record.get('sendTime', 0),
|
|
|
+ 'type': MessageType(msg_type)
|
|
|
})
|
|
|
ret = sorted(ret, key=lambda x: x['timestamp'])
|
|
|
return ret
|