From eb7d02833d05c9d369bf34e7712e29d890832a4c Mon Sep 17 00:00:00 2001 From: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Date: Sat, 25 May 2024 18:28:52 +0200 Subject: [PATCH] implement settings --- cognee/config.py | 6 ------ .../databases/graph/GraphConfig.py | 0 .../databases/vector/VectorConfig.py | 0 cognee/tests/test_library.py | 16 ++++++++-------- 4 files changed, 8 insertions(+), 14 deletions(-) create mode 100644 cognee/infrastructure/databases/graph/GraphConfig.py create mode 100644 cognee/infrastructure/databases/vector/VectorConfig.py diff --git a/cognee/config.py b/cognee/config.py index e70dd32d7..5c6014b2a 100644 --- a/cognee/config.py +++ b/cognee/config.py @@ -44,12 +44,6 @@ class Config: qdrant_url: str = os.getenv("QDRANT_URL", None) qdrant_api_key: str = os.getenv("QDRANT_API_KEY", None) - db_path = str = os.getenv("COGNEE_DB_PATH", "databases") - db_name: str = os.getenv("DB_NAME", "cognee.db") - db_host: str = os.getenv("DB_HOST", "localhost") - db_port: str = os.getenv("DB_PORT", "5432") - db_user: str = os.getenv("DB_USER", "cognee") - db_password: str = os.getenv("DB_PASSWORD", "cognee") sqlalchemy_logging: bool = os.getenv("SQLALCHEMY_LOGGING", True) diff --git a/cognee/infrastructure/databases/graph/GraphConfig.py b/cognee/infrastructure/databases/graph/GraphConfig.py new file mode 100644 index 000000000..e69de29bb diff --git a/cognee/infrastructure/databases/vector/VectorConfig.py b/cognee/infrastructure/databases/vector/VectorConfig.py new file mode 100644 index 000000000..e69de29bb diff --git a/cognee/tests/test_library.py b/cognee/tests/test_library.py index 336b015bd..b1b3d3383 100755 --- a/cognee/tests/test_library.py +++ b/cognee/tests/test_library.py @@ -36,14 +36,14 @@ async def main(): """ - dataset_name = "cs_explanations" - await cognee.add( - [ - text_1, - text_2 - ], - dataset_name - ) + # dataset_name = "cs_explanations" + # await cognee.add( + # [ + # text_1, + # text_2 + # ], + # dataset_name + # ) await cognee.cognify([ "cs_explanations"])