keepalive: read master key from /app/data/.master_key (scheduler runs outside litellm's env)

This commit is contained in:
Inferencebot committed 2026-10-12 19:20:24 -06:00
1 parent acceb43995
commit 0d73c80562
1 file changed
+16 -2
+16 -2
View File
@@ -10,9 +10,23 @@ success; prints + exits non-zero on failure so scheduler logs capture it.
import json, os, urllib.request, urllib.error, sys import json, os, urllib.request, urllib.error, sys
PORT = os.environ.get("PORT", "4000") PORT = os.environ.get("PORT", "4000")
KEY = os.environ.get("LITELLM_MASTER_KEY", "")
def master_key():
"""The gateway's start.sh generates the master key into /app/data/.master_key
and exports LITELLM_MASTER_KEY for the litellm process — scheduler tasks run
in a separate process tree, so read the key file directly."""
key = os.environ.get("LITELLM_MASTER_KEY", "")
if key:
return key
try:
with open("/app/data/.master_key") as f:
return f.read().strip()
except OSError:
return ""
KEY = master_key()
if not KEY: if not KEY:
print("keepalive: LITELLM_MASTER_KEY not set", file=sys.stderr) print("keepalive: no master key (env or /app/data/.master_key)", file=sys.stderr)
sys.exit(1) sys.exit(1)
fail = 0 fail = 0