diff --git a/packages/proxy/schema/secrets.ts b/packages/proxy/schema/secrets.ts index 2056ff2a..923bceff 100644 --- a/packages/proxy/schema/secrets.ts +++ b/packages/proxy/schema/secrets.ts @@ -8,8 +8,8 @@ export const BaseMetadataSchema = z excludeDefaultModels: z.boolean().nullish(), additionalHeaders: z.record(z.string(), z.string()).nullish(), supportsStreaming: z.boolean().default(true), - custom_model: z.string().nullish(), - custom_system_prompt: z.string().nullish(), + overrideModel: z.string().nullish(), + overrideSystemPrompt: z.string().nullish(), }) .strict(); diff --git a/packages/proxy/src/proxy.ts b/packages/proxy/src/proxy.ts index 65dbfbb2..a80645e3 100644 --- a/packages/proxy/src/proxy.ts +++ b/packages/proxy/src/proxy.ts @@ -1182,12 +1182,12 @@ async function fetchModelLoop( const secrets = await getApiSecrets(model); const customModelOverride = - secrets.length > 0 && secrets[0].metadata?.custom_model - ? String(secrets[0].metadata.custom_model) + secrets.length > 0 && secrets[0].metadata?.overrideModel + ? String(secrets[0].metadata.overrideModel) : null; const customSystemPromptContent = - secrets.length > 0 && secrets[0].metadata?.custom_system_prompt - ? String(secrets[0].metadata.custom_system_prompt) + secrets.length > 0 && secrets[0].metadata?.overrideSystemPrompt + ? String(secrets[0].metadata.overrideSystemPrompt) : null; if (customModelOverride) {