feat: support room specific prompt

fix: mobile login modal (Closes #41)

Author: Kerwin

README.en.md

@@ -7,7 +7,18 @@
 </br>
 
 ## Introduction
-> **This project is forked from [Chanzhaoyu/chatgpt-web](https://github.com/Chanzhaoyu/chatgpt-web). In addition to regularly merging this branch, some unique features have been added such as registration and login, setting API key on the front-end page.**
+> **This project is forked from [Chanzhaoyu/chatgpt-web](https://github.com/Chanzhaoyu/chatgpt-web), some unique features have been added:**
+
+[✓] Register & Login & Reset Password
+
+[✓] Sync chat history 
+
+[✓] Front-end page setting apikey
+
+[✓] Custom Sensitive Words
+
+[✓] Set unique prompts for each chat room
+
 </br>
 
 ## Screenshots
@@ -17,6 +28,7 @@
 ![cover](./docs/c1.png)
 ![cover2](./docs/c2.png)
 ![cover3](./docs/basesettings.jpg)
+![cover3](./docs/prompt_en.jpg)
 
 - [ChatGPT Web](#chatgpt-web)
 	- [Introduction](#introduction)

README.md

@@ -7,7 +7,17 @@
 </br>
 
 ## 说明
-> **此项目 Fork 自 [Chanzhaoyu/chatgpt-web](https://github.com/Chanzhaoyu/chatgpt-web), 除了定时合并该分支, 新增了部分特色功能, 注册&登录, 前端页面设置apikey 等**
+> **此项目 Fork 自 [Chanzhaoyu/chatgpt-web](https://github.com/Chanzhaoyu/chatgpt-web), 新增了部分特色功能:**
+
+[✓] 注册&登录&重置密码
+
+[✓] 同步历史会话
+
+[✓] 前端页面设置apikey
+
+[✓] 自定义敏感词
+
+[✓] 每个会话设置独有 Prompt
 </br>
 
 ## 截图
@@ -17,6 +27,7 @@
 ![cover](./docs/c1.png)
 ![cover2](./docs/c2.png)
 ![cover3](./docs/basesettings.jpg)
+![cover3](./docs/prompt.jpg)
 
 - [ChatGPT Web](#chatgpt-web)
 	- [介绍](#介绍)

docs/prompt.jpg

@@ -1,6 +1,6 @@
 {
   "name": "chatgpt-web",
-  "version": "2.12.0",
+  "version": "2.12.3",
   "private": false,
   "description": "ChatGPT Web",
   "author": "ChenZhaoYu <chenzhaoyu1994@gmail.com>",

docs/prompt_en.jpg

@@ -17,6 +17,7 @@ import {
   deleteChatRoom,
   existsChatRoom,
   getChat,
+  getChatRoom,
   getChatRooms,
   getChats,
   getUser,
@@ -26,6 +27,7 @@ import {
   renameChatRoom,
   updateChat,
   updateConfig,
+  updateRoomPrompt,
   updateUserInfo,
   updateUserPassword,
   verifyUser,
@@ -61,6 +63,7 @@ router.get('/chatrooms', auth, async (req, res) => {
         uuid: r.roomId,
         title: r.title,
         isEdit: false,
+        prompt: r.prompt,
       })
     })
     res.send({ status: 'Success', message: null, data: result })
@@ -97,6 +100,22 @@ router.post('/room-rename', auth, async (req, res) => {
   }
 })
 
+router.post('/room-prompt', auth, async (req, res) => {
+  try {
+    const userId = req.headers.userId as string
+    const { prompt, roomId } = req.body as { prompt: string; roomId: number }
+    const success = await updateRoomPrompt(userId, roomId, prompt)
+    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
@@ -272,7 +291,11 @@ router.post('/chat', auth, async (req, res) => {
 router.post('/chat-process', [auth, limiter], async (req, res) => {
   res.setHeader('Content-type', 'application/octet-stream')
 
-  const { roomId, uuid, regenerate, prompt, options = {}, systemMessage, temperature, top_p } = req.body as RequestProps
+  let { roomId, uuid, regenerate, prompt, options = {}, systemMessage, temperature, top_p } = req.body as RequestProps
+  const userId = req.headers.userId as string
+  const room = await getChatRoom(userId, roomId)
+  if (room != null && isNotEmptyString(room.prompt))
+    systemMessage = room.prompt
 
   let lastResponse
   let result

package.json

@@ -37,10 +37,12 @@ export class ChatRoom {
   roomId: number
   userId: string
   title: string
+  prompt: string
   status: Status = Status.Normal
   constructor(userId: string, title: string, roomId: number) {
     this.userId = userId
     this.title = title
+    this.prompt = undefined
     this.roomId = roomId
   }
 }

service/src/index.ts

@@ -77,13 +77,28 @@ export async function deleteChatRoom(userId: string, roomId: number) {
   return result
 }
 
+export async function updateRoomPrompt(userId: string, roomId: number, prompt: string) {
+  const query = { userId, roomId }
+  const update = {
+    $set: {
+      prompt,
+    },
+  }
+  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 = []
   await cursor.forEach(doc => rooms.push(doc))
   return rooms
 }
 
+export async function getChatRoom(userId: string, roomId: number) {
+  return await roomCol.findOne({ userId, roomId, status: { $ne: Status.Deleted } }) as ChatRoom
+}
+
 export async function existsChatRoom(userId: string, roomId: number) {
   const room = await roomCol.findOne({ roomId, userId })
   return !!room

service/src/storage/model.ts

@@ -134,6 +134,13 @@ export function fetchRenameChatRoom<T = any>(title: string, roomId: number) {
   })
 }
 
+export function fetchUpdateChatRoomPrompt<T = any>(prompt: string, roomId: number) {
+  return post<T>({
+    url: '/room-prompt',
+    data: { prompt, roomId },
+  })
+}
+
 export function fetchDeleteChatRoom<T = any>(roomId: number) {
   return post<T>({
     url: '/room-delete',

service/src/storage/mongo.ts

@@ -0,0 +1,73 @@
+<script lang="ts" setup>
+import { computed, ref } from 'vue'
+import { NButton, NInput, NModal, NSpace, useMessage } from 'naive-ui'
+import { t } from '@/locales'
+import { fetchUpdateChatRoomPrompt } from '@/api'
+import { useChatStore } from '@/store'
+
+const props = defineProps<Props>()
+const emit = defineEmits<Emit>()
+
+const chatStore = useChatStore()
+const currentChatHistory = computed(() => chatStore.getChatHistoryByCurrentActive)
+const ms = useMessage()
+const testing = ref(false)
+const title = `Prompt For [${currentChatHistory.value?.title}]`
+
+interface Props {
+  visible: boolean
+  roomId: string
+}
+
+interface Emit {
+  (e: 'update:visible', visible: boolean): void
+}
+
+const show = computed({
+  get() {
+    return props.visible
+  },
+  set(visible: boolean) {
+    emit('update:visible', visible)
+  },
+})
+
+async function handleSaveChatRoomPrompt() {
+  if (!currentChatHistory.value || !currentChatHistory.value)
+    return
+
+  testing.value = true
+  try {
+    const { message } = await fetchUpdateChatRoomPrompt(currentChatHistory.value.prompt ?? '', +props.roomId) as { status: string; message: string }
+    ms.success(message)
+    show.value = false
+  }
+  catch (error: any) {
+    ms.error(error.message)
+  }
+  testing.value = false
+}
+</script>
+
+<template>
+  <NModal
+    v-model:show="show" :auto-focus="false" class="custom-card" preset="card" :style="{ width: '600px' }" :title="title" size="huge"
+    :bordered="false"
+  >
+    <!-- <template #header-extra>
+      噢!
+    </template> -->
+    <NInput
+      :value="currentChatHistory && currentChatHistory.prompt"
+      type="textarea"
+      :autosize="{ minRows: 4, maxRows: 10 }" placeholder="Prompt for this room, If empty will use Settings ->  Advanced -> Role" @input="(val) => { if (currentChatHistory) currentChatHistory.prompt = val }"
+    />
+    <template #footer>
+      <NSpace justify="end">
+        <NButton :loading="testing" type="success" @click="handleSaveChatRoomPrompt">
+          {{ t('common.save') }}
+        </NButton>
+      </NSpace>
+    </template>
+  </NModal>
+</template>