From 54908ebd30077d61df519c66c591d5f21474091e Mon Sep 17 00:00:00 2001 From: liuhua <10215101452@stu.ecnu.edu.cn> Date: Mon, 30 Dec 2024 13:04:51 +0800 Subject: [PATCH] Fix the bug in create_dataset function (#4284) ### What problem does this PR solve? Fix the bug in create_dataset function #4136 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --------- Co-authored-by: liuhua <10215101452@stu.ecun.edu.cn> --- docs/references/python_api_reference.md | 2 +- sdk/python/ragflow_sdk/ragflow.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/docs/references/python_api_reference.md b/docs/references/python_api_reference.md index 281c7bcc2..d94ec444a 100644 --- a/docs/references/python_api_reference.md +++ b/docs/references/python_api_reference.md @@ -28,7 +28,7 @@ RAGFlow.create_dataset( name: str, avatar: str = "", description: str = "", - embedding_model: str = "BAAI/bge-zh-v1.5", + embedding_model: str = "BAAI/bge-large-zh-v1.5", language: str = "English", permission: str = "me", chunk_method: str = "naive", diff --git a/sdk/python/ragflow_sdk/ragflow.py b/sdk/python/ragflow_sdk/ragflow.py index 472cec493..05158f6f0 100644 --- a/sdk/python/ragflow_sdk/ragflow.py +++ b/sdk/python/ragflow_sdk/ragflow.py @@ -46,13 +46,15 @@ class RAGFlow: res = requests.put(url=self.api_url + path, json= json,headers=self.authorization_header) return res - def create_dataset(self, name: str, avatar: str = "", description: str = "", language: str = "English", + def create_dataset(self, name: str, avatar: str = "", description: str = "", embedding_model:str = "BAAI/bge-large-zh-v1.5", + language: str = "English", permission: str = "me",chunk_method: str = "naive", parser_config: DataSet.ParserConfig = None) -> DataSet: if parser_config: parser_config = parser_config.to_json() res = self.post("/datasets", - {"name": name, "avatar": avatar, "description": description, "language": language, + {"name": name, "avatar": avatar, "description": description,"embedding_model":embedding_model, + "language": language, "permission": permission, "chunk_method": chunk_method, "parser_config": parser_config }