Function getAugmentedContext

packages/baseai/src/utils/memory/lib.ts:112–134

({
	similarChunks,
	messages
}: {
	similarChunks: SimilarChunk[];
	messages: Message[];
})

};

export const getAugmentedContext = ({
	similarChunks,
	messages
}: {
	similarChunks: SimilarChunk[];
	messages: Message[];
}) => {
	if (similarChunks.length === 0) return '';

	const memoryContext = similarChunks
		.map(chunk => `${chunk.text} \n\n Source: ${chunk.attributes.docName}`)
		.join('\n\n');

	// Extract Rag prompt from the messages.
	const ragMsg = messages.find(m => m.role === 'system' && m.name === 'rag');
	// If there is no rag prompt, use the default rag prompt.
	const ragPrompt = ragMsg?.content || defaultRagPrompt;

	const contextContent = `"""CONTEXT:\n ${memoryContext}"""`;
	const augmentedContext = `"""${ragPrompt}""" \n\n ${contextContent}`;

	return augmentedContext;
};

export const addContextFromMemory = async ({
	messages,