gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Remove LanceDB fallback

Browse source
This commit is contained in:
Igor Ilic 2025-10-28 14:35:08 +01:00
parent abb1aba517
commit 15a0315375
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cognee/infrastructure/databases/vector/create_vector_engine.py
View file

@ -47,7 +47,7 @@ def create_vector_engine(
embedding_engine=embedding_engine, embedding_engine=embedding_engine,
) )
if vector_db_provider == "pgvector": if vector_db_provider.lower() == "pgvector":
from cognee.infrastructure.databases.relational import get_relational_config from cognee.infrastructure.databases.relational import get_relational_config
# Get configuration for postgres database # Get configuration for postgres database
@ -78,7 +78,7 @@ def create_vector_engine(
embedding_engine, embedding_engine,
) )
elif vector_db_provider == "chromadb": elif vector_db_provider.lower() == "chromadb":
try: try:
import chromadb import chromadb
except ImportError: except ImportError:
@ -94,7 +94,7 @@ def create_vector_engine(
embedding_engine=embedding_engine, embedding_engine=embedding_engine,
) )
elif vector_db_provider == "neptune_analytics": elif vector_db_provider.lower() == "neptune_analytics":
try: try:
from langchain_aws import NeptuneAnalyticsGraph from langchain_aws import NeptuneAnalyticsGraph
except ImportError: except ImportError:
@ -122,7 +122,7 @@ def create_vector_engine(
embedding_engine=embedding_engine, embedding_engine=embedding_engine,
) )
else: elif vector_db_provider.lower() == "lancedb":
from .lancedb.LanceDBAdapter import LanceDBAdapter from .lancedb.LanceDBAdapter import LanceDBAdapter
return LanceDBAdapter( return LanceDBAdapter(
@ -130,3 +130,9 @@ def create_vector_engine(
api_key=vector_db_key, api_key=vector_db_key,
embedding_engine=embedding_engine, embedding_engine=embedding_engine,
) )
else:
raise EnvironmentError(
f"Unsupported graph database provider: {vector_db_provider}. "
f"Supported providers are: {', '.join(list(supported_databases.keys()) + ['LanceDB', 'PGVector', 'neptune_analytics', 'ChromaDB'])}"
)