feat: add gpt-4-1106-preview to AI Assistant (128k context, fresher k…

…nowledge, cheapter than gpt-4!)
chhoumann · Nov 6, 2023 · f404425 · f404425 · vercel · Nov 6, 2023
1 parent feae0a8
commit f404425
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/ai/AIAssistant.ts b/src/ai/AIAssistant.ts
@@ -11,8 +11,9 @@ import { makeNoticeHandler } from "./makeNoticeHandler";
 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
-	const m = model === "gpt-3.5-turbo-16k" ? "gpt-3.5-turbo" : 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;
 
 	return encodingForModel(m).encode(text).length;
 };

diff --git a/src/ai/models.ts b/src/ai/models.ts
@@ -1,4 +1,4 @@
-export const models = ["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4", "gpt-4-32k", "text-davinci-003"] as const;
+export const models = ["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4", "gpt-4-1106-preview", "gpt-4-32k", "text-davinci-003"] as const;
 export const models_and_ask_me = [...models, "Ask me"] as const;
 export type Model = typeof models[number];
 export type Models_And_Ask_Me = typeof models_and_ask_me[number];