diff --git a/cognee/base_config.py b/cognee/base_config.py index a9337001e..78fd7f3e7 100644 --- a/cognee/base_config.py +++ b/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 diff --git a/cognee/config.py b/cognee/config.py index b88431f86..76f58af2c 100644 --- a/cognee/config.py +++ b/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") diff --git a/cognee/infrastructure/databases/vector/embeddings/config.py b/cognee/infrastructure/databases/vector/embeddings/config.py index 4ed8fbc1e..8bdadd587 100644 --- a/cognee/infrastructure/databases/vector/embeddings/config.py +++ b/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")