Function _resolve_embedding_choice

src/cocoindex_code/cli.py:343–415

Resolve the embedding settings per the init control-flow diagram. On a retry, ``previous`` holds the choice from the last attempt; its provider and model become the prompt defaults so the user only edits what was wrong instead of retyping everything.

(
    litellm_model_flag: str | None,
    st_installed: bool,
    tty: bool,
    previous: EmbeddingSettings | None = None,
)

def _resolve_embedding_choice(
    litellm_model_flag: str | None,
    st_installed: bool,
    tty: bool,
    previous: EmbeddingSettings | None = None,
) -> EmbeddingSettings:
    """Resolve the embedding settings per the init control-flow diagram.

    On a retry, ``previous`` holds the choice from the last attempt; its
    provider and model become the prompt defaults so the user only edits
    what was wrong instead of retyping everything.
    """
    if litellm_model_flag is not None:
        return EmbeddingSettings(provider="litellm", model=litellm_model_flag)

    if not tty:
        if st_installed:
            return EmbeddingSettings(provider="sentence-transformers", model=DEFAULT_ST_MODEL)
        _typer.echo(
            "Error: sentence-transformers is not installed and stdin is not a TTY.\n"
            "Either install the extra (`pip install 'cocoindex-code[embeddings-local]'`)\n"
            "or pass `--litellm-model MODEL` to select a LiteLLM model.",
            err=True,
        )
        raise _typer.Exit(code=1)

    # Interactive
    import questionary

    if st_installed:
        provider = questionary.select(
            "Embedding provider",
            choices=[
                questionary.Choice(
                    title="sentence-transformers (local, free — built-in HuggingFace models)",
                    value="sentence-transformers",
                ),
                questionary.Choice(
                    title="litellm (100+ providers — cloud APIs & local Ollama)",
                    value="litellm",
                ),
            ],
            default=previous.provider if previous is not None else None,
        ).ask()
    else:
        _typer.echo(
            "sentence-transformers is not installed — only `litellm` is available.\n"
            "To enable local embeddings, install `cocoindex-code[embeddings-local]`."
        )
        provider = "litellm"

    if provider is None:  # user cancelled (Ctrl-C / Esc)
        raise _typer.Exit(code=1)

    if provider == "sentence-transformers":
        default_model = previous.model if previous is not None else DEFAULT_ST_MODEL
        model = questionary.text(
            "Model name",