From 6df0b02e49ebd0d17a32c95fb186e568a99c1d9d Mon Sep 17 00:00:00 2001 From: kangfenmao Date: Fri, 7 Feb 2025 22:37:18 +0800 Subject: [PATCH] fix: improve reasoning effort handling for different providers --- src/renderer/src/providers/OpenAIProvider.ts | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/src/renderer/src/providers/OpenAIProvider.ts b/src/renderer/src/providers/OpenAIProvider.ts index 6f82579a..0299ea37 100644 --- a/src/renderer/src/providers/OpenAIProvider.ts +++ b/src/renderer/src/providers/OpenAIProvider.ts @@ -142,9 +142,17 @@ export default class OpenAIProvider extends BaseProvider { } private getReasoningEffort(assistant: Assistant, model: Model) { - if (isReasoningModel(model)) return assistant?.settings?.reasoning_effort + if (this.provider.id === 'groq') { + return {} + } - return undefined + if (isReasoningModel(model)) { + return { + reasoning_effort: assistant?.settings?.reasoning_effort + } + } + + return {} } async completions({ messages, assistant, onChunk, onFilterMessages }: CompletionsParams): Promise { @@ -192,7 +200,7 @@ export default class OpenAIProvider extends BaseProvider { max_tokens: maxTokens, keep_alive: this.keepAliveTime, stream: isSupportStreamOutput(), - reasoning_effort: this.getReasoningEffort(assistant, model), + ...this.getReasoningEffort(assistant, model), ...getOpenAIWebSearchParams(assistant, model), ...this.getProviderSpecificParameters(model), ...this.getCustomParameters(assistant)