Function run

packages/plugin/scripts/embedding-baseline.ts:252–401

()

}

async function run() {
    const args = parseArgs();

    // 1. Load embedding config from user's magic-context.jsonc so we match
    //    whatever provider is currently active for the plugin.
    //    Run the raw input through the same Zod schema the plugin uses so we
    //    get the canonical discriminated-union shape initializeEmbedding expects.
    const rawConfig = await loadEmbeddingConfig(args.configPath);
    const parsed = EmbeddingConfigSchema.safeParse(rawConfig);
    if (!parsed.success) {
        console.error(
            `[baseline] embedding config failed validation: ${parsed.error.issues.map((i) => `${i.path.join(".")}: ${i.message}`).join("; ")}`,
        );
        process.exit(1);
    }
    const embeddingConfig = parsed.data;
    console.log("[baseline] embedding config:", embeddingConfig);
    initializeEmbedding(embeddingConfig);

    const modelReady = await ensureEmbeddingModel();
    if (!modelReady) {
        console.error("[baseline] embedding provider failed to initialize. Aborting.");
        process.exit(1);
    }

    const modelId = getEmbeddingModelId();
    console.log(`[baseline] model_id: ${modelId}`);

    // 2. Open the plugin's live context DB (read-only) and load memories + embeddings
    //    for the current project.
    const dbPath = join(homedir(), ".local/share/opencode/storage/plugin/magic-context/context.db");
    if (!existsSync(dbPath)) {
        console.error(`[baseline] plugin DB not found at ${dbPath}`);
        process.exit(1);
    }
    const db = new Database(dbPath, { readonly: true });

    const projectIdentity = resolveProjectIdentity(process.cwd());
    console.log(`[baseline] project_identity: ${projectIdentity}`);

    const memories = getMemoriesByProject(db, projectIdentity, ["active", "permanent"]);
    const embeddings = loadAllEmbeddings(db, projectIdentity, modelId);
    console.log(
        `[baseline] memories_considered=${memories.length} embeddings_loaded=${embeddings.size}`,
    );

    if (embeddings.size === 0) {
        console.error(
            "[baseline] no embeddings in DB for current project. Did the background embed sweep run yet?",
        );
        process.exit(1);
    }

    const memoriesById = new Map(memories.map((m) => [m.id, m]));

    // 3. Embed queries and score.
    const queryLatencies: number[] = [];
    let embeddingDim: number | null = null;