From fa63e2caf2f47d6fc7d932e58ba05b8fb15a0973 Mon Sep 17 00:00:00 2001 From: bjc Date: Mon, 7 Oct 2024 17:57:17 -0700 Subject: [PATCH] refactor(llm): update OpenAI LLM implementation and add support for o1-preview model --- app/lib/.server/llm/openai-llm.ts | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/lib/.server/llm/openai-llm.ts b/app/lib/.server/llm/openai-llm.ts index 621b376..f1ccd25 100644 --- a/app/lib/.server/llm/openai-llm.ts +++ b/app/lib/.server/llm/openai-llm.ts @@ -28,11 +28,11 @@ export class OpenAILLM implements LLM { } const openai = createOpenAI({ apiKey: this.apiKey, compatibility: 'strict' }); - type model_name_t = 'gpt-4o' | 'o1-mini'; + type model_name_t = 'gpt-4o' | 'o1-mini' | 'o1-preview'; const model_name: model_name_t = process.env.OPEN_AI_MODEL as model_name_t; const model = openai(model_name); - if (model_name === 'o1-mini') { + if (model_name === 'o1-mini' || model_name === 'o1-preview') { const o1sysmessage: Message = { role: 'user', content: this.getPrompts().getSystemPrompt()