From d7b3a56c3d858799a8aa1c5c2ae81ba5b57b36cb Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Tue, 7 Jan 2025 20:07:41 +0200
Subject: [PATCH] chore: await before returning

---
 public/scripts/extensions/shared.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/public/scripts/extensions/shared.js b/public/scripts/extensions/shared.js
index 9c39a18d1..a2c15ba41 100644
--- a/public/scripts/extensions/shared.js
+++ b/public/scripts/extensions/shared.js
@@ -251,7 +251,7 @@ export async function countWebLlmTokens(text) {
         return response;
     } catch (error) {
         // Fallback to using current model's tokenizer
-        return getTokenCountAsync(text);
+        return await getTokenCountAsync(text);
     }
 }