From 207592ff57938536eafa99f2632d670d2bb9457e Mon Sep 17 00:00:00 2001
From: ben
Date: Sat, 4 Mar 2023 22:22:22 +0100
Subject: Isolating containers from internet access to enhance security.

---
 src/ollama_provision/init_models.sh | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100755 src/ollama_provision/init_models.sh

(limited to 'src/ollama_provision/init_models.sh')

diff --git a/src/ollama_provision/init_models.sh b/src/ollama_provision/init_models.sh
new file mode 100755
index 0000000..1eae979
--- /dev/null
+++ b/src/ollama_provision/init_models.sh
@@ -0,0 +1,17 @@
+#!/usr/bin/env bash
+
+
+IFS=',' read -r -a models_arr <<< "${MODELS}"
+
+## now loop through the above array
+for m in "${models_arr[@]}"
+do
+  ollama list | tail -n +2 | cut -d' ' -f1 | grep ${m} > /dev/null
+  if [[ $? -ne 0 ]]
+  then
+    echo "download {m}"
+    ollama pull "${m}"
+  else
+    echo "${m} already installed"
+  fi
+done
-- 
cgit v1.2.3