feat: allow bind a chat model for chat role

web/src/views/ChatPlus.vue

@@ -82,7 +82,6 @@
       <el-main v-loading="loading" element-loading-background="rgba(122, 122, 122, 0.3)">
         <div class="chat-head">
           <div class="chat-config">
-            <!--            <span class="role-select-label">聊天角色：</span>-->
             <el-select v-model="roleId" filterable placeholder="角色" class="role-select" @change="_newChat">
               <el-option
                   v-for="item in roles"
@@ -97,7 +96,7 @@
               </el-option>
             </el-select>
 
-            <el-select v-model="modelID" placeholder="模型" @change="_newChat">
+            <el-select v-model="modelID" placeholder="模型" @change="_newChat" :disabled="disableModel">
               <el-option
                   v-for="item in models"
                   :key="item.id"
@@ -445,6 +444,8 @@ const _newChat = () => {
     newChat()
   }
 }
+
+const disableModel = ref(false)
 // 新建会话
 const newChat = () => {
   if (!isLogin.value) {
@@ -452,10 +453,11 @@ const newChat = () => {
     return;
   }
   const role = getRoleById(roleId.value)
-  if (role.key === 'gpt') {
-    showHello.value = true
-  } else {
-    showHello.value = false
+  showHello.value = role.key === 'gpt';
+  // if the role bind a model, disable model change
+  if (role.model_id > 0) {
+    modelID.value = role.model_id
+    disableModel.value = true
   }
   // 已有新开的会话
   if (newChatItem.value !== null && newChatItem.value['role_id'] === roles.value[0]['role_id']) {

web/src/views/ImageSd.vue

@@ -11,7 +11,7 @@
                 <el-form-item label="采样方法">
                   <template #default>
                     <div class="form-item-inner">
-                      <el-select v-model="params.sampler" size="small">
+                      <el-select v-model="params.sampler" size="small" style="width:150px">
                         <el-option v-for="item in samplers" :label="item" :value="item" :key="item"/>
                       </el-select>
                       <el-tooltip
@@ -163,7 +163,7 @@
                   <el-form-item label="放大算法">
                     <template #default>
                       <div class="form-item-inner">
-                        <el-select v-model="params.hd_scale_alg" size="small">
+                        <el-select v-model="params.hd_scale_alg" size="small" style="width:150px">
                           <el-option v-for="item in scaleAlg" :label="item" :value="item" :key="item"/>
                         </el-select>
                         <el-tooltip

web/src/views/admin/Roles.vue

@@ -21,6 +21,7 @@
           </template>
         </el-table-column>
         <el-table-column label="角色标识" prop="key"/>
+        <el-table-column label="绑定模型" prop="model_name"/>
         <el-table-column label="启用状态">
           <template #default="scope">
             <el-switch v-model="scope.row['enable']" @change="roleSet('enable',scope.row)"/>
@@ -47,7 +48,7 @@
 
     <el-dialog
         v-model="showDialog"
-        title="编辑角色"
+        :title="optTitle"
         :close-on-click-modal="false"
         width="50%"
     >
@@ -73,6 +74,21 @@
           />
         </el-form-item>
 
+        <el-form-item label="绑定模型：" prop="model_id">
+          <el-select
+              v-model="role.model_id"
+              filterable
+              placeholder="请选择模型"
+          >
+            <el-option
+                v-for="item in models"
+                :key="item.id"
+                :label="item.name"
+                :value="item.id"
+            />
+          </el-select>
+        </el-form-item>
+
         <el-form-item label="打招呼信息：" prop="hello_msg">
           <el-input
               v-model="role.hello_msg"
@@ -151,7 +167,7 @@ const tableData = ref([])
 const sortedTableData = ref([])
 const role = ref({context: []})
 const formRef = ref(null)
-const editRow = ref({})
+const optTitle = ref({})
 const loading = ref(true)
 
 const rules = reactive({
@@ -165,18 +181,30 @@ const rules = reactive({
   hello_msg: [{required: true, message: '请输入打招呼信息', trigger: 'change',}]
 })
 
-// 获取角色列表
-httpGet('/api/admin/role/list').then((res) => {
-  tableData.value = res.data
-  sortedTableData.value = copyObj(tableData.value)
-  loading.value = false
-}).catch(() => {
-  ElMessage.error("获取聊天角色失败");
+const models = ref([])
+onMounted(() => {
+  fetchData()
+
+  // get chat models
+  httpGet('/api/admin/model/list?enable=1').then((res) => {
+    models.value = res.data
+  }).catch(() => {
+    ElMessage.error("获取AI模型数据失败");
+  })
+
 })
 
-onMounted(() => {
-  const drawBodyWrapper = document.querySelector('.el-table__body tbody')
+const fetchData = () => {
+  // 获取角色列表
+  httpGet('/api/admin/role/list').then((res) => {
+    tableData.value = res.data
+    sortedTableData.value = copyObj(tableData.value)
+    loading.value = false
+  }).catch(() => {
+    ElMessage.error("获取聊天角色失败");
+  })
 
+  const drawBodyWrapper = document.querySelector('.el-table__body tbody')
   // 初始化拖动排序插件
   Sortable.create(drawBodyWrapper, {
     sort: true,
@@ -199,7 +227,7 @@ onMounted(() => {
       })
     }
   })
-})
+}
 
 const roleSet = (filed, row) => {
   httpPost('/api/admin/role/set', {id: row.id, filed: filed, value: row[filed]}).then(() => {
@@ -212,12 +240,14 @@ const roleSet = (filed, row) => {
 // 编辑
 const curIndex = ref(0)
 const rowEdit = function (index, row) {
+  optTitle.value = "修改角色"
   curIndex.value = index
   role.value = copyObj(row)
   showDialog.value = true
 }
 
 const addRole = function () {
+  optTitle.value = "添加新角色"
   role.value = {context: []}
   showDialog.value = true
 }
@@ -226,14 +256,9 @@ const save = function () {
   formRef.value.validate((valid) => {
     if (valid) {
       showDialog.value = false
-      httpPost('/api/admin/role/save', role.value).then((res) => {
+      httpPost('/api/admin/role/save', role.value).then(() => {
         ElMessage.success('操作成功')
-        // 更新当前数据行
-        if (role.value.id) {
-          tableData.value[curIndex.value] = role.value
-        } else {
-          tableData.value.push(res.data)
-        }
+        fetchData()
       }).catch((e) => {
         ElMessage.error('操作失败，' + e.message)
       })
@@ -263,6 +288,7 @@ const removeContext = function (index) {
   role.value.context.splice(index, 1);
 }
 
+
 </script>
 
 <style lang="stylus" scoped>