create_storage

Function

Creates a vector database.

Prototype

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

Parameters

Parameter	Data Type	Required/Optional	Description
**kwargs	Dict[str, Any]	Required	If vector_type is npu_faiss_db, see create. If vector_type is milvus_db, see create. If vector_type is opengauss_db, see create.

Return Value

Data Type	Description
Optional[VectorStore]	Returns the vector database instance that is successfully created. If the creation fails, None is returned.

Example

Creating 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")

Creating milvus_db

import getpass
from mx_rag.storage.vectorstore import VectorStorageFactory
client = MilvusClient("https://x.x.x.x:port", user="xxx", password=getpass.getpass(), token="xxx", secure=True,   client_pem_path="path_to/client.pem",   client_key_path="path_to/client.key",   ca_pem_path="path_to/ca.pem",   server_name="localhost")
storage = VectorStorageFactory.create_storage(vector_type="milvus_db", x_dim=1024,  client=client)

Parent topic: VectorStorageFactory