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