diff --git a/api/server/services/Files/Audio/STTService.js b/api/server/services/Files/Audio/STTService.js index 3a7a51bd8d..abd9c2272b 100644 --- a/api/server/services/Files/Audio/STTService.js +++ b/api/server/services/Files/Audio/STTService.js @@ -186,7 +186,7 @@ class STTService { /** * Prepares the request for the OpenAI STT provider. - * @param {Object} sttSchema - The STT schema for OpenAI. + * @param {Object} sttSchema - The STT schema for OpenAI (includes optional language and extraParams). * @param {Stream} audioReadStream - The audio data to be transcribed. * @param {Object} audioFile - The audio file object (unused in OpenAI provider). * @param {string} language - The language code for the transcription. @@ -206,8 +206,12 @@ class STTService { data.language = validLanguage; } - if (sttSchema?.extraParams) { - Object.assign(data, sttSchema.extraParams); + if (sttSchema.extraParams) { + const reservedFields = new Set(['file', 'model', 'language']); + const safeParams = Object.fromEntries( + Object.entries(sttSchema.extraParams).filter(([key]) => !reservedFields.has(key)), + ); + Object.assign(data, safeParams); } const headers = { diff --git a/packages/data-provider/src/config.ts b/packages/data-provider/src/config.ts index fb89c7c486..7ba0c99249 100644 --- a/packages/data-provider/src/config.ts +++ b/packages/data-provider/src/config.ts @@ -478,10 +478,14 @@ const sttOpenaiSchema = z.object({ url: z.string().optional(), apiKey: z.string(), model: z.string(), - language: z.string().optional(), - extraParams: z.record(z.unknown()).optional(), + language: z + .string() + .regex(/^[a-z]{2}(-[a-z]{2})?$/) + .optional(), + extraParams: z.record(z.union([z.string(), z.number(), z.boolean()])).optional(), }); +/** Note: language and extraParams are only supported for the OpenAI provider. */ const sttAzureOpenAISchema = z.object({ instanceName: z.string(), apiKey: z.string(),