Skip to content

Commit 8295a5d

Browse files
authored
feat: 添加部分环境变量以支持方便快捷获取data目录以及插件数据目录。 (#1485)
好的,这是翻译成中文的 pull request 总结: ## Sourcery 总结 引入环境变量以实现灵活的数据和插件目录解析,并更新配置加载器以使用新的 DATA_PATH。 新特性: - 添加 DATA_PATH 环境变量以配置根数据目录,默认为 $PWD/data - 添加基于 DATA_PATH 的 PLUGIN_PATH,以便按照约定定位插件文件 增强功能: - 重构配置加载器以使用新的 DATA_PATH 变量解析 config.yaml 路径 <details> <summary>Original summary in English</summary> 好的,这是翻译成中文的 pull request 总结: ## Sourcery 总结 引入 DATA_PATH 和 PLUGIN_PATH 环境变量,以便灵活地定位数据和插件目录,并更新配置加载器以使用 DATA_PATH 加载 config.yaml。 新特性: - 添加 DATA_PATH 环境变量,用于指定根数据目录,默认为 $PWD/data。 - 添加基于 DATA_PATH 的 PLUGIN_PATH 变量,用于定位插件文件。 增强功能: - 重构配置加载器,以使用 DATA_PATH 解析 config.yaml 路径。 <details> <summary>Original summary in English</summary> ## Summary by Sourcery Introduce DATA_PATH and PLUGIN_PATH environment variables to flexibly locate the data and plugin directories, and update the config loader to use DATA_PATH for loading config.yaml. New Features: - Add DATA_PATH environment variable for root data directory, defaulting to $PWD/data. - Add PLUGIN_PATH variable based on DATA_PATH to locate plugin files. Enhancements: - Refactor config loader to resolve config.yaml path using DATA_PATH. </details> </details>
2 parents 9dfebd8 + 8ebe4ce commit 8295a5d

File tree

2 files changed

+16
-1
lines changed

2 files changed

+16
-1
lines changed

kirara_ai/config/__init__.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
import os
2+
3+
# 读取DATA_PATH环境变量,若未能找到则以当前工作目录为根文件夹存储在$PWD/data目录下。
4+
DATA_PATH = os.path.abspath(
5+
os.environ.get("DATA_PATH", os.path.join(os.getcwd(), "data"))
6+
)
7+
# 按照规范插件应该在PLUGIN_PATH目录下存储对应的文件。
8+
PLUGIN_PATH = os.path.join(DATA_PATH, "plugins")
9+
10+
if os.path.exists(DATA_PATH) is False:
11+
os.makedirs(DATA_PATH)
12+
13+
if os.path.exists(PLUGIN_PATH) is False:
14+
os.makedirs(PLUGIN_PATH)

kirara_ai/config/config_loader.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,9 @@
88
from ruamel.yaml import YAML
99

1010
from ..logger import get_logger
11+
from . import DATA_PATH
1112

12-
CONFIG_FILE = "data/config.yaml"
13+
CONFIG_FILE = os.path.join(DATA_PATH, "config.yaml")
1314

1415
T = TypeVar("T", bound=BaseModel)
1516

0 commit comments

Comments
 (0)