implement settings

cognee/base_config.py

@@ -3,10 +3,16 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
 
 from cognee.root_dir import get_absolute_path
 
+from cognee.shared.data_models import MonitoringTool
+
+# Monitoring tool
+
+
 
 class BaseConfig(BaseSettings):
     system_root_directory: str = get_absolute_path(".cognee_system")
     data_root_directory: str = get_absolute_path(".data")
+    monitoring_tool: object = MonitoringTool.LANGFUSE
 
 
 
@@ -15,7 +21,8 @@ class BaseConfig(BaseSettings):
     def to_dict(self) -> dict:
         return {
             "system_root_directory": self.system_root_directory,
-            "data_root_directory": self.data_root_directory
+            "data_root_directory": self.data_root_directory,
+            "monitoring_tool": self.monitoring_tool,
         }
 
 @lru_cache

cognee/config.py

@@ -64,10 +64,10 @@ class Config:
     # model_endpoint: str = "openai"
     # llm_api_key: Optional[str] = os.getenv("OPENAI_API_KEY")
     openai_temperature: float = float(os.getenv("OPENAI_TEMPERATURE", 0.0))
-    openai_embedding_model = "text-embedding-3-large"
-    openai_embedding_dimensions = 3072
-    litellm_embedding_model = "text-embedding-3-large"
-    litellm_embedding_dimensions = 3072
+    # openai_embedding_model = "text-embedding-3-large"
+    # openai_embedding_dimensions = 3072
+    # litellm_embedding_model = "text-embedding-3-large"
+    # litellm_embedding_dimensions = 3072
 
     graphistry_username = os.getenv("GRAPHISTRY_USERNAME")
     graphistry_password = os.getenv("GRAPHISTRY_PASSWORD")

cognee/infrastructure/databases/vector/embeddings/config.py

@@ -5,7 +5,7 @@ class EmbeddingConfig(BaseSettings):
     openai_embedding_model: str = "text-embedding-3-large"
     openai_embedding_dimensions: int = 3072
     litellm_embedding_model: str = "text-embedding-3-large"
-    litellm_embedding_dimensions: str = 3072
+    litellm_embedding_dimensions: int = 3072
 
     model_config = SettingsConfigDict(env_file = ".env", extra = "allow")