zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-30 04:29:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix filter dataset by owner_ids not working (#17499)

Browse Source
This commit is contained in:
Wang Qi
2026-07-29 11:48:40 +08:00
committed by GitHub
parent 0eb8e0b393
commit 989529c00a
2 changed files with 36 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/apps/services/dataset_api_service.py
View File

@@ -427,12 +427,18 @@ def list_datasets(tenant_id: str, args: dict):
kbs = KnowledgebaseService.get_kb_by_name(name, tenant_id)
if not kbs:
return False, f"User '{tenant_id}' lacks permission for dataset '{name}'"
if ext_fields.get("owner_ids", []):
tenant_ids = ext_fields["owner_ids"]
owner_ids = [owner_id.strip() for owner_id in ext_fields.get("owner_ids", []) if isinstance(owner_id, str) and owner_id.strip()]
if owner_ids:
tenants = TenantService.get_joined_tenants_by_user_id(tenant_id)
allowed_tenant_ids = {m["tenant_id"] for m in tenants}
allowed_tenant_ids.add(tenant_id)
tenant_ids = [owner_id for owner_id in owner_ids if owner_id in allowed_tenant_ids]
query_user_id = tenant_id if tenant_id in tenant_ids else ""
else:
tenants = TenantService.get_joined_tenants_by_user_id(tenant_id)
tenant_ids = [m["tenant_id"] for m in tenants]
kbs, total = KnowledgebaseService.get_list(tenant_ids, tenant_id, page, page_size, orderby, desc, kb_id, name, keywords, parser_id)
query_user_id = tenant_id
kbs, total = KnowledgebaseService.get_list(tenant_ids, query_user_id, page, page_size, orderby, desc, kb_id, name, keywords, parser_id)
users = UserService.get_by_ids([m["tenant_id"] for m in kbs])
user_map = {m.id: m.to_dict() for m in users}