Function getDocumentsFromMemory

packages/baseai/src/utils/memory/db/lib.ts:181–213

(
	memoryNames: string[]
)

}

export async function getDocumentsFromMemory(
	memoryNames: string[]
): Promise<MemoryChunk[]> {
	const memoryChunks: MemoryChunk[] = [];

	// Load each memory and get the documents
	for (const memoryName of memoryNames) {
		// Load the memory
		const db: Low<Schema> = await loadDb(memoryName);

		// Process the documents
		for (const [docName, document] of Object.entries(db.data.documents)) {
			// Get the chunks for the document
			const chunks = document.chunkIds.map(id => {
				const chunk = db.data.chunks[id];
				return {
					text: chunk.text,
					embedding: chunk.embedding,
					attributes: {
						memoryName,
						docName
					}
				};
			});

			// Add the processed document to the list
			memoryChunks.push(...chunks);
		}
	}

	//  Return the processed documents
	return memoryChunks;
}

export function cosineSimilaritySearch({
	chunks,