fix db engine config

This commit is contained in:
Vasilije 2024-05-25 19:27:05 +02:00
parent 5de96318bd
commit 30c9e5f20f

View file

@ -3,21 +3,21 @@ async def main():
import pathlib import pathlib
import cognee import cognee
print("Working dir: ", str(pathlib.Path(__file__).parent)) # print("Working dir: ", str(pathlib.Path(__file__).parent))
data_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.data")).resolve()) # data_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.data")).resolve())
print("Data dir: ", data_directory_path) # print("Data dir: ", data_directory_path)
cognee.config.data_root_directory(data_directory_path) # cognee.config.data_root_directory(data_directory_path)
#
# cognee_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.cognee_system")).resolve())
# print("System dir: ", cognee_directory_path)
# cognee.config.system_root_directory(cognee_directory_path)
cognee_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.cognee_system")).resolve())
print("System dir: ", cognee_directory_path)
cognee.config.system_root_directory(cognee_directory_path)
dataset_name = "cs_explanations" dataset_name = "cs_explanations"
explanation_file_path = path.abspath( explanation_file_path = "test_data/Natural_language_processing.txt"
"/Users/stef/Documents/dev/cognee/cognee/tests/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"