feat: fixed model for each chat room (Close #300)

Author: Kerwin

service/src/chatgpt/index.ts

@@ -14,7 +14,7 @@ import { getCacheApiKeys, getCacheConfig, getOriginConfig } from '../storage/con
 import { sendResponse } from '../utils'
 import { hasAnyRole, isNotEmptyString } from '../utils/is'
 import type { ChatContext, ChatGPTUnofficialProxyAPIOptions, JWT, ModelConfig } from '../types'
-import { getChatByMessageId, updateRoomAccountId } from '../storage/mongo'
+import { getChatByMessageId, updateRoomAccountId, updateRoomChatModel } from '../storage/mongo'
 import type { RequestOptions } from './types'
 
 const { HttpsProxyAgent } = httpsProxyAgent
@@ -114,6 +114,8 @@ async function chatReplyProcess(options: RequestOptions) {
       throw new Error('无法在一个房间同时使用 AccessToken 以及 Api，请联系管理员，或新开聊天室进行对话 | Unable to use AccessToken and Api at the same time in the same room, please contact the administrator or open a new chat room for conversation')
   }
 
+  updateRoomChatModel(userId, options.room.roomId, model)
+
   const { message, lastContext, process, systemMessage, temperature, top_p } = options
 
   try {

service/src/index.ts

@@ -32,6 +32,7 @@ import {
   updateApiKeyStatus,
   updateChat,
   updateConfig,
+  updateRoomChatModel,
   updateRoomPrompt,
   updateRoomUsingContext,
   updateUser,
@@ -75,6 +76,7 @@ router.get('/chatrooms', auth, async (req, res) => {
         isEdit: false,
         prompt: r.prompt,
         usingContext: r.usingContext === undefined ? true : r.usingContext,
+        chatModel: r.chatModel,
       })
     })
     res.send({ status: 'Success', message: null, data: result })
@@ -127,6 +129,22 @@ router.post('/room-prompt', auth, async (req, res) => {
   }
 })
 
+router.post('/room-chatmodel', auth, async (req, res) => {
+  try {
+    const userId = req.headers.userId as string
+    const { chatModel, roomId } = req.body as { chatModel: CHATMODEL; roomId: number }
+    const success = await updateRoomChatModel(userId, roomId, chatModel)
+    if (success)
+      res.send({ status: 'Success', message: 'Saved successfully', data: null })
+    else
+      res.send({ status: 'Fail', message: 'Saved Failed', data: null })
+  }
+  catch (error) {
+    console.error(error)
+    res.send({ status: 'Fail', message: 'Rename error', data: null })
+  }
+})
+
 router.post('/room-context', auth, async (req, res) => {
   try {
     const userId = req.headers.userId as string

service/src/storage/model.ts

@@ -83,13 +83,15 @@ export class ChatRoom {
   status: Status = Status.Normal
   // only access token used
   accountId?: string
+  chatModel: CHATMODEL
   constructor(userId: string, title: string, roomId: number) {
     this.userId = userId
     this.title = title
     this.prompt = undefined
     this.roomId = roomId
     this.usingContext = true
     this.accountId = null
+    this.chatModel = null
   }
 }
 

service/src/storage/mongo.ts

@@ -120,6 +120,17 @@ export async function updateRoomAccountId(userId: string, roomId: number, accoun
   return result.modifiedCount > 0
 }
 
+export async function updateRoomChatModel(userId: string, roomId: number, chatModel: CHATMODEL) {
+  const query = { userId, roomId }
+  const update = {
+    $set: {
+      chatModel,
+    },
+  }
+  const result = await roomCol.updateOne(query, update)
+  return result.modifiedCount > 0
+}
+
 export async function getChatRooms(userId: string) {
   const cursor = await roomCol.find({ userId, status: { $ne: Status.Deleted } })
   const rooms = []

src/typings/chat.d.ts

@@ -1,4 +1,5 @@
 declare namespace Chat {
+  import { CHATMODEL } from "@/components/common/Setting/model"
 
   interface Chat {
     uuid?: number
@@ -26,6 +27,7 @@ declare namespace Chat {
     all?: boolean
     prompt?: string
     usingContext: boolean
+    chatModel?: CHATMODEL
   }
 
   interface ChatState {

src/views/chat/index.vue

@@ -49,6 +49,8 @@ const firstLoading = ref<boolean>(false)
 const loading = ref<boolean>(false)
 const inputRef = ref<Ref | null>(null)
 const showPrompt = ref(false)
+const nowSelectChatModel = ref<CHATMODEL | null>(null)
+const currentChatModel = computed(() => nowSelectChatModel.value ?? currentChatHistory.value?.chatModel ?? userStore.userInfo.config.chatModel)
 
 let loadingms: MessageReactive
 let allmsg: MessageReactive
@@ -79,6 +81,9 @@ async function onConversation() {
   if (!message || message.trim() === '')
     return
 
+  if (nowSelectChatModel.value && currentChatHistory.value)
+    currentChatHistory.value.chatModel = nowSelectChatModel.value
+
   controller = new AbortController()
 
   const chatUuid = Date.now()
@@ -577,6 +582,7 @@ const footerClass = computed(() => {
 })
 
 async function handleSyncChatModel(chatModel: CHATMODEL) {
+  nowSelectChatModel.value = chatModel
   if (userStore.userInfo.config == null)
     userStore.userInfo.config = new UserConfig()
   userStore.userInfo.config.chatModel = chatModel
@@ -684,7 +690,7 @@ onUnmounted(() => {
             </HoverButton>
             <NSelect
               style="width: 250px"
-              :value="userStore.userInfo.config.chatModel"
+              :value="currentChatModel"
               :options="authStore.session?.chatModels"
               :disabled="!!authStore.session?.auth && !authStore.token"
               @update-value="(val) => handleSyncChatModel(val)"