Sfoglia il codice sorgente

Update dialogue_manager: quick fix for current time

StrayWarrior 1 settimana fa
parent
commit
81ed62ca43
1 ha cambiato i file con 3 aggiunte e 1 eliminazioni
  1. 3 1
      dialogue_manager.py

+ 3 - 1
dialogue_manager.py

@@ -491,7 +491,9 @@ class DialogueManager:
                 fmt_time = self.format_timestamp(dialogue_history[0]['timestamp'])
                 messages.append(cozepy.Message.build_user_question_text(f'[{fmt_time}] '))
             # coze最后一条消息必须为user,且可能吞掉连续的user消息,故强制增加一条空消息(可参与合并)
-            current_time = overwrite_context.get('current_time', datetime.now().strftime('%Y-%m-%d %H:%M:%S'))
+            current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+            if overwrite_context and 'current_time' in overwrite_context:
+                current_time = overwrite_context.get('current_time')
             dialogue_history.append({
                 'role': 'user',
                 'content': ' ',