lambda-agent 0.1.1__tar.gz → 0.1.2__tar.gz

{lambda_agent-0.1.1 → lambda_agent-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lambda-agent
-Version: 0.1.1
+Version: 0.1.2
 Summary: Lambda - A minimal AI coding agent
 Author: Ayush Ranjan
 License: Apache-2.0

{lambda_agent-0.1.1 → lambda_agent-0.1.2}/lambda_agent/agent.py

@@ -169,10 +169,13 @@ class Agent:
         # Log the user message to the full transcript
         self.transcript.log("user", user_input)
 
-        # Send the initial user message
-        with Spinner():
-            response = self.chat_session.send_message(payload)
-        turn_usage = turn_usage + self._accumulate(response)
+        try:
+            # Send the initial user message
+            with Spinner():
+                response = self.chat_session.send_message(payload)
+            turn_usage = turn_usage + self._accumulate(response)
+        except Exception as e:
+            return f"An error occurred while contacting the API: {str(e)}", turn_usage
 
         # The loop will continue as long as Gemini decides to call tools
         while True:

{lambda_agent-0.1.1 → lambda_agent-0.1.2}/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-0.1.1 → lambda_agent-0.1.2}/lambda_agent/config.py

@@ -17,14 +17,16 @@ except ImportError:
     )
 
 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-0.1.1 → lambda_agent-0.1.2}/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,15 @@ 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(
+                    f"\n  [bold red]⚠  An unexpected error occurred: {str(e)}[/bold red]"
+                )
+                continue
 
     except Exception as e:
         console.print(

{lambda_agent-0.1.1 → lambda_agent-0.1.2}/lambda_agent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lambda-agent
-Version: 0.1.1
+Version: 0.1.2
 Summary: Lambda - A minimal AI coding agent
 Author: Ayush Ranjan
 License: Apache-2.0

{lambda_agent-0.1.1 → lambda_agent-0.1.2}/pyproject.toml

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