Configure public base URL env vars and Host/X-Forwarded headers so OmniRoute generates public https URLs for OAuth callbacks

entrypoint.sh

@@ -105,10 +105,19 @@ export HOSTNAME="127.0.0.1"
 # Override DATA_DIR to local ext4 to avoid SQLite FUSE I/O errors on HF /data
 export DATA_DIR="/root/.omniroute"
 export REDIS_URL="redis://127.0.0.1:6379"
-export OMNIROUTE_BASE_PATH="/omniroute"
-export NEXT_PUBLIC_OMNIROUTE_BASE_PATH="/omniroute"
-export NEXT_PUBLIC_BASE_URL="${WEBUI_URL:-https://jishnupg-opencode-cli.hf.space}/omniroute"
+
+# Public Base URL and Origin settings for OAuth and WebSockets
+export APP_URL="${WEBUI_URL:-https://jishnupg-opencode-cli.hf.space}"
+export PUBLIC_URL="${WEBUI_URL:-https://jishnupg-opencode-cli.hf.space}"
+export BASE_URL="${WEBUI_URL:-https://jishnupg-opencode-cli.hf.space}"
+export NEXT_PUBLIC_BASE_URL="${WEBUI_URL:-https://jishnupg-opencode-cli.hf.space}"
+export NEXT_PUBLIC_APP_URL="${WEBUI_URL:-https://jishnupg-opencode-cli.hf.space}"
+export OMNIROUTE_PUBLIC_URL="${WEBUI_URL:-https://jishnupg-opencode-cli.hf.space}"
+export OMNIROUTE_URL="${WEBUI_URL:-https://jishnupg-opencode-cli.hf.space}"
+export OMNIROUTE_BASE_PATH=""
+export NEXT_PUBLIC_OMNIROUTE_BASE_PATH=""
 export LIVE_WS_ALLOWED_ORIGINS="${WEBUI_URL:-https://jishnupg-opencode-cli.hf.space}"
+export OMNIROUTE_AUTO_FREE_FALLBACK_TO_FULL_POOL="true"
 
 # Default admin credentials for OmniRoute
 export INITIAL_PASSWORD="${INITIAL_PASSWORD:-admin}"

gateway/omniroute.py

@@ -31,9 +31,10 @@ router = APIRouter(tags=["OmniRoute"])
 @router.api_route("/omniroute/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD", "OPTIONS"])
 async def omniroute_main_route(request: Request, path: str = ""):
     extra = {
-        "Host": f"127.0.0.1:{OMNIROUTE_PORT}",
+        "Host": PUBLIC_HOST,
         "X-Forwarded-Host": PUBLIC_HOST,
-        "X-Forwarded-Proto": "https"
+        "X-Forwarded-Proto": "https",
+        "X-Forwarded-Port": "443"
     }
     # Strip /omniroute prefix when proxying to OmniRoute backend on port 20128
     target = f"http://127.0.0.1:{OMNIROUTE_PORT}/{path}" if path else f"http://127.0.0.1:{OMNIROUTE_PORT}/"
@@ -74,9 +75,10 @@ async def omniroute_monitoring(request: Request, path: str = ""):
 @router.api_route("/omniroute/_next/{path:path}", methods=["GET", "POST", "PUT", "DELETE", "PATCH", "HEAD", "OPTIONS"])
 async def omniroute_assets(request: Request, path: str = ""):
     extra = {
-        "Host": f"127.0.0.1:{OMNIROUTE_PORT}",
+        "Host": PUBLIC_HOST,
         "X-Forwarded-Host": PUBLIC_HOST,
-        "X-Forwarded-Proto": "https"
+        "X-Forwarded-Proto": "https",
+        "X-Forwarded-Port": "443"
     }
     target = f"http://127.0.0.1:{OMNIROUTE_PORT}/omniroute/_next/{path}"
     res = await proxy_http_request(target, request, default_prefix="/omniroute", extra_headers=extra)
@@ -140,9 +142,10 @@ for p_route in OMNIROUTE_PAGE_ROUTES:
         )
         async def omniroute_page_handler(request: Request, path: str = ""):
             extra = {
-                "Host": f"127.0.0.1:{OMNIROUTE_PORT}",
+                "Host": PUBLIC_HOST,
                 "X-Forwarded-Host": PUBLIC_HOST,
-                "X-Forwarded-Proto": "https"
+                "X-Forwarded-Proto": "https",
+                "X-Forwarded-Port": "443"
             }
             target_path = f"{page_path}/{path}" if path else page_path
             target = f"http://127.0.0.1:{OMNIROUTE_PORT}{target_path}"