aaaa
unknown
diff
9 months ago
2.5 kB
54
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