Skip to content

[Enhancement] Fix error message in datastore/factory.py #388

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
61 changes: 30 additions & 31 deletions datastore/factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,63 +7,62 @@ async def get_datastore() -> DataStore:
assert datastore is not None

match datastore:
case "analyticdb":
from datastore.providers.analyticdb_datastore import AnalyticDBDataStore

return AnalyticDBDataStore()
case "azuresearch":
from datastore.providers.azuresearch_datastore import AzureSearchDataStore

return AzureSearchDataStore()
case "chroma":
from datastore.providers.chroma_datastore import ChromaDataStore

return ChromaDataStore()
case "elasticsearch":
from datastore.providers.elasticsearch_datastore import (
ElasticsearchDataStore,
)

return ElasticsearchDataStore()
case "llama":
from datastore.providers.llama_datastore import LlamaDataStore

return LlamaDataStore()

case "pinecone":
from datastore.providers.pinecone_datastore import PineconeDataStore

return PineconeDataStore()
case "weaviate":
from datastore.providers.weaviate_datastore import WeaviateDataStore

return WeaviateDataStore()
case "milvus":
from datastore.providers.milvus_datastore import MilvusDataStore

return MilvusDataStore()
case "zilliz":
from datastore.providers.zilliz_datastore import ZillizDataStore
case "pinecone":
from datastore.providers.pinecone_datastore import PineconeDataStore

return ZillizDataStore()
case "redis":
from datastore.providers.redis_datastore import RedisDataStore
return PineconeDataStore()
case "postgres":
from datastore.providers.postgres_datastore import PostgresDataStore

return await RedisDataStore.init()
return PostgresDataStore()
case "qdrant":
from datastore.providers.qdrant_datastore import QdrantDataStore

return QdrantDataStore()
case "azuresearch":
from datastore.providers.azuresearch_datastore import AzureSearchDataStore
case "redis":
from datastore.providers.redis_datastore import RedisDataStore

return AzureSearchDataStore()
return await RedisDataStore.init()
case "supabase":
from datastore.providers.supabase_datastore import SupabaseDataStore

return SupabaseDataStore()
case "postgres":
from datastore.providers.postgres_datastore import PostgresDataStore

return PostgresDataStore()
case "analyticdb":
from datastore.providers.analyticdb_datastore import AnalyticDBDataStore
case "weaviate":
from datastore.providers.weaviate_datastore import WeaviateDataStore

return AnalyticDBDataStore()
case "elasticsearch":
from datastore.providers.elasticsearch_datastore import (
ElasticsearchDataStore,
)
return WeaviateDataStore()
case "zilliz":
from datastore.providers.zilliz_datastore import ZillizDataStore

return ElasticsearchDataStore()
return ZillizDataStore()
case _:
raise ValueError(
f"Unsupported vector database: {datastore}. "
f"Try one of the following: llama, elasticsearch, pinecone, weaviate, milvus, zilliz, redis, azuresearch, or qdrant"
f"Try one of the following: analyticdb, azuresearch, chroma, elasticsearch, llama, milvus, pinecone, postgres, qdrant, redis, supabase, weaviate, or zilliz"
)