From db75e1e419a2481a6ccc07f367091bdd01ba0436 Mon Sep 17 00:00:00 2001
From: Matti Luukkainen <mluukkai@iki.fi>
Date: Sat, 28 Sep 2024 14:07:23 +0300
Subject: [PATCH] fix outi bug

---
 src/server/routes/openai.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/server/routes/openai.ts b/src/server/routes/openai.ts
index 0f20927..0c7c740 100644
--- a/src/server/routes/openai.ts
+++ b/src/server/routes/openai.ts
@@ -98,7 +98,11 @@ openaiRouter.post('/stream', upload.single('file'), async (r, res) => {
     (tokenCount / DEFAULT_TOKEN_LIMIT) * 100
   )
 
-  if (tokenCount > 0.1 * DEFAULT_TOKEN_LIMIT && !userConsent) {
+  if (
+    model !== FREE_MODEL &&
+    tokenCount > 0.1 * DEFAULT_TOKEN_LIMIT &&
+    !userConsent
+  ) {
     return res.status(201).json({
       tokenConsumtionWarning: true,
       message: `You are about to use ${tokenUsagePercentage}% of your monthly CurreChat usage`,
@@ -132,7 +136,7 @@ openaiRouter.post('/stream', upload.single('file'), async (r, res) => {
 
   if (courseId) {
     await incrementCourseUsage(userToCharge, courseId, tokenCount)
-  } else if (model === 'gpt-4') {
+  } else if (model !== FREE_MODEL) {
     await incrementUsage(userToCharge, tokenCount)
   }