enh: remove trailing slash from ollama

Co-Authored-By: hi-unc1e <67778054+hi-unc1e@users.noreply.github.com>
This commit is contained in:
Timothy Jaeryang Baek 2024-11-30 18:13:10 -08:00
parent 0b17ff6eef
commit dbf6ec71fe

View File

@ -583,6 +583,12 @@ OLLAMA_API_BASE_URL = os.environ.get(
)
OLLAMA_BASE_URL = os.environ.get("OLLAMA_BASE_URL", "")
if OLLAMA_BASE_URL:
# Remove trailing slash
OLLAMA_BASE_URL = (
OLLAMA_BASE_URL[:-1] if OLLAMA_BASE_URL.endswith("/") else OLLAMA_BASE_URL
)
K8S_FLAG = os.environ.get("K8S_FLAG", "")
USE_OLLAMA_DOCKER = os.environ.get("USE_OLLAMA_DOCKER", "false")