fix(envs): envs of os and .env (#325)

Author: floritange

veadk/utils/misc.py

@@ -137,19 +137,20 @@ def set_envs(config_yaml_path: str, env_from_dotenv: dict = None) -> tuple[dict,
         config_dict = safe_load(yaml_file)
 
     flatten_config_dict = flatten_dict(config_dict)
-
-    veadk_environments = dict(env_from_dotenv)
-    for k, v in flatten_config_dict.items():
-        k = k.upper()
-
-        if k in env_from_dotenv:
+    config_upper_map = {k.upper(): v for k, v in flatten_config_dict.items()}
+    all_keys = {k.upper() for k in flatten_config_dict.keys()} | set(
+        env_from_dotenv.keys() if env_from_dotenv else []
+    )
+    veadk_environments = {}
+    for k in all_keys:
+        if k in os.environ:
             logger.info(
-                f"Environment variable {k} has been set in .env, value in `config.yaml` will be ignored."
+                f"Environment variable {k} has been set, value in `config.yaml` will be ignored."
             )
+            veadk_environments[k] = os.environ[k]
             continue
-        veadk_environments[k] = str(v)
-        os.environ[k] = str(v)
-
+        veadk_environments[k] = str(config_upper_map.get(k))
+        os.environ[k] = str(config_upper_map.get(k))
     return config_dict, veadk_environments