Merge pull request #183 from topoteretes/COG-530-call-osgetenv-in-base-config

Call os.getenv in BaseConfig
This commit is contained in:
Vasilije 2024-11-08 15:21:56 +01:00 committed by GitHub
commit 82b2235f5e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -1,3 +1,4 @@
import os
from typing import Optional
from functools import lru_cache
from pydantic_settings import BaseSettings, SettingsConfigDict
@ -7,8 +8,8 @@ from cognee.shared.data_models import MonitoringTool
class BaseConfig(BaseSettings):
data_root_directory: str = get_absolute_path(".data_storage")
monitoring_tool: object = MonitoringTool.LANGFUSE
graphistry_username: Optional[str] = None
graphistry_password: Optional[str] = None
graphistry_username: Optional[str] = os.getenv("GRAPHISTRY_USERNAME")
graphistry_password: Optional[str] = os.getenv("GRAPHISTRY_PASSWORD")
model_config = SettingsConfigDict(env_file = ".env", extra = "allow")