Minimal proxy test

app.py

@@ -1,119 +1,48 @@
 import os
-import time
-import random
 import httpx
-from typing import Dict
-
-from fastapi import FastAPI, Request, HTTPException
+from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse, StreamingResponse, Response
 
 app = FastAPI()
 
-# CONFIG
-MASTER_API_KEY = os.getenv("MASTER_API_KEY", "changeme")
 BASE_URL = os.getenv("BASE_URL", "https://ollama.com")
 
-# LOAD KEYS - HARDCODED FOR TESTING
-def load_keys(prefix="OLLAMA_KEY_"):
-    # Hardcoded keys - please update!
-    return [
-        "8ca25de51e554c099962b78b7ce0c9e9.Mp5dnqctR2zzq3g-NO_M-cjW",
-        "dbd1d0c534964684a6d4678348ab8d30.ieDfmSYVnf0MmTjR-AIdNrW9",
-        "37e81a6be4104fbfbfbe2ecf557a2c10.GoIbzpHebdM9ZcHarQ9A12Cp"
-    ]
-
-API_KEYS = load_keys()
-print(f"[INIT] Keys: {len(API_KEYS)}")
-
-key_status = {k: {"fail": 0, "cooldown": 0, "status": "active"} for k in API_KEYS}
-
-# AUTH - ACCEPT ALL
-def auth(req: Request):
-    return
-
-def pick_key():
-    now = time.time()
-    valid = [k for k, v in key_status.items() if v["status"] != "dead" and v["cooldown"] < now]
-    return random.choice(valid) if valid else None
-
-def mark_limit(k):
-    key_status[k]["cooldown"] = time.time() + 60
-    key_status[k]["status"] = "limited"
-
-def mark_dead(k):
-    key_status[k]["status"] = "dead"
-
-def mark_ok(k):
-    key_status[k]["fail"] = 0
-    key_status[k]["status"] = "active"
+@app.get("/")
+def root():
+    return {"status": "ok", "base_url": BASE_URL}
 
-@app.api_route("/v1/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
+@app.api_route("/v1/{path:path}", methods=["GET", "POST"])
 async def proxy(req: Request, path: str):
-    print(f"[PROXY] path={path}, auth done")
-    auth(req)
-
+    print(f"[PROXY] request to {path}")
+    
     target_url = f"{BASE_URL}/v1/{path}"
     body = await req.body()
-
+    
     headers = dict(req.headers)
     headers.pop("host", None)
     headers.pop("content-length", None)
-
-    for _ in range(len(API_KEYS)):
-        key = pick_key()
-        if not key:
-            return JSONResponse({"error": "all keys failed"}, status_code=503)
-
-        headers["Authorization"] = f"Bearer {key}"
-        print(f"[PROXY] key={key[:6]}.. url={target_url}")
-
-        try:
-            async with httpx.AsyncClient(timeout=60) as client:
-                test = await client.request(
-                    method=req.method,
-                    url=target_url,
-                    headers=headers,
-                    content=body,
-                    params=req.query_params
-                )
-
-                if test.status_code >= 400:
-                    if test.status_code == 429:
-                        mark_limit(key)
-                    elif test.status_code >= 500:
-                        key_status[key]["fail"] += 1
-                        if key_status[key]["fail"] >= 3:
-                            mark_dead(key)
-                        continue
-                    mark_ok(key)
-                    return Response(content=test.content, status_code=test.status_code, headers=dict(test.headers))
-
-                mark_ok(key)
-
-                stream = await client.stream(
-                    method=req.method,
-                    url=target_url,
-                    headers=headers,
-                    content=body,
-                    params=req.query_params
-                )
-
-                return StreamingResponse(
-                    stream.aiter_bytes(),
-                    media_type=test.headers.get("content-type", "application/json")
-                )
-
-        except Exception as e:
-            key_status[key]["fail"] += 1
-            if key_status[key]["fail"] >= 3:
-                mark_dead(key)
-            print(f"[ERROR] {type(e).__name__}: {e}")
-            continue
-
-    return JSONResponse({"error": "all keys failed"}, status_code=500)
-
-@app.get("/")
-def root():
-    return {"status": "ok", "keys": len(API_KEYS)}
-
-# Rebuild test
+    
+    # Hardcoded key
+    headers["Authorization"] = "Bearer 37e81a6be4104fbfbfbe2ecf557a2c10.GoIbzpHebdM9ZcHarQ9A12Cp"
+    
+    print(f"[PROXY] calling {target_url}")
+    
+    try:
+        async with httpx.AsyncClient(timeout=60) as client:
+            resp = await client.request(
+                method=req.method,
+                url=target_url,
+                headers=headers,
+                content=body,
+                params=req.query_params
+            )
+            print(f"[PROXY] got {resp.status_code}")
+            
+            return Response(
+                content=resp.content,
+                status_code=resp.status_code,
+                headers=dict(resp.headers)
+            )
+    except Exception as e:
+        print(f"[ERROR] {e}")
+        return JSONResponse({"error": str(e)}, status_code=500)