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}