Use env vars MASTER_API_KEY, OLLAMA_KEY_1/2/3

app.py

@@ -1,34 +1,40 @@
 import os
-import random
 import httpx
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import JSONResponse, Response
 
 app = FastAPI()
 
-BASE_URL = "https://ollama.com"
-MASTER_KEY = "ollama-proxy-free"
+# Environment variables
+BASE_URL = os.getenv("BASE_URL", "https://ollama.com")
+MASTER_API_KEY = os.getenv("MASTER_API_KEY", "ollama-proxy-free")
 
-KEYS = [
-    "8ca25de51e554c099962b78b7ce0c9e9.Mp5dnqctR2zzq3g-NO_M-cjW",
-    "dbd1d0c534964684a6d4678348ab8d30.ieDfmSYVnf0MmTjR-AIdNrW9",
-    "37e81a6be4104fbfbfbe2ecf557a2c10.GoIbzpHebdM9ZcHarQ9A12Cp"
-]
+# Load Ollama keys from env
+OLLAMA_KEYS = []
+for i in range(1, 10):
+    key = os.getenv(f"OLLAMA_KEY_{i}")
+    if key:
+        OLLAMA_KEYS.append(key)
 
 @app.get("/")
 def root():
-    return {"status": "ok", "master": MASTER_KEY[:10]}
+    return {
+        "status": "ok",
+        "master_key": MASTER_API_KEY[:10] + "...",
+        "ollama_keys": len(OLLAMA_KEYS)
+    }
 
 @app.post("/v1/chat/completions")
 async def chat(req: Request):
     # Auth
     auth_key = req.headers.get("Authorization", "").replace("Bearer ", "")
-    if auth_key != MASTER_KEY:
-        raise HTTPException(401, "Unauthorized")
+    if auth_key != MASTER_API_KEY:
+        return JSONResponse({"error": "Unauthorized"}, status_code=401)
     
     body = await req.json()
     
-    for key in KEYS:
+    # Try each key
+    for key in OLLAMA_KEYS:
         try:
             async with httpx.AsyncClient(timeout=60) as client:
                 resp = await client.post(
@@ -36,17 +42,20 @@ async def chat(req: Request):
                     json=body,
                     headers={"Authorization": f"Bearer {key}"}
                 )
-                
                 if resp.status_code == 200:
                     return Response(resp.content, status_code=200)
         except Exception as e:
-            print(f"ERR {key[:6]}: {e}")
+            continue
     
     return JSONResponse({"error": "all keys failed"}, status_code=500)
 
 @app.get("/v1/models")
 def models():
-    for key in KEYS:
+    auth_key = req.headers.get("Authorization", "").replace("Bearer ", "")
+    if auth_key != MASTER_API_KEY:
+        return JSONResponse({"error": "Unauthorized"}, status_code=401)
+    
+    for key in OLLAMA_KEYS:
         try:
             resp = httpx.get(
                 f"{BASE_URL}/v1/models",