better:优化可读性
This commit is contained in:
@@ -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
|
||||||
@@ -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模型")
|
||||||
|
|||||||
@@ -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("在进行知识库导入之前")
|
||||||
|
|||||||
Reference in New Issue
Block a user