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

Fix: shared dataset chunk index lookup (#14764)

Browse Source 
### What problem does this PR solve?

shared dataset chunk index lookup

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
buua436
2026-05-11 13:50:08 +08:00
committed by GitHub
parent 024c8cb0b5
commit a03b95f8c4
3 changed files with 69 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
test/testcases/test_http_api/test_file_management_within_dataset/test_doc_sdk_routes_unit.py
View File

@@ -706,6 +706,36 @@ class TestDocRoutesUnit:
assert res["data"]["total"] == 1
assert res["data"]["chunks"][0]["id"] == "chunk-1"
def test_list_chunks_uses_dataset_owner_index_for_team_dataset(self, monkeypatch):
module = _load_restful_chunk_module(monkeypatch)
seen = {}
monkeypatch.setattr(module.KnowledgebaseService, "accessible", lambda **_kwargs: True)
monkeypatch.setattr(
module.KnowledgebaseService,
"get_by_id",
lambda _dataset_id: (True, SimpleNamespace(tenant_id="owner-tenant")),
)
monkeypatch.setattr(module.DocumentService, "query", lambda **_kwargs: [_DummyDoc(kb_id="ds-1")])
monkeypatch.setattr(module, "request", SimpleNamespace(args=_DummyArgs({})))
def _index_exist(index_name, dataset_id):
seen["index_exist"] = (index_name, dataset_id)
return True
class _Retriever:
async def search(self, _query, index_name, dataset_ids, *_args, **_kwargs):
seen["search"] = (index_name, dataset_ids)
return SimpleNamespace(total=0, ids=[], field={}, highlight={})
_patch_docstore(monkeypatch, module, index_exist=_index_exist)
monkeypatch.setattr(module.settings, "retriever", _Retriever())
res = _run(_route_core(module.list_chunks)("member-tenant", "ds-1", "doc-1"))
assert res["code"] == 0
assert seen["index_exist"] == ("idx-owner-tenant", "ds-1")
assert seen["search"] == ("idx-owner-tenant", ["ds-1"])
def test_add_chunk_access_guard(self, monkeypatch):
module = _load_restful_chunk_module(monkeypatch)
monkeypatch.setattr(module.KnowledgebaseService, "accessible", lambda **_kwargs: False)