mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-27 09:13:25 +08:00
84bafe9723
工作空间的配置预设，提供ConfigBasic建造方法产生实例。
该类的实例对象用于存储工作空间的配置信息，如工作空间的路径等
工作空间的配置信息存储在用户的家目录下的.config/chatchat/workspace/workspace_config.json文件中。
注意：不存在则读取默认
提供了操作入口
指令` chatchat-config` 工作空间配置
options:
```
-h, --help show this help message and exit
-v {true,false}, --verbose {true,false}
是否开启详细日志
-d DATA, --data DATA 数据存放路径
-f FORMAT, --format FORMAT
日志格式
--clear 清除配置
```
57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
from pathlib import Path
|
|
|
|
from chatchat.configs import ConfigBasicFactory, ConfigBasic, ConfigWorkSpace
|
|
import os
|
|
|
|
|
|
def test_config_factory_def():
|
|
test_config_factory = ConfigBasicFactory()
|
|
config: ConfigBasic = test_config_factory.get_config()
|
|
assert config is not None
|
|
assert config.log_verbose is False
|
|
assert config.CHATCHAT_ROOT is not None
|
|
assert config.DATA_PATH is not None
|
|
assert config.IMG_DIR is not None
|
|
assert config.NLTK_DATA_PATH is not None
|
|
assert config.LOG_FORMAT is not None
|
|
assert config.LOG_PATH is not None
|
|
assert config.MEDIA_PATH is not None
|
|
|
|
assert os.path.exists(os.path.join(config.MEDIA_PATH, "image"))
|
|
assert os.path.exists(os.path.join(config.MEDIA_PATH, "audio"))
|
|
assert os.path.exists(os.path.join(config.MEDIA_PATH, "video"))
|
|
|
|
|
|
def test_workspace():
|
|
config_workspace = ConfigWorkSpace()
|
|
assert config_workspace.get_config() is not None
|
|
base_root = os.path.join(Path(__file__).absolute().parent, "chatchat")
|
|
config_workspace.set_data_path(os.path.join(base_root, "data"))
|
|
config_workspace.set_log_verbose(True)
|
|
config_workspace.set_log_format(" %(message)s")
|
|
|
|
config: ConfigBasic = config_workspace.get_config()
|
|
assert config.log_verbose is True
|
|
assert config.DATA_PATH == os.path.join(base_root, "data")
|
|
assert config.IMG_DIR is not None
|
|
assert config.NLTK_DATA_PATH == os.path.join(base_root, "data", "nltk_data")
|
|
assert config.LOG_FORMAT == " %(message)s"
|
|
assert config.LOG_PATH == os.path.join(base_root, "data", "logs")
|
|
assert config.MEDIA_PATH == os.path.join(base_root, "data", "media")
|
|
|
|
assert os.path.exists(os.path.join(config.MEDIA_PATH, "image"))
|
|
assert os.path.exists(os.path.join(config.MEDIA_PATH, "audio"))
|
|
assert os.path.exists(os.path.join(config.MEDIA_PATH, "video"))
|
|
config_workspace.clear()
|
|
|
|
|
|
def test_workspace_default():
|
|
from chatchat.configs import (log_verbose, DATA_PATH, IMG_DIR, NLTK_DATA_PATH, LOG_FORMAT, LOG_PATH, MEDIA_PATH)
|
|
assert log_verbose is False
|
|
assert DATA_PATH is not None
|
|
assert IMG_DIR is not None
|
|
assert NLTK_DATA_PATH is not None
|
|
assert LOG_FORMAT is not None
|
|
assert LOG_PATH is not None
|
|
assert MEDIA_PATH is not None
|