15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-31 02:02:48 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: filter applications by publish status in chat step and node

This commit is contained in:
CaptainB 2025-12-30 16:19:40 +08:00
parent 018f4763ea
commit edc8acda21
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py
View File

@ -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

4
apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py
View File

@ -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