fix:修改bat错误描述

This commit is contained in:
SengokuCola
2025-05-02 19:46:04 +08:00
parent deae6f95cb
commit 021ac90ead
2 changed files with 24 additions and 14 deletions

View File

@@ -6,29 +6,39 @@ echo 你需要选择启动方式,输入字母来选择:
echo V = 不知道什么意思就输入 V echo V = 不知道什么意思就输入 V
echo C = 输入 C 使用 Conda 环境 echo C = 输入 C 使用 Conda 环境
echo. echo.
choice /C CV /N /M "在下方输入字母并回车 (C/V)?" /T 10 /D V choice /C CV /N /M "不知道什么意思就输入 V (C/V)?" /T 10 /D V
set "ENV_TYPE=" set "ENV_TYPE="
if %ERRORLEVEL% == 1 set "ENV_TYPE=CONDA" if %ERRORLEVEL% == 1 set "ENV_TYPE=CONDA"
if %ERRORLEVEL% == 2 set "ENV_TYPE=VENV" if %ERRORLEVEL% == 2 set "ENV_TYPE=VENV"
if "%ENV_TYPE%" == "CONDA" ( if "%ENV_TYPE%" == "CONDA" goto activate_conda
if "%ENV_TYPE%" == "VENV" goto activate_venv
REM 如果 choice 超时或返回意外值,默认使用 venv
echo WARN: Invalid selection or timeout from choice. Defaulting to VENV.
set "ENV_TYPE=VENV"
goto activate_venv
:activate_conda
set /p CONDA_ENV_NAME="请输入要使用的 Conda 环境名称: " set /p CONDA_ENV_NAME="请输入要使用的 Conda 环境名称: "
if not defined CONDA_ENV_NAME ( if not defined CONDA_ENV_NAME (
echo 错误: 未输入 Conda 环境名称. echo 错误: 未输入 Conda 环境名称.
pause pause
exit /b 1 exit /b 1
) )
echo 选择: Conda '%CONDA_ENV_NAME%' echo 选择: Conda '!CONDA_ENV_NAME!'
REM 激活Conda环境 REM 激活Conda环境
call conda activate %CONDA_ENV_NAME% call conda activate !CONDA_ENV_NAME!
if %ERRORLEVEL% neq 0 ( if !ERRORLEVEL! neq 0 (
echo 错误: Conda环境 '%CONDA_ENV_NAME%' 激活失败. 请确保Conda已安装并正确配置, 且 '%CONDA_ENV_NAME%' 环境存在. echo 错误: Conda环境 '!CONDA_ENV_NAME!' 激活失败. 请确保Conda已安装并正确配置, 且 '!CONDA_ENV_NAME!' 环境存在.
pause pause
exit /b 1 exit /b 1
) )
) else ( goto env_activated
echo Selected: venv (default)
:activate_venv
echo Selected: venv (default or selected)
REM 查找venv虚拟环境 REM 查找venv虚拟环境
set "venv_path=%~dp0venv\Scripts\activate.bat" set "venv_path=%~dp0venv\Scripts\activate.bat"
if not exist "%venv_path%" ( if not exist "%venv_path%" (
@@ -43,8 +53,9 @@ if "%ENV_TYPE%" == "CONDA" (
pause pause
exit /b 1 exit /b 1
) )
) goto env_activated
:env_activated
echo Environment activated successfully! echo Environment activated successfully!
REM --- 后续脚本执行 --- REM --- 后续脚本执行 ---

View File

@@ -48,11 +48,10 @@ def process_text_file(file_path):
def main(): def main():
# 新增用户确认提示 # 新增用户确认提示
print("=== 重要操作确认,请认真阅读以下内容哦 ===") print("=== 数据预处理脚本 ===")
print("如果你并非第一次导入知识") print(f"本脚本将处理 '{RAW_DATA_PATH}' 目录下的所有 .txt 文件。")
print("请先删除data/import.json文件备份data/openie.json文件") print(f"处理后的段落数据将合并,并以 MM-DD-HH-SS-imported-data.json 的格式保存在 '{IMPORTED_DATA_PATH}' 目录中。")
print("在进行知识库导入之前") print("请确保原始数据已放置在正确的目录中。")
print("请修改config/lpmm_config.toml中的配置项")
confirm = input("确认继续执行?(y/n): ").strip().lower() confirm = input("确认继续执行?(y/n): ").strip().lower()
if confirm != "y": if confirm != "y":
logger.error("操作已取消") logger.error("操作已取消")