Spaces:
Running
Running
Update api_usage.py
Browse files- api_usage.py +5 -1
api_usage.py
CHANGED
@@ -200,7 +200,11 @@ def get_models(session, key, org: str = None):
|
|
200 |
headers = get_headers(key)
|
201 |
rq = session.get(f"{BASE_URL}/models", headers=headers, timeout=10)
|
202 |
avai_models = rq.json()
|
203 |
-
|
|
|
|
|
|
|
|
|
204 |
|
205 |
def check_key_availability(session, key):
|
206 |
try:
|
|
|
200 |
headers = get_headers(key)
|
201 |
rq = session.get(f"{BASE_URL}/models", headers=headers, timeout=10)
|
202 |
avai_models = rq.json()
|
203 |
+
try:
|
204 |
+
list_models = [model["id"] for model in avai_models["data"]] #[model["id"] for model in avai_models["data"] if model["id"] in GPT_TYPES]
|
205 |
+
except:
|
206 |
+
list_models = []
|
207 |
+
return list_models
|
208 |
|
209 |
def check_key_availability(session, key):
|
210 |
try:
|