llama-index

1
from enum import Enum
2
from typing import Dict, Type
3

4
from llama_index.legacy.vector_stores.bagel import BagelVectorStore
5
from llama_index.legacy.vector_stores.cassandra import CassandraVectorStore
6
from llama_index.legacy.vector_stores.chatgpt_plugin import ChatGPTRetrievalPluginClient
7
from llama_index.legacy.vector_stores.chroma import ChromaVectorStore
8
from llama_index.legacy.vector_stores.deeplake import DeepLakeVectorStore
9
from llama_index.legacy.vector_stores.epsilla import EpsillaVectorStore
10
from llama_index.legacy.vector_stores.faiss import FaissVectorStore
11
from llama_index.legacy.vector_stores.jaguar import JaguarVectorStore
12
from llama_index.legacy.vector_stores.lancedb import LanceDBVectorStore
13
from llama_index.legacy.vector_stores.milvus import MilvusVectorStore
14
from llama_index.legacy.vector_stores.myscale import MyScaleVectorStore
15
from llama_index.legacy.vector_stores.opensearch import OpensearchVectorStore
16
from llama_index.legacy.vector_stores.pinecone import PineconeVectorStore
17
from llama_index.legacy.vector_stores.qdrant import QdrantVectorStore
18
from llama_index.legacy.vector_stores.redis import RedisVectorStore
19
from llama_index.legacy.vector_stores.rocksetdb import RocksetVectorStore
20
from llama_index.legacy.vector_stores.simple import SimpleVectorStore
21
from llama_index.legacy.vector_stores.supabase import SupabaseVectorStore
22
from llama_index.legacy.vector_stores.txtai import TxtaiVectorStore
23
from llama_index.legacy.vector_stores.types import VectorStore
24
from llama_index.legacy.vector_stores.upstash import UpstashVectorStore
25
from llama_index.legacy.vector_stores.weaviate import WeaviateVectorStore
26

27

28
class VectorStoreType(str, Enum):
29
    SIMPLE = "simple"
30
    REDIS = "redis"
31
    WEAVIATE = "weaviate"
32
    QDRANT = "qdrant"
33
    PINECONE = "pinecone"
34
    OPENSEARCH = "opensearch"
35
    FAISS = "faiss"
36
    TXTAI = "txtai"
37
    CASSANDRA = "cassandra"
38
    CHROMA = "chroma"
39
    CHATGPT_PLUGIN = "chatgpt_plugin"
40
    LANCEDB = "lancedb"
41
    MILVUS = "milvus"
42
    DEEPLAKE = "deeplake"
43
    MYSCALE = "myscale"
44
    SUPABASE = "supabase"
45
    ROCKSET = "rockset"
46
    BAGEL = "bagel"
47
    EPSILLA = "epsilla"
48
    JAGUAR = "jaguar"
49
    UPSTASH = "upstash"
50

51

52
VECTOR_STORE_TYPE_TO_VECTOR_STORE_CLASS: Dict[VectorStoreType, Type[VectorStore]] = {
53
    VectorStoreType.SIMPLE: SimpleVectorStore,
54
    VectorStoreType.REDIS: RedisVectorStore,
55
    VectorStoreType.WEAVIATE: WeaviateVectorStore,
56
    VectorStoreType.QDRANT: QdrantVectorStore,
57
    VectorStoreType.LANCEDB: LanceDBVectorStore,
58
    VectorStoreType.SUPABASE: SupabaseVectorStore,
59
    VectorStoreType.MILVUS: MilvusVectorStore,
60
    VectorStoreType.PINECONE: PineconeVectorStore,
61
    VectorStoreType.OPENSEARCH: OpensearchVectorStore,
62
    VectorStoreType.FAISS: FaissVectorStore,
63
    VectorStoreType.TXTAI: TxtaiVectorStore,
64
    VectorStoreType.CASSANDRA: CassandraVectorStore,
65
    VectorStoreType.CHROMA: ChromaVectorStore,
66
    VectorStoreType.CHATGPT_PLUGIN: ChatGPTRetrievalPluginClient,
67
    VectorStoreType.DEEPLAKE: DeepLakeVectorStore,
68
    VectorStoreType.MYSCALE: MyScaleVectorStore,
69
    VectorStoreType.ROCKSET: RocksetVectorStore,
70
    VectorStoreType.BAGEL: BagelVectorStore,
71
    VectorStoreType.EPSILLA: EpsillaVectorStore,
72
    VectorStoreType.JAGUAR: JaguarVectorStore,
73
    VectorStoreType.UPSTASH: UpstashVectorStore,
74
}
75

76
VECTOR_STORE_CLASS_TO_VECTOR_STORE_TYPE: Dict[Type[VectorStore], VectorStoreType] = {
77
    cls_: type_ for type_, cls_ in VECTOR_STORE_TYPE_TO_VECTOR_STORE_CLASS.items()
78
}
79