From 084cf266a2bf296aaa26e2acf52867216f6a0f32 Mon Sep 17 00:00:00 2001 From: Danny Avila Date: Fri, 7 Jun 2024 20:51:43 -0400 Subject: [PATCH] =?UTF-8?q?=F0=9F=97=83=EF=B8=8F=20fix:=20revise=20RAG=20p?= =?UTF-8?q?rompt=20(#3006)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../clients/prompts/createContextHandlers.js | 39 ++++++++++--------- 1 file changed, 20 insertions(+), 19 deletions(-) diff --git a/api/app/clients/prompts/createContextHandlers.js b/api/app/clients/prompts/createContextHandlers.js index e48dfd8e67..4dcfaf68e4 100644 --- a/api/app/clients/prompts/createContextHandlers.js +++ b/api/app/clients/prompts/createContextHandlers.js @@ -8,8 +8,6 @@ In your response, remember to follow these guidelines: - If you don't know the answer, simply say that you don't know. - If you are unsure how to answer, ask for clarification. - Avoid mentioning that you obtained the information from the context. - -Answer appropriately in the user's language. `; function createContextHandlers(req, userMessageContent) { @@ -94,37 +92,40 @@ function createContextHandlers(req, userMessageContent) { const resolvedQueries = await Promise.all(queryPromises); - const context = resolvedQueries - .map((queryResult, index) => { - const file = processedFiles[index]; - let contextItems = queryResult.data; + const context = + resolvedQueries.length === 0 + ? '\n\tThe semantic search did not return any results.' + : resolvedQueries + .map((queryResult, index) => { + const file = processedFiles[index]; + let contextItems = queryResult.data; - const generateContext = (currentContext) => - ` + const generateContext = (currentContext) => + ` ${file.filename} ${currentContext} `; - if (useFullContext) { - return generateContext(`\n${contextItems}`); - } + if (useFullContext) { + return generateContext(`\n${contextItems}`); + } - contextItems = queryResult.data - .map((item) => { - const pageContent = item[0].page_content; - return ` + contextItems = queryResult.data + .map((item) => { + const pageContent = item[0].page_content; + return ` `; + }) + .join(''); + + return generateContext(contextItems); }) .join(''); - return generateContext(contextItems); - }) - .join(''); - if (useFullContext) { const prompt = `${header} ${context}