Jelajahi Sumber

Update agent_service: add protection for testing period

StrayWarrior 2 minggu lalu
induk
melakukan
91bc9fe7a1
1 mengubah file dengan 6 tambahan dan 1 penghapusan
  1. 6 1
      agent_service.py

+ 6 - 1
agent_service.py

@@ -182,6 +182,11 @@ class AgentService:
         if response := agent.generate_response(chat_response):
             logging.warning(f"staff[{agent.staff_id}] user[{user_id}]: response: {response}")
             current_ts = int(time.time() * 1000)
+            # FIXME(zhoutian)
+            # 测试期间临时逻辑,只发送特定的用户
+            if agent.staff_id not in set(['1688854492669990']):
+                logging.warning(f"skip message from sender [{agent.staff_id}]")
+                return
             self.send_queue.produce(
                 Message.build(MessageType.TEXT, MessageChannel.CORP_WECHAT,
                               agent.staff_id, user_id, response, current_ts)
@@ -283,7 +288,7 @@ if __name__ == "__main__":
         if not text:
             continue
         message_id += 1
-        sender = '7881301263964433'
+        sender = '7881302581935903'
         receiver = '1688854492669990'
         if text == MessageType.AGGREGATION_TRIGGER.name:
             message = Message.build(MessageType.AGGREGATION_TRIGGER, MessageChannel.CORP_WECHAT,