Spaces:

GGSheng
/

apple

Paused

File size: 9,027 Bytes

62f6d9a

#!/usr/bin/env bash
# ============================================================
# CCMR (Claude Code Model Router) 配置自动生成脚本
#
# 功能说明:
#   读取 CCMR_* 环境变量，生成 CCMR 网关所需的 .env 和 models.yaml
#   支持两种配置方式，优先级：文件 > Docker/HF Spaces 环境变量
#
# 方式1 - 配置文件（推荐）:
#   创建 /root/.env.d/ccmr.env，填入以下变量即可
#   修改后立即生效，无需重启容器（由 ccmr-wrapper.sh 监控热加载）
#
# 方式2 - Docker/HF Spaces 环境变量:
#   在 docker run -e 或 HF Spaces Secrets 中设置 CCMR_* 变量
#
# 运行原理:
#   1. 检测 /root/.env.d/ccmr.env 是否存在，存在则 source 加载
#   2. 读取当前 Shell 中所有 CCMR_* 环境变量（来自文件或 Docker）
#   3. 将 CCMR_DEEPSEEK_API_KEY -> DEEPSEEK_API_KEY 映射写入 .env
#   4. 如果 /root/.ccmr/models.yaml 不存在，生成默认模型配置
#
# ============================================================
# 所有支持的环境变量（共 10 个 API Key + 3 个系统设置）
# ============================================================
#
# 系统设置:
#   CCMR_ENABLED            - 是否启用 CCMR (true/false, 默认 false)
#   CCMR_PORT               - 网关监听端口 (默认 8080)
#   CCMR_CONFIG_DIR         - 配置存放目录 (默认 ~/.ccmr)
#
# API Key 列表（按平台分组）:
#
# 1) DeepSeek（https://platform.deepseek.com/）
#   CCMR_DEEPSEEK_API_KEY=sk-xxx
#   支持模型: deepseek-v4-pro, deepseek-v4-flash
#   可用别名: deepseek, deepseek-pro, deepseek-flash, ds-flash
#
# 2) 通义千问 / Qwen（https://dashscope.console.aliyun.com/）
#   CCMR_QWEN_API_KEY=sk-xxx
#   支持模型: qwen3.5-plus, qwen3.5-flash, qwen3-max
#   可用别名: qwen, qwen-plus, qwen-flash, qw-turbo, qwen-max
#
# 3) Kimi / Moonshot（https://platform.kimi.ai/）
#   CCMR_KIMI_API_KEY=sk-xxx
#   支持模型: kimi-k2.6
#   可用别名: kimi, moonshot
#
# 4) 智谱 GLM（https://open.bigmodel.cn/）
#   CCMR_GLM_API_KEY=xxx
#   支持模型: glm-5.1, glm-5.1-flash
#   可用别名: glm, glm4, glm-flash
#
# 5) MiniMax CN（https://platform.minimaxi.com/）
#   CCMR_MINIMAX_API_KEY=xxx
#   支持模型: minimax-m2.7
#   可用别名: minimax, mmax
#
# 6) MiniMax Global（https://platform.minimax.io/）
#   CCMR_MINIMAX_GLOBAL_API_KEY=xxx
#   支持模型: minimax-global-m2.7
#   可用别名: minimax-global, mmax-global
#
# 7) MiMo Token Plan SGP 集群（https://platform.xiaomimimo.com/）
#   CCMR_MIMO_API_KEY=tp-xxx
#   支持模型: mimo-v2.5-pro - SGP 集群 (Singapore)
#   可用别名: mimo, mimo-pro
#
# 8) MiMo Token Plan CN 集群
#   CCMR_MIMO_TOKEN_CN_API_KEY=tp-xxx
#   支持模型: mimo-v2.5-pro - CN 集群
#
# 9) MiMo Token Plan AMS 集群
#   CCMR_MIMO_TOKEN_AMS_API_KEY=tp-xxx
#   支持模型: mimo-v2.5-pro - AMS 集群 (Amsterdam)
#
# 10) MiMo 按量付费（https://platform.xiaomimimo.com/）
#   CCMR_MIMO_PAYG_API_KEY=sk-xxx
#   支持模型: mimo-v2.5-pro - Pay-as-you-go
#
# ============================================================

set -euo pipefail

# ============================================================
# 从文件加载 CCMR 环境变量
# ============================================================
CCMR_CONFIG_FILE="${CCMR_CONFIG_FILE:-/root/.env.d/ccmr.env}"

if [[ -f "$CCMR_CONFIG_FILE" ]]; then
  echo "[CCMR-SETUP] Loading configuration from: $CCMR_CONFIG_FILE"
  # shellcheck source=/dev/null
  source "$CCMR_CONFIG_FILE"
else
  echo "[CCMR-SETUP] No config file at $CCMR_CONFIG_FILE, using env vars only"
  echo "[CCMR-SETUP] To use file-based config, create: $CCMR_CONFIG_FILE"
fi

CCMR_CONFIG_DIR="${CCMR_CONFIG_DIR:-$HOME/.ccmr}"
mkdir -p "$CCMR_CONFIG_DIR"

echo "[CCMR-SETUP] Generating configuration in: $CCMR_CONFIG_DIR"

# ============================================================
# 生成 .env 文件
# ============================================================
ENV_FILE="$CCMR_CONFIG_DIR/.env"

