Retry configuration options?

[mcantrell]

I've poked around the documentation and the source code a little and I was not able to find anything regarding retry configuration in the lib. Is this something that I've overlooked or is it left as a responsibility for the developer?

[sararob]

We're working on adding support for retries and will comment on this issue when there's an update. Thanks!

[mcantrell]

For those looking for an interim solution, I ended up going with tenacity for the following errors:

from google.genai.errors import APIError
from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_exception

def _is_retryable_error(exception: BaseException) -> bool:
    """
    Checks if the exception is a retryable error based on the criteria.
    """
    if isinstance(exception, APIError):
        return exception.code in [429, 502, 503, 504]
    if isinstance(exception, requests.exceptions.ConnectionError):
        return True
    return False


retry_gemini_call = retry(
    retry=retry_if_exception(_is_retryable_error),
    wait=wait_exponential(min=1, max=10),
    stop=stop_after_attempt(3),
    reraise=True
)

@retry_gemini_call
def test_gemini_completion(self, prompt: str) -> str:
  # example usage