ilia/nanobot
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix: unify exit cleanup, conditionally show spinner with --logs flag

Browse Source
This commit is contained in:
Re-bin 2026-02-09 05:15:26 +00:00
parent 391ee21275
commit d47219ef6a
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
nanobot/cli/commands.py
View File

@ -144,7 +144,7 @@ def _print_agent_response(response: str, render_markdown: bool) -> None:
console.print( console.print(
Panel( Panel(
body, body,
title=f"{__logo__} Nanobot", title=f"{__logo__} nanobot",
title_align="left", title_align="left",
border_style="cyan", border_style="cyan",
padding=(0, 1), padding=(0, 1),
@ -464,10 +464,17 @@ def agent(
restrict_to_workspace=config.tools.restrict_to_workspace, restrict_to_workspace=config.tools.restrict_to_workspace,
) )
# Show spinner when logs are off (no output to miss); skip when logs are on
def _thinking_ctx():
if logs:
from contextlib import nullcontext
return nullcontext()
return console.status("[dim]nanobot is thinking...[/dim]", spinner="dots")
if message: if message:
# Single message mode # Single message mode
async def run_once(): async def run_once():
with console.status("[dim]Nanobot is thinking...[/dim]", spinner="dots"): with _thinking_ctx():
response = await agent_loop.process_direct(message, session_id) response = await agent_loop.process_direct(message, session_id)
_print_agent_response(response, render_markdown=markdown) _print_agent_response(response, render_markdown=markdown)
@ -475,7 +482,7 @@ def agent(
else: else:
# Interactive mode # Interactive mode
_enable_line_editing() _enable_line_editing()
console.print(f"{__logo__} Interactive mode (Ctrl+C to exit)\n") console.print(f"{__logo__} Interactive mode (type [bold]exit[/bold] or [bold]Ctrl+C[/bold] to quit)\n")
# input() runs in a worker thread that can't be cancelled. # input() runs in a worker thread that can't be cancelled.
# Without this handler, asyncio.run() would hang waiting for it. # Without this handler, asyncio.run() would hang waiting for it.
@ -497,10 +504,12 @@ def agent(
continue continue
if _is_exit_command(command): if _is_exit_command(command):
_save_history()
_restore_terminal()
console.print("\nGoodbye!") console.print("\nGoodbye!")
break break
with console.status("[dim]Nanobot is thinking...[/dim]", spinner="dots"): with _thinking_ctx():
response = await agent_loop.process_direct(user_input, session_id) response = await agent_loop.process_direct(user_input, session_id)
_print_agent_response(response, render_markdown=markdown) _print_agent_response(response, render_markdown=markdown)
except KeyboardInterrupt: except KeyboardInterrupt:
@ -509,6 +518,8 @@ def agent(
console.print("\nGoodbye!") console.print("\nGoodbye!")
break break
except EOFError: except EOFError:
_save_history()
_restore_terminal()
console.print("\nGoodbye!") console.print("\nGoodbye!")
break break