wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-02 22:02:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: robust environment variable resolution in config

Browse Source
This commit is contained in:
amszuidas
2026-01-23 17:01:38 +08:00
parent 3f4bcd9433
commit 82a6ae81bd
1 changed files with 10 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
backend/src/config/app_config.py
View File

@@ -1,6 +1,6 @@
import os
from pathlib import Path
from typing import Self
from typing import Any, Self
import yaml
from dotenv import load_dotenv
@@ -90,7 +90,7 @@ class AppConfig(BaseModel):
return result
@classmethod
def resolve_env_variables(cls, config: dict) -> dict:
def resolve_env_variables(cls, config: Any) -> Any:
"""Recursively resolve environment variables in the config.
Environment variables are resolved using the `os.getenv` function. Example: $OPENAI_API_KEY
@@ -101,18 +101,14 @@ class AppConfig(BaseModel):
Returns:
The config with environment variables resolved.
"""
for key, value in config.items():
if isinstance(value, str):
if value.startswith("$"):
env_value = os.getenv(value[1:], None)
if env_value is not None:
config[key] = env_value
else:
config[key] = value
elif isinstance(value, dict):
config[key] = cls.resolve_env_variables(value)
elif isinstance(value, list):
config[key] = [cls.resolve_env_variables(item) for item in value]
if isinstance(config, str):
if config.startswith("$"):
return os.getenv(config[1:], config)
return config
elif isinstance(config, dict):
return {k: cls.resolve_env_variables(v) for k, v in config.items()}
elif isinstance(config, list):
return [cls.resolve_env_variables(item) for item in config]
return config
def get_model_config(self, name: str) -> ModelConfig | None: