9729547f5a
<!-- .github/pull_request_template.md --> ## Description <!-- Provide a clear description of the changes in this PR --> ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin. --------- Co-authored-by: Hande <159312713+hande-k@users.noreply.github.com> Co-authored-by: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
30 lines
1.2 KiB
Python
30 lines
1.2 KiB
Python
import os
|
|
from typing import Optional
|
|
from functools import lru_cache
|
|
from cognee.root_dir import get_absolute_path
|
|
from cognee.modules.observability.observers import Observer
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
|
|
class BaseConfig(BaseSettings):
|
|
data_root_directory: str = get_absolute_path(".data_storage")
|
|
monitoring_tool: object = Observer.LANGFUSE
|
|
graphistry_username: Optional[str] = os.getenv("GRAPHISTRY_USERNAME")
|
|
graphistry_password: Optional[str] = os.getenv("GRAPHISTRY_PASSWORD")
|
|
langfuse_public_key: Optional[str] = os.getenv("LANGFUSE_PUBLIC_KEY")
|
|
langfuse_secret_key: Optional[str] = os.getenv("LANGFUSE_SECRET_KEY")
|
|
langfuse_host: Optional[str] = os.getenv("LANGFUSE_HOST")
|
|
default_user_email: Optional[str] = os.getenv("DEFAULT_USER_EMAIL")
|
|
default_user_password: Optional[str] = os.getenv("DEFAULT_USER_PASSWORD")
|
|
model_config = SettingsConfigDict(env_file=".env", extra="allow")
|
|
|
|
def to_dict(self) -> dict:
|
|
return {
|
|
"data_root_directory": self.data_root_directory,
|
|
"monitoring_tool": self.monitoring_tool,
|
|
}
|
|
|
|
|
|
@lru_cache
|
|
def get_base_config():
|
|
return BaseConfig()
|