fix model list

lobehub · Jan 9, 2025 · aa6186b · aa6186b
1 parent 79ade8f
commit aa6186b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/database/repositories/aiInfra/index.ts b/src/database/repositories/aiInfra/index.ts
@@ -120,7 +120,9 @@ export class AiInfraRepos {
   ): Promise<AiProviderModelListItem[] | undefined> => {
     try {
       const { default: providerModels } = await import(`@/config/aiModels/${providerId}`);
-      return (providerModels as AIChatModelCard[]).map<AiProviderModelListItem>((m) => ({
+
+      const presetList = this.providerConfigs[providerId]?.serverModelLists || providerModels;
+      return (presetList as AIChatModelCard[]).map<AiProviderModelListItem>((m) => ({
         ...m,
         enabled: m.enabled || false,
         source: AiModelSourceEnum.Builtin,