Update api_usage.py

api_usage.py

@@ -22,37 +22,25 @@ def get_subscription(key):
     #results = r.json()
     body_turbo = {"model": "gpt-3.5-turbo", "max_tokens": 1, "messages": [{'role':'user', 'content': ''}]}
     body_gpt4 = {"model": "gpt-4", "max_tokens": 1, "messages": [{'role':'user', 'content': ''}]}
+    gpt4_avai = check_gpt4_availability()
 
     if check_key_availability():
         rpm = ""
         org = ""
         quota = ""
-        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 gpt4_avai else body_turbo)
         result = r.json()
         if "id" in result:
             rpm = r.headers['x-ratelimit-limit-requests']
             org = r.headers['openai-organization']
-            quota = "yes"            
+            quota = check_key_type("gpt-4" if gpt4_avai else "gpt-3.5-turbo", rpm)         
         else:
             e = result["error"]["code"]
-            rpm = "";
-            org = "";
             quota = f"Error: {e}"
-        
-        #has_payment_method = results["has_payment_method"]
-        # hard_limit = results["hard_limit"]
-        #soft_limit_usd = results["soft_limit_usd"]
-        #hard_limit_usd = results["hard_limit_usd"]
-        #plan_title = results["plan"]["title"]
-        #plan_id = results["plan"]["id"]
-        #account_name = results["account_name"]
+            
         return {"organization": org,
                 "rpm": rpm,
-                "quota": quota}
-                #"has_payment_method": has_payment_method,
-                #"soft_limit_usd": soft_limit_usd,
-                #"hard_limit_usd": hard_limit_usd,
-                #"plan": plan_title + ", " + plan_id}               
+                "quota": quota}           
     else:
         return {"organization": "",
                 "rpm": "",
@@ -72,6 +60,14 @@ def get_subscription(key):
 #    else:
 #        return ""
 
+def check_key_type(model, rpm):
+    if rpm < rate_limit_per_model[model]:
+        return "yes | trial"
+    elif rpm == rate_limit_per_model[model]:
+        return "yes | pay"
+    else:
+        return "yes | pay, possibly big key."
+    
 def check_gpt4_availability():
     if check_key_availability():
         available_models = [model["root"] for model in openai.Model.list()["data"]]