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

fix db engine config

Browse source
This commit is contained in:
Vasilije 2024-05-25 20:05:25 +02:00
parent 7f43d12102
commit 2e7cfd8ab7
2 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/modules/tasks/create_task_status_table.py
View file

@ -1,7 +1,7 @@
from cognee.infrastructure.databases.relational.config import get_relationaldb_config from cognee.infrastructure.InfrastructureConfig import infrastructure_config
config = get_relationaldb_config()
def create_task_status_table(): def create_task_status_table():
db_engine = config.db_engine db_engine = infrastructure_config.get_config("database_engine")
db_engine.create_table("cognee_task_status", [ db_engine.create_table("cognee_task_status", [
dict(name = "data_id", type = "STRING"), dict(name = "data_id", type = "STRING"),

1
cognee/tests/test_library.py
View file

@ -25,7 +25,6 @@ async def main():
dataset_name = "cs_explanations" dataset_name = "cs_explanations"
explanation_file_path = os.path.join(pathlib.Path(__file__).parent, "test_data/Natural_language_processing.txt") explanation_file_path = os.path.join(pathlib.Path(__file__).parent, "test_data/Natural_language_processing.txt")
#
await cognee.add([explanation_file_path], dataset_name) await cognee.add([explanation_file_path], dataset_name)
# dataset_name = "short_stories" # dataset_name = "short_stories"