From 9a5e4aaea8d4a1d6a42b5ad08ebbce13d17efc66 Mon Sep 17 00:00:00 2001 From: Ashwin Krishna Kumar Date: Tue, 2 Jul 2024 16:22:00 +0530 Subject: [PATCH] Add database operations to MilvusClient (#2152) Allows the following operations to be performed through the `MilvusClient` interface: - create_database - drop_database - list_databases Resolves https://github.com/milvus-io/pymilvus/issues/2151 Signed-off-by: Ashwin Krishna Kumar --- pymilvus/milvus_client/milvus_client.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/pymilvus/milvus_client/milvus_client.py b/pymilvus/milvus_client/milvus_client.py index 072044efd..79430dcb1 100644 --- a/pymilvus/milvus_client/milvus_client.py +++ b/pymilvus/milvus_client/milvus_client.py @@ -970,3 +970,15 @@ def list_aliases( 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): + conn = self._get_connection() + conn.create_database(db_name, **kwargs) + + def drop_database(self, db_name: str, **kwargs): + conn = self._get_connection() + conn.drop_database(db_name, **kwargs) + + def list_databases(self, **kwargs) -> List[str]: + conn = self._get_connection() + return conn.list_database(**kwargs)