Feature: add password and extraction parameters for document translation

Author: ikhimenko

examples/memories_management.py

@@ -26,7 +26,6 @@ def main():
     lara = Translator(credentials)
 
     memory_id = None
-    external_memory_id = None
     memory_2_to_delete = None
 
     try:
@@ -147,12 +146,6 @@ def main():
             except Exception as e:
                 print(f"Error deleting memory: {e}")
 
-        if external_memory_id:
-            try:
-                deleted_external_memory = lara.memories.delete(external_memory_id)
-                print(f"🗑️  Deleted external memory: {deleted_external_memory.name}")
-            except Exception as e:
-                print(f"Error deleting external memory: {e}")
 
         if memory_2_to_delete:
             try:

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, Memories, Translator, TranslatePriority, UseCache, Documents, Document, DocumentStatus
+from ._translator import Memory, MemoryImport, TextBlock, TextResult, 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

@@ -67,6 +67,23 @@ class DocumentOptions:
     no_trace: Optional[bool] = None
     style: Optional[TranslationStyle] = None
 
+# Extraction parameters for DOCX files
+@dataclass
+class DocxExtractionParams:
+    extract_comments: Optional[bool] = None
+    accept_revisions: Optional[bool] = None
+
+    def to_dict(self) -> Dict[str, Union[str, bool]]:
+        result = {}
+        if self.extract_comments is not None:
+            result['extract_comments'] = self.extract_comments
+        if self.accept_revisions is not None:
+            result['accept_revisions'] = self.accept_revisions
+        return result
+
+# Union type for all extraction parameter types
+DocumentExtractionParams = Union[DocxExtractionParams]
+
 class Document(LaraObject):
 
     def __init__(self, **kwargs):
@@ -300,7 +317,8 @@ def __init__(self, client: LaraClient):
 
     def upload(self, file_path: str, filename: str, target: str, source: Optional[str] = None,
                adapt_to: Optional[List[str]] = None, glossaries: Optional[List[str]] = None, no_trace: bool = False,
-               style: Optional[TranslationStyle] = None) -> Document:
+               style: Optional[TranslationStyle] = None, password: Optional[str] = None,
+               extraction_params: Optional[DocumentExtractionParams] = None) -> Document:
         with open(file_path, 'rb') as file_payload:
             response_data = self._client.get('/documents/upload-url', {'filename': filename})
 
@@ -322,9 +340,15 @@ def upload(self, file_path: str, filename: str, target: str, source: Optional[st
         if glossaries is not None:
             body['glossaries'] = glossaries
 
+        if password is not None:
+            body['password'] = password
+
         if style is not None:
             body['style'] = style
 
+        if extraction_params is not None:
+            body['extraction_params'] = extraction_params.to_dict()
+
         headers = None
         if no_trace is True:
             headers = {'X-No-Trace': 'true'}
@@ -343,10 +367,11 @@ def download(self, id: str, output_format: Optional[str] = None) -> bytes:
 
     def translate(self, file_path: str, filename: str, target: str, source: Optional[str] = None,
                   adapt_to: Optional[List[str]] = None, glossaries: Optional[List[str]] = None, output_format: Optional[str] = None,
-                  no_trace: bool = False, style: Optional[TranslationStyle] = None) -> bytes:
+                  no_trace: bool = False, style: Optional[TranslationStyle] = None, password: Optional[str] = None,
+                  extraction_params: Optional[DocumentExtractionParams] = None) -> bytes:
 
         document = self.upload(file_path=file_path, filename=filename, target=target, source=source, adapt_to=adapt_to,
-                               glossaries=glossaries, no_trace=no_trace, style=style)
+                               glossaries=glossaries, no_trace=no_trace, style=style, password=password, extraction_params=extraction_params)
 
         max_wait_time = 60 * 15 # 15 minutes
         start = time.time()