This commit is contained in:
archer
2023-06-14 14:33:26 +08:00
parent 85e6a0f38d
commit 8036ed6143
3 changed files with 7 additions and 7 deletions

View File

@@ -103,7 +103,7 @@ export async function appKbSearch({
.map((item) => `'${item}'`) .map((item) => `'${item}'`)
.join(',')}) AND vector <#> '[${promptVector[0]}]' < -${similarity} order by vector <#> '[${ .join(',')}) AND vector <#> '[${promptVector[0]}]' < -${similarity} order by vector <#> '[${
promptVector[0] promptVector[0]
}]' limit 15; }]' limit 10;
COMMIT;` COMMIT;`
); );

View File

@@ -29,14 +29,14 @@ const fileExtension = '.txt,.doc,.docx,.pdf,.md';
const modeMap = { const modeMap = {
[TrainingModeEnum.qa]: { [TrainingModeEnum.qa]: {
maxLen: 9000, maxLen: 8000,
slideLen: 3000, slideLen: 3000,
price: ChatModelMap[OpenAiChatEnum.GPT35].price, price: ChatModelMap[OpenAiChatEnum.GPT35].price,
isPrompt: true isPrompt: true
}, },
[TrainingModeEnum.index]: { [TrainingModeEnum.index]: {
maxLen: 2000, maxLen: 1400,
slideLen: 600, slideLen: 700,
price: embeddingPrice, price: embeddingPrice,
isPrompt: false isPrompt: false
} }

View File

@@ -94,9 +94,9 @@ export async function generateQA(): Promise<any> {
messages: [ messages: [
{ {
obj: ChatRoleEnum.System, obj: ChatRoleEnum.System,
value: `你是出题人 value: `你是出题人.
${data.prompt || '下面是"一段长文本"'} ${data.prompt || '用户会发送一段长文本'}.
从中选出15至30个题目和答案.答案详细.按格式回: Q1: 从中选出 25 个问题和答案. 答案详细完整. 按格式回: Q1:
A1: A1:
Q2: Q2:
A2: A2: