ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-22 04:06:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add answer quote number (#4658)

Browse Source
This commit is contained in:
heheer
2025-04-25 10:34:37 +08:00
committed by GitHub
parent 2a54be4d91
commit 25dc8984be
9 changed files with 215 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
packages/service/core/workflow/dispatch/chat/oneapi.ts
View File

@@ -338,7 +338,17 @@ async function filterDatasetQuote({
model: LLMModelItemType;
quoteTemplate: string;
}) {
function getValue(item: SearchDataResponseItemType, index: number) {
function getValue({
item,
index,
sourceList
}: {
item: SearchDataResponseItemType;
index: number;
sourceList: { sourceName: string; sourceId: string; sourceIndex: number }[];
}) {
const source = sourceList.find((source) => source.sourceId === item.sourceId);
return replaceVariable(quoteTemplate, {
id: item.id,
q: item.q,
@@ -346,6 +356,7 @@ async function filterDatasetQuote({
updateTime: formatTime2YMDHM(item.updateTime),
source: item.sourceName,
sourceId: String(item.sourceId || ''),
sourceIndex: source?.sourceIndex || 1,
index: index + 1
});
}
@@ -353,9 +364,24 @@ async function filterDatasetQuote({
// slice filterSearch
const filterQuoteQA = await filterSearchResultsByMaxChars(quoteQA, model.quoteMaxToken);
const sourceList = Object.values(
filterQuoteQA.reduce((acc: Record<string, SearchDataResponseItemType[]>, cur) => {
if (!acc[cur.collectionId]) {
acc[cur.collectionId] = [cur];
}
return acc;
}, {})
)
.flat()
.map((item, index) => ({
sourceName: item.sourceName || '',
sourceId: item.sourceId || '',
sourceIndex: index + 1
}));
const datasetQuoteText =
filterQuoteQA.length > 0
? `${filterQuoteQA.map((item, index) => getValue(item, index).trim()).join('\n------\n')}`
? `${filterQuoteQA.map((item, index) => getValue({ item, index, sourceList }).trim()).join('\n------\n')}`
: '';
return {