Fix tiktoken mock: use class not instance, remove TESTING check

2026-01-11 09:17:49 -05:00 · 2026-01-11 09:17:49 -05:00 · cf63e5c6cd
commit cf63e5c6cd
parent 9fb7047866
2 changed files with 7 additions and 6 deletions
--- a/models.py
+++ b/models.py
@ -125,7 +125,7 @@ class Message(db.Model):
        if self.token_count is None:
            if self.is_base64_image():
                self.token_count = 0
-            elif not TIKTOKEN_AVAILABLE or os.environ.get("TESTING"):
+            elif not TIKTOKEN_AVAILABLE:
                # Fallback: estimate ~4 chars per token when tiktoken unavailable
                self.token_count = len(self.content) // 4 + 1
            else: