@lobehub/chat 1.134.2 → 1.134.3

package/CHANGELOG.md

@@ -2,6 +2,31 @@
 
 # Changelog
 
+### [Version 1.134.3](https://github.com/lobehub/lobe-chat/compare/v1.134.2...v1.134.3)
+
+<sup>Released on **2025-10-05**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Type not preserved when model is sorted.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Type not preserved when model is sorted, closes [#9561](https://github.com/lobehub/lobe-chat/issues/9561) ([5fe2518](https://github.com/lobehub/lobe-chat/commit/5fe2518))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.134.2](https://github.com/lobehub/lobe-chat/compare/v1.134.1...v1.134.2)
 
 <sup>Released on **2025-10-05**</sup>

package/changelog/v1.json

@@ -1,4 +1,13 @@
 [
+  {
+    "children": {
+      "fixes": [
+        "Type not preserved when model is sorted."
+      ]
+    },
+    "date": "2025-10-05",
+    "version": "1.134.3"
+  },
   {
     "children": {
       "improvements": [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.134.2",
+  "version": "1.134.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",

package/packages/database/src/models/__tests__/aiModel.test.ts

@@ -315,5 +315,14 @@ describe('AiModelModel', () => {
       expect(models[0].id).toBe('model2');
       expect(models[1].id).toBe('model1');
     });
+
+    it('should preserve model type when inserting order records', async () => {
+      const sortMap = [{ id: 'image-model', sort: 0, type: 'image' as const }];
+
+      await aiProviderModel.updateModelsOrder('openai', sortMap);
+
+      const model = await aiProviderModel.findById('image-model');
+      expect(model?.type).toBe('image');
+    });
   });
 });

package/packages/database/src/models/aiModel.ts

@@ -223,20 +223,32 @@ export class AiModelModel {
 
   updateModelsOrder = async (providerId: string, sortMap: AiModelSortMap[]) => {
     await this.db.transaction(async (tx) => {
-      const updates = sortMap.map(({ id, sort }) => {
+      const updates = sortMap.map(({ id, sort, type }) => {
+        const now = new Date();
+        const insertValues: typeof aiModels.$inferInsert = {
+          enabled: true,
+          id,
+          providerId,
+          sort,
+          // source: isBuiltin ? 'builtin' : 'custom',
+          updatedAt: now,
+          userId: this.userId,
+        };
+
+        if (type) insertValues.type = type;
+
+        const updateValues: Partial<typeof aiModels.$inferInsert> = {
+          sort,
+          updatedAt: now,
+        };
+
+        if (type) updateValues.type = type;
+
         return tx
           .insert(aiModels)
-          .values({
-            enabled: true,
-            id,
-            providerId,
-            sort,
-            // source: isBuiltin ? 'builtin' : 'custom',
-            updatedAt: new Date(),
-            userId: this.userId,
-          })
+          .values(insertValues)
           .onConflictDoUpdate({
-            set: { sort, updatedAt: new Date() },
+            set: updateValues,
             target: [aiModels.id, aiModels.userId, aiModels.providerId],
           });
       });

package/packages/model-bank/src/types/aiModel.ts

@@ -371,6 +371,7 @@ export type UpdateAiModelParams = z.infer<typeof UpdateAiModelSchema>;
 export interface AiModelSortMap {
   id: string;
   sort: number;
+  type?: AiModelType;
 }
 
 export const ToggleAiModelEnableSchema = z.object({

package/packages/model-runtime/src/providers/google/index.ts

@@ -75,9 +75,10 @@ const getThinkingModelCategory = (model?: string): ThinkingModelCategory => {
   const normalized = model.toLowerCase();
 
   if (normalized.includes('robotics-er-1.5-preview')) return 'robotics';
-  if (normalized.includes('-2.5-flash-lite')) return 'flashLite';
-  if (normalized.includes('-2.5-flash')) return 'flash';
-  if (normalized.includes('-2.5-pro')) return 'pro';
+  if (normalized.includes('-2.5-flash-lite') || normalized.includes('flash-lite-latest'))
+    return 'flashLite';
+  if (normalized.includes('-2.5-flash') || normalized.includes('flash-latest')) return 'flash';
+  if (normalized.includes('-2.5-pro') || normalized.includes('pro-latest')) return 'pro';
 
   return 'other';
 };

package/src/app/[variants]/(main)/settings/provider/features/ModelList/SortModelModal/index.tsx

@@ -6,8 +6,8 @@ import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
 import { useAiInfraStore } from '@/store/aiInfra';
-import { AiProviderModelListItem } from '../../../../../../../../../packages/model-bank/src/types/aiModel';
 
+import { AiProviderModelListItem } from '../../../../../../../../../packages/model-bank/src/types/aiModel';
 import ListItem from './ListItem';
 
 const useStyles = createStyles(({ css, token }) => ({
@@ -76,6 +76,7 @@ const SortModelModal = memo<SortModelModalProps>(({ open, onCancel, defaultItems
             const sortMap = items.map((item, index) => ({
               id: item.id,
               sort: index,
+              type: item.type,
             }));
 
             setLoading(true);

package/src/server/routers/lambda/aiModel.ts

@@ -7,13 +7,15 @@ import { authedProcedure, router } from '@/libs/trpc/lambda';
 import { serverDatabase } from '@/libs/trpc/lambda/middleware';
 import { getServerGlobalConfig } from '@/server/globalConfig';
 import { KeyVaultsGateKeeper } from '@/server/modules/KeyVaultsEncrypt';
+import { ProviderConfig } from '@/types/user/settings';
+
 import {
+  AiModelTypeSchema,
   AiProviderModelListItem,
   CreateAiModelSchema,
   ToggleAiModelEnableSchema,
   UpdateAiModelSchema,
 } from '../../../../packages/model-bank/src/types/aiModel';
-import { ProviderConfig } from '@/types/user/settings';
 
 const aiModelProcedure = authedProcedure.use(serverDatabase).use(async (opts) => {
   const { ctx } = opts;
@@ -121,6 +123,7 @@ export const aiModelRouter = router({
           z.object({
             id: z.string(),
             sort: z.number(),
+            type: AiModelTypeSchema.optional(),
           }),
         ),
       }),