chore: 每个对话上下文关系独立

Author: Kerwin

service/src/index.ts

@@ -30,6 +30,7 @@ import {
   updateChat,
   updateConfig,
   updateRoomPrompt,
+  updateRoomUsingContext,
   updateUserInfo,
   updateUserPassword,
   verifyUser,
@@ -66,6 +67,7 @@ router.get('/chatrooms', auth, async (req, res) => {
         title: r.title,
         isEdit: false,
         prompt: r.prompt,
+        usingContext: r.usingContext === undefined ? true : r.usingContext,
       })
     })
     res.send({ status: 'Success', message: null, data: result })
@@ -118,6 +120,22 @@ router.post('/room-prompt', auth, async (req, res) => {
   }
 })
 
+router.post('/room-context', auth, async (req, res) => {
+  try {
+    const userId = req.headers.userId as string
+    const { using, roomId } = req.body as { using: boolean; roomId: number }
+    const success = await updateRoomUsingContext(userId, roomId, using)
+    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-delete', auth, async (req, res) => {
   try {
     const userId = req.headers.userId as string

service/src/storage/model.ts

@@ -38,12 +38,14 @@ export class ChatRoom {
   userId: string
   title: string
   prompt: string
+  usingContext: boolean
   status: Status = Status.Normal
   constructor(userId: string, title: string, roomId: number) {
     this.userId = userId
     this.title = title
     this.prompt = undefined
     this.roomId = roomId
+    this.usingContext = true
   }
 }
 

service/src/storage/mongo.ts

@@ -95,6 +95,17 @@ export async function updateRoomPrompt(userId: string, roomId: number, prompt: s
   return result.modifiedCount > 0
 }
 
+export async function updateRoomUsingContext(userId: string, roomId: number, using: boolean) {
+  const query = { userId, roomId }
+  const update = {
+    $set: {
+      usingContext: using,
+    },
+  }
+  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/api/index.ts

@@ -148,6 +148,13 @@ export function fetchUpdateChatRoomPrompt<T = any>(prompt: string, roomId: numbe
   })
 }
 
+export function fetchUpdateChatRoomUsingContext<T = any>(using: boolean, roomId: number) {
+  return post<T>({
+    url: '/room-context',
+    data: { using, roomId },
+  })
+}
+
 export function fetchDeleteChatRoom<T = any>(roomId: number) {
   return post<T>({
     url: '/room-delete',

src/store/modules/chat/helper.ts

@@ -7,7 +7,7 @@ export function defaultState(): Chat.ChatState {
   return {
     active: uuid,
     usingContext: true,
-    history: [{ uuid, title: 'New Chat', isEdit: false }],
+    history: [{ uuid, title: 'New Chat', isEdit: false, usingContext: true }],
     chat: [{ uuid, data: [] }],
   }
 }

src/store/modules/chat/index.ts

@@ -1,7 +1,7 @@
 import { defineStore } from 'pinia'
 import { getLocalState, setLocalState } from './helper'
 import { router } from '@/router'
-import { fetchClearChat, fetchCreateChatRoom, fetchDeleteChat, fetchDeleteChatRoom, fetchGetChatHistory, fetchGetChatRooms, fetchRenameChatRoom } from '@/api'
+import { fetchClearChat, fetchCreateChatRoom, fetchDeleteChat, fetchDeleteChatRoom, fetchGetChatHistory, fetchGetChatRooms, fetchRenameChatRoom, fetchUpdateChatRoomUsingContext } from '@/api'
 
 export const useChatStore = defineStore('chat-store', {
   state: (): Chat.ChatState => getLocalState(),
@@ -39,7 +39,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 })
+        await this.addHistory({ title: 'New Chat', uuid: Date.now(), isEdit: false, usingContext: true })
       }
       else {
         this.active = uuid
@@ -89,8 +89,8 @@ export const useChatStore = defineStore('chat-store', {
       }
     },
 
-    setUsingContext(context: boolean) {
-      this.usingContext = context
+    async  setUsingContext(context: boolean, roomId: number) {
+      await fetchUpdateChatRoomUsingContext(context, roomId)
       this.recordState()
     },
 
@@ -118,7 +118,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 })
+        await this.addHistory({ title: 'New Chat', uuid: Date.now(), isEdit: false, usingContext: true })
         return
       }
 
@@ -166,7 +166,7 @@ export const useChatStore = defineStore('chat-store', {
         if (this.history.length === 0) {
           const uuid = Date.now()
           fetchCreateChatRoom(chat.text, uuid)
-          this.history.push({ uuid, title: chat.text, isEdit: false })
+          this.history.push({ uuid, title: chat.text, isEdit: false, usingContext: true })
           this.chat.push({ uuid, data: [chat] })
           this.active = uuid
           this.recordState()

src/typings/chat.d.ts

@@ -25,6 +25,7 @@ declare namespace Chat {
     loading?: boolean
     all?: boolean
     prompt?: string
+    usingContext: boolean
   }
 
   interface ChatState {

src/views/chat/hooks/useUsingContext.ts

@@ -9,7 +9,6 @@ import html2canvas from 'html2canvas'
 import { Message } from './components'
 import { useScroll } from './hooks/useScroll'
 import { useChat } from './hooks/useChat'
-import { useUsingContext } from './hooks/useUsingContext'
 import HeaderComponent from './components/Header/index.vue'
 import { HoverButton, SvgIcon } from '@/components/common'
 import { useBasicLayout } from '@/hooks/useBasicLayout'
@@ -34,10 +33,11 @@ const chatStore = useChatStore()
 const { isMobile } = useBasicLayout()
 const { addChat, updateChat, updateChatSome, getChatByUuidAndIndex } = useChat()
 const { scrollRef, scrollToBottom, scrollToBottomIfAtBottom, scrollTo } = useScroll()
-const { usingContext, toggleUsingContext } = useUsingContext()
 
 const { uuid } = route.params as { uuid: string }
 
+const currentChatHistory = computed(() => chatStore.getChatHistoryByCurrentActive)
+const usingContext = computed(() => currentChatHistory?.value?.usingContext ?? true)
 const dataSources = computed(() => chatStore.getChatByUuid(+uuid))
 const conversationList = computed(() => dataSources.value.filter(item => (!item.inversion && !!item.conversationOptions)))
 
@@ -513,6 +513,18 @@ async function handleScroll(event: any) {
   prevScrollTop = scrollTop
 }
 
+async function handleToggleUsingContext() {
+  if (!currentChatHistory.value)
+    return
+
+  currentChatHistory.value.usingContext = !currentChatHistory.value.usingContext
+  chatStore.setUsingContext(currentChatHistory.value.usingContext, +uuid)
+  if (currentChatHistory.value.usingContext)
+    ms.success(t('chat.turnOnContext'))
+  else
+    ms.warning(t('chat.turnOffContext'))
+}
+
 // 可优化部分
 // 搜索选项计算，这里使用value作为索引项，所以当出现重复value时渲染异常(多项同时出现选中效果)
 // 理想状态下其实应该是key作为索引项,但官方的renderOption会出现问题，所以就需要value反renderLabel实现
@@ -577,7 +589,7 @@ onUnmounted(() => {
       v-if="isMobile"
       :using-context="usingContext"
       :show-prompt="showPrompt"
-      @export="handleExport" @toggle-using-context="toggleUsingContext"
+      @export="handleExport" @toggle-using-context="handleToggleUsingContext"
       @toggle-show-prompt="showPrompt = true"
     />
     <main class="flex-1 overflow-hidden">
@@ -642,7 +654,7 @@ onUnmounted(() => {
               <IconPrompt class="w-[20px] m-auto" />
             </span>
           </HoverButton>
-          <HoverButton v-if="!isMobile" @click="toggleUsingContext">
+          <HoverButton v-if="!isMobile" @click="handleToggleUsingContext">
             <span class="text-xl" :class="{ 'text-[#4b9e5f]': usingContext, 'text-[#a8071a]': !usingContext }">
               <SvgIcon icon="ri:chat-history-line" />
             </span>

src/views/chat/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 })
+  await chatStore.addHistory({ title: 'New Chat', uuid: Date.now(), isEdit: false, usingContext: true })
   if (isMobile.value)
     appStore.setSiderCollapsed(true)
 }