|
19 | 19 | from rich.table import Column |
20 | 20 |
|
21 | 21 | from beeai_cli.api import api_request |
22 | | -from beeai_cli.async_typer import AsyncTyper, console, create_table, err_console |
| 22 | +from beeai_cli.async_typer import AsyncTyper, console, create_table |
23 | 23 | from beeai_cli.configuration import Configuration |
24 | 24 | from beeai_cli.utils import format_error, parse_env_var, run_command, verbosity |
25 | 25 |
|
@@ -387,7 +387,7 @@ async def setup( |
387 | 387 | ) |
388 | 388 | response_text = test_response.json().get("choices", [{}])[0].get("message", {}).get("content", "") |
389 | 389 | if "Hello" not in response_text: |
390 | | - err_console.print(format_error("Error", "Model did not provide a proper response.")) |
| 390 | + console.print(format_error("Error", "Model did not provide a proper response.")) |
391 | 391 | return False |
392 | 392 |
|
393 | 393 | selected_embedding_model = None |
@@ -442,7 +442,7 @@ async def setup( |
442 | 442 | return False |
443 | 443 |
|
444 | 444 | except Exception as e: |
445 | | - err_console.print(format_error("Error", f"Error during model test: {e!s}")) |
| 445 | + console.print(format_error("Error", f"Error during model test: {e!s}")) |
446 | 446 | return False |
447 | 447 |
|
448 | 448 | if not use_true_localhost: |
@@ -489,10 +489,8 @@ async def ensure_llm_env(): |
489 | 489 | console.print("[bold]Welcome to 🐝 [red]BeeAI[/red]![/bold]") |
490 | 490 | console.print("Let's start by configuring your LLM environment.\n") |
491 | 491 | if not await setup(): |
492 | | - err_console.print( |
493 | | - format_error("Error", "Could not continue because the LLM environment is not properly set up.") |
494 | | - ) |
495 | | - err_console.print( |
| 492 | + console.print(format_error("Error", "Could not continue because the LLM environment is not properly set up.")) |
| 493 | + console.print( |
496 | 494 | "💡 [yellow]HINT[/yellow]: Try re-entering your LLM API details with: [green]beeai env setup[/green]" |
497 | 495 | ) |
498 | 496 | sys.exit(1) |
|
0 commit comments