implement settings

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()

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():

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):