Implement `aind_session.configure(retries=..., )` with a manager like this: https://github.com/google-gemini/generative-ai-python/blob/a04fcd185eefeb13955f0468c86c927d52436590/google/generativeai/client.py#L302