"Claude全系列模型对比：Opus vs Sonnet vs Haiku怎么选？"

from openai import OpenAI

# 通过Crazyrouter统一API访问所有Claude模型
client = OpenAI(
    api_key="your-crazyrouter-key",
    base_url="https://api.crazyrouter.com/v1"
)

# 使用Opus - 复杂推理任务
opus_response = client.chat.completions.create(
    model="claude-opus-4-5",
    messages=[
        {"role": "user", "content": "设计一个分布式缓存系统的架构，要求支持百万级QPS，考虑一致性、可用性和分区容错性的权衡。"}
    ]
)

# 使用Sonnet - 日常开发任务
sonnet_response = client.chat.completions.create(
    model="claude-sonnet-4-5",
    messages=[
        {"role": "user", "content": "帮我写一个Python装饰器，实现函数调用的自动重试和指数退避。"}
    ]
)

# 使用Haiku - 简单快速任务
haiku_response = client.chat.completions.create(
    model="claude-haiku-4-5",
    messages=[
        {"role": "user", "content": "把这段JSON格式化一下：{\"name\":\"test\",\"value\":123}"}
    ]
)

def smart_route(task_description, complexity="auto"):
    """根据任务复杂度自动选择最合适的Claude模型"""

    if complexity == "auto":
        # 简单启发式：根据任务描述长度和关键词判断
        complex_keywords = ["设计", "架构", "分析", "优化", "推理", "证明", "比较"]
        simple_keywords = ["格式化", "翻译", "总结", "提取", "分类"]

        desc_lower = task_description.lower()
        if any(kw in desc_lower for kw in complex_keywords) and len(task_description) > 200:
            complexity = "high"
        elif any(kw in desc_lower for kw in simple_keywords) or len(task_description) < 100:
            complexity = "low"
        else:
            complexity = "medium"

    model_map = {
        "high": "claude-opus-4-5",
        "medium": "claude-sonnet-4-5",
        "low": "claude-haiku-4-5"
    }

    model = model_map[complexity]
    print(f"Selected model: {model} (complexity: {complexity})")

    response = client.chat.completions.create(
        model=model,
        messages=[{"role": "user", "content": task_description}]
    )

    return response.choices[0].message.content

# 自动路由到Haiku
result1 = smart_route("把hello翻译成中文")

# 自动路由到Sonnet
result2 = smart_route("写一个用户认证的REST API，包含注册、登录和JWT验证")

# 自动路由到Opus
result3 = smart_route("设计一个支持百万用户的实时聊天系统架构，需要考虑消息持久化、在线状态管理、消息推送、群聊支持等功能，并给出详细的技术选型和架构图")