Apply suggestions from code review

Co-authored-by: Bartosz Sławecki <bartosz@ilikepython.com>

Author: ambv

Lib/_pyrepl/commands.py

@@ -464,7 +464,8 @@ def do(self) -> None:
 
         with self.reader.suspend():
             self.reader.msg = _sitebuiltins._Helper()()  # type: ignore[assignment]
-
+        self.reader.invalidate_prompt()
+
 
 class invalid_key(Command):
     def do(self) -> None:

Lib/_pyrepl/reader.py

@@ -194,33 +194,33 @@ def buffer_rebuild_from_pos(self) -> int | None:
             return 0
         return self.buffer_from_pos
 
-    def with_cursor(self) -> RefreshInvalidation:
+    def with_cursor(self) -> Self:
         if self.needs_screen_refresh:
             return self
         return replace(self, cursor_only=True)
 
-    def with_buffer(self, from_pos: int) -> RefreshInvalidation:
+    def with_buffer(self, from_pos: int) -> Self:
         current = from_pos
         if self.buffer_from_pos is not None:
             current = min(current, self.buffer_from_pos)
         return replace(self, cursor_only=False, buffer_from_pos=current)
 
-    def with_prompt(self) -> RefreshInvalidation:
+    def with_prompt(self) -> Self:
         return replace(self, cursor_only=False, prompt=True)
 
-    def with_layout(self) -> RefreshInvalidation:
+    def with_layout(self) -> Self:
         return replace(self, cursor_only=False, layout=True)
 
-    def with_theme(self) -> RefreshInvalidation:
+    def with_theme(self) -> Self:
         return replace(self, cursor_only=False, theme=True)
 
-    def with_message(self) -> RefreshInvalidation:
+    def with_message(self) -> Self:
         return replace(self, cursor_only=False, message=True)
 
-    def with_overlay(self) -> RefreshInvalidation:
+    def with_overlay(self) -> Self:
         return replace(self, cursor_only=False, overlay=True)
 
-    def with_full(self) -> RefreshInvalidation:
+    def with_full(self) -> Self:
         return replace(self, cursor_only=False, full=True)