Browse Source

enh: remove trailing slash from ollama

Co-Authored-By: hi-unc1e <67778054+hi-unc1e@users.noreply.github.com>
Timothy Jaeryang Baek 5 months ago
parent
commit
dbf6ec71fe
1 changed files with 6 additions and 0 deletions
  1. 6 0
      backend/open_webui/config.py

+ 6 - 0
backend/open_webui/config.py

@@ -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")