#!/bin/bash
# download-model.sh - 下载适合 8GB 内存的 Qwen 模型

MODEL_DIR="${HOME}/.llama/models"
mkdir -p "$MODEL_DIR"

echo "=== 下载 Qwen 模型 (适合 8GB 内存) ==="
echo ""
echo "可选模型:"
echo "1) Qwen3-1.8B-Q4_K_M (推荐, ~1.2GB, 速度最快)"
echo "2) Qwen3-4B-Q4_K_M (~2.5GB, 质量更好)"
echo "3) Qwen2.5-1.5B-Q4_K_M (备用, ~1GB)"
echo ""

# 默认下载 1.8B 版本
MODEL_URL="https://huggingface.co/Qwen/Qwen3-1.8B-GGUF/resolve/main/qwen3-1.8b-q4_k_m.gguf"
MODEL_NAME="qwen3-1.8b-q4_k_m.gguf"

cd "$MODEL_DIR"

if [ -f "$MODEL_NAME" ]; then
    echo "✅ 模型已存在: $MODEL_NAME"
else
    echo "⬇️ 正在下载 $MODEL_NAME..."
    echo "(如下载慢可用浏览器访问 https://huggingface.co/Qwen/Qwen3-1.8B-GGUF)"
    
    # 使用 curl 下载，支持断点续传
    curl -L -C - -o "$MODEL_NAME" "$MODEL_URL" || {
        echo "❌ 下载失败，请手动下载到: $MODEL_DIR/$MODEL_NAME"
        echo "下载链接: $MODEL_URL"
        exit 1
    }
fi

echo ""
echo "✅ 模型已准备: $MODEL_DIR/$MODEL_NAME"
echo ""
echo "=== 下一步 ==="
echo "启动服务: ./start-server.sh"