Claude + Codex + DeepSeek 多模型路由最佳实践

输入任务
  │
  ├─ 是代码补全 / 单函数生成？
  │   └─ 是 → DeepSeek V4 Flash（免费，Teamorouter 免代理直连）
  │
  ├─ 是 Agent 自动化 / 多步骤编排？
  │   └─ 是 → Codex（工程化 Agent，工具链完善）
  │
  ├─ 需要深度推理？（Bug 定位、架构设计、安全审计）
  │   └─ 是 → Claude Fable 5 / Opus（复杂推理最强）
  │
  ├─ 中等复杂度工程任务？
  │   └─ 是 → DeepSeek V4 Pro 或 GPT-5.6 Sol
  │
  └─ 默认兜底 → DeepSeek V4 Flash（便宜省成本）

import teamorouter

# 初始化客户端 —— 一个 Key 接入所有模型
client = teamorouter.Client(api_key="tr-xxxxx")

def route_by_complexity(task: str, file_count: int, task_type: str) -> str:
    """
    根据任务复杂度选择模型
    返回模型 ID
    """
    # 统计复杂度信号
    complexity_score = 0

    # 任务描述越长，通常越复杂
    if len(task) > 500:
        complexity_score += 2
    elif len(task) > 200:
        complexity_score += 1

    # 涉及文件越多，越复杂
    if file_count > 5:
        complexity_score += 3
    elif file_count > 2:
        complexity_score += 1

    # 任务类型加权
    if task_type in ("architecture", "refactor", "security_audit"):
        complexity_score += 3
    elif task_type in ("bug_fix", "code_review"):
        complexity_score += 2
    elif task_type in ("agent_workflow", "multi_step"):
        return "codex"  # Agent 编排任务直接走 Codex

    # 根据综合得分路由
    if complexity_score >= 5:
        return "claude-opus-4-5"
    elif complexity_score >= 3:
        return "claude-fable-5"
    elif complexity_score >= 1:
        return "deepseek-v4-pro"
    else:
        return "deepseek-v4-flash"


def smart_completion(task: str, files: list[str], task_type: str):
    """智能完成——自动选择最佳模型"""
    model = route_by_complexity(task, len(files), task_type)

    response = client.chat.completions.create(
        model=model,
        messages=[
            {"role": "system", "content": "你是一个专业的软件工程师。"},
            {"role": "user", "content": task}
        ],
        context_files=files  # Teamorouter 自动注入文件上下文
    )

    print(f"[路由] 任务类型: {task_type} → 模型: {model}")
    print(f"[成本] 预估: ${response.usage.cost_estimate:.4f}")
    return response


# 使用示例
# 简单补全 → DS V4 Flash（极低成本）
result = smart_completion(
    task="写一个 Python 函数，计算两个日期的间隔天数",
    files=[],
    task_type="function_gen"
)

# 复杂重构 → Claude Fable 5（高质量推理）
result = smart_completion(
    task="重构 user_service.py，将认证逻辑抽离为独立模块，保持向后兼容",
    files=["user_service.py", "auth.py", "middleware.py"],
    task_type="refactor"
)