From 2ef450a43cc5857578fdbaecc4048e856fefbcce Mon Sep 17 00:00:00 2001 From: David Zhang Date: Sat, 8 Feb 2025 19:38:21 -0800 Subject: [PATCH] simplified using other oai models --- src/ai/providers.ts | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) diff --git a/src/ai/providers.ts b/src/ai/providers.ts index c4eced0..6ce3d2b 100644 --- a/src/ai/providers.ts +++ b/src/ai/providers.ts @@ -13,20 +13,14 @@ const openai = createOpenAI({ baseURL: process.env.OPENAI_ENDPOINT || 'https://api.openai.com/v1', } as CustomOpenAIProviderSettings); -const isCustomEndpoint = - process.env.OPENAI_ENDPOINT && - process.env.OPENAI_ENDPOINT !== 'https://api.openai.com/v1'; -const customModel = process.env.OPENAI_MODEL; +const customModel = process.env.OPENAI_MODEL || 'o3-mini'; // Models -export const o3MiniModel = openai( - isCustomEndpoint && customModel ? customModel : 'o3-mini', - { - reasoningEffort: 'medium', - structuredOutputs: true, - }, -); +export const o3MiniModel = openai(customModel, { + reasoningEffort: customModel.startsWith('o') ? 'medium' : undefined, + structuredOutputs: true, +}); const MinChunkSize = 140; const encoder = getEncoding('o200k_base');