@lobehub/chat 0.142.1 → 0.142.3

package/.env.example

@@ -98,6 +98,7 @@ OPENAI_API_KEY=sk-xxxxxxxxx
 ########################################
 
 #OPENROUTER_API_KEY=xxxxxxxxxxxxxxxxxxxxxxxxxxxxx
+#OPENROUTER_CUSTOM_MODELS=model1,model2,model3
 
 ########################################
 ######### 01.AI Service ##########

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 0.142.3](https://github.com/lobehub/lobe-chat/compare/v0.142.2...v0.142.3)
+
+<sup>Released on **2024-03-26**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Pin `next-auth` temporary to fix build error.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Pin `next-auth` temporary to fix build error, closes [#1739](https://github.com/lobehub/lobe-chat/issues/1739) ([e9ece9f](https://github.com/lobehub/lobe-chat/commit/e9ece9f))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 0.142.2](https://github.com/lobehub/lobe-chat/compare/v0.142.1...v0.142.2)
+
+<sup>Released on **2024-03-25**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Support openrouter custom models env.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Support openrouter custom models env, closes [#1647](https://github.com/lobehub/lobe-chat/issues/1647) ([78baa16](https://github.com/lobehub/lobe-chat/commit/78baa16))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 0.142.1](https://github.com/lobehub/lobe-chat/compare/v0.142.0...v0.142.1)
 
 <sup>Released on **2024-03-25**</sup>

package/Dockerfile

@@ -94,6 +94,7 @@ ENV MISTRAL_API_KEY ""
 
 # OpenRouter
 ENV OPENROUTER_API_KEY ""
+ENV OPENROUTER_CUSTOM_MODELS ""
 
 # 01.AI
 ENV ZEROONE_API_KEY ""

package/docs/self-hosting/environment-variables/model-provider.mdx

@@ -188,5 +188,12 @@ When using the `turn` mode, the API Keys will be retrieved in a round-robin mann
 - Default: -
 - Example: `sk-or-v1-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx=`
 
+### `OPENROUTER_CUSTOM_MODELS`
+
+- Type: Optional
+- Description: Used to control the model list, use `+` to add a model, use `-` to hide a model, use `model_name=display_name` to customize the display name of a model, separated by commas.
+- Default: `-`
+- Example: `-all,+01-ai/yi-34b-chat,+huggingfaceh4/zephyr-7b-beta`
+
 
 [azure-api-verion-url]: https://docs.microsoft.com/zh-cn/azure/developer/javascript/api-reference/es-modules/azure-sdk/ai-translation/translationconfiguration?view=azure-node-latest#api-version

package/docs/self-hosting/environment-variables/model-provider.zh-CN.mdx

@@ -186,6 +186,13 @@ LobeChat 在部署时提供了丰富的模型服务商相关的环境变量，
 - 默认值：-
 - 示例：`sk-or-v1-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx=`
 
+### `OPENROUTER_CUSTOM_MODELS`
+
+- 类型：可选
+- 描述：用来控制模型列表，使用 `+` 增加一个模型，使用 `-` 来隐藏一个模型，使用 `模型名=展示名` 来自定义模型的展示名，用英文逗号隔开。
+- 默认值：`-`
+- 示例：`-all,+01-ai/yi-34b-chat,+huggingfaceh4/zephyr-7b-beta`
+
 ## 01 AI
 
 ### `ZEROONE_API_KEY`

package/docs/self-hosting/platform/netlify.zh-CN.mdx

@@ -13,7 +13,7 @@
 
   前往 [OpenAI API Key](https://platform.openai.com/account/api-keys) 获取你的 OpenAI API Key
 
-  ### 在 Netflix 工作台导入
+  ### 在 Netlify 工作台导入
 
   <Callout>经过测试，暂不支持类似 Vercel/Zeabur 的一键部署按钮，原因未知。因此需要手动导入</Callout>
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "0.142.1",
+  "version": "0.142.3",
   "description": "Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",
@@ -80,7 +80,7 @@
   "dependencies": {
     "@ant-design/icons": "^5",
     "@anthropic-ai/sdk": "^0.18.0",
-    "@auth/core": "latest",
+    "@auth/core": "0.28.0",
     "@aws-sdk/client-bedrock-runtime": "^3.525.0",
     "@azure/openai": "^1.0.0-beta.11",
     "@cfworker/json-schema": "^1",
@@ -119,7 +119,7 @@
     "modern-screenshot": "^4",
     "nanoid": "^5",
     "next": "^14.1",
-    "next-auth": "beta",
+    "next-auth": "5.0.0-beta.15",
     "next-sitemap": "^4",
     "numeral": "^2",
     "nuqs": "^1",

package/src/app/api/config/route.ts

@@ -26,6 +26,7 @@ export const GET = async () => {
     ENABLED_ZEROONE,
     DEFAULT_AGENT_CONFIG,
     OLLAMA_CUSTOM_MODELS,
+    OPENROUTER_CUSTOM_MODELS,
   } = getServerConfig();
 
   const config: GlobalServerConfig = {
@@ -43,7 +44,7 @@ export const GET = async () => {
       mistral: { enabled: ENABLED_MISTRAL },
       moonshot: { enabled: ENABLED_MOONSHOT },
       ollama: { customModelName: OLLAMA_CUSTOM_MODELS, enabled: ENABLE_OLLAMA },
-      openrouter: { enabled: ENABLED_OPENROUTER },
+      openrouter: { customModelName: OPENROUTER_CUSTOM_MODELS, enabled: ENABLED_OPENROUTER },
       perplexity: { enabled: ENABLED_PERPLEXITY },
       zeroone: { enabled: ENABLED_ZEROONE },
       zhipu: { enabled: ENABLED_ZHIPU },

package/src/config/server/provider.ts

@@ -45,6 +45,7 @@ declare global {
 
       // OpenRouter Provider
       OPENROUTER_API_KEY?: string;
+      OPENROUTER_CUSTOM_MODELS?: string;
 
       // ZeroOne Provider
       ZEROONE_API_KEY?: string;
@@ -118,6 +119,7 @@ export const getProviderConfig = () => {
 
     ENABLED_OPENROUTER: !!OPENROUTER_API_KEY,
     OPENROUTER_API_KEY,
+    OPENROUTER_CUSTOM_MODELS: process.env.OPENROUTER_CUSTOM_MODELS,
 
     ENABLED_MOONSHOT: !!MOONSHOT_API_KEY,
     MOONSHOT_API_KEY,

package/src/store/global/slices/settings/selectors/__snapshots__/modelProvider.test.ts.snap

@@ -86,3 +86,145 @@ exports[`modelProviderSelectors > CUSTOM_MODELS > should work correct with gpt-4
   },
 ]
 `;
+
+exports[`modelProviderSelectors > OPENROUTER_CUSTOM_MODELS > custom deletion, addition, and renaming of models 1`] = `
+[
+  {
+    "chatModels": [
+      {
+        "description": "GPT 3.5 Turbo，适用于各种文本生成和理解任务",
+        "displayName": "GPT-3.5 Turbo",
+        "functionCall": true,
+        "id": "gpt-3.5-turbo",
+        "tokens": 16385,
+      },
+      {
+        "displayName": "GPT-3.5 Turbo (0125)",
+        "functionCall": true,
+        "hidden": true,
+        "id": "gpt-3.5-turbo-0125",
+        "tokens": 16385,
+      },
+      {
+        "displayName": "GPT-3.5 Turbo (1106)",
+        "functionCall": true,
+        "hidden": true,
+        "id": "gpt-3.5-turbo-1106",
+        "tokens": 16385,
+      },
+      {
+        "displayName": "GPT-3.5 Turbo Instruct",
+        "hidden": true,
+        "id": "gpt-3.5-turbo-instruct",
+        "tokens": 4096,
+      },
+      {
+        "displayName": "GPT-3.5 Turbo 16K",
+        "hidden": true,
+        "id": "gpt-3.5-turbo-16k",
+        "tokens": 16385,
+      },
+      {
+        "displayName": "GPT-3.5 Turbo (0613)",
+        "hidden": true,
+        "id": "gpt-3.5-turbo-0613",
+        "legacy": true,
+        "tokens": 4096,
+      },
+      {
+        "displayName": "GPT-3.5 Turbo 16K (0613)",
+        "hidden": true,
+        "id": "gpt-3.5-turbo-16k-0613",
+        "legacy": true,
+        "tokens": 4096,
+      },
+      {
+        "displayName": "GPT-4 Turbo Preview",
+        "functionCall": true,
+        "id": "gpt-4-turbo-preview",
+        "tokens": 128000,
+      },
+      {
+        "displayName": "GPT-4 Turbo Preview (0125)",
+        "functionCall": true,
+        "hidden": true,
+        "id": "gpt-4-0125-preview",
+        "tokens": 128000,
+      },
+      {
+        "description": "GPT-4 视觉预览版，支持视觉任务",
+        "displayName": "GPT-4 Turbo Vision Preview",
+        "id": "gpt-4-vision-preview",
+        "tokens": 128000,
+        "vision": true,
+      },
+      {
+        "displayName": "GPT-4 Turbo Preview (1106)",
+        "functionCall": true,
+        "hidden": true,
+        "id": "gpt-4-1106-preview",
+        "tokens": 128000,
+      },
+      {
+        "displayName": "GPT-4",
+        "functionCall": true,
+        "hidden": true,
+        "id": "gpt-4",
+        "tokens": 8192,
+      },
+      {
+        "displayName": "GPT-4 (0613)",
+        "functionCall": true,
+        "hidden": true,
+        "id": "gpt-4-0613",
+        "tokens": 8192,
+      },
+      {
+        "displayName": "GPT-4 32K",
+        "functionCall": true,
+        "hidden": true,
+        "id": "gpt-4-32k",
+        "tokens": 32768,
+      },
+      {
+        "displayName": "GPT-4 32K (0613)",
+        "functionCall": true,
+        "hidden": true,
+        "id": "gpt-4-32k-0613",
+        "tokens": 32768,
+      },
+      {
+        "displayName": "GPT-4 ALL",
+        "files": true,
+        "functionCall": true,
+        "hidden": true,
+        "id": "gpt-4-all",
+        "tokens": 32768,
+        "vision": true,
+      },
+    ],
+    "enabled": true,
+    "id": "openai",
+  },
+  {
+    "chatModels": [
+      {
+        "displayName": "google/gemma-7b-it",
+        "functionCall": true,
+        "id": "google/gemma-7b-it",
+        "isCustom": true,
+        "vision": true,
+      },
+      {
+        "displayName": "Mistral-7B-Instruct",
+        "functionCall": true,
+        "id": "mistralai/mistral-7b-instruct",
+        "isCustom": true,
+        "vision": true,
+      },
+    ],
+    "enabled": true,
+    "id": "openrouter",
+  },
+]
+`;

package/src/store/global/slices/settings/selectors/modelProvider.test.ts

@@ -131,6 +131,32 @@ describe('modelProviderSelectors', () => {
     });
   });
 
+  describe('OPENROUTER_CUSTOM_MODELS', () => {
+    it('custom deletion, addition, and renaming of models', () => {
+      const s = merge(initialSettingsState, {
+        settings: {
+          languageModel: {
+            openrouter: {
+              enabled: true,
+            },
+          }
+        },
+        serverConfig: {
+          languageModel: {
+            openrouter: {
+              apiKey: 'test-openrouter-api-key',
+              customModelName:
+                '-all,+google/gemma-7b-it,+mistralai/mistral-7b-instruct=Mistral-7B-Instruct',
+            },
+          }
+        },
+      }) as unknown as GlobalStore;
+
+      const result = modelProviderSelectors.modelSelectList(s).filter((r) => r.enabled);
+      expect(result).toMatchSnapshot();
+    });
+  });
+
   describe('modelEnabledVision', () => {
     it('should return true if the model has vision ability', () => {
       const hasAbility = modelProviderSelectors.modelEnabledVision('gpt-4-vision-preview')(

package/src/store/global/slices/settings/selectors/modelProvider.ts

@@ -149,9 +149,15 @@ const modelSelectList = (s: GlobalStore): ModelProviderCard[] => {
 
   const ollamaChatModels = processChatModels(ollamaModelConfig, OllamaProvider.chatModels);
 
-  const openrouterModelConfig = parseModelString(
-    currentSettings(s).languageModel.openrouter.customModelName,
-  )
+  const openrouterModelString = [
+    s.serverConfig.languageModel?.openrouter?.customModelName,
+    currentSettings(s).languageModel.openrouter.customModelName
+  ]
+    .filter(Boolean)
+    .join(',');
+  
+  const openrouterModelConfig = parseModelString(openrouterModelString);
+  
   const openrouterChatModels = processChatModels(openrouterModelConfig, OpenRouterProvider.chatModels);
 
   return [