瀏覽代碼

Update dialogue_manager: increase dialogue history to recover

StrayWarrior 1 月之前
父節點
當前提交
4402ec5828
共有 1 個文件被更改,包括 3 次插入2 次删除
  1. 3 2
      pqai_agent/dialogue_manager.py

+ 3 - 2
pqai_agent/dialogue_manager.py

@@ -150,7 +150,9 @@ class DialogueManager:
         self.current_state, self.previous_state = self.state_cache.get_state(self.staff_id, self.user_id)
 
         # 从数据库恢复对话状态
-        self.dialogue_history = self.history_dialogue_service.get_dialogue_history(self.staff_id, self.user_id)
+        minutes_to_get = 5 * 24 * 60
+        self.dialogue_history = self.history_dialogue_service.get_dialogue_history(
+            self.staff_id, self.user_id, minutes_to_get)
         if self.dialogue_history:
             self.last_interaction_time = self.dialogue_history[-1]['timestamp']
             if self.current_state == DialogueState.MESSAGE_AGGREGATING:
@@ -476,7 +478,6 @@ class DialogueManager:
 
         current_datetime = datetime.now()
         context = {
-            "user_profile": self.user_profile,
             "current_state": self.current_state.name,
             "previous_state": self.previous_state.name,
             "current_time_period": time_context.description,