Skip to content

Commit 0ee1b06

Browse files
committed
Add language detection support
1 parent a10c136 commit 0ee1b06

1 file changed

Lines changed: 21 additions & 0 deletions

File tree

src/modernmt/modernmt.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,22 @@ def __init__(self, license_key, platform="modernmt-python", platform_version="1.
2323
def list_supported_languages(self):
2424
return self.__send("get", "/translate/languages")
2525

26+
def detect_language(self, q, format=None):
27+
data = {"q": q}
28+
if format is not None:
29+
data["format"] = format
30+
31+
res = self.__send("get", "/translate/detect", data=data)
32+
33+
if not isinstance(q, list):
34+
return DetectedLanguage(res)
35+
36+
languages = []
37+
for el in res:
38+
languages.append(DetectedLanguage(el))
39+
40+
return languages
41+
2642
def translate(self, source, target, q, hints=None, context_vector=None, options=None):
2743
data = {"target": target, "q": q}
2844
if source is not None:
@@ -217,3 +233,8 @@ def __init__(self, data) -> None:
217233
class ImportJob(_Model):
218234
def __init__(self, data) -> None:
219235
super().__init__(data, ["id", "memory", "size", "progress"])
236+
237+
238+
class DetectedLanguage(_Model):
239+
def __init__(self, data) -> None:
240+
super().__init__(data, ["billedCharacters", "detectedLanguage"])

0 commit comments

Comments
 (0)