aaaa

 avatar
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