docs: update docstrings

winstxnhdw · Jun 2, 2024 · 0b06095 · 0b06095
1 parent 363a60e
commit 0b06095
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/server/features/translator.py b/server/features/translator.py
@@ -57,8 +57,8 @@ async def translate(cls, text: str, source_language: Languages, target_language:
         Parameters
         ----------
         input (str) : the input to translate
-        source_language (str) : the source language
-        target_language (str) : the target language
+        source_language (Languages) : the source language
+        target_language (Languages) : the target language
 
         Returns
         -------

diff --git a/server/schemas/v1/translation.py b/server/schemas/v1/translation.py
@@ -12,8 +12,8 @@ class Translation(BaseModel):
     Attributes
     ----------
     text (str) : the text to translate
-    source (str) : the source language
-    target (str) : the target language
+    source (Languages) : the source language
+    target (Languages) : the target language
     """
 
     text: str = Field(examples=['Hello, world!'])