From f56049f74752232393f6947d663f8d0d383ad0b2 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Sat, 25 May 2024 18:56:03 +0200
Subject: [PATCH] implement settings

---
 cognee/base_config.py                         | 23 +++++++++++++++++++
 cognee/infrastructure/InfrastructureConfig.py |  2 +-
 .../{RelationalConfig.py => config.py}        |  0
 .../embeddings/DefaultEmbeddingEngine.py      |  2 +-
 4 files changed, 25 insertions(+), 2 deletions(-)
 create mode 100644 cognee/base_config.py
 rename cognee/infrastructure/databases/relational/{RelationalConfig.py => config.py} (100%)

diff --git a/cognee/base_config.py b/cognee/base_config.py
new file mode 100644
index 000000000..a9337001e
--- /dev/null
+++ b/cognee/base_config.py
@@ -0,0 +1,23 @@
+from functools import lru_cache
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+from cognee.root_dir import get_absolute_path
+
+
+class BaseConfig(BaseSettings):
+    system_root_directory: str = get_absolute_path(".cognee_system")
+    data_root_directory: str = get_absolute_path(".data")
+
+
+
+    model_config = SettingsConfigDict(env_file = ".env", extra = "allow")
+
+    def to_dict(self) -> dict:
+        return {
+            "system_root_directory": self.system_root_directory,
+            "data_root_directory": self.data_root_directory
+        }
+
+@lru_cache
+def get_llm_config():
+    return BaseConfig()
\ No newline at end of file
diff --git a/cognee/infrastructure/InfrastructureConfig.py b/cognee/infrastructure/InfrastructureConfig.py
index d0cad9afc..5cb046e1c 100644
--- a/cognee/infrastructure/InfrastructureConfig.py
+++ b/cognee/infrastructure/InfrastructureConfig.py
@@ -15,7 +15,7 @@ from ..shared.data_models import GraphDBType, DefaultContentPrediction, Knowledg
 logging.basicConfig(level=logging.DEBUG)
 config = Config()
 config.load()
-from cognee.infrastructure.databases.relational.RelationalConfig import get_relationaldb_config
+from cognee.infrastructure.databases.relational.config import get_relationaldb_config
 
 relational = get_relationaldb_config()
 class InfrastructureConfig():
diff --git a/cognee/infrastructure/databases/relational/RelationalConfig.py b/cognee/infrastructure/databases/relational/config.py
similarity index 100%
rename from cognee/infrastructure/databases/relational/RelationalConfig.py
rename to cognee/infrastructure/databases/relational/config.py
diff --git a/cognee/infrastructure/databases/vector/embeddings/DefaultEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/DefaultEmbeddingEngine.py
index 29f99dc74..b25536d52 100644
--- a/cognee/infrastructure/databases/vector/embeddings/DefaultEmbeddingEngine.py
+++ b/cognee/infrastructure/databases/vector/embeddings/DefaultEmbeddingEngine.py
@@ -12,7 +12,7 @@ from litellm import aembedding
 import litellm
 
 litellm.set_verbose = True
-from cognee.infrastructure.databases.vector.embeddings.EmbeddingEngine import get_embedding_config
+from cognee.infrastructure.databases.vector.embeddings.config import get_embedding_config
 config = get_embedding_config()
 
 class DefaultEmbeddingEngine(EmbeddingEngine):