feat: support built-in prompt type (#628)

* feat: add built-in prompt in prompt-list api

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

* feat: support prompt type tag

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

* refactor: prompt store remove use local storage

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

---------

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

Author: BobDu

package.json

@@ -49,7 +49,7 @@
     "lint-staged": "^16.1.0",
     "markdown-it": "^14.1.0",
     "markdown-it-link-attributes": "^4.0.1",
-    "naive-ui": "~2.40.4",
+    "naive-ui": "^2.41.1",
     "pinia": "^3.0.2",
     "qrcode.vue": "^3.6.0",
     "rimraf": "^6.0.1",

pnpm-lock.yaml

@@ -5,6 +5,7 @@ import { auth } from '../middleware/auth'
 import {
   clearUserPrompt,
   deleteUserPrompt,
+  getBuiltInPromptList,
   getUserPromptList,
   importUserPrompt,
   upsertUserPrompt,
@@ -15,16 +16,45 @@ export const router = Router()
 router.get('/prompt-list', auth, async (req, res) => {
   try {
     const userId = req.headers.userId as string
-    const prompts = await getUserPromptList(userId)
-    const result = []
-    prompts.data.forEach((p) => {
-      result.push({
+
+    // 获取用户自定义提示词
+    const userPrompts = await getUserPromptList(userId)
+    const userResult = []
+    userPrompts.data.forEach((p) => {
+      userResult.push({
+        _id: p._id,
+        title: p.title,
+        value: p.value,
+        type: 'user-defined',
+      })
+    })
+
+    // 获取平台预置提示词
+    const builtInPrompts = await getBuiltInPromptList()
+    const builtInResult = []
+    builtInPrompts.data.forEach((p) => {
+      builtInResult.push({
         _id: p._id,
         title: p.title,
         value: p.value,
+        type: 'built-in',
       })
     })
-    res.send({ status: 'Success', message: null, data: { data: result, total: prompts.total } })
+
+    // 合并两种类型的提示词
+    const allPrompts = [...userResult, ...builtInResult]
+    const totalCount = userPrompts.total + builtInPrompts.total
+
+    res.send({
+      status: 'Success',
+      message: null,
+      data: {
+        data: allPrompts,
+        total: totalCount,
+        userTotal: userPrompts.total,
+        builtInTotal: builtInPrompts.total,
+      },
+    })
   }
   catch (error) {
     res.send({ status: 'Fail', message: error.message, data: null })

service/src/routes/prompt.ts

@@ -302,6 +302,16 @@ export class KeyConfig {
   }
 }
 
+export class BuiltInPrompt {
+  _id: ObjectId
+  title: string
+  value: string
+  constructor(title: string, value: string) {
+    this.title = title
+    this.value = value
+  }
+}
+
 export class UserPrompt {
   _id: ObjectId
   userId: string

service/src/storage/model.ts

@@ -1,6 +1,7 @@
 import type { WithId } from 'mongodb'
 import type {
   AdvancedConfig,
+  BuiltInPrompt,
   ChatOptions,
   Config,
   GiftCard,
@@ -39,6 +40,7 @@ const userCol = client.db(dbName).collection<UserInfo>('user')
 const configCol = client.db(dbName).collection<Config>('config')
 const usageCol = client.db(dbName).collection<ChatUsage>('chat_usage')
 const keyCol = client.db(dbName).collection<KeyConfig>('key_config')
+const builtInPromptCol = client.db(dbName).collection<BuiltInPrompt>('built_in_prompt')
 const userPromptCol = client.db(dbName).collection<UserPrompt>('user_prompt')
 // 新增兑换券的数据库
 // {
@@ -664,6 +666,13 @@ export async function updateApiKeyStatus(id: string, status: Status) {
   await keyCol.updateOne({ _id: new ObjectId(id) }, { $set: { status } })
 }
 
+export async function getBuiltInPromptList(): Promise<{ data: BuiltInPrompt[], total: number }> {
+  const total = await builtInPromptCol.countDocuments()
+  const cursor = builtInPromptCol.find().sort({ _id: -1 })
+  const data = await cursor.toArray()
+  return { data, total }
+}
+
 export async function upsertUserPrompt(userPrompt: UserPrompt): Promise<UserPrompt> {
   if (userPrompt._id === undefined) {
     const doc = await userPromptCol.insertOne(userPrompt)

service/src/storage/mongo.ts

@@ -13,8 +13,10 @@ interface DataProps {
   _id?: string
   renderKey: string
   renderValue: string
+  renderType: string
   title: string
   value: string
+  type: 'built-in' | 'user-defined'
 }
 
 interface Props {
@@ -280,9 +282,11 @@ function renderTemplate() {
     return {
       renderKey: item.title.length <= keyLimit ? item.title : `${item.title.substring(0, keyLimit)}...`,
       renderValue: item.value.length <= valueLimit ? item.value : `${item.value.substring(0, valueLimit)}...`,
+      renderType: item.type === 'built-in' ? t('store.builtIn') : t('store.userDefined'),
       title: item.title,
       value: item.value,
       _id: item._id,
+      type: item.type,
     }
   })
 }
@@ -298,6 +302,10 @@ const pagination = computed(() => {
 // table相关
 function createColumns(): DataTableColumns<DataProps> {
   return [
+    {
+      title: 'type',
+      key: 'renderType',
+    },
     {
       title: t('store.title'),
       key: 'renderKey',
@@ -312,6 +320,9 @@ function createColumns(): DataTableColumns<DataProps> {
       width: 100,
       align: 'center',
       render(row) {
+        if (row.type === 'built-in') {
+          return ''
+        }
         return h('div', { class: 'flex items-center flex-col gap-2' }, {
           default: () => [h(
             NButton,
@@ -349,12 +360,10 @@ watch(
 )
 
 onMounted(async () => {
-  if (!!authStore.session?.auth && !authStore.token)
+  if (!authStore.session?.auth)
     return
-  if (promptStore.getPromptList().promptList.length === 0) {
-    await handleGetUserPromptList()
-    promptStore.updatePromptList(promptList.value)
-  }
+  await handleGetUserPromptList()
+  promptStore.updatePromptList(promptList.value)
 })
 
 const dataSource = computed(() => {

src/components/common/PromptStore/index.vue

@@ -0,0 +1,25 @@
+<script setup lang="ts">
+import { t } from '@/locales'
+
+interface Props {
+  type: 'built-in' | 'user-defined'
+}
+
+defineProps<Props>()
+</script>
+
+<template>
+  <NTag
+    size="small"
+    :type="type === 'built-in' ? 'success' : 'info'"
+    round
+    style="
+      width: 85px;
+      display: inline-flex;
+      justify-content: center;
+      align-items: center;
+    "
+  >
+    {{ type === 'built-in' ? t('store.builtIn') : t('store.userDefined') }}
+  </NTag>
+</template>

src/components/common/PromptTypeTag/index.vue

@@ -115,6 +115,7 @@ export class UserPrompt {
   _id?: string
   title: string
   value: string
+  type?: 'built-in' | 'user-defined'
   constructor(title: string, value: string) {
     this.title = title
     this.value = value

src/components/common/Setting/model.ts

@@ -2,9 +2,10 @@ import GithubSite from './GitHubSite/index.vue'
 import HoverButton from './HoverButton/index.vue'
 import NaiveProvider from './NaiveProvider/index.vue'
 import PromptStore from './PromptStore/index.vue'
+import PromptTypeTag from './PromptTypeTag/index.vue'
 import Setting from './Setting/index.vue'
 import SvgIcon from './SvgIcon/index.vue'
 import UserAvatar from './UserAvatar/index.vue'
 import Watermark from './Watermark/index.vue'
 
-export { GithubSite, HoverButton, NaiveProvider, PromptStore, Setting, SvgIcon, UserAvatar, Watermark }
+export { GithubSite, HoverButton, NaiveProvider, PromptStore, PromptTypeTag, Setting, SvgIcon, UserAvatar, Watermark }

src/components/common/index.ts

@@ -241,5 +241,7 @@ export default {
     importRepeatContent: 'Content is repeatedly skipped: {msg}',
     onlineImportWarning: 'Note: Please check the JSON file source!',
     downloadError: 'Please check the network status and JSON file validity',
+    builtIn: 'Built-in',
+    userDefined: 'User defined',
   },
 }