superdup95 commited on
Commit
597023b
·
verified ·
1 Parent(s): b5f6bbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -152,9 +152,13 @@ def get_key_mistral_info(key):
152
  "key_type": "Mistral AI",
153
  "key_availability": True if key_avai else False,
154
  "has_quota": "",
 
155
  "models": ""}
156
  if key_avai:
157
- info_dict['has_quota'] = check_mistral_quota(key)
 
 
 
158
  info_dict['models'] = key_avai
159
  return info_dict
160
 
 
152
  "key_type": "Mistral AI",
153
  "key_availability": True if key_avai else False,
154
  "has_quota": "",
155
+ "limits": "",
156
  "models": ""}
157
  if key_avai:
158
+ quota_info = check_mistral_quota(key)
159
+ info_dict['has_quota'] = quota_info[0]
160
+ if quota_info[1]:
161
+ info_dict['limits'] = quota_info[1]
162
  info_dict['models'] = key_avai
163
  return info_dict
164