estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Feat/add knowledge include all filter (#12537)

Browse Source
This commit is contained in:
Jyong
2025-01-09 20:21:25 +08:00
committed by GitHub
parent 2e97ba5700
commit 14ee51aead
3 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/services/dataset_service.py
View File

@@ -71,7 +71,7 @@ from tasks.sync_website_document_indexing_task import sync_website_document_inde
class DatasetService:
@staticmethod
def get_datasets(page, per_page, tenant_id=None, user=None, search=None, tag_ids=None):
def get_datasets(page, per_page, tenant_id=None, user=None, search=None, tag_ids=None, include_all=False):
query = Dataset.query.filter(Dataset.tenant_id == tenant_id).order_by(Dataset.created_at.desc())
if user:
@@ -86,7 +86,7 @@ class DatasetService:
else:
return [], 0
else:
if user.current_role != TenantAccountRole.OWNER:
if user.current_role != TenantAccountRole.OWNER or not include_all:
# show all datasets that the user has permission to access
if permitted_dataset_ids:
query = query.filter(