fix: Added if statement to check that the model is included

casual-simulation · Nov 1, 2024 · 1494b82 · 1494b82
1 parent 8ea80f6
commit 1494b82
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/aux-records/AIController.ts b/src/aux-records/AIController.ts
@@ -612,12 +612,14 @@ export class AIController {
 
             if (allowedFeatures.ai.chat.allowedModels) {
                 const allowedModels = allowedFeatures.ai.chat.allowedModels;
-                return {
-                    success: false,
-                    errorCode: 'not_authorized',
-                    errorMessage:
-                        'The subscription does not permit the given model for AI Chat features.',
-                };
+                if (!allowedModels.includes(model)) {
+                    return {
+                        success: false,
+                        errorCode: 'not_authorized',
+                        errorMessage:
+                            'The subscription does not permit the given model for AI Chat features.',
+                    };
+                }
             }
 
             let maxTokens: number = undefined;