Commit 3e945a54 authored by dmMaze's avatar dmMaze
Browse files

#379

parent ab7bd210

modules/translators/trans_chatgpt.py

+5 −1
Original line number Diff line number Diff line
@@ -222,6 +222,10 @@ class GPTTranslator(BaseTranslator): 
                    response = self._request_translation(prompt, chat_sample)

                    new_translations = re.split(r'<\|\d+\|>', response)[-num_src:]

                    if len(new_translations) != num_src:

                        _tr2 = response.split('\n') # https://github.com/dmMaze/BallonsTranslator/issues/379

                        if len(_tr2) == num_src:

                            new_translations = _tr2

                        else:

                            raise InvalidNumTranslations

                    break

                except InvalidNumTranslations: