Skip to content

Commit 2b5e922

Browse files
perf: Assistant basic applications use the default private data source mechanism
1 parent 6de26a3 commit 2b5e922

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

backend/apps/system/crud/assistant.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,9 +40,14 @@ def get_assistant_ds(session: Session, llm_service) -> list[dict]:
4040
stmt = select(CoreDatasource.id, CoreDatasource.name, CoreDatasource.description).where(
4141
CoreDatasource.oid == oid)
4242
if not assistant.online:
43-
private_list: list[int] = config.get('private_list') or None
43+
public_list: list[int] = config.get('public_list') or None
44+
if public_list:
45+
stmt = stmt.where(CoreDatasource.id.in_(public_list))
46+
else:
47+
raise RuntimeError('no public datasource valid')
48+
""" private_list: list[int] = config.get('private_list') or None
4449
if private_list:
45-
stmt = stmt.where(~CoreDatasource.id.in_(private_list))
50+
stmt = stmt.where(~CoreDatasource.id.in_(private_list)) """
4651
db_ds_list = session.exec(stmt)
4752

4853
result_list = [

0 commit comments

Comments
 (0)