aaaa
unknown
diff
3 months ago
2.5 kB
51
No Index
diff --git a/src/modules/llms/server/gemini/gemini.router.ts b/src/modules/llms/server/gemini/gemini.router.ts index 666a7c1d..f65fbab6 100644 --- a/src/modules/llms/server/gemini/gemini.router.ts +++ b/src/modules/llms/server/gemini/gemini.router.ts @@ -21,8 +21,7 @@ const DEFAULT_GEMINI_HOST = 'https://generativelanguage.googleapis.com'; // Mappers export function geminiAccess(access: GeminiAccessSchema, modelRefId: string | null, apiPath: string, useV1Alpha: boolean): { headers: HeadersInit, url: string } { - const geminiHost = fixupHost(access.geminiHost || DEFAULT_GEMINI_HOST, apiPath); + const geminiHost = fixupHost(access.geminiHost || env.GEMINI_API_HOST || DEFAULT_GEMINI_HOST, apiPath); let geminiKey = access.geminiKey || env.GEMINI_API_KEY || ''; // multi-key with random selection - https://github.com/enricoros/big-AGI/issues/653 @@ -44,7 +43,6 @@ export function geminiAccess(access: GeminiAccessSchema, modelRefId: string | nu // [Gemini, 2025-01-23] CoT support - requires `v1alpha` Gemini API if (useV1Alpha) apiPath = apiPath.replaceAll('v1beta', 'v1alpha'); return { headers: { 'Content-Type': 'application/json', diff --git a/src/modules/llms/server/openai/openai.router.ts b/src/modules/llms/server/openai/openai.router.ts index 5e11e7fe..51049782 100644 --- a/src/modules/llms/server/openai/openai.router.ts +++ b/src/modules/llms/server/openai/openai.router.ts @@ -374,7 +374,7 @@ export function openAIAccess(access: OpenAIAccessSchema, modelRefId: string | nu case 'deepseek': // https://platform.deepseek.com/api-docs/ const deepseekKey = access.oaiKey || env.DEEPSEEK_API_KEY || ''; - const deepseekHost = fixupHost(access.oaiHost || DEFAULT_DEEPSEEK_HOST, apiPath); + const deepseekHost = fixupHost(access.oaiHost || env.DEEPSEEK_API_HOST || DEFAULT_DEEPSEEK_HOST, apiPath); if (!deepseekKey || !deepseekHost) throw new Error('Missing Deepseek API Key or Host. Add it on the UI (Models Setup) or server side (your deployment).'); diff --git a/src/server/env.mjs b/src/server/env.mjs index ccecd58f..f4a2b9e8 100644 --- a/src/server/env.mjs +++ b/src/server/env.mjs @@ -31,10 +31,11 @@ export const env = createEnv({ // LLM: Deepseek AI DEEPSEEK_API_KEY: z.string().optional(), + DEEPSEEK_API_HOST: z.string().optional(), // LLM: Google AI's Gemini GEMINI_API_KEY: z.string().optional(), + GEMINI_API_HOST: z.string().optional(), // LLM: Groq GROQ_API_KEY: z.string().optional(),
Editor is loading...
Leave a Comment