Google Gemini

pip install "rail-score-sdk[google]"

import asyncio
from rail_score_sdk.integrations import RAILGemini

async def main():
    client = RAILGemini(
        google_api_key="AIza...",
        rail_api_key="YOUR_RAIL_API_KEY",
        rail_threshold=7.0,
        rail_policy="regenerate",
        rail_mode="basic",
    )

    response = await client.generate_content(
        model="gemini-1.5-pro",
        contents="Explain the risks of algorithmic bias in hiring.",
    )

    print(f"Content: {response.content}")
    print(f"RAIL Score: {response.rail_score}/10")
    print(f"Threshold met: {response.threshold_met}")
    print(f"Was regenerated: {response.was_regenerated}")

asyncio.run(main())