e1a25caa52
Co-authored-by: Bastien <bastien.baret@gmail.com> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Julien Legrand <72564015+JulienLGRD@users.noreply.github.com> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
|
|
from rich import print as rprint
|
|
from textual.app import App
|
|
|
|
from vibe.core.paths import GLOBAL_ENV_FILE
|
|
from vibe.setup.onboarding.screens import ApiKeyScreen, WelcomeScreen
|
|
|
|
|
|
class OnboardingApp(App[str | None]):
|
|
CSS_PATH = "onboarding.tcss"
|
|
|
|
def on_mount(self) -> None:
|
|
self.theme = "textual-ansi"
|
|
|
|
self.install_screen(WelcomeScreen(), "welcome")
|
|
self.install_screen(ApiKeyScreen(), "api_key")
|
|
self.push_screen("welcome")
|
|
|
|
|
|
def run_onboarding(app: App | None = None) -> None:
|
|
result = (app or OnboardingApp()).run()
|
|
match result:
|
|
case None:
|
|
rprint("\n[yellow]Setup cancelled. See you next time![/]")
|
|
sys.exit(0)
|
|
case str() as s if s.startswith("env_var_error:"):
|
|
env_key = s.removeprefix("env_var_error:")
|
|
rprint(
|
|
"\n[yellow]Could not save the API key because this provider is "
|
|
f"configured with an invalid environment variable name: {env_key}.[/]"
|
|
"\n[dim]The API key was not saved for this session. "
|
|
"Update the provider's `api_key_env_var` setting in your config and try again.[/]\n"
|
|
)
|
|
sys.exit(1)
|
|
case str() as s if s.startswith("save_error:"):
|
|
err = s.removeprefix("save_error:")
|
|
rprint(
|
|
f"\n[yellow]Warning: Could not save API key to .env file: {err}[/]"
|
|
"\n[dim]The API key is set for this session only. "
|
|
f"You may need to set it manually in {GLOBAL_ENV_FILE.path}[/]\n"
|
|
)
|
|
case "completed":
|
|
rprint(
|
|
'\nSetup complete 🎉. Run "vibe" to start using the Mistral Vibe CLI.\n'
|
|
)
|