bug: fix wrong var used

chhoumann · Nov 6, 2023 · 80fe081 · 80fe081 · vercel · Nov 6, 2023
1 parent 6c0a6bd
commit 80fe081
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ai/AIAssistant.ts b/src/ai/AIAssistant.ts
@@ -13,7 +13,7 @@ import { getModelMaxTokens } from "./getModelMaxTokens";
 export const getTokenCount = (text: string, model: Model) => {
 	// gpt-3.5-turbo-16k is a special case - it isn't in the library list yet. Same with gpt-4-1106-preview.
 	let m = model === "gpt-3.5-turbo-16k" ? "gpt-3.5-turbo" : model;
-	m = model === "gpt-4-1106-preview" ? "gpt-4" : model;
+	m = m === "gpt-4-1106-preview" ? "gpt-4" : m;
 
 	return encodingForModel(m).encode(text).length;
 };