feat: fix finish_reason check in translator engines

rss-translator · Jun 24, 2024 · ac50081 · ac50081
1 parent a7a947c
commit ac50081
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/translator/models/base.py b/translator/models/base.py
@@ -170,7 +170,8 @@ def translate(
                 presence_penalty=self.presence_penalty,
                 max_tokens=self.max_tokens,
             )
-            if res.choices[0].finish_reason.lower() == "stop" or res.choices[0].message.content:
+            #if res.choices[0].finish_reason.lower() == "stop" or res.choices[0].message.content:
+            if res.choices and res.choices[0].message.content:
                 translated_text = res.choices[0].message.content
                 logging.info("OpenAITranslator->%s: %s", res.choices[0].finish_reason, translated_text)
             # else:

diff --git a/translator/models/doubao.py b/translator/models/doubao.py
@@ -68,7 +68,8 @@ def translate(self, text: str, target_language: str, system_prompt: str = None,
                 ],
                 max_tokens=self.max_tokens,
             )
-            if res.choices[0].finish_reason.lower() == "stop" or res.choices[0].message.content:
+            #if res.choices[0].finish_reason.lower() == "stop" or res.choices[0].message.content:
+            if res.choices and res.choices[0].message.content:
                 translated_text = res.choices[0].message.content
                 logging.info("DoubaoTranslator->%s: %s", res.choices[0].finish_reason, translated_text)
             tokens = res.usage.total_tokens