File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree 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