Update api_usage.py

api_usage.py

@@ -7,9 +7,10 @@ import openai
 queryUrl = 'https://api.openai.com/v1/chat/completions'
 GPT_TYPES = ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k"]
 rate_limit_per_model = {
-    "gpt-3.5-turbo": 3500,
+    "gpt-3.5-turbo-new": 2000,
+    "gpt-3.5-turbo-old": 3500,
     "gpt-4": 200,
-    "gpt-4-32k": 10  # No actual clue, rare enough
+    "gpt-4-32k": 1000  # No actual clue, rare enough
 }
 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': ''}]}
@@ -60,12 +61,20 @@ def get_subscription(key):
 #        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"
+    if model == "gpt-3.5-turbo":
+        if rpm > rate_limit_per_model['gpt-3.5-turbo-old']:
+              return "yes | pay, possibly big key"
+        elif rpm == rate_limit_per_model['gpt-3.5-turbo-new'] or rpm == rate_limit_per_model['gpt-3.5-turbo-old']:
+            return "yes | pay"
+        else:
+            return "yes | trial"
     else:
-        return "yes | pay, possibly big key"
+        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():