diff --git a/docs/references/python_api_reference.md b/docs/references/python_api_reference.md index 281c7bcc24a..d94ec444a27 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 472cec49384..05158f6f0f8 100644 --- a/sdk/python/ragflow_sdk/ragflow.py +++ b/sdk/python/ragflow_sdk/ragflow.py @@ -46,13 +46,15 @@ def put(self, path, json): 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 }