Function embedMemory

packages/baseai/src/memory/embed.ts:15–108

({
	memoryName: memoryNameInput,
	overwrite,
	useLocalEmbeddings
}: {
	memoryName: string;
	overwrite?: boolean;
	useLocalEmbeddings?: boolean;
})

import color from 'picocolors';

export async function embedMemory({
	memoryName: memoryNameInput,
	overwrite,
	useLocalEmbeddings
}: {
	memoryName: string;
	overwrite?: boolean;
	useLocalEmbeddings?: boolean;
}) {
	// Spinner to show current action.
	const s = p.spinner();

	try {
		p.intro(
			heading({
				text: 'EMBED',
				sub: `Creating embeddings of ${color.cyan(memoryNameInput)}`
			})
		);

		if (!memoryNameInput) {
			p.cancel(
				'Memory name is required. Use --memory or -m flag to specify.'
			);
			process.exit(1);
		}

		// 1- Check memory exists.
		const memoryName = validateMemoryName(memoryNameInput);
		await checkMemoryExists(memoryName);

		// 2- Load memory data.
		s.start('Processing memory docs...');
		let memoryFiles = await loadMemoryFiles(memoryName);

		if (memoryFiles.length === 0) {
			p.cancel(`No valid documents found in memory '${memoryName}'.`);
			process.exit(1);
		}

		// 3- Get memory config.
		const memoryConfig = await loadMemoryConfig(memoryName);

		let filesToEmbed: string[] = [];
		let filesToDelete: string[] = [];

		if (memoryConfig.git.enabled) {
			const { filesToDeploy, filesToDelete: gitFilesToDelete } =
				await handleGitSyncMemories({
					memoryName: memoryName,
					config: memoryConfig
				});

			filesToEmbed = filesToDeploy;
			filesToDelete = gitFilesToDelete;

			// Filter memory files to emebed
			memoryFiles = memoryFiles.filter(doc =>