Skip to content

Commit 121178d

Browse files
committed
Add gemini exception
1 parent 62a9746 commit 121178d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

stream_translator_gpt/llm_translator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from datetime import datetime, timedelta
88

99
import google.generativeai as genai
10-
from google.api_core.exceptions import InternalServerError
10+
from google.api_core.exceptions import InternalServerError, ResourceExhausted
1111
from google.generativeai.types import HarmCategory, HarmBlockThreshold
1212
from openai import OpenAI, APITimeoutError, APIConnectionError
1313

@@ -118,7 +118,7 @@ def _translate_by_gemini(self, translation_task: TranslationTask):
118118
generation_config=config,
119119
safety_settings=safety_settings)
120120
translation_task.translated_text = _parse_json_completion(response.text)
121-
except (ValueError, InternalServerError) as e:
121+
except (ValueError, InternalServerError, ResourceExhausted) as e:
122122
print(e)
123123
return
124124
if self.history_size:

0 commit comments

Comments
 (0)