klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

perf: Assistant basic applications use the default private data source mechanism

Browse Source
This commit is contained in:
fit2cloud-chenyw
2025-09-02 14:34:50 +08:00
parent 6de26a39c1
commit 2b5e922a1f
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/system/crud/assistant.py
+7 -2
View File
@@ -40,9 +40,14 @@ def get_assistant_ds(session: Session, llm_service) -> list[dict]:
stmt = select(CoreDatasource.id, CoreDatasource.name, CoreDatasource.description).where(
CoreDatasource.oid == oid)
if not assistant.online:
private_list: list[int] = config.get('private_list') or None
public_list: list[int] = config.get('public_list') or None
if public_list:
stmt = stmt.where(CoreDatasource.id.in_(public_list))
else:
raise RuntimeError('no public datasource valid')
""" private_list: list[int] = config.get('private_list') or None
if private_list:
stmt = stmt.where(~CoreDatasource.id.in_(private_list))
stmt = stmt.where(~CoreDatasource.id.in_(private_list)) """
db_ds_list = session.exec(stmt)
result_list = [