perf: Set default model

Author: fit2cloud-chenyw

backend/apps/system/api/aimodel.py

@@ -2,7 +2,7 @@
 from typing import List, Union
 from apps.system.schemas.ai_model_schema import AiModelConfigItem, AiModelCreator, AiModelEditor, AiModelGridItem
 from fastapi import APIRouter, Query
-from sqlmodel import func, select
+from sqlmodel import func, select, update
 
 from apps.system.models.system_model import AiModelDetail
 from common.core.deps import SessionDep
@@ -23,7 +23,7 @@ async def query(
                        AiModelDetail.default_model)
     if keyword is not None:
         statement = statement.where(AiModelDetail.name.like(f"%{keyword}%"))
-    
+    statement = statement.order_by(AiModelDetail.create_time.asc())
     items = session.exec(statement).all()
     return items
 
@@ -85,6 +85,27 @@ async def delete_model(
         id: int
 ):
     item = session.get(AiModelDetail, id)
+    if item.default_model:
+        raise RuntimeError(f"Can not delete [${item.name}], because it is default model!")
     session.delete(item)
     session.commit()
+    
+@router.put("/default/{id}")
+async def set_default(session: SessionDep, id: int):
+    db_model = session.get(AiModelDetail, id)
+    if not db_model:
+        raise ValueError(f"AiModelDetail with id {id} not found")
+    if db_model.default_model:
+        return
+
+    try:
+        session.exec(
+            update(AiModelDetail).values(default_model=False)
+        )
+        db_model.default_model = True
+        session.add(db_model)
+        session.commit()
+    except Exception as e:
+        session.rollback()
+        raise e
 

frontend/src/api/system.ts

@@ -7,5 +7,5 @@ export const modelApi = {
   edit: (data: any) => request.put('/system/aimodel', data),
   delete: (id: number) => request.delete(`/system/aimodel/${id}`),
   query: (id: number) => request.get(`/system/aimodel/${id}`),
-  status: (data: any) => request.patch('/system/aimodel/status', data),
+  setDefault: (id: number) => request.put(`/system/aimodel/default/${id}`),
 }

frontend/src/router/index.ts

@@ -9,7 +9,6 @@ import Datasource from '@/views/ds/Datasource.vue'
 import DashboardEditor from '@/views/dashboard/editor/index.vue'
 import DashboardPreview from '@//views/dashboard/preview/SQPreviewSingle.vue'
 import Dashboard from '@/views/dashboard/index.vue'
-// import Model from '@/views/system/model/index.vue'
 import Model from '@/views/system/model/Model.vue'
 // import User from "@/views/system/user/index.vue";
 import { watchRouter } from './watch'

frontend/src/views/system/model/Card.vue

@@ -59,14 +59,12 @@ const handleDel = () => {
           <edit></edit>
         </el-icon>
       </el-tooltip>
-      <template v-if="!isDefault">
-        <span class="divide"></span>
-        <el-tooltip :offset="14" effect="dark" :content="$t('dashboard.delete')" placement="top">
-          <el-icon size="16" @click="handleDel">
-            <delIcon></delIcon>
-          </el-icon>
-        </el-tooltip>
-      </template>
+      <span class="divide"></span>
+      <el-tooltip :offset="14" effect="dark" :content="$t('dashboard.delete')" placement="top">
+        <el-icon size="16" @click="handleDel">
+          <delIcon></delIcon>
+        </el-icon>
+      </el-tooltip>
     </div>
   </div>
 </template>

frontend/src/views/system/model/Model.vue

@@ -69,6 +69,10 @@ const defaultModelListWithSearch = computed(() => {
 })
 
 const handleDefaultModelChange = (item: any) => {
+  const current_default_node = modelList.value.find((ele: Model) => ele.default_model)
+  if (current_default_node?.id === item.id) {
+    return
+  }
   ElMessageBox.confirm(`是否设置 ${item.name} 为系统默认模型？`, {
     confirmButtonType: 'primary',
     tip: '系统默认模型被替换后，智能问数的结果将会受到影响，请谨慎操作。',
@@ -78,10 +82,10 @@ const handleDefaultModelChange = (item: any) => {
     autofocus: false,
     callback: (val: string) => {
       if (val === 'confirm') {
-        modelList.value.forEach((ele: any) => {
-          ele.default_model = ele.id === item.id
+        modelApi.setDefault(item.id).then(() => {
+          ElMessage.success('设置成功')
+          search()
         })
-        ElMessage.success('设置成功')
       }
     },
   })