diff --git a/cognee/api/v1/cognify/cognify.py b/cognee/api/v1/cognify/cognify.py
index 19ca263f1..5f509ec69 100644
--- a/cognee/api/v1/cognify/cognify.py
+++ b/cognee/api/v1/cognify/cognify.py
@@ -148,6 +148,8 @@ async def process_batch(files_batch):
 
     for (dataset_name, files) in dataset_files:
         for file_metadata in files:
+            from cognee.infrastructure.databases.graph.config import get_graph_config
+            graph_config = get_graph_config()
             graph_topology = graph_config.graph_model
 
             if graph_topology == SourceCodeGraph: