fix: correct parsing and client binding for AI message types

Author: SychO3

pyrogram/methods/messages/transcribe_audio.py

@@ -26,7 +26,7 @@ async def transcribe_audio(
         chat_id: str,
         message_id: int,
     ) -> "types.TranscribedAudio":
-        """Transcribe an audio message to text.
+        """Transcribe an audio or voice message to text.
 
         .. include:: /_includes/usable-by/users.rst
 
@@ -37,7 +37,7 @@ async def transcribe_audio(
                 For a contact that exists in your Telegram address book you can use his phone number (str).
 
             message_id (``int``):
-                Identifier of the audio message.
+                Identifier of the audio or voice message.
 
         Returns:
             :obj:`~pyrogram.types.TranscribedAudio`: On success, the transcription result is returned.
@@ -53,4 +53,4 @@ async def transcribe_audio(
             )
         )
 
-        return types.TranscribedAudio._parse(r)
+        return types.TranscribedAudio._parse(self, r)

pyrogram/types/messages_and_media/composed_message_with_ai.py

@@ -38,10 +38,11 @@ class ComposedMessageWithAI(Object):
     def __init__(
         self,
         *,
+        client: "pyrogram.Client" = None,
         result_text: "types.FormattedText",
         diff_text: Optional["types.FormattedText"] = None,
     ):
-        super().__init__()
+        super().__init__(client)
 
         self.result_text = result_text
         self.diff_text = diff_text
@@ -55,6 +56,7 @@ def _parse(
             return None
 
         return ComposedMessageWithAI(
+            client=client,
             result_text=types.FormattedText._parse(
                 client, composed_message.result_text
             ),

pyrogram/types/messages_and_media/message_entity.py

@@ -175,7 +175,7 @@ async def write(self):
             args["collapsed"] = self.expandable
 
         if self.old_text is None:
-            args.pop("old_text")
+            args.pop("old_text", None)
 
         entity = self.type.value
 

pyrogram/types/messages_and_media/transcribed_audio.py

@@ -18,6 +18,9 @@
 
 from typing import Optional
 
+import pyrogram
+from pyrogram import raw
+
 from ..object import Object
 
 
@@ -44,13 +47,14 @@ class TranscribedAudio(Object):
     def __init__(
         self,
         *,
+        client: "pyrogram.Client" = None,
         transcription_id: int,
         text: str,
         pending: Optional[bool] = None,
         trial_remains_num: Optional[int] = None,
         trial_remains_until_date: Optional[int] = None,
     ):
-        super().__init__()
+        super().__init__(client)
 
         self.transcription_id = transcription_id
         self.text = text
@@ -59,11 +63,15 @@ def __init__(
         self.trial_remains_until_date = trial_remains_until_date
 
     @staticmethod
-    def _parse(transcribed_audio) -> Optional["TranscribedAudio"]:
-        if transcribed_audio is None:
+    def _parse(
+        client: "pyrogram.Client",
+        transcribed_audio: "raw.types.messages.TranscribedAudio",
+    ) -> Optional["TranscribedAudio"]:
+        if not isinstance(transcribed_audio, raw.types.messages.TranscribedAudio):
             return None
 
         return TranscribedAudio(
+            client=client,
             transcription_id=transcribed_audio.transcription_id,
             text=transcribed_audio.text,
             pending=transcribed_audio.pending,