Implement non-blocking TCP connection (

RknDeveloper · web-flow · commit eea9f125e608 · 2024-06-26T11:10:40.000+05:45
diff --git a/pyrogram/session/session.py b/pyrogram/session/session.py
@@ -31,7 +31,7 @@
 from pyrogram.crypto import mtproto
 from pyrogram.errors import (
     RPCError, InternalServerError, AuthKeyDuplicated,
-    FloodWait, FloodPremiumWait,
+    FloodWait,
     ServiceUnavailable, BadMsgNotification,
     SecurityCheckMismatch
 )
@@ -407,7 +407,7 @@ async def invoke(
         while True:
             try:
                 return await self.send(query, timeout=timeout)
-            except (FloodWait, FloodPremiumWait) as e:
+            except (FloodWait) as e:
                 amount = e.value
 
                 if amount > sleep_threshold >= 0:
