Browse Source

Update dialogue_manager: add is_valid

StrayWarrior 1 month ago
parent
commit
a5db13ce0c
2 changed files with 8 additions and 0 deletions
  1. 3 0
      pqai_agent/agent_service.py
  2. 5 0
      pqai_agent/dialogue_manager.py

+ 3 - 0
pqai_agent/agent_service.py

@@ -218,6 +218,9 @@ class AgentService:
         # 获取用户信息和Agent实例
         user_profile = self.user_manager.get_user_profile(user_id)
         agent = self._get_agent_instance(staff_id, user_id)
+        if not agent.is_valid():
+            logger.error(f"staff[{staff_id}] user[{user_id}]: agent is invalid")
+            return
 
         # 更新对话状态
         logger.debug("process message: {}".format(message))

+ 5 - 0
pqai_agent/dialogue_manager.py

@@ -140,6 +140,11 @@ class DialogueManager:
         else:
             return TimeContext.NIGHT
 
+    def is_valid(self):
+        if not self.staff_profile.get('agent_name', None):
+            return False
+        return True
+
     def _recover_state(self):
         self.current_state, self.previous_state = self.state_cache.get_state(self.staff_id, self.user_id)