fix outi bug

UniversityOfHelsinkiCS · Sep 28, 2024 · db75e1e · db75e1e
1 parent c6b3bf9
commit db75e1e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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)
   }