free trial logic fixes

continuedev · Jan 20, 2025 · 442b863 · 442b863
1 parent 709e597
commit 442b863
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 23 deletions.
diff --git a/core/config/onboarding.ts b/core/config/onboarding.ts
@@ -9,7 +9,7 @@ export const LOCAL_ONBOARDING_CHAT_MODEL = "llama3.1:8b";
 export const LOCAL_ONBOARDING_CHAT_TITLE = "Llama 3.1 8B";
 
 /**
- * We set the "best" chat + autocopmlete models by default
+ * We set the "best" chat + autocomplete models by default
  * whenever a user doesn't have a config.json
  */
 export function setupBestConfig(

diff --git a/extensions/vscode/package-lock.json b/extensions/vscode/package-lock.json
diff --git a/gui/src/components/Layout.tsx b/gui/src/components/Layout.tsx
@@ -151,11 +151,7 @@ const Layout = () => {
     "incrementFtc",
     async () => {
       const u = getLocalStorage("ftc");
-      if (u) {
-        setLocalStorage("ftc", u + 1);
-      } else {
-        setLocalStorage("ftc", 1);
-      }
+      setLocalStorage("ftc", (u ?? 0) + 1);
     },
     [],
   );

diff --git a/gui/src/pages/gui/Chat.tsx b/gui/src/pages/gui/Chat.tsx
@@ -247,21 +247,18 @@ export function Chat() {
       editorToClearOnSend?: Editor,
     ) => {
       if (defaultModel?.provider === "free-trial") {
-        const u = getLocalStorage("ftc");
-        if (u) {
-          setLocalStorage("ftc", u + 1);
-
-          if (u >= FREE_TRIAL_LIMIT_REQUESTS) {
-            onboardingCard.open("Best");
-            posthog?.capture("ftc_reached");
-            ideMessenger.ide.showToast(
-              "info",
-              "You've reached the free trial limit. Please configure a model to continue.",
-            );
-            return;
-          }
-        } else {
-          setLocalStorage("ftc", 1);
+        const u = getLocalStorage("ftc") ?? 0;
+        setLocalStorage("ftc", u + 1);
+        if (u === FREE_TRIAL_LIMIT_REQUESTS) {
+          posthog?.capture("ftc_reached");
+        }
+        if (u >= FREE_TRIAL_LIMIT_REQUESTS) {
+          onboardingCard.open("Best");
+          ideMessenger.ide.showToast(
+            "info",
+            "You've reached the free trial limit. Please configure a model to continue.",
+          );
+          return;
         }
       }