enhance createdatabase support properties

Signed-off-by: Xianhui.Lin <[email protected]>
milvus-io · Dec 13, 2024 · d3bdfd3 · d3bdfd3
1 parent be184b5
commit d3bdfd3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pymilvus/client/grpc_handler.py b/pymilvus/client/grpc_handler.py
@@ -1359,8 +1359,8 @@ def get_loading_progress(
         return response.progress
 
     @retry_on_rpc_failure()
-    def create_database(self, db_name: str, timeout: Optional[float] = None, **kwargs):
-        request = Prepare.create_database_req(db_name, **kwargs)
+    def create_database(self, db_name: str, properties: Optional[dict] = None, timeout: Optional[float] = None, **kwargs):
+        request = Prepare.create_database_req(db_name, properties=properties, **kwargs)
         status = self._stub.CreateDatabase(request, timeout=timeout)
         check_status(status)
 

diff --git a/pymilvus/milvus_client/milvus_client.py b/pymilvus/milvus_client/milvus_client.py
@@ -1167,9 +1167,9 @@ def using_database(self, db_name: str, **kwargs):
         conn = self._get_connection()
         conn.reset_db_name(db_name)
 
-    def create_database(self, db_name: str, **kwargs):
+    def create_database(self, db_name: str, properties: Optional[dict] = None, **kwargs):
         conn = self._get_connection()
-        conn.create_database(db_name, **kwargs)
+        conn.create_database(db_name, properties, **kwargs)
 
     def drop_database(self, db_name: str, **kwargs):
         conn = self._get_connection()