better:优化可读性

This commit is contained in:
SengokuCola
2025-05-02 19:17:59 +08:00
parent edda834538
commit a859f9238f
3 changed files with 51 additions and 20 deletions

View File

@@ -2,26 +2,57 @@
CHCP 65001 > nul CHCP 65001 > nul
setlocal enabledelayedexpansion setlocal enabledelayedexpansion
REM 查找venv虚拟环境 echo 你需要选择启动方式,输入字母来选择:
set "venv_path=%~dp0venv\Scripts\activate.bat" echo V = 不知道什么意思就输入 V
if not exist "%venv_path%" ( echo C = 输入 C 使用 Conda 环境
echo 错误: 未找到虚拟环境请确保venv目录存在 echo.
pause choice /C CV /N /M "在下方输入字母并回车 (C/V)?" /T 10 /D V
exit /b 1
set "ENV_TYPE="
if %ERRORLEVEL% == 1 set "ENV_TYPE=CONDA"
if %ERRORLEVEL% == 2 set "ENV_TYPE=VENV"
if "%ENV_TYPE%" == "CONDA" (
set /p CONDA_ENV_NAME="请输入要使用的 Conda 环境名称: "
if not defined CONDA_ENV_NAME (
echo 错误: 未输入 Conda 环境名称.
pause
exit /b 1
)
echo 选择: Conda '%CONDA_ENV_NAME%'
REM 激活Conda环境
call conda activate %CONDA_ENV_NAME%
if %ERRORLEVEL% neq 0 (
echo 错误: Conda环境 '%CONDA_ENV_NAME%' 激活失败. 请确保Conda已安装并正确配置, 且 '%CONDA_ENV_NAME%' 环境存在.
pause
exit /b 1
)
) else (
echo Selected: venv (default)
REM 查找venv虚拟环境
set "venv_path=%~dp0venv\Scripts\activate.bat"
if not exist "%venv_path%" (
echo Error: venv not found. Ensure the venv directory exists alongside the script.
pause
exit /b 1
)
REM 激活虚拟环境
call "%venv_path%"
if %ERRORLEVEL% neq 0 (
echo Error: Failed to activate venv virtual environment.
pause
exit /b 1
)
) )
REM 激活虚拟环境 echo Environment activated successfully!
call "%venv_path%"
if %ERRORLEVEL% neq 0 ( REM --- 后续脚本执行 ---
echo 错误: 虚拟环境激活失败
pause
exit /b 1
)
REM 运行预处理脚本 REM 运行预处理脚本
python "%~dp0scripts\raw_data_preprocessor.py" python "%~dp0scripts\raw_data_preprocessor.py"
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo 错误: raw_data_preprocessor.py 执行失败 echo Error: raw_data_preprocessor.py execution failed.
pause pause
exit /b 1 exit /b 1
) )
@@ -29,7 +60,7 @@ if %ERRORLEVEL% neq 0 (
REM 运行信息提取脚本 REM 运行信息提取脚本
python "%~dp0scripts\info_extraction.py" python "%~dp0scripts\info_extraction.py"
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo 错误: info_extraction.py 执行失败 echo Error: info_extraction.py execution failed.
pause pause
exit /b 1 exit /b 1
) )
@@ -37,10 +68,10 @@ if %ERRORLEVEL% neq 0 (
REM 运行OpenIE导入脚本 REM 运行OpenIE导入脚本
python "%~dp0scripts\import_openie.py" python "%~dp0scripts\import_openie.py"
if %ERRORLEVEL% neq 0 ( if %ERRORLEVEL% neq 0 (
echo 错误: import_openie.py 执行失败 echo Error: import_openie.py execution failed.
pause pause
exit /b 1 exit /b 1
) )
echo 所有处理步骤完成! echo All processing steps completed!
pause pause

View File

@@ -87,8 +87,8 @@ def main():
signal.signal(signal.SIGINT, signal_handler) signal.signal(signal.SIGINT, signal_handler)
# 新增用户确认提示 # 新增用户确认提示
print("=== 重要操作确认 ===") print("=== 重要操作确认,请认真阅读以下内容哦 ===")
print("实体提取操作将会花费较多资金和时间,建议在空闲时段执行。") print("实体提取操作将会花费较多api余额和时间,建议在空闲时段执行。")
print("举例600万字全剧情提取选用deepseek v3 0324消耗约40元约3小时。") print("举例600万字全剧情提取选用deepseek v3 0324消耗约40元约3小时。")
print("建议使用硅基流动的非Pro模型") print("建议使用硅基流动的非Pro模型")
print("或者使用可以用赠金抵扣的Pro模型") print("或者使用可以用赠金抵扣的Pro模型")

View File

@@ -44,7 +44,7 @@ def process_text_file(file_path):
def main(): def main():
# 新增用户确认提示 # 新增用户确认提示
print("=== 重要操作确认 ===") print("=== 重要操作确认,请认真阅读以下内容哦 ===")
print("如果你并非第一次导入知识") print("如果你并非第一次导入知识")
print("请先删除data/import.json文件备份data/openie.json文件") print("请先删除data/import.json文件备份data/openie.json文件")
print("在进行知识库导入之前") print("在进行知识库导入之前")