fix: env var migration issue

e3fc9724 · Timothy J. Baek · b4c770d7 · e3fc9724
Commit e3fc9724 authored Mar 10, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 4 deletions

backend/config.py backend/config.py +5 -4

No files found.
--- a/backend/config.py
+++ b/backend/config.py
@@ -209,10 +209,6 @@ OLLAMA_API_BASE_URL = os.environ.get(

 OLLAMA_BASE_URL = os.environ.get("OLLAMA_BASE_URL", "")

-if ENV == "prod":
-    if OLLAMA_BASE_URL == "/ollama":
-        OLLAMA_BASE_URL = "http://host.docker.internal:11434"
-

 if OLLAMA_BASE_URL == "" and OLLAMA_API_BASE_URL != "":
    OLLAMA_BASE_URL = (
@@ -221,6 +217,11 @@ if OLLAMA_BASE_URL == "" and OLLAMA_API_BASE_URL != "":
        else OLLAMA_API_BASE_URL
    )

+if ENV == "prod":
+    if OLLAMA_BASE_URL == "/ollama":
+        OLLAMA_BASE_URL = "http://host.docker.internal:11434"
+
+
 OLLAMA_BASE_URLS = os.environ.get("OLLAMA_BASE_URLS", "")
 OLLAMA_BASE_URLS = OLLAMA_BASE_URLS if OLLAMA_BASE_URLS != "" else OLLAMA_BASE_URL