feat: quote source

src/pages/api/chat/history/updateHistoryQuote.ts

@@ -27,7 +27,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       },
       {
         $set: {
-          'content.$.quote.$[quoteElem].isEdit': true
+          'content.$.quote.$[quoteElem].source': '手动修改'
         }
       },
       {

src/pages/api/chat/saveChat.ts

@@ -58,11 +58,7 @@ export async function saveChat({
     obj: item.obj,
     value: item.value,
     systemPrompt: item.systemPrompt,
-    quote:
+    quote: item.quote || []
-      item.quote?.map((item) => ({
-        ...item,
-        isEdit: false
-      })) || []
   }));
 
   const [id] = await Promise.all([

src/pages/api/openapi/kb/appKbSearch.ts

@@ -12,7 +12,7 @@ import { ChatRoleEnum } from '@/constants/chat';
 import { openaiEmbedding } from '../plugin/openaiEmbedding';
 import { modelToolMap } from '@/utils/plugin';
 
-export type QuoteItemType = { id: string; q: string; a: string; isEdit: boolean };
+export type QuoteItemType = { id: string; q: string; a: string; source?: string };
 type Props = {
   prompts: ChatItemSimpleType[];
   similarity: number;
@@ -92,7 +92,7 @@ export async function appKbSearch({
 
   // search kb
   const { rows: searchRes } = await PgClient.select<QuoteItemType>('modelData', {
-    fields: ['id', 'q', 'a'],
+    fields: ['id', 'q', 'a', 'source'],
     where: [
       `kb_id IN (${model.chat.relatedKbs.map((item) => `'${item}'`).join(',')})`,
       'AND',

src/pages/chat/components/QuoteModal.tsx

@@ -126,7 +126,7 @@ const QuoteModal = ({
                 position={'relative'}
                 _hover={{ '& .edit': { display: 'flex' } }}
               >
-                {item.isEdit && <Box color={'myGray.600'}>(编辑过)</Box>}
+                {item.source && <Box color={'myGray.600'}>({item.source})</Box>}
                 <Box>{item.q}</Box>
                 <Box>{item.a}</Box>
                 <Box

src/service/models/chat.ts

@@ -60,7 +60,7 @@ const ChatSchema = new Schema({
               id: String,
               q: String,
               a: String,
-              isEdit: Boolean
+              source: String
             }
           ],
           default: []