107 lines
3.1 KiB
Batchfile
107 lines
3.1 KiB
Batchfile
|
@echo off
|
|||
|
|
chcp 65001 >nul
|
|||
|
|
echo ========================================
|
|||
|
|
echo RAG知识库系统 - 自动化测试脚本
|
|||
|
|
echo ========================================
|
|||
|
|
echo.
|
|||
|
|
|
|||
|
|
set BASE_URL=http://localhost:8080
|
|||
|
|
|
|||
|
|
REM 检查服务是否运行
|
|||
|
|
echo [1/8] 检查系统健康状态...
|
|||
|
|
curl -s %BASE_URL%/psychology/rag-test/health >nul 2>&1
|
|||
|
|
if errorlevel 1 (
|
|||
|
|
echo [失败] 系统未启动，请先启动Spring Boot应用
|
|||
|
|
pause
|
|||
|
|
exit /b 1
|
|||
|
|
)
|
|||
|
|
echo [成功] 系统运行正常
|
|||
|
|
echo.
|
|||
|
|
|
|||
|
|
REM 测试向量化
|
|||
|
|
echo [2/8] 测试向量化功能...
|
|||
|
|
curl -s -X POST "%BASE_URL%/psychology/rag-test/test-embedding" ^
|
|||
|
|
-H "Content-Type: application/x-www-form-urlencoded" ^
|
|||
|
|
-d "text=这是一个测试文本" | findstr "embedding_dimension" >nul
|
|||
|
|
if errorlevel 1 (
|
|||
|
|
echo [失败] 向量化测试失败
|
|||
|
|
) else (
|
|||
|
|
echo [成功] 向量化功能正常
|
|||
|
|
)
|
|||
|
|
echo.
|
|||
|
|
|
|||
|
|
REM 测试文本生成
|
|||
|
|
echo [3/8] 测试文本生成功能...
|
|||
|
|
curl -s -X POST "%BASE_URL%/psychology/rag-test/test-generation" ^
|
|||
|
|
-H "Content-Type: application/x-www-form-urlencoded" ^
|
|||
|
|
-d "prompt=什么是心理学？" | findstr "response" >nul
|
|||
|
|
if errorlevel 1 (
|
|||
|
|
echo [失败] 文本生成测试失败
|
|||
|
|
) else (
|
|||
|
|
echo [成功] 文本生成功能正常
|
|||
|
|
)
|
|||
|
|
echo.
|
|||
|
|
|
|||
|
|
REM 测试ChromaDB
|
|||
|
|
echo [4/8] 测试ChromaDB存储和查询...
|
|||
|
|
curl -s -X POST "%BASE_URL%/psychology/rag-test/test-chromadb" | findstr "query_results" >nul
|
|||
|
|
if errorlevel 1 (
|
|||
|
|
echo [失败] ChromaDB测试失败
|
|||
|
|
) else (
|
|||
|
|
echo [成功] ChromaDB功能正常
|
|||
|
|
)
|
|||
|
|
echo.
|
|||
|
|
|
|||
|
|
REM 创建测试文档
|
|||
|
|
echo [5/8] 创建测试文档...
|
|||
|
|
echo 心理学基础知识 > test_doc.txt
|
|||
|
|
echo. >> test_doc.txt
|
|||
|
|
echo 人格心理学是研究个体在不同情境下表现出的稳定行为模式的学科。 >> test_doc.txt
|
|||
|
|
echo 主要理论包括特质理论、精神分析理论和人本主义理论。 >> test_doc.txt
|
|||
|
|
echo. >> test_doc.txt
|
|||
|
|
echo 认知心理学关注人类的思维过程，包括注意、记忆、语言和问题解决。 >> test_doc.txt
|
|||
|
|
echo [成功] 测试文档已创建: test_doc.txt
|
|||
|
|
echo.
|
|||
|
|
|
|||
|
|
REM 测试文档上传
|
|||
|
|
echo [6/8] 测试文档上传功能...
|
|||
|
|
curl -s -X POST "%BASE_URL%/psychology/knowledge/upload" ^
|
|||
|
|
-F "file=@test_doc.txt" ^
|
|||
|
|
-F "category=综合心理学" | findstr "doc_id" >nul
|
|||
|
|
if errorlevel 1 (
|
|||
|
|
echo [失败] 文档上传测试失败
|
|||
|
|
) else (
|
|||
|
|
echo [成功] 文档上传功能正常
|
|||
|
|
)
|
|||
|
|
echo.
|
|||
|
|
|
|||
|
|
REM 等待处理完成
|
|||
|
|
echo [7/8] 等待文档处理完成...
|
|||
|
|
timeout /t 3 /nobreak >nul
|
|||
|
|
echo [成功] 文档处理完成
|
|||
|
|
echo.
|
|||
|
|
|
|||
|
|
REM 测试智能问答
|
|||
|
|
echo [8/8] 测试智能问答功能...
|
|||
|
|
curl -s -X POST "%BASE_URL%/psychology/ai/chat" ^
|
|||
|
|
-H "Content-Type: application/json" ^
|
|||
|
|
-d "{\"question\":\"什么是人格心理学？\",\"context\":\"\"}" | findstr "content" >nul
|
|||
|
|
if errorlevel 1 (
|
|||
|
|
echo [失败] 智能问答测试失败
|
|||
|
|
) else (
|
|||
|
|
echo [成功] 智能问答功能正常
|
|||
|
|
)
|
|||
|
|
echo.
|
|||
|
|
|
|||
|
|
echo ========================================
|
|||
|
|
echo 测试完成！
|
|||
|
|
echo ========================================
|
|||
|
|
echo.
|
|||
|
|
echo 详细测试结果请查看上方输出
|
|||
|
|
echo 如需查看完整响应，请手动运行curl命令
|
|||
|
|
echo.
|
|||
|
|
echo 清理测试文件...
|
|||
|
|
del test_doc.txt >nul 2>&1
|
|||
|
|
echo.
|
|||
|
|
pause
|