Add DetectPrediction class and predictions to DetectResult

Author: davc0n

src/lara_sdk/__init__.py

@@ -1,7 +1,7 @@
 from ._client import LaraObject
 from ._credentials import Credentials
 from ._errors import LaraApiError, LaraError
-from ._translator import Memory, MemoryImport, TextBlock, TextResult, DetectResult, Memories, Translator, TranslatePriority, UseCache, Documents, Document, DocumentStatus, DocxExtractionParams, DocumentExtractionParams
+from ._translator import Memory, MemoryImport, TextBlock, TextResult, DetectPrediction, DetectResult, Memories, Translator, TranslatePriority, UseCache, Documents, Document, DocumentStatus, DocxExtractionParams, DocumentExtractionParams
 
 # This constant is auto-generated by the build script.
 # Manual modifications will be overwritten and may cause unexpected behavior.

src/lara_sdk/_translator.py

@@ -141,10 +141,16 @@ def __init__(self, **kwargs):
                 self.translation = [TextBlock(**e) for e in translation]
 
 
+class DetectPrediction(LaraObject):
+    def __init__(self, **kwargs):
+        self.language: str = kwargs.get('language')
+        self.confidence: float = kwargs.get('confidence')
+
 class DetectResult(LaraObject):
     def __init__(self, **kwargs):
         self.language: str = kwargs.get('language')
         self.content_type: str = kwargs.get('content_type')
+        self.predictions: List[DetectPrediction] = [DetectPrediction(**p) for p in kwargs.get('predictions', [])]
 
 
 # Translator SDK -------------------------------------------------------------------------------------------------------
@@ -480,5 +486,5 @@ def detect(self, text: Union[str, List[str]], *, hint: Optional[str] = None,
             'hint': hint,
             'passlist': passlist
         }
-        
+
         return DetectResult(**self._client.post('/detect', body))