Commit
•
ef0da58
1
Parent(s):
9fba7a1
Update api_usage.py
Browse files- api_usage.py +1 -1
api_usage.py
CHANGED
|
@@ -61,7 +61,7 @@ def get_subscription(key, available_models):
|
|
| 61 |
def get_org_name(key):
|
| 62 |
headers=get_headers(key)
|
| 63 |
r = requests.post(f"{BASE_URL}/images/generations", headers=headers)
|
| 64 |
-
return r.headers
|
| 65 |
|
| 66 |
def check_key_type(model, rpm):
|
| 67 |
if rpm < RATE_LIMIT_PER_MODEL[model]:
|
|
|
|
| 61 |
def get_org_name(key):
|
| 62 |
headers=get_headers(key)
|
| 63 |
r = requests.post(f"{BASE_URL}/images/generations", headers=headers)
|
| 64 |
+
return r.headers.get("openai-organization", "")
|
| 65 |
|
| 66 |
def check_key_type(model, rpm):
|
| 67 |
if rpm < RATE_LIMIT_PER_MODEL[model]:
|