fix:introduced gracefull handling

Author: ayusrjn

lambda_agent/cli_setup.py

@@ -17,7 +17,7 @@ def run_setup() -> tuple[str, str]:
         if not api_key:
             print("API Key cannot be empty. Please try again.")
 
-    default_model = "gemini-3.1-flash-lite-preview"
+    default_model = "gemini-3.1-pro-preview"
     model_name = input(f"Enter model name (default: {default_model}): ").strip()
     if not model_name:
         model_name = default_model
@@ -37,7 +37,7 @@ def run_setup() -> tuple[str, str]:
 
         print(f"\n✅ Setup complete! Configuration saved to {config_file}\n")
     except Exception as e:
-        print(f"\n❌ Error saving configuration: {e}")
+        print(f"\n Error saving configuration: {e}")
         print("Continuing with in-memory configuration for this session.\n")
 
     return api_key, model_name

lambda_agent/config.py

@@ -17,14 +17,16 @@
     )
 
 API_KEY = os.getenv("API_KEY")
-MODEL_NAME = os.getenv("MODEL_NAME", "gemini-3.1-flash-lite-preview")
+MODEL_NAME = os.getenv("MODEL_NAME", "gemini-3.1-pro-preview")
 
 # Models available for /models switching
 AVAILABLE_MODELS = [
+    "gemini-3.1-pro-preview",
+    "gemini-3-flash-preview",
     "gemini-3.1-flash-lite-preview",
+    "gemini-2.5-pro",
     "gemini-2.5-flash",
-    "gemini-3.1-pro-preview",
-    "gemini-2.5-pro-preview-05-06",
-    "gemini-2.0-flash",
-    "gemini-2.0-flash-lite",
+    "gemini-2.5-flash-lite",
+    "gemma-4-26b-a4b-it",
+    "gemma-4-31b-it",
 ]

lambda_agent/main.py

@@ -238,13 +238,43 @@ def main():
         )
 
         while True:
+            # Inner loop logic to handle prompt input vs command execution
             try:
-                # Styled prompt — uses plain input to keep cursor on same line
                 user_input = Prompt.ask(
                     "\n[bold bright_yellow]  You[/bold bright_yellow]",
                     console=console,
                 )
+            except KeyboardInterrupt:
+                console.print()
+                # Show session token summary before quitting
+                if agent.token_usage.total > 0:
+                    console.print(
+                        Panel(
+                            Text.assemble(
+                                ("Session token usage\n", "bold white"),
+                                ("  Prompt (in):      ", "dim"),
+                                (f"{agent.token_usage.prompt:>10,}\n", "cyan"),
+                                ("  Completion (out): ", "dim"),
+                                (f"{agent.token_usage.completion:>10,}\n", "cyan"),
+                                ("  Total:            ", "dim"),
+                                (f"{agent.token_usage.total:>10,}", "bold cyan"),
+                            ),
+                            border_style="cyan",
+                            box=box.ROUNDED,
+                            title="[bold cyan]⚡ Token Summary[/bold cyan]",
+                            title_align="left",
+                        )
+                    )
+                console.print(
+                    Panel(
+                        "[bold cyan]Goodbye! Lambda signing off.[/bold cyan]",
+                        border_style="cyan",
+                        box=box.ROUNDED,
+                    )
+                )
+                break
 
+            try:
                 if user_input.lower() in ["exit", "quit"]:
                     console.print()
                     # Show session token summary before quitting
@@ -300,9 +330,10 @@ def main():
                 print_token_stats(turn_usage, agent.token_usage)
 
             except KeyboardInterrupt:
-                console.print()
-                console.print("[bold cyan]\nGoodbye![/bold cyan]")
-                break
+                console.print(
+                    "\n  [bold yellow]⚠  Action cancelled by user.[/bold yellow]"
+                )
+                continue
 
     except Exception as e:
         console.print(

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "lambda-agent"
-version = "0.1.0"
+version = "0.1.1"
 description = "Lambda - A minimal AI coding agent"
 readme = "README.md"
 requires-python = ">=3.10"