Spaces:
Sleeping
Sleeping
Update proxy_cerebras.py
Browse files- proxy_cerebras.py +13 -2
proxy_cerebras.py
CHANGED
|
@@ -98,13 +98,24 @@ def mark_ok(k):
|
|
| 98 |
# =====================================================
|
| 99 |
@app.get("/")
|
| 100 |
async def root():
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 101 |
return {
|
| 102 |
"status": "ok",
|
| 103 |
"keys": len(OLLAMA_KEYS),
|
| 104 |
-
"detail":
|
| 105 |
}
|
| 106 |
|
| 107 |
-
|
| 108 |
# =====================================================
|
| 109 |
# /v1/models
|
| 110 |
# =====================================================
|
|
|
|
| 98 |
# =====================================================
|
| 99 |
@app.get("/")
|
| 100 |
async def root():
|
| 101 |
+
safe = {}
|
| 102 |
+
|
| 103 |
+
for k, v in key_status.items():
|
| 104 |
+
masked = k[:4] + "****" + k[-4:]
|
| 105 |
+
safe[masked] = {
|
| 106 |
+
"index": v["index"],
|
| 107 |
+
"healthy": v["healthy"],
|
| 108 |
+
"busy": v["busy"],
|
| 109 |
+
"success": v["success"],
|
| 110 |
+
"fail": v["fail"],
|
| 111 |
+
}
|
| 112 |
+
|
| 113 |
return {
|
| 114 |
"status": "ok",
|
| 115 |
"keys": len(OLLAMA_KEYS),
|
| 116 |
+
"detail": safe
|
| 117 |
}
|
| 118 |
|
|
|
|
| 119 |
# =====================================================
|
| 120 |
# /v1/models
|
| 121 |
# =====================================================
|