feat: make translate_to_language_code optional (#309)

Author: BreezeKun

pyrogram/methods/messages/compose_text_with_ai.py

@@ -26,7 +26,7 @@ class ComposeTextWithAI:
     async def compose_text_with_ai(
         self: "pyrogram.Client",
         text: Union[str, "types.FormattedText"],
-        translate_to_language_code: Optional[str],
+        translate_to_language_code: Optional[str] = None,
         style_name: Optional[str] = None,
         add_emojis: Optional[bool] = None,
     ) -> "types.FormattedText":
@@ -44,6 +44,7 @@ async def compose_text_with_ai(
                 "fi", "fr", "fy", "gl", "ka", "de", "el", "gu", "ht", "ha", "haw", "he", "iw", "hi", "hmn", "hu", "is", "ig", "id", "in", "ga", "it", "ja", "jv", "kn", "kk", "km", "rw", "ko",
                 "ku", "ky", "lo", "la", "lv", "lt", "lb", "mk", "mg", "ms", "ml", "mt", "mi", "mr", "mn", "my", "ne", "no", "ny", "or", "ps", "fa", "pl", "pt", "pa", "ro", "ru", "sm", "gd", "sr",
                 "st", "sn", "sd", "si", "sk", "sl", "so", "es", "su", "sw", "sv", "tl", "tg", "ta", "tt", "te", "th", "tr", "tk", "uk", "ur", "ug", "uz", "vi", "cy", "xh", "yi", "ji", "yo", "zu"
+                Defaults to the client's language code.
 
             style_name (``str``, *optional*):
                 Name of the style of the resulted text.
@@ -53,14 +54,24 @@ async def compose_text_with_ai(
 
         Returns:
             :obj:`~pyrogram.types.FormattedText`: On success, information about the composed text is returned.
+
+        Example: 
+            .. code-block:: python 
+            
+                await app.compose_text_with_ai( 
+                    "hello, how are you?",
+                    translate_to_language_code="ru",
+                    style_name="formal",
+                    add_emojis=True 
+                )
         """
         if isinstance(text, str):
             text = types.FormattedText(text=text)
 
         r = await self.invoke(
             raw.functions.messages.ComposeMessageWithAI(
                 text=await text.write(self),
-                translate_to_lang=translate_to_language_code,
+                translate_to_lang=translate_to_language_code or self.lang_code,
                 tone=raw.types.InputAiComposeToneDefault(tone=style_name) if style_name else None,
                 emojify=add_emojis,
             )

pyrogram/methods/messages/fix_text_with_ai.py

@@ -36,6 +36,13 @@ async def fix_text_with_ai(
 
         Returns:
             :obj:`~pyrogram.types.FormattedText`: On success, information about the fixed text is returned.
+
+        Example: 
+            .. code-block:: python
+            
+                app.fix_text_with_ai(
+                    "This statement maay havve feew mistakes"
+                )
         """
         if isinstance(text, str):
             text = types.FormattedText(text=text)

pyrogram/methods/messages/summarize_message.py

@@ -57,6 +57,17 @@ async def summarize_message(
 
         Returns:
             :obj:`~pyrogram.types.FormattedText`: On success, information about the summarized text is returned.
+
+        Example:
+            .. code-block:: python
+            
+                await app.summarize_message(
+                    chat_id,
+                    message_id,
+                    translate_to_language_code="ru",
+                    tone="neutral"
+                )
+            
         """
         r = await self.invoke(
             raw.functions.messages.SummarizeText(

pyrogram/methods/messages/translate_message_text.py

@@ -51,6 +51,7 @@ async def translate_message_text(
                 "fi", "fr", "fy", "gl", "ka", "de", "el", "gu", "ht", "ha", "haw", "he", "iw", "hi", "hmn", "hu", "is", "ig", "id", "in", "ga", "it", "ja", "jv", "kn", "kk", "km", "rw", "ko",
                 "ku", "ky", "lo", "la", "lv", "lt", "lb", "mk", "mg", "ms", "ml", "mt", "mi", "mr", "mn", "my", "ne", "no", "ny", "or", "ps", "fa", "pl", "pt", "pa", "ro", "ru", "sm", "gd", "sr",
                 "st", "sn", "sd", "si", "sk", "sl", "so", "es", "su", "sw", "sv", "tl", "tg", "ta", "tt", "te", "th", "tr", "tk", "uk", "ur", "ug", "uz", "vi", "cy", "xh", "yi", "ji", "yo", "zu"
+                Defaults to the client's language code.
 
             tone (``str``, *optional*):
                 Tone of the translation.
@@ -66,7 +67,7 @@ async def translate_message_text(
         """
         r = await self.invoke(
             raw.functions.messages.TranslateText(
-                to_lang=to_language_code,
+                to_lang=to_language_code or self.lang_code,
                 peer=await self.resolve_peer(chat_id),
                 id=[message_id],
                 tone=tone

pyrogram/methods/messages/translate_text.py

@@ -45,6 +45,7 @@ async def translate_text(
                 "fi", "fr", "fy", "gl", "ka", "de", "el", "gu", "ht", "ha", "haw", "he", "iw", "hi", "hmn", "hu", "is", "ig", "id", "in", "ga", "it", "ja", "jv", "kn", "kk", "km", "rw", "ko",
                 "ku", "ky", "lo", "la", "lv", "lt", "lb", "mk", "mg", "ms", "ml", "mt", "mi", "mr", "mn", "my", "ne", "no", "ny", "or", "ps", "fa", "pl", "pt", "pa", "ro", "ru", "sm", "gd", "sr",
                 "st", "sn", "sd", "si", "sk", "sl", "so", "es", "su", "sw", "sv", "tl", "tg", "ta", "tt", "te", "th", "tr", "tk", "uk", "ur", "ug", "uz", "vi", "cy", "xh", "yi", "ji", "yo", "zu"
+                Defaults to the client's language code.
 
             tone (``str``, *optional*):
                 Tone of the translation.
@@ -63,7 +64,7 @@ async def translate_text(
 
         r = await self.invoke(
             raw.functions.messages.TranslateText(
-                to_lang=to_language_code,
+                to_lang=to_language_code or self.lang_code,
                 text=[await text.write(self)],
                 tone=tone
             )