#!/bin/bash
# setup_ollama_docker.sh

# Pull the official Ollama Docker image
docker pull ollama/ollama

# Create a directory for Ollama data
mkdir -p ~/ollama_data

# Run Ollama in Docker with data persistence
docker run -d \
  --name ollama \
  -p 11434:11434 \
  -v ~/ollama_data:/root/.ollama \
  --restart unless-stopped \
  ollama/ollama

echo "Ollama is running in Docker. Wait a moment for it to initialize..."
sleep 10

# Pull the DeepSeek V2 Lite model
docker exec ollama ollama pull deepseek-coder:6.7b

echo "Setup complete! Ollama is running with DeepSeek V2 Lite model."