Function select_model

deeptutor_cli/init_wizard.py:729–761

Numbered-list model picker with ``[c] Custom`` escape.

(
    console: Console,
    strings: dict[str, str],
    *,
    models: list[str],
    current: str = "",
    custom_prompt_label: str | None = None,
)

def select_model(
    console: Console,
    strings: dict[str, str],
    *,
    models: list[str],
    current: str = "",
    custom_prompt_label: str | None = None,
) -> str:
    """Numbered-list model picker with ``[c] Custom`` escape."""
    if not models:
        return typer.prompt(
            custom_prompt_label or strings["init.custom_model"],
            default=current or "",
        )

    options = [(m, m, "") for m in models]
    extra = {"c": strings["init.custom_model"]}
    default_key = current if current in models else models[0]
    pick = select_from_options(
        console,
        title=strings["init.pick_model"].format(marker="[c]"),
        options=options,
        default_key=default_key,
        extra_keys=extra,
        prompt_label=strings["init.choice"],
        invalid_label=strings["init.choice_invalid"],
    )
    if pick == "c":
        return typer.prompt(
            custom_prompt_label or strings["init.custom_model"],
            default=current or "",
        )
    return pick


# --- Connectivity probe --------------------------------------------------------