o1Mini

Technoculture · Sep 13, 2024 · 5c0fb77 · 5c0fb77
1 parent 6881e60
commit 5c0fb77
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/app/api/chatmodel/[chatid]/route.ts b/src/app/api/chatmodel/[chatid]/route.ts
@@ -85,7 +85,8 @@ export async function POST(
     chatType === "ella" ? ellaPrompt : systemPrompt,
   );
 
-  const model = OPEN_AI_MODELS.gpt4Turbo;
+  // const model = OPEN_AI_MODELS.gpt4Turbo;
+  const model = OPEN_AI_MODELS.o1Mini;
 
   const { stream, handlers } = LangChainStream({
     onCompletion: async (fullResponse: string) => {

diff --git a/src/utils/apiHelper.ts b/src/utils/apiHelper.ts
@@ -30,6 +30,8 @@ export const OPEN_AI_MODELS = {
   gptTurbo: "gpt-3.5-turbo" as const,
   gptTurbo16k: "gpt-3.5-turbo-16k" as const,
   gpt4Turbo: "gpt-4-turbo" as const,
+  o1Mini: "01-mini" as const
+  //TODO: gpt-40-mini
 };
 
 const TOKEN_SIZE = {