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

try few things

Browse source
This commit is contained in:
Vasilije 2024-05-25 16:57:44 +02:00
parent 1d33ed938b
commit cf249a522e
3 changed files with 15 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cognee/config.py
View file

@ -35,6 +35,7 @@ class Config:
system_root_directory = get_absolute_path(".cognee_system")
logging.info("system_root_directory: %s", system_root_directory)
data_root_directory = os.getenv("DATA_PATH", get_absolute_path(".data"))
vectordb: str = os.getenv("VECTORDB", "weaviate")

7
cognee/root_dir.py
View file

@ -1,6 +1,11 @@
from os import path
import logging
ROOT_DIR = path.dirname(path.abspath(__file__))
logging.debug("ROOT_DIR: ", ROOT_DIR)
def get_absolute_path(path_from_root: str) -> str:
logging.debug("abspath: ", path.abspath(path.join(ROOT_DIR, path_from_root)))
return path.abspath(path.join(ROOT_DIR, path_from_root))

16
cognee/tests/test_library.py
View file

@ -3,14 +3,14 @@ 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)
#
# 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)
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)
dataset_name = "cs_explanations"