mirror of
https://github.com/csunny/DB-GPT.git
synced 2026-01-25 23:04:03 +00:00
42 lines
901 B
Python
42 lines
901 B
Python
from typing import Any
|
|
|
|
|
|
def _import_pgvector() -> Any:
|
|
from pilot.vector_store.pgvector_store import PGVectorStore
|
|
|
|
return PGVectorStore
|
|
|
|
|
|
def _import_milvus() -> Any:
|
|
from pilot.vector_store.milvus_store import MilvusStore
|
|
|
|
return MilvusStore
|
|
|
|
|
|
def _import_chroma() -> Any:
|
|
from pilot.vector_store.chroma_store import ChromaStore
|
|
|
|
return ChromaStore
|
|
|
|
|
|
def _import_weaviate() -> Any:
|
|
from pilot.vector_store.weaviate_store import WeaviateStore
|
|
|
|
return WeaviateStore
|
|
|
|
|
|
def __getattr__(name: str) -> Any:
|
|
if name == "Chroma":
|
|
return _import_chroma()
|
|
elif name == "Milvus":
|
|
return _import_milvus()
|
|
elif name == "Weaviate":
|
|
return _import_weaviate()
|
|
elif name == "PGVector":
|
|
return _import_pgvector()
|
|
else:
|
|
raise AttributeError(f"Could not find: {name}")
|
|
|
|
|
|
__all__ = ["Chroma", "Milvus", "Weaviate", "PGVector"]
|