fix:added graceful handling

Author: ayusrjn

lambda_agent/agent.py

@@ -169,10 +169,13 @@ def chat(self, user_input: str) -> tuple[str, TokenUsage]:
         # 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/main.py

@@ -334,6 +334,11 @@ def main():
                     "\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(