better:优化可读性
This commit is contained in:
@@ -2,26 +2,57 @@
|
||||
CHCP 65001 > nul
|
||||
setlocal enabledelayedexpansion
|
||||
|
||||
echo 你需要选择启动方式,输入字母来选择:
|
||||
echo V = 不知道什么意思就输入 V
|
||||
echo C = 输入 C 使用 Conda 环境
|
||||
echo.
|
||||
choice /C CV /N /M "在下方输入字母并回车 (C/V)?" /T 10 /D V
|
||||
|
||||
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 错误: 未找到虚拟环境,请确保venv目录存在
|
||||
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 错误: 虚拟环境激活失败
|
||||
echo Error: Failed to activate venv virtual environment.
|
||||
pause
|
||||
exit /b 1
|
||||
)
|
||||
)
|
||||
|
||||
echo Environment activated successfully!
|
||||
|
||||
REM --- 后续脚本执行 ---
|
||||
|
||||
REM 运行预处理脚本
|
||||
python "%~dp0scripts\raw_data_preprocessor.py"
|
||||
if %ERRORLEVEL% neq 0 (
|
||||
echo 错误: raw_data_preprocessor.py 执行失败
|
||||
echo Error: raw_data_preprocessor.py execution failed.
|
||||
pause
|
||||
exit /b 1
|
||||
)
|
||||
@@ -29,7 +60,7 @@ if %ERRORLEVEL% neq 0 (
|
||||
REM 运行信息提取脚本
|
||||
python "%~dp0scripts\info_extraction.py"
|
||||
if %ERRORLEVEL% neq 0 (
|
||||
echo 错误: info_extraction.py 执行失败
|
||||
echo Error: info_extraction.py execution failed.
|
||||
pause
|
||||
exit /b 1
|
||||
)
|
||||
@@ -37,10 +68,10 @@ if %ERRORLEVEL% neq 0 (
|
||||
REM 运行OpenIE导入脚本
|
||||
python "%~dp0scripts\import_openie.py"
|
||||
if %ERRORLEVEL% neq 0 (
|
||||
echo 错误: import_openie.py 执行失败
|
||||
echo Error: import_openie.py execution failed.
|
||||
pause
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
echo 所有处理步骤完成!
|
||||
echo All processing steps completed!
|
||||
pause
|
||||
@@ -87,8 +87,8 @@ def main():
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
# 新增用户确认提示
|
||||
print("=== 重要操作确认 ===")
|
||||
print("实体提取操作将会花费较多资金和时间,建议在空闲时段执行。")
|
||||
print("=== 重要操作确认,请认真阅读以下内容哦 ===")
|
||||
print("实体提取操作将会花费较多api余额和时间,建议在空闲时段执行。")
|
||||
print("举例:600万字全剧情,提取选用deepseek v3 0324,消耗约40元,约3小时。")
|
||||
print("建议使用硅基流动的非Pro模型")
|
||||
print("或者使用可以用赠金抵扣的Pro模型")
|
||||
|
||||
@@ -44,7 +44,7 @@ def process_text_file(file_path):
|
||||
|
||||
def main():
|
||||
# 新增用户确认提示
|
||||
print("=== 重要操作确认 ===")
|
||||
print("=== 重要操作确认,请认真阅读以下内容哦 ===")
|
||||
print("如果你并非第一次导入知识")
|
||||
print("请先删除data/import.json文件,备份data/openie.json文件")
|
||||
print("在进行知识库导入之前")
|
||||
|
||||
Reference in New Issue
Block a user