From 3f42982290e7209e265a47ec649e7f60901ca5d8 Mon Sep 17 00:00:00 2001 From: based Date: Wed, 6 Nov 2024 06:25:14 +1000 Subject: [PATCH] openrouter fix --- AWSAsync.py | 3 +++ OpenRouter.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/AWSAsync.py b/AWSAsync.py index 4a73ed3..38d9365 100644 --- a/AWSAsync.py +++ b/AWSAsync.py @@ -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): diff --git a/OpenRouter.py b/OpenRouter.py index ffd035a..559df80 100644 --- a/OpenRouter.py +++ b/OpenRouter.py @@ -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