google()

def google(model_id: str) -> LanguageModel

export GOOGLE_API_KEY="..."

from ai_query import generate_text
from ai_query.providers import google

result = await generate_text(
    model=google("gemini-2.5-flash"),
    prompt="Hello, world!"
)

result = await generate_text(
    model=google("gemini-2.5-flash"),
    prompt="Be creative.",
    provider_options={
        "google": {
            "temperature": 0.8,
            "max_output_tokens": 2000,
            "top_p": 0.9,
            "top_k": 40,
            "safety_settings": {
                "HARM_CATEGORY_VIOLENCE": "BLOCK_ONLY_HIGH"
            }
        }
    }
)