Add reasoning support

Author: spirokristollari

src/lara_sdk/_client.py

@@ -117,6 +117,17 @@ def put(self, path: str, body: Dict = None, files: Dict = None, headers: Dict =
         """
         return self._request('PUT', path, body, files, headers)
 
+    def post_and_get_stream(self, path: str, body: Dict = None, files: Dict = None, headers: Dict = None):
+        """
+        Sends a POST request to the Lara API and yields streaming responses.
+        :param path: The path to send the request to.
+        :param body: The parameters to send with the request.
+        :param files: The files to send with the request. If present, request will be sent as multipart/form-data.
+        :param headers: Additional headers to include in the request.
+        :return: A generator yielding streaming responses.
+        """
+        return self._request_stream('POST', path, body, files, headers)
+
     def _request(self, method: str, path: str, body: Dict = None, files: Dict = None, headers: Dict = None) -> Optional[Union[Dict, List, bytes]]:
         if not path.startswith('/'):
             path = '/' + path
@@ -148,4 +159,52 @@ def _request(self, method: str, path: str, body: Dict = None, files: Dict = None
             if "text/csv" in response.headers.get('Content-Type', ''):
                 return response.content
             return response.json().get('content', None)
-        raise LaraApiError.from_response(response)
+        raise LaraApiError.from_response(response)
+
+    def _request_stream(self, method: str, path: str, body: Dict = None, files: Dict = None, headers: Dict = None):
+        if not path.startswith('/'):
+            path = '/' + path
+
+        _headers = {
+            'X-HTTP-Method-Override': method,
+            'Date': datetime.datetime.now(datetime.timezone.utc).isoformat(),
+            'X-Lara-SDK-Name': self.sdk_name,
+            'X-Lara-SDK-Version': self.sdk_version
+        }
+
+        if headers is not None:
+            _headers.update(headers)
+
+        if body is not None:
+            body = {k: v for k, v in body.items() if v is not None}
+
+            if len(body) > 0:
+                encoded_body = json.dumps(body, ensure_ascii=False, separators=(',', ':')).encode('UTF-8')
+                _headers['Content-MD5'] = hashlib.md5(encoded_body).hexdigest()
+
+        if files is not None:
+            response = self.session.request('POST', f'{self.base_url}{path}', headers=_headers, data=body, files=files, stream=True)
+        else:
+            response = self.session.request('POST', f'{self.base_url}{path}', headers=_headers, json=body, stream=True)
+
+        buffer = ''
+        for chunk in response.iter_content(chunk_size=None, decode_unicode=True):
+            if chunk:
+                buffer += chunk
+                lines = buffer.split('\n')
+                buffer = lines.pop()
+
+                for line in lines:
+                    if line.strip():
+                        try:
+                            parsed = json.loads(line)
+                            yield parsed.get('data', parsed).get('content')
+                        except (json.JSONDecodeError, AttributeError):
+                            pass
+
+        if buffer.strip():
+            try:
+                parsed = json.loads(buffer)
+                yield parsed.get('data', parsed).get('content')
+            except (json.JSONDecodeError, AttributeError):
+                pass

src/lara_sdk/_translator.py

@@ -429,7 +429,8 @@ def translate(self, text: Union[str, Iterable[str], Iterable[TextBlock]], *,
                   multiline: bool = True, timeout_ms: int = None, priority: TranslatePriority = None,
                   use_cache: Union[bool, UseCache] = None, cache_ttl_s: int = None,
                   no_trace: bool = False, verbose: bool = False, style: Optional[TranslationStyle] = None,
-                  headers: Optional[Dict[str, str]] = None) -> TextResult:
+                  headers: Optional[Dict[str, str]] = None, reasoning: bool = False,
+                  callback: Optional[Callable[[TextResult], None]] = None) -> TextResult:
         if isinstance(text, str):
             q = text
         elif hasattr(text, '__iter__'):
@@ -452,7 +453,7 @@ def translate(self, text: Union[str, Iterable[str], Iterable[TextBlock]], *,
             'multiline': multiline, 'adapt_to': adapt_to, 'instructions': instructions, 'timeout': timeout_ms, 'q': q,
             'priority': priority.value if priority is not None else None,
             'use_cache': use_cache.value if use_cache is not None else None, 'cache_ttl': cache_ttl_s,
-            'glossaries': glossaries, 'verbose': verbose, 'style': style
+            'glossaries': glossaries, 'verbose': verbose, 'style': style, 'reasoning': reasoning
         }
 
         request_headers = {}
@@ -461,7 +462,16 @@ def translate(self, text: Union[str, Iterable[str], Iterable[TextBlock]], *,
         if no_trace is True:
             request_headers['X-No-Trace'] = 'true'
 
-        return TextResult(**self._client.post('/translate', body, headers=request_headers))
+        last_result = None
+        for partial in self._client.post_and_get_stream('/translate', body, headers=request_headers):
+            last_result = partial
+            if callback is not None and reasoning:
+                callback(TextResult(**partial))
+
+        if last_result is None:
+            raise ValueError('No translation result received.')
+
+        return TextResult(**last_result)
 
     def detect(self, text: Union[str, List[str]], *, hint: Optional[str] = None,
                passlist: Optional[List[str]] = None) -> DetectResult: