From 6bb44b95b04d6d59e44b64d21f7aa0bc2a11cfc5 Mon Sep 17 00:00:00 2001
From: SillyLossy <titan0599@gmail.com>
Date: Tue, 16 May 2023 00:53:33 +0300
Subject: [PATCH] Fix OAI key usage

---
 server.js | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/server.js b/server.js
index 7aee9bb00..2308c2a16 100644
--- a/server.js
+++ b/server.js
@@ -2437,7 +2437,13 @@ app.post("/openai_bias", jsonParser, async function (request, response) {
 // Shamelessly stolen from Agnai
 app.post("/openai_usage", jsonParser, async function (request, response) {
     if (!request.body) return response.sendStatus(400);
-    const key = request.body.key;
+    const key = readSecret(SECRET_KEYS.OPENAI);
+
+    if (!key) {
+        console.warn('Get key usage failed: Missing OpenAI API key.');
+        return response.sendStatus(401);
+    }
+
     const api_url = new URL(request.body.reverse_proxy || api_openai).toString();
 
     const headers = {