"""向知识库上传示例保险产品文档。""" import requests import base64 import json import sys import time # ============ 保险产品数据 ============ PRODUCTS = [ { "name": "康乐一生重疾险", "company": "复星联合健康", "type": "重疾险", "coverage": ["30万", "50万", "100万"], "premium": {"30岁男": "4500元/年", "30岁女": "4200元/年", "35岁男": "5800元/年", "35岁女": "5400元/年"}, "coverage_period": "终身", "waiting_period": "90天", "diseases": "110种重疾 + 25种中症 + 50种轻症", "features": "重疾不分组赔付3次，中症赔付60%保额，轻症赔付30%保额，自带被保人豁免", "age_range": "0-55岁", }, { "name": "达尔文6号重疾险", "company": "国富人寿", "type": "重疾险", "coverage": ["30万", "40万", "50万"], "premium": {"30岁男": "5200元/年", "30岁女": "4800元/年", "35岁男": "6800元/年", "35岁女": "6300元/年"}, "coverage_period": "终身", "waiting_period": "180天", "diseases": "110种重疾 + 25种中症 + 50种轻症", "features": "60岁前首次重疾额外赔付80%保额，癌症二次赔付间隔期3年", "age_range": "0-55岁", }, { "name": "超级玛丽7号重疾险", "company": "和泰人寿", "type": "重疾险", "coverage": ["30万", "50万", "60万"], "premium": {"30岁男": "5500元/年", "30岁女": "5100元/年", "35岁男": "7200元/年", "35岁女": "6700元/年"}, "coverage_period": "终身", "waiting_period": "180天", "diseases": "110种重疾 + 25种中症 + 55种轻症", "features": "癌症二次赔付间隔期仅1年，赔付比例120%保额，心脑血管二次赔付", "age_range": "0-55岁", }, { "name": "e享护-医享无忧百万医疗险", "company": "太平洋健康", "type": "医疗险", "coverage": ["400万"], "premium": {"30岁男": "268元/年", "30岁女": "288元/年", "35岁男": "358元/年", "35岁女": "388元/年"}, "coverage_period": "1年（保证续保20年）", "waiting_period": "30天", "diseases": "住院医疗、特殊门诊、门诊手术、住院前后门急诊", "features": "保证续保20年，涵盖外购药、质子重离子治疗，100%报销", "age_range": "0-65岁", "deductible": "1万元", }, { "name": "蓝医保长期医疗险", "company": "太平洋健康", "type": "医疗险", "coverage": ["400万"], "premium": {"30岁男": "298元/年", "30岁女": "318元/年", "35岁男": "398元/年", "35岁女": "428元/年"}, "coverage_period": "1年（保证续保20年）", "waiting_period": "30天", "diseases": "住院医疗、特殊门诊、门诊手术、住院前后门急诊、重疾住院津贴", "features": "保证续保20年，重疾0免赔，涵盖外购药、质子重离子，含重疾住院津贴100元/天", "age_range": "0-65岁", "deductible": "1万元（重疾0免赔）", }, { "name": "平安e生保长期医疗险", "company": "平安健康", "type": "医疗险", "coverage": ["200万", "400万"], "premium": {"30岁男": "320元/年", "30岁女": "340元/年", "35岁男": "420元/年", "35岁女": "450元/年"}, "coverage_period": "1年（保证续保20年）", "waiting_period": "30天", "diseases": "住院医疗、特殊门诊、门诊手术、住院前后门急诊", "features": "保证续保20年，平安品牌，服务体系完善，支持垫付医疗费", "age_range": "0-65岁", "deductible": "1万元", }, { "name": "小蜜蜂3号意外险", "company": "鼎和保险", "type": "意外险", "coverage": ["50万", "100万"], "premium": {"30岁男": "158元/年", "30岁女": "158元/年", "35岁男": "158元/年", "35岁女": "158元/年"}, "coverage_period": "1年", "waiting_period": "无", "diseases": "意外身故/伤残、意外医疗、意外住院津贴、猝死保障", "features": "含猝死保障50万，意外医疗0免赔100%报销，不限社保", "age_range": "18-60岁", }, { "name": "大护甲5号意外险", "company": "人保财险", "type": "意外险", "coverage": ["50万", "100万"], "premium": {"30岁男": "168元/年", "30岁女": "168元/年", "35岁男": "168元/年", "35岁女": "168元/年"}, "coverage_period": "1年", "waiting_period": "无", "diseases": "意外身故/伤残、意外医疗、意外住院津贴、交通意外额外赔付", "features": "交通意外额外赔付，意外医疗0免赔，不限社保，含住院津贴150元/天", "age_range": "18-60岁", }, { "name": "华贵大麦2024定期寿险", "company": "华贵人寿", "type": "寿险", "coverage": ["50万", "100万", "200万"], "premium": {"30岁男": "590元/年", "30岁女": "310元/年", "35岁男": "810元/年", "35岁女": "450元/年"}, "coverage_period": "20年/30年/至60岁/至65岁/至70岁", "waiting_period": "90天", "diseases": "身故/全残保障", "features": "免责条款仅3条，健康告知宽松，最高可投保300万，支持减保", "age_range": "18-60岁", }, { "name": "擎天柱9号定期寿险", "company": "横琴人寿", "type": "寿险", "coverage": ["50万", "100万", "300万"], "premium": {"30岁男": "620元/年", "30岁女": "330元/年", "35岁男": "850元/年", "35岁女": "480元/年"}, "coverage_period": "20年/30年/至60岁/至65岁/至70岁", "waiting_period": "90天", "diseases": "身故/全残保障，可选猝死额外赔付", "features": "可选猝死额外赔付50%保额，健康告知宽松，支持月缴", "age_range": "18-60岁", }, { "name": "增多多3号增额终身寿险", "company": "弘康人寿", "type": "储蓄险", "coverage": ["按保费确定保额"], "premium": {"趸交": "10万起", "3年交": "3万/年起", "5年交": "2万/年起", "10年交": "1万/年起"}, "coverage_period": "终身", "waiting_period": "无", "diseases": "身故/全残保障", "features": "保额每年递增3.5%，现金价值高，支持减保取现，可作为养老储备", "age_range": "0-70岁", }, { "name": "年金险-如意享", "company": "信泰人寿", "type": "年金险", "coverage": ["按缴费确定"], "premium": {"趸交": "5万起", "3年交": "2万/年起", "5年交": "1万/年起", "10年交": "5000元/年起"}, "coverage_period": "终身", "waiting_period": "无", "diseases": "身故保障 + 年金领取", "features": "55岁/60岁/65岁开始领取年金，保证领取20年，活多久领多久，可附加万能账户", "age_range": "0-60岁", }, ] def build_document(product: dict) -> str: """将产品信息转为文档文本。""" lines = [ f"# {product['name']}", f"", f"## 基本信息", f"- 产品名称：{product['name']}", f"- 所属保险公司：{product['company']}", f"- 险种类型：{product['type']}", f"- 投保年龄：{product['age_range']}", f"- 保障期限：{product['coverage_period']}", f"- 等待期：{product['waiting_period']}", f"", f"## 保额选择", f"可选保额：{'、'.join(product['coverage'])}", f"", f"## 费率表", ] for age, premium in product['premium'].items(): lines.append(f"- {age}：{premium}") lines.extend([ f"", f"## 保障内容", f"- 保障范围：{product['diseases']}", f"", f"## 产品特色", f"{product['features']}", ]) if 'deductible' in product: lines.extend([f"", f"## 免赔额", f"{product['deductible']}"]) return "\n".join(lines) def main(): session = requests.Session() # Login password_b64 = base64.b64encode('taiyi1224'.encode()).decode() resp = session.post('http://localhost:5001/console/api/login', json={ 'email': 'taiyi@baodan.com', 'password': password_b64 }) if resp.status_code != 200: sys.stdout.buffer.write(f'Login failed: {resp.status_code}\n'.encode('utf-8')) return csrf_token = session.cookies.get('csrf_token') sys.stdout.buffer.write(b'Login OK\n') DATASET_ID = 'ec920b75-758a-4fa1-9132-22cb5e22126e' # Build documents docs = [] for p in PRODUCTS: docs.append({ "name": f"{p['name']}（{p['company']}）", "text": build_document(p), }) # Upload as one batch via create_by_text # Dify API: POST /console/api/datasets/{dataset_id}/documents/create-by-text for doc in docs: resp = session.post( f'http://localhost:5001/console/api/datasets/{DATASET_ID}/document/create-by-text', json={ "name": doc["name"], "text": doc["text"], "indexing_technique": "high_quality", "process_rule": { "mode": "automatic", }, }, headers={'X-CSRF-Token': csrf_token}, ) status = resp.status_code if status in (200, 201): sys.stdout.buffer.write(f" ✓ {doc['name']} uploaded\n".encode('utf-8')) else: sys.stdout.buffer.write(f" ✗ {doc['name']} failed ({status}): {resp.text[:200]}\n".encode('utf-8')) time.sleep(0.5) # rate limit sys.stdout.buffer.write(b'\nDone!\n') if __name__ == "__main__": main()