Function chooseEmbedding

packages/cli/src/commands/setup-pi.ts:192–224

(prompts: PromptIO)

}

async function chooseEmbedding(prompts: PromptIO): Promise<EmbeddingChoice> {
    const provider = await prompts.selectOne("Select embedding provider", [
        {
            label: "Local embeddings — no API key required",
            value: "local",
            recommended: true,
        },
        { label: "OpenAI-compatible endpoint", value: "openai-compatible" },
    ]);

    if (provider === "local") {
        return { provider: "local", model: "Xenova/all-MiniLM-L6-v2" };
    }

    const endpoint = await prompts.text("Embedding endpoint URL", {
        placeholder: "https://api.openai.com/v1",
        validate: (value) => (value.trim().length === 0 ? "Endpoint is required" : undefined),
    });
    const model = await prompts.text("Embedding model", {
        initialValue: "text-embedding-3-small",
        validate: (value) => (value.trim().length === 0 ? "Model is required" : undefined),
    });
    const apiKey = await prompts.text("Embedding API key (optional; leave blank to use env)", {
        placeholder: "optional",
    });

    return compactObject({
        provider: "openai-compatible" as const,
        endpoint: endpoint.trim(),
        model: model.trim(),
        api_key: apiKey.trim() || undefined,
    });
}

export async function runSetup(options: RunSetupOptions = {}): Promise<number> {
    const prompts = options.prompts ?? (await getDefaultPrompts());