From 1da39e5242ea7e44dd673933149a0a7b6302ddfe Mon Sep 17 00:00:00 2001 From: Game_Time <108236317+RayBytes@users.noreply.github.com> Date: Sat, 12 Aug 2023 21:05:53 +0500 Subject: [PATCH] a few more nice changes --- api/provider_auth.py | 13 ++++++++----- api/streaming.py | 6 ++---- 2 files changed, 10 insertions(+), 9 deletions(-) diff --git a/api/provider_auth.py b/api/provider_auth.py index bc499d7..5b19ec7 100644 --- a/api/provider_auth.py +++ b/api/provider_auth.py @@ -3,11 +3,14 @@ import asyncio async def invalidate_key(provider_and_key: str) -> none: - """Invalidates a key stored in the secret/ folder by storing it in the associated .invalid.txt file. -The schmea in which should be passed is: -, e.g. -closed4>sk-... -""" + """ + + Invalidates a key stored in the secret/ folder by storing it in the associated .invalid.txt file. + The schmea in which should be passed is: + , e.g. + closed4>sk-... + + """ if not provider_and_key: return diff --git a/api/streaming.py b/api/streaming.py index 295b06b..a1c7402 100644 --- a/api/streaming.py +++ b/api/streaming.py @@ -96,12 +96,11 @@ async def stream( webhook = dhooks.Webhook(os.getenv('DISCORD_WEBHOOK__API_ISSUE')) webhook.send(content=f'API Issue: **`{exc}`**\nhttps://i.imgflip.com/7uv122.jpg') - error = await errors.yield_error( + yield await errors.yield_error( 500, 'Sorry, the API has no working keys anymore.', 'The admins have been messaged automatically.' ) - yield error return for k, v in target_request.get('headers', {}).items(): @@ -180,12 +179,11 @@ async def stream( except Exception as exc: if 'Connection closed' in str(exc): - error = await errors.yield_error( + yield await errors.yield_error( 500, 'Sorry, there was an issue with the connection.', 'Please first check if the issue on your end. If this error repeats, please don\'t heistate to contact the staff!.' ) - yield error return break