Skip to content

Instantly share code, notes, and snippets.

@marttp

marttp/genai_client.py

Created June 19, 2025 13:47
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save marttp/fe9b17b07cbd077e21e2a780ee425dbe to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save marttp/fe9b17b07cbd077e21e2a780ee425dbe to your computer and use it in GitHub Desktop.
Download ZIP
Raw
genai_client.py
from google import genai
from google.api_core import retry
import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from helper.get_env import GEMINI_API_KEY
is_retriable = lambda e: (isinstance(e, genai.errors.APIError) and e.code in {429, 503})
genai.models.Models.generate_content = retry.Retry(predicate=is_retriable)(
genai.models.Models.generate_content
)
client = genai.Client(api_key=GEMINI_API_KEY)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment