From c3f0cb95da2d61dd9523079c1977a67dd031c7c6 Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Tue, 28 Oct 2025 18:06:04 +0100 Subject: [PATCH] fix: delete unnecessary comments, add to config --- cognee/context_global_variables.py | 19 +++---------------- 1 file changed, 3 insertions(+), 16 deletions(-) diff --git a/cognee/context_global_variables.py b/cognee/context_global_variables.py index ee2e37030..09a351f15 100644 --- a/cognee/context_global_variables.py +++ b/cognee/context_global_variables.py @@ -59,32 +59,19 @@ async def set_database_global_context_variables(dataset: Union[str, UUID], user_ vector_config = { "vector_db_provider": dataset_database.vector_database_provider, "vector_db_url": dataset_database.vector_database_url, - # TODO: Maybe add key to dataset_database, and put it here?? - "vector_db_key": "", + "vector_db_key": dataset_database.vector_database_key, + "vector_db_name": dataset_database.vector_database_name, } - # vector_config = { - # "vector_db_url": os.path.join( - # databases_directory_path, dataset_database.vector_database_name - # ), - # "vector_db_key": "", - # "vector_db_provider": "lancedb", - # } - graph_config = { "graph_database_provider": dataset_database.graph_database_provider, "graph_database_url": dataset_database.graph_database_url, "graph_database_name": dataset_database.graph_database_name, + "graph_database_key": dataset_database.graph_database_key, "graph_file_path": os.path.join( databases_directory_path, dataset_database.graph_database_name ), } - # graph_config = { - # "graph_database_provider": "kuzu", - # "graph_file_path": os.path.join( - # databases_directory_path, dataset_database.graph_database_name - # ), - # } storage_config = { "data_root_directory": data_root_directory,