🐛 fix: Fixed jump error when requesting desktop website

lobehub · May 28, 2024 · fa6e750 · fa6e750
1 parent 920a70d
commit fa6e750
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/src/const/url.ts b/src/const/url.ts
@@ -6,7 +6,21 @@ import { withBasePath } from '@/utils/basePath';
 import pkg from '../../package.json';
 import { INBOX_SESSION_ID } from './session';
 
-export const OFFICIAL_URL = 'https://chat-preview.lobehub.com/';
+const isVercelPreview = process.env.VERCEL === '1' && process.env.VERCEL_ENV !== 'production';
+
+const inVercelProduction = process.env.VERCEL === '1' && process.env.VERCEL_ENV === 'production';
+
+const vercelPreviewUrl = `https://${process.env.VERCEL_URL}`;
+
+const vercelProductionUrl = `https://${process.env.VERCEL_PROJECT_PRODUCTION_URL}`;
+
+export const siteUrl = inVercelProduction
+  ? vercelProductionUrl
+  : isVercelPreview
+    ? vercelPreviewUrl
+    : 'https://chat-preview.lobehub.com';
+
+export const OFFICIAL_URL = siteUrl;
 export const OFFICIAL_SITE = 'https://lobehub.com/';
 
 export const getCanonicalUrl = (path: string) => urlJoin(OFFICIAL_URL, path);

diff --git a/src/libs/agent-runtime/openrouter/index.ts b/src/libs/agent-runtime/openrouter/index.ts
@@ -9,7 +9,7 @@ export const LobeOpenRouterAI = LobeOpenAICompatibleFactory({
   baseURL: 'https://openrouter.ai/api/v1',
   constructorOptions: {
     defaultHeaders: {
-      'HTTP-Referer': 'https://chat-preview.lobehub.com',
+      'HTTP-Referer': 'siteUrl',
       'X-Title': 'Lobe Chat',
     },
   },

diff --git a/src/libs/agent-runtime/togetherai/index.ts b/src/libs/agent-runtime/togetherai/index.ts
@@ -10,7 +10,7 @@ export const LobeTogetherAI = LobeOpenAICompatibleFactory({
   baseURL: `${baseURL}/v1`,
   constructorOptions: {
     defaultHeaders: {
-      'HTTP-Referer': 'https://chat-preview.lobehub.com',
+      'HTTP-Referer': 'siteUrl',
       'X-Title': 'Lobe Chat',
     },
   },