fix: correct the model list on the model provider's page

lobehub · Dec 16, 2024 · 5859ddd · 5859ddd
1 parent 321b7a1
commit 5859ddd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/server/services/discover/index.ts b/src/server/services/discover/index.ts
@@ -328,7 +328,10 @@ export class DiscoverService {
 
   getModelById = async (locale: Locales, id: string): Promise<DiscoverModelItem | undefined> => {
     const list = await this.getModelList(locale);
-    let model = list.find((item) => item.identifier === id);
+    let model = list.find((item) => {
+      const ids = item.identifier.split('/');
+      return ids[1] || item.identifier === id;
+    });
 
     if (!model) return;