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

feat: support entity and topic wiki modes (#18216)

Browse Source
This commit is contained in:
buua436
2026-08-13 19:03:24 +08:00
committed by GitHub
parent 8f6b5d0e99
commit a4e819504c
21 changed files with 278 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/integration/wiki/test_wiki_incremental.py
View File

@@ -249,12 +249,12 @@ def test_wiki_plan_toggle_resets_state(wiki_dataset):
import asyncio
from rag.svr.task_executor_refactor import dataset_wiki_generator as dwg
# Mode-A (plan=off) build records plan_kwd=false in the mode meta row.
asyncio.run(dwg._wiki_save_mode_plan(tenant_id, ds_id, False))
loaded = asyncio.run(dwg._wiki_load_mode_plan(tenant_id, ds_id))
assert loaded is False, f"expected recorded mode plan=false, got {loaded!r}"
# Entity mode records its mode in the mode meta row.
asyncio.run(dwg._wiki_save_mode(tenant_id, ds_id, "entity"))
loaded = asyncio.run(dwg._wiki_load_mode(tenant_id, ds_id))
assert loaded == "entity", f"expected recorded entity mode, got {loaded!r}"
# Toggling to plan=true (Mode B) is a config change: run_wiki_incremental
# Switching to topic mode is a config change: run_wiki_incremental
# detects prev != new and resets all wiki-derived state so the next build
# rebuilds cleanly in the new mode (no mixing of A/B page structures).
asyncio.run(dwg._wiki_reset_all_wiki_state(tenant_id, ds_id))
@@ -262,4 +262,4 @@ def test_wiki_plan_toggle_resets_state(wiki_dataset):
assert pages_after == 0, "full reset did not clear wiki state"
# After reset the mode meta is gone too (first build of the new mode).
assert asyncio.run(dwg._wiki_load_mode_plan(tenant_id, ds_id)) is None, "mode meta was not cleared by reset"
assert asyncio.run(dwg._wiki_load_mode(tenant_id, ds_id)) is None, "mode meta was not cleared by reset"