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); + } } }