From 276ac8d011e6cc53a6bbe2a7e94b31f38b9c4071 Mon Sep 17 00:00:00 2001 From: Danny Avila Date: Fri, 13 Feb 2026 11:21:53 -0500 Subject: [PATCH] =?UTF-8?q?=F0=9F=9B=B0=EF=B8=8F=20feat:=20Add=20Bedrock?= =?UTF-8?q?=20Parameter=20Settings=20for=20MoonshotAI=20and=20Z.AI=20Model?= =?UTF-8?q?s=20(#11783)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Introduced new model entries for 'moonshotai.kimi' and 'moonshotai.kimi-k2.5' in tokens.ts. - Updated parameterSettings.ts to include configurations for MoonshotAI and ZAI providers. - Enhanced schemas.ts by adding MoonshotAI and ZAI to the BedrockProviders enum for better integration. --- packages/api/src/utils/tokens.ts | 2 ++ packages/data-provider/src/parameterSettings.ts | 7 +++++++ packages/data-provider/src/schemas.ts | 2 ++ 3 files changed, 11 insertions(+) diff --git a/packages/api/src/utils/tokens.ts b/packages/api/src/utils/tokens.ts index 49f1640a7a..53d1ec8059 100644 --- a/packages/api/src/utils/tokens.ts +++ b/packages/api/src/utils/tokens.ts @@ -197,6 +197,8 @@ const moonshotModels = { 'moonshot.kimi-k2.5': 262144, 'moonshot.kimi-k2-thinking': 262144, 'moonshot.kimi-k2-0711': 131072, + 'moonshotai.kimi': 262144, + 'moonshotai.kimi-k2.5': 262144, }; const metaModels = { diff --git a/packages/data-provider/src/parameterSettings.ts b/packages/data-provider/src/parameterSettings.ts index b3baa7fb50..c9f85f9b98 100644 --- a/packages/data-provider/src/parameterSettings.ts +++ b/packages/data-provider/src/parameterSettings.ts @@ -952,6 +952,8 @@ export const paramSettings: Record = [`${EModelEndpoint.bedrock}-${BedrockProviders.Amazon}`]: bedrockGeneral, [`${EModelEndpoint.bedrock}-${BedrockProviders.DeepSeek}`]: bedrockGeneral, [`${EModelEndpoint.bedrock}-${BedrockProviders.Moonshot}`]: bedrockMoonshot, + [`${EModelEndpoint.bedrock}-${BedrockProviders.MoonshotAI}`]: bedrockMoonshot, + [`${EModelEndpoint.bedrock}-${BedrockProviders.ZAI}`]: bedrockGeneral, [EModelEndpoint.google]: googleConfig, }; @@ -1000,6 +1002,11 @@ export const presetSettings: Record< col1: bedrockMoonshotCol1, col2: bedrockMoonshotCol2, }, + [`${EModelEndpoint.bedrock}-${BedrockProviders.MoonshotAI}`]: { + col1: bedrockMoonshotCol1, + col2: bedrockMoonshotCol2, + }, + [`${EModelEndpoint.bedrock}-${BedrockProviders.ZAI}`]: bedrockGeneralColumns, [EModelEndpoint.google]: { col1: googleCol1, col2: googleCol2, diff --git a/packages/data-provider/src/schemas.ts b/packages/data-provider/src/schemas.ts index a429758c19..803d970477 100644 --- a/packages/data-provider/src/schemas.ts +++ b/packages/data-provider/src/schemas.ts @@ -101,7 +101,9 @@ export enum BedrockProviders { Meta = 'meta', MistralAI = 'mistral', Moonshot = 'moonshot', + MoonshotAI = 'moonshotai', StabilityAI = 'stability', + ZAI = 'zai', } export const getModelKey = (endpoint: EModelEndpoint | string, model: string) => {