feat: add Ollama WebUI Just script (#1224)

ublue-os · May 3, 2024 · 6b58f98 · 6b58f98
1 parent b9cf82c
commit 6b58f98
Showing 1 changed file with 51 additions and 0 deletions.
diff --git a/just/bluefin-tools.just b/just/bluefin-tools.just
@@ -85,6 +85,7 @@ ollama:
     RemapUsers=keep-id
     RunInit=yes
     NoNewPrivileges=no
+    Network=ollama.network
     Volume=%h/.ollama:/.ollama
     PodmanArgs=--userns=keep-id
     PodmanArgs=--group-add=keep-groups
@@ -104,6 +105,56 @@ ollama:
     else
         echo "Ollama container already exists, skipping..."
     fi
+
+    read -r -d '' QUADLET_NETWORK <<-EOF
+    [Network]
+    NetworkName=ollama
+    EOF
+    if [  ! -f ~/.config/containers/systemd/ollama.network ]; then
+        mkdir -p ~/.config/containers/systemd
+        echo "${QUADLET_NETWORK}" > ~/.config/containers/systemd/ollama.network
+    else
+        echo "Ollama network already exists, skipping..."
+    fi
+
     systemctl --user daemon-reload
     systemctl --user start ollama.service || echo "Error starting Ollama Quadlet."
     echo "Please install the ollama cli via \`brew install ollama\`"
+
+# Setup a local Ollama WebUI in a container
+ollama-web: ollama
+    #!/usr/bin/env bash
+
+    read -r -d '' QUADLET <<-EOF
+    [Unit]
+    Description=An Ollama WebUI container
+    After=network-online.target ollama.service
+    Requires=ollama.service
+
+    [Container]
+    Image=ghcr.io/open-webui/open-webui:latest
+    AutoUpdate=registry
+    ContainerName=ollama-web
+    Environment=OLLAMA_BASE_URL=http://ollama:11434
+    Environment=WEBUI_SECRET_KEY=abc123
+    Environment=DEFAULT_USER_ROLE=admin
+    # Open WebUI does not allow access without a user account, nor does it allow
+    # account creation via environment variables.
+    Environment=ENABLE_SIGNUP=true
+    PublishPort=8080:8080
+    Network=ollama.network
+
+    [Service]
+    TimeoutStartSec=900
+
+    [Install]
+    WantedBy=multi-user.target
+    EOF
+    if [  ! -f ~/.config/containers/systemd/ollama-web.container ]; then
+        mkdir -p ~/.config/containers/systemd
+        echo "${QUADLET}" > ~/.config/containers/systemd/ollama-web.container
+    else
+        echo "Ollama WebUI container already exists, skipping..."
+    fi
+    systemctl --user daemon-reload
+    systemctl --user start ollama-web.service