Initial: integrated 2025 LawGraph (graphrag_pipeline) + 2026 kg_project
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,75 @@
|
||||
# 测试文件说明
|
||||
|
||||
本目录包含所有测试脚本。
|
||||
|
||||
## 测试文件列表
|
||||
|
||||
### 1. `test_basic.py`
|
||||
基础功能测试,包括:
|
||||
- 模块导入测试
|
||||
- 本体模型测试
|
||||
- Prompt模板测试
|
||||
|
||||
运行方式:
|
||||
```bash
|
||||
uv run python tests/test_basic.py
|
||||
```
|
||||
|
||||
### 2. `test_siliconflow.py`
|
||||
测试硅基流动API配置,验证API密钥和连接是否正常。
|
||||
|
||||
运行方式:
|
||||
```bash
|
||||
uv run python tests/test_siliconflow.py
|
||||
```
|
||||
|
||||
### 3. `test_documents.py`
|
||||
测试文档解析功能,包括:
|
||||
- 文档解析(.docx文件)
|
||||
- TextUnit切分
|
||||
- 不调用LLM,仅测试解析逻辑
|
||||
|
||||
运行方式:
|
||||
```bash
|
||||
uv run python tests/test_documents.py
|
||||
```
|
||||
|
||||
### 4. `test_model.py`
|
||||
测试硅基流动平台上的多个模型,找出可用模型。
|
||||
|
||||
运行方式:
|
||||
```bash
|
||||
uv run python tests/test_model.py
|
||||
```
|
||||
|
||||
### 5. `test_my_models.py`
|
||||
测试项目配置使用的两个指定模型:
|
||||
- `deepseek-ai/DeepSeek-R1-Distill-Qwen-7B`
|
||||
- `Qwen/Qwen2.5-7B-Instruct`
|
||||
|
||||
运行方式:
|
||||
```bash
|
||||
uv run python tests/test_my_models.py
|
||||
```
|
||||
|
||||
### 6. `test_quick.py`
|
||||
快速测试完整流程,处理少量TextUnit(1个文档的前2个),验证:
|
||||
- 文档解析
|
||||
- 实体识别(NER)
|
||||
- 关系抽取(RE)
|
||||
- 知识图谱构建
|
||||
- 社区检测
|
||||
|
||||
运行方式:
|
||||
```bash
|
||||
uv run python tests/test_quick.py
|
||||
```
|
||||
|
||||
## 注意事项
|
||||
|
||||
- 测试脚本需要使用配置的API密钥(`.env`文件)
|
||||
- 部分测试会调用LLM API,可能需要一些时间
|
||||
- 确保已安装所有依赖包:`uv sync`
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user