From 46dc3f1c483abfd85791278bce726bb481de141e Mon Sep 17 00:00:00 2001 From: Liu An Date: Mon, 11 Aug 2025 17:15:48 +0800 Subject: [PATCH] Fix: Update test assertions and add GraphRAG config in dataset tests (#9386) ### What problem does this PR solve? - Modify error message assertion in chunk update test to check for document ownership - Add GraphRAG configuration with `use_graphrag: False` in dataset update tests - Fix actions: https://github.com/infiniflow/ragflow/actions/runs/16863637898/job/47767511582 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- .../test_chunk_management_within_dataset/test_update_chunk.py | 2 +- .../test_sdk_api/test_dataset_mangement/test_update_dataset.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/test/testcases/test_sdk_api/test_chunk_management_within_dataset/test_update_chunk.py b/test/testcases/test_sdk_api/test_chunk_management_within_dataset/test_update_chunk.py index dc85d6385..34d73b99e 100644 --- a/test/testcases/test_sdk_api/test_chunk_management_within_dataset/test_update_chunk.py +++ b/test/testcases/test_sdk_api/test_chunk_management_within_dataset/test_update_chunk.py @@ -151,4 +151,4 @@ class TestUpdatedChunk: with pytest.raises(Exception) as excinfo: chunks[0].update({}) - assert f"Can't find this chunk {chunks[0].id}" in str(excinfo.value), str(excinfo.value) + assert f"You don't own the document {chunks[0].document_id}" in str(excinfo.value), str(excinfo.value) diff --git a/test/testcases/test_sdk_api/test_dataset_mangement/test_update_dataset.py b/test/testcases/test_sdk_api/test_dataset_mangement/test_update_dataset.py index 0d57daaca..cc9d67023 100644 --- a/test/testcases/test_sdk_api/test_dataset_mangement/test_update_dataset.py +++ b/test/testcases/test_sdk_api/test_dataset_mangement/test_update_dataset.py @@ -693,6 +693,7 @@ class TestDatasetUpdate: client, { "raptor": {"use_raptor": False}, + "graphrag": {"use_graphrag": False}, }, ) dataset.update({"chunk_method": "qa"}) @@ -708,6 +709,7 @@ class TestDatasetUpdate: client, { "raptor": {"use_raptor": False}, + "graphrag": {"use_graphrag": False}, }, ) dataset.update({"chunk_method": "qa", "parser_config": None})