feat: Add style and tones to write API

Author: JanEbbing

deepl/api_data.py

@@ -452,3 +452,49 @@ class ModelType(Enum):
 
     def __str__(self):
         return self.value
+
+
+class WritingStyle(Enum):
+    """Options for the `style` parameter of the Write API.
+    Sets the style the improved text should be in. Note that currently, only
+    a style OR a tone is supported.
+
+    When using a style starting with `prefer_`, the style will only be used
+    if the chosen or detected language supports it.
+    """
+
+    ACADEMIC = "academic"
+    BUSINESS = "business"
+    CASUAL = "casual"
+    DEFAULT = "default"
+    PREFER_ACADEMIC = "prefer_academic"
+    PREFER_BUSINESS = "prefer_business"
+    PREFER_CASUAL = "prefer_casual"
+    PREFER_SIMPLE = "prefer_simple"
+    SIMPLE = "simple"
+
+    def __str__(self):
+        return self.value
+
+
+class WritingTone(Enum):
+    """Options for the `tone` parameter of the Write API.
+    Sets the tone the improved text should be in. Note that currently, only
+    a style OR a tone is supported.
+
+    When using a tone starting with `prefer_`, the tone will only be used
+    if the chosen or detected language supports it.
+    """
+
+    CONFIDENT = "confident"
+    DEFAULT = "default"
+    DIPLOMATIC = "diplomatic"
+    ENTHUSIASTIC = "enthusiastic"
+    FRIENDLY = "friendly"
+    PREFER_CONFIDENT = "prefer_confident"
+    PREFER_DIPLOMATIC = "prefer_diplomatic"
+    PREFER_ENTHUSIASTIC = "prefer_enthusiastic"
+    PREFER_FRIENDLY = "prefer_friendly"
+
+    def __str__(self):
+        return self.value

deepl/deepl_client.py

@@ -40,7 +40,9 @@ def rephrase_text(
         self,
         text: Union[str, Iterable[str]],
         *,
-        target_lang: Union[str, Language],
+        target_lang: Union[None, str, Language] = None,
+        style: Optional[str] = None,
+        tone: Optional[str] = None,
     ) -> Union[WriteResult, List[WriteResult]]:
         """Improve the text(s) and optionally convert them to the variant of
         the `target_lang` (requires source lang to match target_lang, excluding
@@ -51,6 +53,10 @@ def rephrase_text(
             generator)
         :param target_lang: language code the final text should be in, for
             example "DE", "EN-US", "FR".
+        :param style: Writing style to be used for the improvement. Either
+            style OR tone can be used.
+        :param tone: Tone to be used for the improvement. Either style OR tone
+            can be used.
         :return: List of WriteResult objects containing results, unless input
             text was one string, then a single WriteResult object is returned.
         """
@@ -68,7 +74,13 @@ def rephrase_text(
                 "text parameter must be a string or an iterable of strings"
             )
 
-        request_data = {"target_lang": target_lang, "text": text}
+        request_data: dict = {"text": text}
+        if target_lang:
+            request_data["target_lang"] = target_lang
+        if style:
+            request_data["writing_style"] = style
+        if tone:
+            request_data["tone"] = tone
 
         status, content, json = self._api_call(
             "v2/write/rephrase", json=request_data