#!/bin/bash
# Update Mortdecai model to a new version
# Usage: ./update-model.sh [url] [name]
# Example: ./update-model.sh https://mortdec.ai/dl/m5gguf/mortdecai-0.5.0.gguf mortdecai:0.5.0

set -e

URL="${1:-https://mortdec.ai/dl/m4gguf/mortdecai-0.4.0.gguf}"
NAME="${2:-mortdecai:0.4.0}"

echo "=== Mortdecai Model Update ==="
echo "  URL:  $URL"
echo "  Name: $NAME"
echo ""

# Download
echo "Downloading..."
mkdir -p models
curl -L -o "models/${NAME}.gguf" "$URL" --progress-bar
echo "Download complete"

# Load into Ollama
echo "Loading into Ollama..."
docker exec mortdecai-ollama ollama create "$NAME" -f /models/Modelfile
echo "Model loaded as $NAME"

# Verify
echo ""
echo "Verifying..."
docker exec mortdecai-ollama ollama list | grep "$NAME"

echo ""
echo "=== Update complete ==="
echo "Model $NAME is ready"