Elysiadev11 commited on
Commit
574c72e
·
verified ·
1 Parent(s): f1c283c

Update proxy_cerebras.py

Browse files
Files changed (1) hide show
  1. 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": key_status
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
  # =====================================================