移除空格，增加custom_models 容错性

ChatGPTNextWeb · Dec 30, 2024 · 69fcb92 · 69fcb92
1 parent 3b5b496
commit 69fcb92
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/app/utils/model.ts b/app/utils/model.ts
@@ -76,6 +76,7 @@ export function collectModelTable(
   // server custom models
   customModels
     .split(",")
+    .map((v) => v.trim())
     .filter((v) => !!v && v.length > 0)
     .forEach((m) => {
       const available = !m.startsWith("-");

diff --git a/test/model.test.ts b/test/model.test.ts
@@ -127,7 +127,7 @@ describe('collectModelTable', () => {
   });
 
   test('All models except OpenAI should be excluded, and additional models should be added when customized as "-all, +*openai,gpt-4o@azure"', () => {
-    const customModels = '-all, +*openai,gpt-4o@azure';
+    const customModels = '-all,+*openai,gpt-4o@azure';
     const result = collectModelTable(mockModels, customModels);
 
     expect(result['gpt-3.5-turbo@openai'].available).toBe(true);