Allow port override

wandb · Oct 21, 2024 · 663c60b · 663c60b
1 parent 4f6f531
commit 663c60b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/backend/openui/__main__.py b/backend/openui/__main__.py
@@ -68,7 +68,7 @@ def is_running_in_docker():
             "openui.server:app",
             host="0.0.0.0" if is_running_in_docker() else "127.0.0.1",
             log_config=str(config_file) if ui else None,
-            port=7878,
+            port=config.PORT,
             reload=reload,
         )
     )
@@ -111,7 +111,7 @@ def is_running_in_docker():
             uvicorn.run(
                 "openui.server:app",
                 host="0.0.0.0" if is_running_in_docker() else "127.0.0.1",
-                port=7878,
+                port=config.PORT,
                 reload=reload,
             )
         else:

diff --git a/backend/openui/config.py b/backend/openui/config.py
@@ -66,3 +66,4 @@ class Env(Enum):
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 LITELLM_API_KEY = os.getenv("LITELLM_API_KEY", os.getenv("LITELLM_MASTER_KEY"))
 LITELLM_BASE_URL = os.getenv("LITELLM_BASE_URL", "http://0.0.0.0:4000")
+PORT = int(os.getenv("PORT", 7878))