tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

added comments

Browse source
This commit is contained in:
Jannik Streidl 2024-03-22 09:50:01 +01:00
parent fde0139bf7
commit 6dc4b748fb
3 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
Dockerfile
View file

@ -1,5 +1,6 @@
# syntax=docker/dockerfile:1
# Initialize device type args
# use buiild args in the docker build commmand with --build-arg="BUILDARG=true"
ARG USE_CUDA=false
ARG USE_MPS=false
ARG INCLUDE_OLLAMA=false

2
backend/config.py
View file

@ -221,6 +221,8 @@ if OLLAMA_BASE_URL == "" and OLLAMA_API_BASE_URL != "":
if ENV == "prod":
if OLLAMA_BASE_URL == "/ollama":
if INCLUDE_OLLAMA:
# If you use all in one Docker container (WEbUI + Ollama)
# with the docker build arg INCLUDE_OLLAMA=true (--build-arg="INCLUDE_OLLAMA=true") this only works with http://localhost:11434
OLLAMA_BASE_URL = "http://localhost:11434"
else:
OLLAMA_BASE_URL = "http://host.docker.internal:11434"

2
backend/start.sh
View file

@ -1,5 +1,7 @@
#!/usr/bin/env bash
# Get the INCLUDE_OLLAMA_ENV environment variable whcih is set in the Dockerfile
# This includes the ollama in the image
INCLUDE_OLLAMA=${INCLUDE_OLLAMA_ENV:-false}
SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )