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 cognee
print("Working dir: ", str(pathlib.Path(__file__).parent))
data_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.data")).resolve())
print("Data dir: ", data_directory_path)
cognee.config.data_root_directory(data_directory_path)
# print("Working dir: ", str(pathlib.Path(__file__).parent))
# data_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.data")).resolve())
# print("Data dir: ", 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"
explanation_file_path = path.abspath(
"/Users/stef/Documents/dev/cognee/cognee/tests/test_data/Natural_language_processing.txt"
)
explanation_file_path = "test_data/Natural_language_processing.txt"
await cognee.add([explanation_file_path], dataset_name)
# dataset_name = "short_stories"