aboutsummaryrefslogtreecommitdiffstats
path: root/src/llm_provision/init_models.sh
blob: 0afbbd0e63a7219e8e877d6773cbdcadfbb7c7fb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#!/usr/bin/env bash

OLLAMA_HOST="http://ollama:11434"

IFS=',' read -r -a models_arr <<< "${MODELS}"

## now loop through the above array
for m in "${models_arr[@]}"
do
  curl -s "${OLLAMA_HOST}/api/tags" | jq '.models[].name' | grep ${m} > /dev/null
  if [[ $? -ne 0 ]]
  then
    curl -s "${OLLAMA_HOST}/api/pull" -d "{\"model\": \"${m}\"}"
  else
    echo "${m} already installed"
  fi
done