#!/bin/bash

set -euo pipefail
source source ../../functions/base.sh

mark_done

exit 0

URL="https://github.com/ollama/ollama/releases/download/v0.11.6/ollama-linux-amd64.tgz"
TGZ="/tmp/ollama.tgz"
INSTALL_PATH="/usr/bin/ollama"

echo "[*] Checking for running ollama serve..."
if pgrep -x "ollama" > /dev/null; then
    echo "[*] Stopping running ollama process..."
    pkill -9 ollama
    sleep 2
fi

echo "[*] Downloading ollama..."
curl -L "$URL" -o "$TGZ"

echo "[*] Extracting..."
tar -xzf "$TGZ" -C /tmp

echo "[*] Installing to $INSTALL_PATH..."
sudo mv /tmp/ollama "$INSTALL_PATH"
sudo chmod +x "$INSTALL_PATH"

pkill -9 ollama

SESSION=$(tmux display-message -p '#S')

echo "[*] Using tmux session: $SESSION"


echo "[*] Started ollama pulls in tmux windows."

ollama pull adhishtanaka/llama_3.2_1b-SQL

mark_done