create_storage

功能描述

类方法，用于创建向量数据库。

函数原型

@classmethod
def create_storage(cls, **kwargs) -> Optional[VectorStore]

参数说明

参数名	数据类型	是否必选	说明
**kwargs	Dict[str, Any]	必选	如果vector_type为“npu_faiss_db”，则参见create。如果vector_type为“milvus_db”，则参见create。如果vector_type为“opengauss_db”，则参见create。

返回值说明

数据类型	说明
Optional[VectorStore]	返回构造成功的向量数据库实例，如果创建失败则会返回None

调用示例

创建npu_faiss_db

from mx_rag.storage.vectorstore import VectorStorageFactory
storage = VectorStorageFactory.create_storage(vector_type="npu_faiss_db", x_dim = 1024,  devs[1], load_local_index="faiss.index")

创建milvus_db

from mx_rag.storage.vectorstore import VectorStorageFactory
client = MilvusClient("https://x.x.x.x:port")
storage = VectorStorageFactory.create_storage(vector_type="milvus_db", x_dim=1024,  client=client)

父主题： VectorStorageFactory