Update api_usage.py

api_usage.py

@@ -377,7 +377,7 @@ def check_mistral_quota(key):
         url = 'https://api.mistral.ai/v1/chat/completions'
         headers = {'Authorization': f'Bearer {key}'}    
         data = {
-            'model': 'mistral-tiny',
+            'model': 'mistral-small-latest',
             'messages': [{ "role": "user", "content": "" }],
             'max_tokens': -1
         }
@@ -390,23 +390,30 @@ def check_mistral_quota(key):
 
 def check_key_replicate_availability(key):
     try:
+        quota = False
+        s = requests.Session()
         url = 'https://api.replicate.com/v1/account'
         headers = {'Authorization': f'Token {key}'}
         
-        rq = requests.get(url, headers=headers)
+        rq = s.get(url, headers=headers)
         info = rq.json()
         if rq.status_code == 401:
             return False, "", ""
             
         url = 'https://api.replicate.com/v1/hardware'
-        rq = requests.get(url, headers=headers)
+        rq = s.get(url, headers=headers)
         result = rq.json()
         hardware = []
         if result:
             hardware = [res['name'] for res in result]
-        return True, info, hardware
+        url = 'https://api.replicate.com/v1/predictions'
+        data = {"version": "5c7d5dc6dd8bf75c1acaa8565735e7986bc5b66206b55cca93cb72c9bf15ccaa", "input": {}}
+        rq = s.post(url, headers=headers, json=data)
+        if rq.status_code == 422: # 422 have quota, 402 out of quota
+            quota = True
+        return True, info, quota, hardware
     except:
-        return "Unknown", "", "Error while making request"
+        return "Unknown", "", "", "Error while making request"
 
 def check_key_aws_availability(key):
     access_id = key.split(':')[0]