From 2509495cdc66f474971970fee5fb801ac6dbb736 Mon Sep 17 00:00:00 2001 From: Davidlasky Date: Tue, 25 Mar 2025 18:01:23 -0500 Subject: [PATCH] try to add o1 as a vision model --- app/client/platforms/openai.ts | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/app/client/platforms/openai.ts b/app/client/platforms/openai.ts index c6f3fc425..a4ac638c9 100644 --- a/app/client/platforms/openai.ts +++ b/app/client/platforms/openai.ts @@ -196,9 +196,9 @@ export class ChatGPTApi implements LLMApi { let requestPayload: RequestPayload | DalleRequestPayload; const isDalle3 = _isDalle3(options.config.model); - const isO1OrO3 = - options.config.model.startsWith("o1") || - options.config.model.startsWith("o3"); + const isO1 = options.config.model.startsWith("o1"); + const isO3 = options.config.model.startsWith("o3"); + const isO1OrO3 = isO1 || isO3; if (isDalle3) { const prompt = getMessageTextContent( options.messages.slice(-1)?.pop() as any, @@ -244,7 +244,12 @@ export class ChatGPTApi implements LLMApi { // add max_tokens to vision model if (visionModel) { - requestPayload["max_tokens"] = Math.max(modelConfig.max_tokens, 4000); + if (isO1) { + requestPayload["max_completion_tokens"] = modelConfig.max_tokens; + + } else { + requestPayload["max_tokens"] = Math.max(modelConfig.max_tokens, 4000); + } } }