openrouter fix

This commit is contained in:
based
2024-11-06 06:25:14 +10:00
parent 6f664b9964
commit 3f42982290
2 changed files with 4 additions and 1 deletions
+3
View File
@@ -192,6 +192,9 @@ async def invoke_model(key: APIKey, session, region, model):
resp = await response.json()
if resp['message'] and 'Malformed input request' in resp['message']:
key.models[region].append(model_name)
elif resp['message'] and 'Operation not' in resp['message']:
key.useless = True
key.useless_reasons.append('Quarantined Key')
async def retrieve_models(key: APIKey, session, region):
+1 -1
View File
@@ -15,7 +15,7 @@ async def check_openrouter(key: APIKey, session):
key.bought_credits = not data['is_free_tier']
key.limit_reached = key.credit_limit is not None and key.usage >= key.credit_limit
key.rpm = int(data['rate_limit']['requests']) // int(data['rate_limit']['interval'].replace('s', '')) * 60
key.balance = await get_key_balance(key, session)
# key.balance = await get_key_balance(key, session)
return True