fix db engine config

This commit is contained in:
Vasilije 2024-05-25 19:17:34 +02:00
parent 3a7abb3853
commit a585938128

View file

@ -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"),