File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff 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 = [
You can’t perform that action at this time.
0 commit comments