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) => {