"llm/vscode:/vscode.git/clone" did not exist on "7e9a2da097268e33f2b2bd935d6effe2a1a98d46"
start.sh 1008 Bytes
Newer Older
1
2
#!/usr/bin/env bash

Jannik Streidl's avatar
Jannik Streidl committed
3
4
# Get the INCLUDE_OLLAMA_ENV environment variable whcih is set in the Dockerfile
# This includes the ollama in the image
5
6
INCLUDE_OLLAMA=${INCLUDE_OLLAMA_ENV:-false}

Mike Slinn's avatar
Mike Slinn committed
7
8
9
SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
cd "$SCRIPT_DIR" || exit

10
11
KEY_FILE=.webui_secret_key

12
PORT="${PORT:-8080}"
Tim Farrell's avatar
Tim Farrell committed
13
if test "$WEBUI_SECRET_KEY $WEBUI_JWT_SECRET_KEY" = " "; then
14
  echo "No WEBUI_SECRET_KEY provided"
15
16

  if ! [ -e "$KEY_FILE" ]; then
17
    echo "Generating WEBUI_SECRET_KEY"
18
    # Generate a random value to use as a WEBUI_SECRET_KEY in case the user didn't provide one.
19
    echo $(head -c 12 /dev/random | base64) > "$KEY_FILE"
20
21
  fi

22
23
24
25
26
27
28
  echo "Loading WEBUI_SECRET_KEY from $KEY_FILE"
  WEBUI_SECRET_KEY=$(cat "$KEY_FILE")
fi

if [ "$INCLUDE_OLLAMA" = "true" ]; then
    echo "INCLUDE_OLLAMA is set to true, starting ollama serve."
    ollama serve &
29
30
fi

31
WEBUI_SECRET_KEY="$WEBUI_SECRET_KEY" exec uvicorn main:app --host 0.0.0.0 --port "$PORT" --forwarded-allow-ips '*'