import Config

# ── 模型注册表 ─────────────────────────────────────────────────────────────
#
# 每个条目是一个带 :name atom 键的关键字列表。
# 列表顺序即优先级：第一个为系统默认模型。
#
# 必填字段
#   :name     - atom，在 iex 中用作 Eai.Chat.talk(model: :name) 的参数
#   :model    - 实际传给 API 的模型字符串
#   :url      - 聊天补全接口地址
#   :provider - :openai_compat | :anthropic（决定鉴权头格式与消息体结构）
#
# 可选字段
#   :api_key_env   - 读取 API Key 的环境变量名（默认 "OPENAI_API_KEY"）
#   :vision        - true 表示此模型支持图像输入（用于 read_media_file）
#   :reasoning_effort - "low" | "medium" | "high"（仅部分模型支持）
#   :receive_timeout  - 毫秒，覆盖全局超时

config :eai, :models, [
  # ── 主力对话模型（默认）─────────────────────────────────────────
  [
    name: :deepseek,
    model: "deepseek-v4-pro",
    url: "https://api.deepseek.com/chat/completions",
    provider: :openai_compat,
    api_key_env: "DEEPSEEK_API_KEY",
    reasoning_effort: "high",
    receive_timeout: 120_000
  ],

  # ── OpenAI ──────────────────────────────────────────────────────
  [
    name: :gpt4o,
    model: "gpt-4o",
    url: "https://api.openai.com/v1/chat/completions",
    provider: :openai_compat,
    api_key_env: "OPENAI_API_KEY",
    vision: true,
    receive_timeout: 60_000
  ],
  [
    name: :gpt4o_mini,
    model: "gpt-4o-mini",
    url: "https://api.openai.com/v1/chat/completions",
    provider: :openai_compat,
    api_key_env: "OPENAI_API_KEY",
    vision: true,
    receive_timeout: 30_000
  ],

  # ── Anthropic ───────────────────────────────────────────────────
  [
    name: :claude_opus,
    model: "claude-opus-4-6",
    url: "https://api.anthropic.com/v1/messages",
    provider: :anthropic,
    api_key_env: "ANTHROPIC_API_KEY",
    vision: true,
    receive_timeout: 120_000
  ],
  [
    name: :claude_sonnet,
    model: "claude-sonnet-4-6",
    url: "https://api.anthropic.com/v1/messages",
    provider: :anthropic,
    api_key_env: "ANTHROPIC_API_KEY",
    vision: true,
    receive_timeout: 60_000
  ],

  # ── 本地 / 自托管（Ollama） ─────────────────────────────────────
  [
    name: :llava,
    model: "llava",
    url: "http://localhost:11434/v1/chat/completions",
    provider: :openai_compat,
    # Ollama 无需 key
    api_key_env: nil,
    vision: true,
    receive_timeout: 120_000
  ],
  [
    name: :llama3,
    model: "llama3",
    url: "http://localhost:11434/v1/chat/completions",
    provider: :openai_compat,
    api_key_env: nil,
    receive_timeout: 120_000
  ]
]