diff --git a/api/apps/sdk/doc.py b/api/apps/sdk/doc.py index a71cc7328..f76065c1c 100644 --- a/api/apps/sdk/doc.py +++ b/api/apps/sdk/doc.py @@ -856,6 +856,12 @@ def list_chunks(tenant_id, dataset_id, document_id): required: false default: 30 description: Number of items per page. + - in: query + name: id + type: string + required: false + default: "" + description: Chunk Id. - in: header name: Authorization type: string diff --git a/sdk/python/ragflow_sdk/modules/document.py b/sdk/python/ragflow_sdk/modules/document.py index 60a6e064b..187d08901 100644 --- a/sdk/python/ragflow_sdk/modules/document.py +++ b/sdk/python/ragflow_sdk/modules/document.py @@ -66,8 +66,8 @@ class Document(Base): except json.JSONDecodeError: return res.content - def list_chunks(self, page=1, page_size=30, keywords=""): - data = {"keywords": keywords, "page": page, "page_size": page_size} + def list_chunks(self, page=1, page_size=30, keywords="", id = ""): + data = {"keywords": keywords, "page": page, "page_size": page_size, "id": id} res = self.get(f'/datasets/{self.dataset_id}/documents/{self.id}/chunks', data) res = res.json() if res.get("code") == 0: