# 模型配置说明

## 当前使用的模型

项目已配置为使用以下两个硅基流动模型：

### 1. NER（命名实体识别）
- **模型**: `deepseek-ai/DeepSeek-R1-Distill-Qwen-7B`
- **用途**: 从法规文本中提取实体（目标要素、原则要素、空间要素等）
- **特点**: DeepSeek R1系列，具有推理能力

### 2. RE（关系抽取）
- **模型1**: `deepseek-ai/DeepSeek-R1-Distill-Qwen-7B`
- **模型2**: `Qwen/Qwen2.5-7B-Instruct`
- **用途**: 从文本中提取实体之间的关系
- **特点**: 多模型并行，提高准确性和覆盖率

### 3. 社区摘要生成
- **模型**: `deepseek-ai/DeepSeek-R1-Distill-Qwen-7B`
- **用途**: 为知识图谱的社区生成摘要

## 配置位置

模型配置在以下文件中：
- `src/extraction/ner.py`: NER模型配置
- `src/extraction/re.py`: RE模型配置（多模型）
- `src/kg_builder/indexer.py`: 社区摘要生成模型配置

## 如何修改模型

如果要修改使用的模型，请编辑上述文件中的模型名称字符串。

## 测试模型可用性

运行以下命令测试模型：
```bash
uv run python tests/test_model.py

# 测试指定的两个模型
uv run python tests/test_my_models.py
```