From c7cf7aad4e55acb4a25293b7069b67587c16f384 Mon Sep 17 00:00:00 2001 From: Magicbook1108 Date: Mon, 22 Dec 2025 11:09:56 +0800 Subject: [PATCH] Fix: update RAGFlow SDK for consistency (#12065) ### What problem does this PR solve? Fix: update RAGFlow SDK for consistency #12059 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- sdk/python/ragflow_sdk/ragflow.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/sdk/python/ragflow_sdk/ragflow.py b/sdk/python/ragflow_sdk/ragflow.py index e46023177..da8a3d336 100644 --- a/sdk/python/ragflow_sdk/ragflow.py +++ b/sdk/python/ragflow_sdk/ragflow.py @@ -201,6 +201,8 @@ class RAGFlow: keyword: bool = False, cross_languages: list[str]|None = None, metadata_condition: dict | None = None, + use_kg: bool = False, + toc_enhance: bool = False, ): if document_ids is None: document_ids = [] @@ -216,7 +218,9 @@ class RAGFlow: "dataset_ids": dataset_ids, "document_ids": document_ids, "cross_languages": cross_languages, - "metadata_condition": metadata_condition + "metadata_condition": metadata_condition, + "use_kg": use_kg, + "toc_enhance": toc_enhance } # Send a POST request to the backend service (using requests library as an example, actual implementation may vary) res = self.post("/retrieval", json=data_json)