forked from open-webui/open-webui
16 lines
360 B
Python
16 lines
360 B
Python
|
import sys
|
||
|
|
import os
|
||
|
|
from dotenv import load_dotenv, find_dotenv
|
||
|
|
|
||
|
|
load_dotenv(find_dotenv())
|
||
|
|
|
||
|
|
ENV = os.environ.get("ENV", "dev")
|
||
|
|
|
||
|
|
OLLAMA_API_BASE_URL = os.environ.get(
|
||
|
|
"OLLAMA_API_BASE_URL", "http://localhost:11434/api"
|
||
|
|
)
|
||
|
|
|
||
|
|
if ENV == "prod":
|
||
|
|
if OLLAMA_API_BASE_URL == "/ollama/api":
|
||
|
|
OLLAMA_API_BASE_URL = "http://host.docker.internal:11434/api"
|