Update gemini_client/core.py

gemini_client/core.py

@@ -846,6 +846,26 @@ class AsyncChatbot:
                         # Process all parts in this JSON array
                         for part_index, part in enumerate(parsed_json):
                             try:
+                                # Check for error frame ["e", code, ...]
+                                if isinstance(part, list) and len(part) >= 2 and part[0] == "e" and isinstance(part[1], int):
+                                    error_code = part[1]
+                                    error_status = "Unknown"
+                                    # Map common gRPC error codes
+                                    if error_code == 3: error_status = "INVALID_ARGUMENT"
+                                    elif error_code == 4: error_status = "DEADLINE_EXCEEDED (Request timed out on server)"
+                                    elif error_code == 8: error_status = "RESOURCE_EXHAUSTED"
+                                    elif error_code == 14: error_status = "UNAVAILABLE"
+                                    elif error_code == 16: error_status = "UNAUTHENTICATED"
+                                    
+                                    error_msg = f"Server returned error: {error_status} (Code {error_code})"
+                                    if len(part) > 4 and part[4]:
+                                        error_msg += f" - Details: {part[4]}"
+                                        
+                                    console.log(f"[bold red]{error_msg}[/bold red]")
+                                    # Don't return immediately, look for other parts? No, "e" typically ends the stream or indicates failure.
+                                    # Return error result
+                                    return {"content": error_msg, "error": True}
+
                                 if isinstance(part, list) and len(part) > 2:
                                     # part[2] might be a string that needs parsing
                                     if isinstance(part[2], str):
@@ -861,7 +881,8 @@ class AsyncChatbot:
                                     if main_part and isinstance(main_part, list) and len(main_part) > 4 and main_part[4]:
                                         body = main_part
                                         body_index = part_index
-                            except (IndexError, TypeError, json.JSONDecodeError, AttributeError):
+                            except (IndexError, TypeError, json.JSONDecodeError, AttributeError) as e:
+                                # console.log(f"[yellow]Debug: Error parsing part {part_index}: {e}[/yellow]")
                                 continue
                         
                         # If we found a body, stop looking