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) graph_db_config = ContextVar("graph_db_config", default=None)
session_user = ContextVar("session_user", default=None) session_user = ContextVar("session_user", default=None)
VECTOR_DBS_WITH_MULTI_USER_SUPPORT = ["lancedb"] VECTOR_DBS_WITH_MULTI_USER_SUPPORT = ["lancedb", "falkor"]
GRAPH_DBS_WITH_MULTI_USER_SUPPORT = ["kuzu"] GRAPH_DBS_WITH_MULTI_USER_SUPPORT = ["kuzu", "falkor"]
async def set_session_user_context_variable(user): 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": if graph_config.graph_database_provider == "kuzu":
graph_db_name = f"{dataset_id}.pkl" graph_db_name = f"{dataset_id}.pkl"
else: else:
graph_db_name = dataset_id graph_db_name = f"{dataset_id}"
if vector_config.vector_db_provider == "lancedb": if vector_config.vector_db_provider == "lancedb":
vector_db_name = f"{dataset_id}.lance.db" vector_db_name = f"{dataset_id}.lance.db"
else: else:
vector_db_name = dataset_id vector_db_name = f"{dataset_id}"
async with db_engine.get_async_session() as session: async with db_engine.get_async_session() as session:
# Create dataset if it doesn't exist # Create dataset if it doesn't exist