From d3d9ead8f448e90a25bc8acd77b9c9eb154c2375 Mon Sep 17 00:00:00 2001 From: henceiusegentoo Date: Fri, 22 Sep 2023 02:38:21 +0200 Subject: [PATCH] Added /v1/accounts/credits --- api/handler.py | 5 ++++- api/main.py | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/api/handler.py b/api/handler.py index a4a9aa0..e6d5993 100644 --- a/api/handler.py +++ b/api/handler.py @@ -71,6 +71,9 @@ async def handle(incoming_request: fastapi.Request): if ban_reason: return await errors.error(403, f'Your NovaAI account has been banned. Reason: \'{ban_reason}\'.', 'Contact the staff for an appeal.') + if 'account/credits' in path: + return fastapi.responses.JSONResponse({'credits': user['credits']}) + costs = config['costs'] cost = costs['other'] @@ -150,7 +153,7 @@ async def handle(incoming_request: fastapi.Request): media_type = 'text/event-stream' if payload.get('stream', False) else 'application/json' - if payload['model'] not in models: + if (model := payload.get('model')) not in models and model is not None: return await errors.error(404, 'Model not found.', 'Check the model name and try again.') return fastapi.responses.StreamingResponse( diff --git a/api/main.py b/api/main.py index 9e6fd9e..8290360 100644 --- a/api/main.py +++ b/api/main.py @@ -66,4 +66,4 @@ async def root(): @app.route('/v1/{path:path}', methods=['GET', 'POST', 'PUT', 'DELETE', 'PATCH']) async def v1_handler(request: fastapi.Request): res = await handler.handle(incoming_request=request) - return res + return res \ No newline at end of file