# API Key 映射表: CCMR_xxx -> .env 中的键名
declare -A API_KEY_MAP=(
  ["CCMR_DEEPSEEK_API_KEY"]="DEEPSEEK_API_KEY"
  ["CCMR_QWEN_API_KEY"]="QWEN_API_KEY"
  ["CCMR_KIMI_API_KEY"]="KIMI_API_KEY"
  ["CCMR_GLM_API_KEY"]="GLM_API_KEY"
  ["CCMR_MINIMAX_API_KEY"]="MINIMAX_API_KEY"
  ["CCMR_MINIMAX_GLOBAL_API_KEY"]="MINIMAX_GLOBAL_API_KEY"
  ["CCMR_MIMO_API_KEY"]="MIMO_API_KEY"
  ["CCMR_MIMO_TOKEN_CN_API_KEY"]="MIMO_TOKEN_CN_API_KEY"
  ["CCMR_MIMO_TOKEN_AMS_API_KEY"]="MIMO_TOKEN_AMS_API_KEY"
  ["CCMR_MIMO_PAYG_API_KEY"]="MIMO_PAYG_API_KEY"
)

# 创建 .env 文件
{
  echo "# ============================================================"
  echo "# Claude Code Model Router - 自动生成的配置"
  echo "# ============================================================"
  echo ""

  has_any_key=false
  for ccmr_var in "${!API_KEY_MAP[@]}"; do
    env_key="${API_KEY_MAP[$ccmr_var]}"
    env_value="${!ccmr_var:-}"
    if [[ -n "$env_value" ]]; then
      echo "${env_key}=${env_value}"
      has_any_key=true
    fi
  done

  if [[ "$has_any_key" == "false" ]]; then
    echo "# 未检测到 API Key 环境变量"
    echo "# 请设置环境变量后重新运行，例如:"
    echo "#   CCMR_DEEPSEEK_API_KEY=sk-xxx"
    echo "# 支持的 Key 列表:"
    for ccmr_var in "${!API_KEY_MAP[@]}"; do
      env_key="${API_KEY_MAP[$ccmr_var]}"
      echo "#   ${ccmr_var}=xxx  ->  ${env_key}=xxx"
    done
  fi
} > "$ENV_FILE"

echo "[CCMR-SETUP] ✓ .env written with $([ "$has_any_key" == "true" ] && echo "$(grep -c '=' "$ENV_FILE") key(s) configured" || echo "no API keys (template mode)")"

# ============================================================
# 生成 models.yaml 配置文件
# ============================================================
MODELS_FILE="$CCMR_CONFIG_DIR/models.yaml"

if [[ ! -f "$MODELS_FILE" ]]; then
  echo "[CCMR-SETUP] Creating default models.yaml..."
  cat > "$MODELS_FILE" << 'YAMLEOF'
# ============================================================
# Claude Code Model Router - 模型配置
# ============================================================

providers:
  deepseek:
    base_url: "https://api.deepseek.com"
    api_key_env: "DEEPSEEK_API_KEY"
    variants:
      deepseek-v4-pro:
        model: "deepseek-chat"
        context_window: 65536
        max_tokens: 8192
        aliases:
          - deepseek
          - deepseek-pro
      deepseek-v4-flash:
        model: "deepseek-chat"
        context_window: 65536
        max_tokens: 8192
        aliases:
          - deepseek-flash
          - ds-flash

  qwen:
    base_url: "https://dashscope.aliyuncs.com/compatible-mode/v1"
    api_key_env: "QWEN_API_KEY"
    variants:
      qwen3.5-plus:
        model: "qwen-plus"
        context_window: 131072
        max_tokens: 8192
        aliases:
          - qwen
          - qwen-plus
      qwen3.5-flash:
        model: "qwen-turbo"
        context_window: 131072
        max_tokens: 8192
        aliases:
          - qwen-flash
          - qw-turbo
      qwen3-max:
        model: "qwen-max"
        context_window: 32768
        max_tokens: 8192
        aliases:
          - qwen-max

  kimi:
    base_url: "https://api.moonshot.cn/v1"
    api_key_env: "KIMI_API_KEY"
    variants:
      kimi-k2.6:
        model: "moonshot-v1-8k"
        context_window: 8192
        max_tokens: 4096
        aliases:
          - kimi
          - moonshot

  glm:
    base_url: "https://open.bigmodel.cn/api/paas/v4"
    api_key_env: "GLM_API_KEY"
    variants:
      glm-5.1:
        model: "glm-4"
        context_window: 131072
        max_tokens: 4096
        aliases:
          - glm
          - glm4
      glm-5.1-flash:
        model: "glm-4-flash"
        context_window: 131072
        max_tokens: 4096
        aliases:
          - glm-flash

  minimax:
    base_url: "https://api.minimax.chat/v1"
    api_key_env: "MINIMAX_API_KEY"
    variants:
      minimax-m2.7:
        model: "minimax-m2.7"
        context_window: 131072
        max_tokens: 8192
        aliases:
          - minimax
          - mmax

  minimax_global:
    base_url: "https://api.minimax.io/v1"
    api_key_env: "MINIMAX_GLOBAL_API_KEY"
    variants:
      minimax-global-m2.7:
        model: "minimax-m2.7"
        context_window: 131072
        max_tokens: 8192
        aliases:
          - minimax-global
          - mmax-global

  mimo:
    base_url: "https://api.miMo.com/v1"
    api_key_env: "MIMO_API_KEY"
    variants:
      mimo-v2.5-pro:
        model: "mimo-v2.5-pro"
        context_window: 131072
        max_tokens: 8192
        aliases:
          - mimo
          - mimo-pro
YAMLEOF
  echo "[CCMR-SETUP] ✓ models.yaml created"
else
  echo "[CCMR-SETUP] → models.yaml already exists, skipping"
fi

# ============================================================
# 完成
# ============================================================
echo "[CCMR-SETUP] Config directory: $CCMR_CONFIG_DIR"
echo "[CCMR-SETUP] Files:"
ls -la "$CCMR_CONFIG_DIR/"
echo "[CCMR-SETUP] Setup complete!"