Sfoglia il codice sorgente

Update history_dialogue_service: support message type

StrayWarrior 3 giorni fa
parent
commit
4731488723
1 ha cambiato i file con 4 aggiunte e 1 eliminazioni
  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 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