feat: set chat model by per chat room

Signed-off-by: BobDu <i@bobdu.cc>

Author: BobDu

service/src/chatgpt/index.ts

@@ -15,7 +15,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, updateRoomChatModel } from '../storage/mongo'
+import { getChatByMessageId, updateRoomAccountId } from '../storage/mongo'
 import type { RequestOptions } from './types'
 
 const { HttpsProxyAgent } = httpsProxyAgent
@@ -113,8 +113,8 @@ export async function initApi(key: KeyConfig, chatModel: string, maxContextCount
 }
 const processThreads: { userId: string; abort: AbortController; messageId: string }[] = []
 async function chatReplyProcess(options: RequestOptions) {
-  const model = options.user.config.chatModel
-  const key = await getRandomApiKey(options.user, options.user.config.chatModel, options.room.accountId)
+  const model = options.room.chatModel
+  const key = await getRandomApiKey(options.user, model, options.room.accountId)
   const userId = options.user._id.toString()
   const maxContextCount = options.user.advanced.maxContextCount ?? 20
   const messageId = options.messageId
@@ -130,8 +130,6 @@ 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

@@ -95,8 +95,8 @@ router.get('/chatrooms', auth, async (req, res) => {
 router.post('/room-create', auth, async (req, res) => {
   try {
     const userId = req.headers.userId as string
-    const { title, roomId } = req.body as { title: string; roomId: number }
-    const room = await createChatRoom(userId, title, roomId)
+    const { title, roomId, chatModel } = req.body as { title: string; roomId: number; chatModel: string }
+    const room = await createChatRoom(userId, title, roomId, chatModel)
     res.send({ status: 'Success', message: null, data: room })
   }
   catch (error) {

service/src/storage/model.ts

@@ -67,14 +67,14 @@ export class ChatRoom {
   // only access token used
   accountId?: string
   chatModel: string
-  constructor(userId: string, title: string, roomId: number) {
+  constructor(userId: string, title: string, roomId: number, chatModel: string) {
     this.userId = userId
     this.title = title
     this.prompt = undefined
     this.roomId = roomId
     this.usingContext = true
     this.accountId = null
-    this.chatModel = null
+    this.chatModel = chatModel
   }
 }
 

service/src/storage/mongo.ts

@@ -70,8 +70,8 @@ export async function insertChatUsage(userId: ObjectId, roomId: number, chatId:
   return chatUsage
 }
 
-export async function createChatRoom(userId: string, title: string, roomId: number) {
-  const room = new ChatRoom(userId, title, roomId)
+export async function createChatRoom(userId: string, title: string, roomId: number, chatModel: string) {
+  const room = new ChatRoom(userId, title, roomId, chatModel)
   await roomCol.insertOne(room)
   return room
 }

src/api/index.ts

@@ -185,10 +185,10 @@ export function fetchGetChatRooms<T = any>() {
   })
 }
 
-export function fetchCreateChatRoom<T = any>(title: string, roomId: number) {
+export function fetchCreateChatRoom<T = any>(title: string, roomId: number, chatModel?: string) {
   return post<T>({
     url: '/room-create',
-    data: { title, roomId },
+    data: { title, roomId, chatModel },
   })
 }
 
@@ -206,6 +206,13 @@ export function fetchUpdateChatRoomPrompt<T = any>(prompt: string, roomId: numbe
   })
 }
 
+export function fetchUpdateChatRoomChatModel<T = any>(chatModel: string, roomId: number) {
+  return post<T>({
+    url: '/room-chatmodel',
+    data: { chatModel, roomId },
+  })
+}
+
 export function fetchUpdateChatRoomUsingContext<T = any>(using: boolean, roomId: number) {
   return post<T>({
     url: '/room-context',

src/store/modules/chat/index.ts

@@ -1,7 +1,18 @@
 import { defineStore } from 'pinia'
 import { getLocalState, setLocalState } from './helper'
+import { useUserStore } from '@/store'
 import { router } from '@/router'
-import { fetchClearChat, fetchCreateChatRoom, fetchDeleteChat, fetchDeleteChatRoom, fetchGetChatHistory, fetchGetChatRooms, fetchRenameChatRoom, fetchUpdateChatRoomUsingContext } from '@/api'
+import {
+  fetchClearChat,
+  fetchCreateChatRoom,
+  fetchDeleteChat,
+  fetchDeleteChatRoom,
+  fetchGetChatHistory,
+  fetchGetChatRooms,
+  fetchRenameChatRoom,
+  fetchUpdateChatRoomChatModel,
+  fetchUpdateChatRoomUsingContext,
+} from '@/api'
 
 export const useChatStore = defineStore('chat-store', {
   state: (): Chat.ChatState => getLocalState(),
@@ -39,7 +50,7 @@ export const useChatStore = defineStore('chat-store', {
         this.chat.unshift({ uuid: r.uuid, data: [] })
       }
       if (uuid == null) {
-        await this.addHistory({ title: 'New Chat', uuid: Date.now(), isEdit: false, usingContext: true })
+        await this.addNewHistory()
       }
       else {
         this.active = uuid
@@ -94,12 +105,27 @@ export const useChatStore = defineStore('chat-store', {
       this.recordState()
     },
 
+    async setChatModel(chatModel: string, roomId: number) {
+      await fetchUpdateChatRoomChatModel(chatModel, roomId)
+    },
+
     async addHistory(history: Chat.History, chatData: Chat.Chat[] = []) {
-      await fetchCreateChatRoom(history.title, history.uuid)
+      await fetchCreateChatRoom(history.title, history.uuid, history.chatModel)
       this.history.unshift(history)
       this.chat.unshift({ uuid: history.uuid, data: chatData })
       this.active = history.uuid
-      this.reloadRoute(history.uuid)
+      await this.reloadRoute(history.uuid)
+    },
+
+    async addNewHistory() {
+      const userStore = useUserStore()
+      await this.addHistory({
+        title: 'New Chat',
+        uuid: Date.now(),
+        isEdit: false,
+        usingContext: true,
+        chatModel: userStore.userInfo.config.chatModel,
+      })
     },
 
     updateHistory(uuid: number, edit: Partial<Chat.History>) {
@@ -118,7 +144,7 @@ export const useChatStore = defineStore('chat-store', {
       this.chat.splice(index, 1)
 
       if (this.history.length === 0) {
-        await this.addHistory({ title: 'New Chat', uuid: Date.now(), isEdit: false, usingContext: true })
+        await this.addNewHistory()
         return
       }
 

src/views/chat/index.vue

@@ -13,11 +13,14 @@ import HeaderComponent from './components/Header/index.vue'
 import { HoverButton, SvgIcon } from '@/components/common'
 import { useBasicLayout } from '@/hooks/useBasicLayout'
 import { useAuthStore, useChatStore, usePromptStore, useUserStore } from '@/store'
-import { fetchChatAPIProcess, fetchChatResponseoHistory, fetchChatStopResponding, fetchUpdateUserChatModel } from '@/api'
+import {
+  fetchChatAPIProcess,
+  fetchChatResponseoHistory,
+  fetchChatStopResponding,
+} from '@/api'
 import { t } from '@/locales'
 import { debounce } from '@/utils/functions/debounce'
 import IconPrompt from '@/icons/Prompt.vue'
-import { UserConfig } from '@/components/common/Setting/model'
 const Prompt = defineAsyncComponent(() => import('@/components/common/Setting/Prompt.vue'))
 
 let controller = new AbortController()
@@ -582,11 +585,7 @@ const footerClass = computed(() => {
 
 async function handleSyncChatModel(chatModel: string) {
   nowSelectChatModel.value = chatModel
-  if (userStore.userInfo.config == null)
-    userStore.userInfo.config = new UserConfig()
-  userStore.userInfo.config.chatModel = chatModel
-  userStore.recordState()
-  await fetchUpdateUserChatModel(chatModel)
+  await chatStore.setChatModel(chatModel, +uuid)
 }
 
 onMounted(() => {

src/views/chat/layout/sider/index.vue

@@ -18,7 +18,7 @@ const show = ref(false)
 const collapsed = computed(() => appStore.siderCollapsed)
 
 async function handleAdd() {
-  await chatStore.addHistory({ title: 'New Chat', uuid: Date.now(), isEdit: false, usingContext: true })
+  await chatStore.addNewHistory()
   if (isMobile.value)
     appStore.setSiderCollapsed(true)
 }