From e6e2601d4b992a5615b479f41463ee2c2def0734 Mon Sep 17 00:00:00 2001
From: JRoot Junior <jroot.junior@gmail.com>
Date: Tue, 13 Feb 2024 22:09:48 +0200
Subject: [PATCH] Fixed typehints in keyboard builder

---
 aiogram/utils/keyboard.py | 52 ++++++++++++++++++++++-----------------
 1 file changed, 29 insertions(+), 23 deletions(-)

diff --git a/aiogram/utils/keyboard.py b/aiogram/utils/keyboard.py
index 3457ec13..ef9b2f52 100644
--- a/aiogram/utils/keyboard.py
+++ b/aiogram/utils/keyboard.py
@@ -312,19 +312,22 @@ class InlineKeyboardBuilder(KeyboardBuilder[InlineKeyboardButton]):
         callback_game: Optional[CallbackGame] = None,
         pay: Optional[bool] = None,
         **kwargs: Any,
-    ) -> "KeyboardBuilder[InlineKeyboardButton]":
-        return self._button(
-            text=text,
-            url=url,
-            callback_data=callback_data,
-            web_app=web_app,
-            login_url=login_url,
-            switch_inline_query=switch_inline_query,
-            switch_inline_query_current_chat=switch_inline_query_current_chat,
-            switch_inline_query_chosen_chat=switch_inline_query_chosen_chat,
-            callback_game=callback_game,
-            pay=pay,
-            **kwargs,
+    ) -> "InlineKeyboardBuilder":
+        return cast(
+            InlineKeyboardBuilder,
+            self._button(
+                text=text,
+                url=url,
+                callback_data=callback_data,
+                web_app=web_app,
+                login_url=login_url,
+                switch_inline_query=switch_inline_query,
+                switch_inline_query_current_chat=switch_inline_query_current_chat,
+                switch_inline_query_chosen_chat=switch_inline_query_chosen_chat,
+                callback_game=callback_game,
+                pay=pay,
+                **kwargs,
+            ),
         )
 
     if TYPE_CHECKING:
@@ -377,16 +380,19 @@ class ReplyKeyboardBuilder(KeyboardBuilder[KeyboardButton]):
         request_poll: Optional[KeyboardButtonPollType] = None,
         web_app: Optional[WebAppInfo] = None,
         **kwargs: Any,
-    ) -> "KeyboardBuilder[KeyboardButton]":
-        return self._button(
-            text=text,
-            request_users=request_users,
-            request_chat=request_chat,
-            request_contact=request_contact,
-            request_location=request_location,
-            request_poll=request_poll,
-            web_app=web_app,
-            **kwargs,
+    ) -> "ReplyKeyboardBuilder":
+        return cast(
+            ReplyKeyboardBuilder,
+            self._button(
+                text=text,
+                request_users=request_users,
+                request_chat=request_chat,
+                request_contact=request_contact,
+                request_location=request_location,
+                request_poll=request_poll,
+                web_app=web_app,
+                **kwargs,
+            ),
         )
 
     if TYPE_CHECKING: