From 4375742a8209a95138e594dd799ed95761257850 Mon Sep 17 00:00:00 2001 From: Leon Luithlen Date: Fri, 8 Nov 2024 13:54:33 +0100 Subject: [PATCH] Call os.getenv in BaseConfig --- cognee/base_config.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/cognee/base_config.py b/cognee/base_config.py index d2245ef6e..0e70b7652 100644 --- a/cognee/base_config.py +++ b/cognee/base_config.py @@ -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")