Add more debug

app.py

@@ -9,95 +9,81 @@ from fastapi.responses import JSONResponse, StreamingResponse, Response
 app = FastAPI()
 
 BASE_URL = os.getenv("BASE_URL", "https://ollama.com")
-
-# CLIENT API KEY (for auth)
 MASTER_API_KEY = "ollama-proxy-free"
 
-# BACKEND API KEYS (for Ollama)
 API_KEYS = [
     "8ca25de51e554c099962b78b7ce0c9e9.Mp5dnqctR2zzq3g-NO_M-cjW",
     "dbd1d0c534964684a6d4678348ab8d30.ieDfmSYVnf0MmTjR-AIdNrW9",
     "37e81a6be4104fbfbfbe2ecf557a2c10.GoIbzpHebdM9ZcHarQ9A12Cp"
 ]
 
-# KEY STATUS
-key_status: Dict[str, Dict] = {
-    k: {"fail": 0, "cooldown": 0, "status": "active"}
-    for k in API_KEYS
-}
+key_status = {k: {"fail": 0, "cooldown": 0, "status": "active"} for k in API_KEYS}
 
 def auth(req: Request):
-    """Validate client API key"""
-    client_key = req.headers.get("Authorization", "").replace("Bearer ", "")
-    if client_key != MASTER_API_KEY:
+    key = req.headers.get("Authorization", "").replace("Bearer ", "")
+    if key != MASTER_API_KEY:
+        print(f"[AUTH] reject: {key[:8]} != {MASTER_API_KEY[:8]}")
         raise HTTPException(401, "Unauthorized")
+    print(f"[AUTH] ok")
 
-def pick_key():
+def pick_key() -> str:
     now = time.time()
-    valid = [k for k, v in key_status.items() 
-             if v["status"] != "dead" and v["cooldown"] < now]
-    
-    if valid:
-        return random.choice(valid)
-    
-    # Reset if all failed
-    for k in key_status:
-        key_status[k]["status"] = "active"
-        key_status[k]["fail"] = 0
-    
-    return random.choice(API_KEYS)
+    valid = [k for k, v in key_status.items() if v["status"] != "dead" and v["cooldown"] < now]
+    return random.choice(valid) if valid else random.choice(API_KEYS)
 
-def mark_fail(key: str):
-    key_status[key]["fail"] += 1
-    if key_status[key]["fail"] >= 3:
-        key_status[key]["cooldown"] = time.time() + 60
+def mark_fail(k):
+    key_status[k]["fail"] += 1
+    if key_status[k]["fail"] >= 3:
+        key_status[k]["cooldown"] = time.time() + 60
 
-def mark_ok(key: str):
-    key_status[key]["fail"] = 0
-    key_status[key]["status"] = "active"
+def mark_ok(k):
+    key_status[k]["fail"] = 0
+    key_status[k]["status"] = "active"
 
 @app.get("/")
 def root():
-    active = sum(1 for v in key_status.values() if v["status"] == "active")
-    return {
-        "status": "ok",
-        "mode": "openai-proxy",
-        "keys_total": len(API_KEYS),
-        "keys_active": active
-    }
+    return {"status": "ok", "keys": len(API_KEYS), "master_set": MASTER_API_KEY[:10]}
+
+# Simple route for testing
+@app.post("/test")
+def test(req: Request):
+    auth(req)
+    key = pick_key()
+    return {"key": key[:10], "url": BASE_URL}
 
 @app.api_route("/v1/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
 async def proxy(req: Request, path: str):
-    # Auth first
-    auth(req)
+    try:
+        auth(req)
+    except HTTPException:
+        raise
     
-    target_url = f"{BASE_URL}/v1/{path}"
+    target = f"{BASE_URL}/v1/{path}"
     body = await req.body()
-    
     headers = dict(req.headers)
     headers.pop("host", None)
     headers.pop("content-length", None)
+
+    print(f"[PROXY] {req.method} {path}")
     
-    # Try each key
-    tried = set()
-    for _ in range(len(API_KEYS)):
+    for i in range(len(API_KEYS)):
         key = pick_key()
-        if key in tried:
-            continue
-        tried.add(key)
-        
         headers["Authorization"] = f"Bearer {key}"
         
+        print(f"[TRY] key={key[:6]}...")
+        
         try:
-            async with httpx.AsyncClient(timeout=60) as client:
+            async with httpx.AsyncClient(timeout=120.0, follow_redirects=True) as client:
                 resp = await client.request(
                     method=req.method,
-                    url=target_url,
+                    url=target,
                     headers=headers,
                     content=body,
                     params=req.query_params
                 )
                 
+                print(f"[RESP] {key[:6]} <- {resp.status_code}")
+                
                 if resp.status_code < 400:
                     mark_ok(key)
                     return Response(
@@ -107,12 +93,9 @@ async def proxy(req: Request, path: str):
                     )
                 
                 mark_fail(key)
-                continue
-                
         except Exception as e:
+            print(f"[ERR] {key[:6]} {type(e).__name__}: {e}")
             mark_fail(key)
-            continue
-    
+
+    print("[FAIL] all keys failed")
     return JSONResponse({"error": "all keys failed"}, status_code=500)
-# rebuild
-# Testing build