瀏覽代碼

Update push_service: reduce database read

StrayWarrior 3 天之前
父節點
當前提交
3c17aa8ee4
共有 1 個文件被更改,包括 3 次插入3 次删除
  1. 3 3
      pqai_agent/push_service.py

+ 3 - 3
pqai_agent/push_service.py

@@ -62,12 +62,12 @@ class PushScanThread:
             # if abtest_params.get('agent_push_enabled', 'false').lower() != 'true':
             #     logger.debug(f"User {user_id} not enabled agent push, skipping.")
             #     continue
-            agent = self.service.get_agent_instance(staff_id, user_id)
-            should_initiate = agent.should_initiate_conversation()
             user_tags = self.service.user_relation_manager.get_user_tags(user_id)
-
             if configs.get_env() != 'dev' and not white_list_tags.intersection(user_tags):
                 should_initiate = False
+            else:
+                agent = self.service.get_agent_instance(staff_id, user_id)
+                should_initiate = agent.should_initiate_conversation()
             if should_initiate:
                 logger.info(f"user[{user_id}], tags{user_tags}: generate a generation task for conversation initiation")
                 rmq_msg = generate_task_rmq_message(self.rmq_topic, staff_id, user_id, TaskType.GENERATE)