fix user enabled models

lobehub · Jan 9, 2025 · 46e260a · 46e260a
1 parent ccbfe90
commit 46e260a
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/src/database/repositories/aiInfra/index.ts b/src/database/repositories/aiInfra/index.ts
@@ -81,17 +81,16 @@ export class AiInfraRepos {
           .map<EnabledAiModel & { enabled?: boolean | null }>((item) => {
             const user = allModels.find((m) => m.id === item.id && m.providerId === provider.id);
 
-            const enabled = !!user ? user.enabled : item.enabled;
-            const sort = !!user ? user.sort : undefined;
-            const displayName = !!user ? user.displayName : item.displayName;
-
             return {
-              ...item,
-              abilities: item.abilities || {},
-              displayName,
-              enabled,
+              abilities: !!user ? user.abilities : item.abilities || {},
+              config: !!user ? user.config : item.config,
+              contextWindowTokens: !!user ? user.contextWindowTokens : item.contextWindowTokens,
+              displayName: !!user ? user.displayName : item.displayName,
+              enabled: !!user ? user.enabled : item.enabled,
+              id: item.id,
               providerId: provider.id,
-              sort,
+              sort: !!user ? user.sort : undefined,
+              type: item.type,
             };
           })
           .filter((i) => i.enabled);