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

Merge branch 'feature/cog-3245-enable-multi-user-for-falkor' of github.com:topoteretes/cognee into feature/cog-3245-enable-multi-user-for-falkor

Browse source
This commit is contained in:
Igor Ilic 2025-11-11 15:13:17 +01:00
commit a8706a2fd2
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

@ -43,12 +43,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}"
base_config = get_base_config() base_config = get_base_config()
databases_directory_path = os.path.join( databases_directory_path = os.path.join(