From edc8acda21df759c0846ef96c7f8d5696952c7d5 Mon Sep 17 00:00:00 2001 From: CaptainB Date: Tue, 30 Dec 2025 16:19:40 +0800 Subject: [PATCH] fix: filter applications by publish status in chat step and node --- .../chat_pipeline/step/chat_step/impl/base_chat_step.py | 4 +++- .../flow/step_node/ai_chat_step_node/impl/base_chat_node.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py b/apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py index f42241c68..57c55784f 100644 --- a/apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py +++ b/apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py @@ -269,7 +269,9 @@ class BaseChatStep(IChatStep): if application_ids and len(application_ids) > 0: self.context['application_ids'] = application_ids for application_id in application_ids: - app = QuerySet(Application).filter(id=application_id).first() + app = QuerySet(Application).filter(id=application_id, is_publish=True).first() + if app is None: + continue app_key = QuerySet(ApplicationApiKey).filter(application_id=application_id, is_active=True).first() if app_key is not None: api_key = app_key.secret_key diff --git a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py index 980cdf648..231b4139f 100644 --- a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py +++ b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py @@ -256,7 +256,9 @@ class BaseChatNode(IChatNode): if application_ids and len(application_ids) > 0: self.context['application_ids'] = application_ids for application_id in application_ids: - app = QuerySet(Application).filter(id=application_id).first() + app = QuerySet(Application).filter(id=application_id, is_publish=True).first() + if app is None: + continue app_key = QuerySet(ApplicationApiKey).filter(application_id=application_id, is_active=True).first() if app_key is not None: api_key = app_key.secret_key