From 8036ed61434163a7e065f14fedf3be0cc2890191 Mon Sep 17 00:00:00 2001 From: archer <545436317@qq.com> Date: Wed, 14 Jun 2023 14:33:26 +0800 Subject: [PATCH] perf: qa --- client/src/pages/api/openapi/kb/appKbSearch.ts | 2 +- client/src/pages/kb/components/SelectFileModal.tsx | 6 +++--- client/src/service/events/generateQA.ts | 6 +++--- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/client/src/pages/api/openapi/kb/appKbSearch.ts b/client/src/pages/api/openapi/kb/appKbSearch.ts index 74e6bac21..69be8cd89 100644 --- a/client/src/pages/api/openapi/kb/appKbSearch.ts +++ b/client/src/pages/api/openapi/kb/appKbSearch.ts @@ -103,7 +103,7 @@ export async function appKbSearch({ .map((item) => `'${item}'`) .join(',')}) AND vector <#> '[${promptVector[0]}]' < -${similarity} order by vector <#> '[${ promptVector[0] - }]' limit 15; + }]' limit 10; COMMIT;` ); diff --git a/client/src/pages/kb/components/SelectFileModal.tsx b/client/src/pages/kb/components/SelectFileModal.tsx index 1018d5cee..925b218f6 100644 --- a/client/src/pages/kb/components/SelectFileModal.tsx +++ b/client/src/pages/kb/components/SelectFileModal.tsx @@ -29,14 +29,14 @@ const fileExtension = '.txt,.doc,.docx,.pdf,.md'; const modeMap = { [TrainingModeEnum.qa]: { - maxLen: 9000, + maxLen: 8000, slideLen: 3000, price: ChatModelMap[OpenAiChatEnum.GPT35].price, isPrompt: true }, [TrainingModeEnum.index]: { - maxLen: 2000, - slideLen: 600, + maxLen: 1400, + slideLen: 700, price: embeddingPrice, isPrompt: false } diff --git a/client/src/service/events/generateQA.ts b/client/src/service/events/generateQA.ts index 7714f98ce..f603a6bcd 100644 --- a/client/src/service/events/generateQA.ts +++ b/client/src/service/events/generateQA.ts @@ -94,9 +94,9 @@ export async function generateQA(): Promise { messages: [ { obj: ChatRoleEnum.System, - value: `你是出题人 -${data.prompt || '下面是"一段长文本"'} -从中选出15至30个题目和答案.答案详细.按格式返回: Q1: + value: `你是出题人. +${data.prompt || '用户会发送一段长文本'}. +从中选出 25 个问题和答案. 答案详细完整. 按格式回答: Q1: A1: Q2: A2: