#!/bin/bash
# Ollama 記憶體管理腳本

case "$1" in
  status)
    echo "=== Ollama 狀態 ==="
    curl -s http://127.0.0.1:11434/api/tags 2>/dev/null | python3 -c "import sys,json; d=json.load(sys.stdin); [print(f'  {m["name"]} ({m["details"]["parameter_size"]})') for m in d.get("models",[])]" 2>/dev/null || echo "  Ollama 沒有回應"
    echo "=== 記憶體使用 ==="
    ps aux | grep -E 'ollama|llama|qwen' | grep -v grep | awk '{printf "  %s %s MB
", \$11, \$6/1024}' 2>/dev/null
    ;;
  restart)
    echo "重啟 Ollama..."
    killall ollama 2>/dev/null
    sleep 2
    open -a Ollama
    echo "Ollama 已重啟"
    ;;
  pull)
    echo "下載模型: $2"
    curl -s -X POST http://127.0.0.1:11434/api/pull -d "{"name":"$2"}"
    ;;
  *)
    echo "用法: ollama-manage.sh {status|restart|pull <model>}"
    ;;
esac