瀏覽代碼

Update history_dialogue_service: support message type

StrayWarrior 3 天之前
父節點
當前提交
4731488723
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      history_dialogue_service.py

+ 4 - 1
history_dialogue_service.py

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