From 64b9367ca10b297158349f734e3c39881babc4c5 Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Mon, 28 Aug 2023 13:47:33 +0800
Subject: [PATCH] perf: quote output prompt

---
 .../src/pages/kb/detail/components/DataCard.tsx |  2 +-
 .../kb/detail/components/Import/Manual.tsx      |  4 ++--
 .../kb/detail/components/InputDataModal.tsx     |  4 ++--
 .../src/service/moduleDispatch/chat/oneapi.ts   | 17 +++++++++++------
 4 files changed, 16 insertions(+), 11 deletions(-)

diff --git a/client/src/pages/kb/detail/components/DataCard.tsx b/client/src/pages/kb/detail/components/DataCard.tsx
index d489affb7..a9fec1df5 100644
--- a/client/src/pages/kb/detail/components/DataCard.tsx
+++ b/client/src/pages/kb/detail/components/DataCard.tsx
@@ -163,7 +163,7 @@ const DataCard = ({ kbId }: { kbId: string }) => {
           maxW={['60%', '300px']}
           size={'sm'}
           value={searchText}
-          placeholder="根据匹配知识，补充知识和来源进行搜索"
+          placeholder="根据匹配知识，预期答案和来源进行搜索"
           onChange={(e) => {
             setSearchText(e.target.value);
             getFirstData();
diff --git a/client/src/pages/kb/detail/components/Import/Manual.tsx b/client/src/pages/kb/detail/components/Import/Manual.tsx
index c7ede324c..731c39f4f 100644
--- a/client/src/pages/kb/detail/components/Import/Manual.tsx
+++ b/client/src/pages/kb/detail/components/Import/Manual.tsx
@@ -95,7 +95,7 @@ const ManualImport = ({ kbId }: { kbId: string }) => {
         </Box>
         <Box flex={1} h={['50%', '100%']}>
           <Flex>
-            <Box h={'30px'}>{'补充知识'}</Box>
+            <Box h={'30px'}>{'预期答案'}</Box>
             <MyTooltip
               label={'匹配的知识点被命中后，这部分内容会随匹配知识点一起注入模型，引导模型回答'}
             >
@@ -104,7 +104,7 @@ const ManualImport = ({ kbId }: { kbId: string }) => {
           </Flex>
           <Textarea
             placeholder={
-              '补充知识。这部分内容不会被搜索，但会作为"匹配的知识点"的内容补充，你可以讲一些细节的内容填写在这里。总和最多 3000 字。'
+              '预期答案。这部分内容不会被搜索，但会作为"匹配的知识点"的内容补充，通常是问题的答案。总和最多 3000 字。'
             }
             h={['250px', '500px']}
             maxLength={3000}
diff --git a/client/src/pages/kb/detail/components/InputDataModal.tsx b/client/src/pages/kb/detail/components/InputDataModal.tsx
index 73856a581..47b3753b4 100644
--- a/client/src/pages/kb/detail/components/InputDataModal.tsx
+++ b/client/src/pages/kb/detail/components/InputDataModal.tsx
@@ -164,7 +164,7 @@ const InputDataModal = ({
           </Box>
           <Box flex={1} h={['50%', '100%']}>
             <Flex>
-              <Box h={'30px'}>{'补充知识'}</Box>
+              <Box h={'30px'}>{'预期答案'}</Box>
               <MyTooltip
                 label={'匹配的知识点被命中后，这部分内容会随匹配知识点一起注入模型，引导模型回答'}
               >
@@ -173,7 +173,7 @@ const InputDataModal = ({
             </Flex>
             <Textarea
               placeholder={
-                '补充知识。这部分内容不会被搜索，但会作为"匹配的知识点"的内容补充，你可以讲一些细节的内容填写在这里。总和最多 3000 字。'
+                '预期答案。这部分内容不会被搜索，但会作为"匹配的知识点"的内容补充，通常是问题的答案。总和最多 3000 字。'
               }
               maxLength={3000}
               resize={'none'}
diff --git a/client/src/service/moduleDispatch/chat/oneapi.ts b/client/src/service/moduleDispatch/chat/oneapi.ts
index 7443903f2..44560074b 100644
--- a/client/src/service/moduleDispatch/chat/oneapi.ts
+++ b/client/src/service/moduleDispatch/chat/oneapi.ts
@@ -68,7 +68,7 @@ export const dispatchChatCompletion = async (props: Record<string, any>): Promis
     return Promise.reject('The chat model is undefined, you need to select a chat model.');
   }
 
-  const { filterQuoteQA, quotePrompt } = filterQuote({
+  const { filterQuoteQA, quotePrompt, hasQuoteOutput } = filterQuote({
     quoteQA,
     model: modelConstantsData
   });
@@ -89,7 +89,8 @@ export const dispatchChatCompletion = async (props: Record<string, any>): Promis
     quotePrompt,
     userChatInput,
     systemPrompt,
-    limitPrompt
+    limitPrompt,
+    hasQuoteOutput
   });
   const { max_tokens } = getMaxTokens({
     model: modelConstantsData,
@@ -219,7 +220,8 @@ function filterQuote({
 
   return {
     filterQuoteQA,
-    quotePrompt
+    quotePrompt,
+    hasQuoteOutput: !!filterQuoteQA.find((item) => item.a)
   };
 }
 function getChatMessages({
@@ -228,7 +230,8 @@ function getChatMessages({
   systemPrompt,
   limitPrompt,
   userChatInput,
-  model
+  model,
+  hasQuoteOutput
 }: {
   quotePrompt: string;
   history: ChatProps['history'];
@@ -236,13 +239,15 @@ function getChatMessages({
   limitPrompt: string;
   userChatInput: string;
   model: ChatModelItemType;
+  hasQuoteOutput: boolean;
 }) {
   const limitText = (() => {
     if (!quotePrompt) {
       return limitPrompt;
     }
-    const defaultPrompt =
-      '三引号是我提供给你的专属知识，它们拥有最高优先级。instruction 是相关介绍，output 是预期回答，使用引用内容来回答我下面的问题。';
+    const defaultPrompt = `三引号引用的内容是我提供给你的知识，它们拥有最高优先级。instruction 是相关介绍${
+      hasQuoteOutput ? '，output 是预期回答或补充' : ''
+    }，使用引用内容来回答我下面的问题。`;
     if (limitPrompt) {
       return `${defaultPrompt}${limitPrompt}`;
     }