# 合同审查数据集

## 1、数据整理
在审查维度合并Excel与JSON，每个审查点对应一个JSON文件

## 2、数据增广
基于Excel合同审查结果标注JSON结果，Few-Shot-Prompt更大模型

## 3、原文拓展
为保证长上下文注意力效果，对原文进行拓展，以符合分段合同分布

## 4、训练
基于数据集进行SFT训练

## 5、统一导出
合并 `merge_excel.py` 与 `merge_json.py` 的导出内容：

```bash
OPENAI_API_KEY=your_key python main.py --excel datasets/excel --json-dir datasets/json --out datasets/merged/merged_all.json
```

说明：`main.py` 会使用每条数据的 `prompt` 作为 system、`original_text` 作为 user 调用 OpenAI 接口；
解析返回 JSON 中的 `result` 与 `ground_truth` 对比，不一致会重试，默认最多 `10` 次。
可选参数：`--model`、`--max-retries`、`--openai-base-url`。

依赖安装：`pip install openai tenacity pandas`