File tree Expand file tree Collapse file tree 2 files changed +19
-1
lines changed
Expand file tree Collapse file tree 2 files changed +19
-1
lines changed Original file line number Diff line number Diff line change 1515import os
1616from typing import Any
1717
18- from dotenv import find_dotenv
18+ from dotenv import find_dotenv , load_dotenv
1919from pydantic import BaseModel , Field
2020
2121from veadk .configs .database_configs import (
3333 PrometheusConfig ,
3434 TLSConfig ,
3535)
36+ from veadk .utils .logger import get_logger
3637from veadk .utils .misc import set_envs
3738
39+ logger = get_logger (__name__ )
40+
41+ if load_dotenv (find_dotenv (usecwd = True )):
42+ logger .info (f"Find `.env` file in { find_dotenv (usecwd = True )} , load envs." )
43+ else :
44+ logger .info ("No env file found." )
45+
3846
3947class VeADKConfig (BaseModel ):
4048 model : ModelConfig = Field (default_factory = ModelConfig )
@@ -89,7 +97,10 @@ def getenv(
8997veadk_environments = {}
9098
9199if config_yaml_path :
100+ logger .info (f"Find `config.yaml` file in { config_yaml_path } " )
92101 config_dict , _veadk_environments = set_envs (config_yaml_path = config_yaml_path )
93102 veadk_environments .update (_veadk_environments )
103+ else :
104+ logger .warning ("No `config.yaml` file found." )
94105
95106settings = VeADKConfig ()
Original file line number Diff line number Diff line change @@ -128,6 +128,10 @@ def getenv(
128128
129129
130130def set_envs (config_yaml_path : str ) -> tuple [dict , dict ]:
131+ from veadk .utils .logger import get_logger
132+
133+ logger = get_logger (__name__ )
134+
131135 with open (config_yaml_path , "r" , encoding = "utf-8" ) as yaml_file :
132136 config_dict = safe_load (yaml_file )
133137
@@ -138,6 +142,9 @@ def set_envs(config_yaml_path: str) -> tuple[dict, dict]:
138142 k = k .upper ()
139143
140144 if k in os .environ :
145+ logger .info (
146+ f"Environment variable { k } has been set, value in `config.yaml` will be ignored."
147+ )
141148 veadk_environments [k ] = os .environ [k ]
142149 continue
143150 veadk_environments [k ] = str (v )
You can’t perform that action at this time.
0 commit comments