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

fix: fix names, add falkor to constants

Browse source
This commit is contained in:
Andrej Milicevic 2025-11-11 15:07:59 +01:00
parent cfc131307f
commit ac6c3ef9de
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/context_global_variables.py
View file

@ -16,8 +16,8 @@ vector_db_config = ContextVar("vector_db_config", default=None)
graph_db_config = ContextVar("graph_db_config", default=None)
session_user = ContextVar("session_user", default=None)
VECTOR_DBS_WITH_MULTI_USER_SUPPORT = ["lancedb"]
GRAPH_DBS_WITH_MULTI_USER_SUPPORT = ["kuzu"]
VECTOR_DBS_WITH_MULTI_USER_SUPPORT = ["lancedb", "falkor"]
GRAPH_DBS_WITH_MULTI_USER_SUPPORT = ["kuzu", "falkor"]
async def set_session_user_context_variable(user):

4
cognee/infrastructure/databases/utils/get_or_create_dataset_database.py
View file

@ -40,12 +40,12 @@ async def get_or_create_dataset_database(
if graph_config.graph_database_provider == "kuzu":
graph_db_name = f"{dataset_id}.pkl"
else:
graph_db_name = dataset_id
graph_db_name = f"{dataset_id}"
if vector_config.vector_db_provider == "lancedb":
vector_db_name = f"{dataset_id}.lance.db"
else:
vector_db_name = dataset_id
vector_db_name = f"{dataset_id}"
async with db_engine.get_async_session() as session:
# Create dataset if it doesn't exist