Zum Hauptinhalt springen

Installation

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

Verwendung

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="Erklären Sie die Risiken von algorithmischer Voreingenommenheit bei Einstellungen.",
    )

    print(f"Inhalt: {response.content}")
    print(f"RAIL Score: {response.rail_score}/10")
    print(f"Schwellenwert erreicht: {response.threshold_met}")
    print(f"Wurde regeneriert: {response.was_regenerated}")

asyncio.run(main())

Vertex AI

Um Gemini über Vertex AI anstelle von Google AI Studio zu verwenden, übergeben Sie use_vertex=True:
from rail_score_sdk.integrations import RAILGemini

client = RAILGemini(
    use_vertex=True,
    gcp_project="your-gcp-project-id",
    gcp_region="us-central1",
    rail_api_key="YOUR_RAIL_API_KEY",
    rail_threshold=7.0,
)

response = await client.generate_content(
    model="gemini-1.5-pro",
    contents="...",
)
Vertex AI verwendet Anwendungsstandardanmeldeinformationen. Führen Sie gcloud auth application-default login aus oder setzen Sie GOOGLE_APPLICATION_CREDENTIALS auf Ihre Dienstkontoschlüsseldatei.

Konfigurationsoptionen

ParameterTypStandardBeschreibung
google_api_keystrGoogle AI Studio API-Schlüssel (nicht erforderlich, wenn use_vertex=True)
rail_api_keystrIhr RAIL API-Schlüssel
rail_thresholdfloatNoneBlockieren oder regenerieren unterhalb dieses Scores
rail_policystr"log_only""log_only" | "block" | "regenerate"
rail_modestr"basic""basic" | "deep"
use_vertexboolFalseÜber Vertex AI anstelle von Google AI Studio leiten
gcp_projectstrGCP-Projekt-ID (erforderlich, wenn use_vertex=True)
gcp_regionstr"us-central1"Vertex AI-Region
Übergeben Sie rail_skip=True an generate_content(), um die RAIL-Bewertung für einen einzelnen Aufruf zu umgehen.

Was kommt als Nächstes

OpenAI-Integration

RAILOpenAI-Wrapper für Chat-Vervollständigungen.

Anthropic-Integration

RAILAnthropic-Wrapper für Claude.

Konzepte: Middleware

Wie Anbieter-Wrapper im Hintergrund funktionieren.

Python SDK-Übersicht

Vollständige Installation und Referenz des Python SDK.