INNER CODE UNIT · Python

_default_backend

htdt/godogen · asset-gen/tools/asset_gen.py:86

def _default_backend() -> str | None:
    """Gemini when its key is set (faster, same quality), else Grok."""
    if os.environ.get("GEMINI_API_KEY") or os.environ.get("GOOGLE_API_KEY"):
        return "gemini"
    if os.environ.get("XAI_API_KEY"):
        return "grok"
    return None


def _generate_gemini(args, output: Path, cost: int):
    config = types.GenerateContentConfig(
        response_modalities=["IMAGE"],
        image_config=types.ImageConfig(
            image_size=args.size,
            aspect_ratio=args.aspect_ratio,
        ),
    )

View source record →