Google Gemini

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

npm install @responsible-ai-labs/rail-score @google/generative-ai

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())

import { RAILGemini } from "@responsible-ai-labs/rail-score";

const client = new RAILGemini({
  googleApiKey: "AIza...",
  railApiKey: "YOUR_RAIL_API_KEY",
  railThreshold: 7.0,
  railPolicy: "regenerate",
});

const response = await client.generateContent({
  model: "gemini-1.5-pro",
  contents: "Explain the risks of algorithmic bias in hiring.",
});

console.log(`RAIL Score: ${response.railScore}/10`);