HTS Tariff Intelligence — LLM 驱动的跨境关税分类与节税优化

"""
HTS Tariff Intelligence — LLM 驱动的 HTS 关税编码分类
母婴跨境场景：5 个典型 SKU 的分层分类 + GRI 规则链输出
依赖：openai>=1.0.0 或可替换为任意 LLM API
"""

import json
import re
from dataclasses import dataclass, field
from typing import Optional


# ─── 数据结构 ─────────────────────────────────────────────

@dataclass
class HTSResult:
    """HTS 分类结果"""
    sku_name: str
    hts_code: str           # 10 位 HTS 编码（示例：9019.20.0000）
    duty_rate: float        # 关税税率（百分比）
    gri_rule: str           # 适用 GRI 规则（如 "GRI Rule 1"）
    gri_reasoning: str      # GRI 规则适用理由
    section: str            # HTS Section（如 "Section XVI"）
    chapter: str            # Chapter（如 "Chapter 90"）
    confidence: float       # 置信度 0-1
    alternative_codes: list = field(default_factory=list)  # 备选编码
    cbp_ruling_ref: Optional[str] = None  # CBP 裁定参考编号


# ─── HTS 关键章节/税率数据（简化版，实际应接 USITC 数据库） ──

HTS_CHAPTER_MAP = {
    "84": {"name": "核反应堆、锅炉、机械器具", "section": "Section XVI"},
    "85": {"name": "电气机械及设备", "section": "Section XVI"},
    "90": {"name": "光学、医疗、精密仪器", "section": "Section XVIII"},
    "94": {"name": "家具、寝具、灯具", "section": "Section XX"},
    "95": {"name": "玩具、游戏品、运动用品", "section": "Section XX"},
    "87": {"name": "车辆及其零件", "section": "Section XVII"},
    "62": {"name": "纺织服装", "section": "Section XI"},
}

# 示例税率库（实际应接 USITC Tariff Schedule API）
DUTY_RATE_DB = {
    "9019.20.0000": 0.0,    # 呼吸治疗器具 — 免税
    "8413.20.0000": 3.4,    # 液体泵
    "8509.80.5050": 4.2,    # 家用电动器具
    "8516.79.0050": 0.0,    # 电热器具（部分）
    "9403.20.0018": 0.0,    # 金属家具（婴儿床架）
    "8714.99.8000": 10.0,   # 童车零件
    "8715.00.0000": 0.0,    # 童车及婴儿车 — 免税
    "9021.90.8100": 0.0,    # 矫形器具
    "3406.00.0000": 0.0,    # 蜡烛（儿童夜灯蜡烛）
    "8471.60.9000": 0.0,    # 输入/输出装置
}

# GRI 规则摘要（简化）
GRI_RULES = {
    "GRI Rule 1": "按税则各类的条文及各节或章的注释归类，税则条文未另规定者，依后续规则处理",
    "GRI Rule 2a": "非完整品/未制成品归类同完整品，前提是具备完整品的基本特征",
    "GRI Rule 2b": "混合材料或物质归类，按 Rule 3 处理",