diff --git a/src/providers/cerebras/index.ts b/src/providers/cerebras/index.ts index 7270563cb..e0aa742ac 100644 --- a/src/providers/cerebras/index.ts +++ b/src/providers/cerebras/index.ts @@ -4,14 +4,20 @@ import { ProviderConfigs } from '../types'; import { cerebrasAPIConfig } from './api'; export const cerebrasProviderAPIConfig: ProviderConfigs = { - chatComplete: chatCompleteParams([ - 'frequency_penalty', - 'logit_bias', - 'logprobs', - 'presence_penalty', - 'parallel_tool_calls', - 'service_tier', - ]), + chatComplete: chatCompleteParams( + [ + 'frequency_penalty', + 'logit_bias', + 'logprobs', + 'presence_penalty', + 'parallel_tool_calls', + 'service_tier', + ], + {}, + { + max_completion_tokens: { param: 'max_tokens', min: 1 }, + } + ), api: cerebrasAPIConfig, responseTransforms: responseTransformers(CEREBRAS, { chatComplete: true, diff --git a/src/providers/groq/index.ts b/src/providers/groq/index.ts index 7c008f664..54f655af8 100644 --- a/src/providers/groq/index.ts +++ b/src/providers/groq/index.ts @@ -16,6 +16,7 @@ const GroqConfig: ProviderConfigs = { { service_tier: { param: 'service_tier', required: false }, reasoning_effort: { param: 'reasoning_effort', required: false }, + max_completion_tokens: { param: 'max_tokens', min: 1 }, } ), createTranscription: {},