// Memory Core plugin module implements embedding mocks behavior.

import { vi, type Mock } from "vitest";

import { vi } from "vitest";

import "./test-runtime-mocks.js";

// Avoid exporting vitest mock types (TS2742 under pnpm + d.ts emit).

type EmbedBatchMock = Mock<(texts: string[]) => Promise<number[][]>>;

const hoisted = vi.hoisted(() => ({

embedBatch: vi.fn(async (texts: string[]) => texts.map(() => [0, 1, 0])),

embedQuery: vi.fn(async () => [0, 1, 0]),

}));

export function getEmbedBatchMock(): EmbedBatchMock {

return hoisted.embedBatch;

}

export function resetEmbeddingMocks(): void {

hoisted.embedBatch.mockReset();

hoisted.embedQuery.mockReset();

return listBundledChannelPluginIdsForRoot(rootScope.cacheKey);

}

/**

* Iterates bundled bootstrap channel plugins that can be loaded successfully.

*/

export function* iterateBootstrapChannelPlugins(): IterableIterator<ChannelPlugin> {

for (const id of listBootstrapChannelPluginIds()) {

const plugin = getBootstrapChannelPlugin(id);

if (plugin) {

yield plugin;

}

}

}

/**

* Loads a bundled channel plugin for bootstrap, merging runtime and setup artifacts.

*/