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:
|