Update api_usage.py

api_usage.py

@@ -28,7 +28,12 @@ def get_subscription(key):
         org = ""
         quota = ""
         try:
-            r = requests.post(queryUrl, headers=headers, json=body_gpt4 if check_gpt4_availability() else body_turbo)            
+            r = requests.post(queryUrl, headers=headers, json=body_gpt4 if check_gpt4_availability() else body_turbo)
+            if (r.headers['x-ratelimit-limit-requests']):
+                rpm = r.headers['x-ratelimit-limit-requests']
+                org = r.headers['openai-organization']
+                quota = ""
+                
         except Exception as e:
             error_message = str(e)
             if "You exceeded your current quota" in error_message:
@@ -43,9 +48,7 @@ def get_subscription(key):
                 rpm = "";
                 org = "";
                 quota = f"Unexpected Error at check_key: {error_message}"
-        rpm = r.headers['x-ratelimit-limit-requests']
-        org = r.headers['openai-organization']
-        quota = ""
+        
         #has_payment_method = results["has_payment_method"]
         # hard_limit = results["hard_limit"]
         #soft_limit_usd = results["soft_limit_usd"]