feat: Make knowledge base reference document count configurable

This commit is contained in:
kangfenmao 2025-02-08 20:56:59 +08:00
parent 3d2e209550
commit 220d11a414

View File

@ -90,8 +90,10 @@ export const getKnowledgeReferences = async (base: KnowledgeBase, message: Messa
}) })
) )
const documentCount = base.documentCount || 6
const references = await Promise.all( const references = await Promise.all(
take(_searchResults, 6).map(async (item, index) => { take(_searchResults, documentCount).map(async (item, index) => {
const baseItem = base.items.find((i) => i.uniqueId === item.metadata.uniqueLoaderId) const baseItem = base.items.find((i) => i.uniqueId === item.metadata.uniqueLoaderId)
return { return {
id: index, id: index,