feat(config): support dotenv file to init env variables (#261)

* feat(config): support dotenv file to init env variables

* update version

Author: yaozheng-fang

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "veadk-python"
-version = "0.2.15"
+version = "0.2.16"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"

veadk/config.py

@@ -15,7 +15,7 @@
 import os
 from typing import Any
 
-from dotenv import find_dotenv
+from dotenv import find_dotenv, load_dotenv
 from pydantic import BaseModel, Field
 
 from veadk.configs.database_configs import (
@@ -33,8 +33,16 @@
     PrometheusConfig,
     TLSConfig,
 )
+from veadk.utils.logger import get_logger
 from veadk.utils.misc import set_envs
 
+logger = get_logger(__name__)
+
+if load_dotenv(find_dotenv(usecwd=True)):
+    logger.info(f"Find `.env` file in {find_dotenv(usecwd=True)}, load envs.")
+else:
+    logger.info("No env file found.")
+
 
 class VeADKConfig(BaseModel):
     model: ModelConfig = Field(default_factory=ModelConfig)
@@ -89,7 +97,10 @@ def getenv(
 veadk_environments = {}
 
 if config_yaml_path:
+    logger.info(f"Find `config.yaml` file in {config_yaml_path}")
     config_dict, _veadk_environments = set_envs(config_yaml_path=config_yaml_path)
     veadk_environments.update(_veadk_environments)
+else:
+    logger.warning("No `config.yaml` file found.")
 
 settings = VeADKConfig()

veadk/utils/misc.py

@@ -128,6 +128,10 @@ def getenv(
 
 
 def set_envs(config_yaml_path: str) -> tuple[dict, dict]:
+    from veadk.utils.logger import get_logger
+
+    logger = get_logger(__name__)
+
     with open(config_yaml_path, "r", encoding="utf-8") as yaml_file:
         config_dict = safe_load(yaml_file)
 
@@ -138,6 +142,9 @@ def set_envs(config_yaml_path: str) -> tuple[dict, dict]:
         k = k.upper()
 
         if k in os.environ:
+            logger.info(
+                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)

veadk/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-VERSION = "0.2.15"
+VERSION = "0.2.16"