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 14:17:36 +02:00
parent acc982ae79
commit fb316a1a0d
2 changed files with 14 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/infrastructure/InfrastructureConfig.py
View file

@ -45,10 +45,10 @@ class InfrastructureConfig():
def get_config(self, config_entity: str = None) -> dict: def get_config(self, config_entity: str = None) -> dict:
if (config_entity is None or config_entity == "database_engine") and self.database_engine is None: if (config_entity is None or config_entity == "database_engine") and self.database_engine is None:
if self.system_root_directory is None: # if self.system_root_directory is None:
self.system_root_directory ='/Users/runner/work/cognee/cognee/.cognee_system' # self.system_root_directory ='/Users/runner/work/cognee/cognee/.cognee_system'
if config.db_path is None: # if config.db_path is None:
config.db_path = "databases" # config.db_path = "databases"
db_path = self.system_root_directory + "/" + config.db_path db_path = self.system_root_directory + "/" + config.db_path

20
cognee/tests/test_library.py
View file

@ -3,23 +3,23 @@ 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()) # cognee_directory_path = str(pathlib.Path(path.join(pathlib.Path(__file__).parent, "../../.cognee_system")).resolve())
print("System dir: ", cognee_directory_path) # print("System dir: ", cognee_directory_path)
cognee.config.system_root_directory(cognee_directory_path) # cognee.config.system_root_directory(cognee_directory_path)
dataset_name = "cs_explanations" dataset_name = "cs_explanations"
explanation_file_path = path.join(pathlib.Path(__file__).parent, "test_data/Natural_language_processing.txt") explanation_file_path = path.join("/Users/runner/work/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"
# data_directory_path is defined above # data_directory_path is defined above
await cognee.add("data://" + data_directory_path, dataset_name) await cognee.add("data://" + "/Users/runner/work/cognee/cognee/tests/", dataset_name)
text_1 = """A quantum computer is a computer that takes advantage of quantum mechanical phenomena. text_1 = """A quantum computer is a computer that takes advantage of quantum mechanical phenomena.
At small scales, physical matter exhibits properties of both particles and waves, and quantum computing leverages this behavior, specifically quantum superposition and entanglement, using specialized hardware that supports the preparation and manipulation of quantum states. At small scales, physical matter exhibits properties of both particles and waves, and quantum computing leverages this behavior, specifically quantum superposition and entanglement, using specialized hardware that supports the preparation and manipulation of quantum states.