From a5859381282aa1b4154e8f4cb447f5a23af35f79 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Sat, 25 May 2024 19:17:34 +0200
Subject: [PATCH] fix db engine config

---
 cognee/modules/tasks/create_task_status_table.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/cognee/modules/tasks/create_task_status_table.py b/cognee/modules/tasks/create_task_status_table.py
index 26beceb1c..5fa99e9bf 100644
--- a/cognee/modules/tasks/create_task_status_table.py
+++ b/cognee/modules/tasks/create_task_status_table.py
@@ -1,7 +1,7 @@
-from cognee.infrastructure.InfrastructureConfig import infrastructure_config
-
+from cognee.infrastructure.databases.relational.config import get_relationaldb_config
+config = get_relationaldb_config()
 def create_task_status_table():
-    db_engine = infrastructure_config.get_config("database_engine")
+    db_engine = config.database_engine
 
     db_engine.create_table("cognee_task_status", [
         dict(name = "data_id", type = "STRING"),