Skip to main content

Google GenAI Plugin

This Genkit plugin provides a unified interface for Google AI (Gemini) and Vertex AI models, embedding, and other services.

Setup environment

uv add genkit genkit-google-genai

Configuration

Google AI (AI Studio)

To use Google AI models, obtain an API key from Google AI Studio and set it in your environment:

export GEMINI_API_KEY='<your-api-key>'

Vertex AI (Google Cloud)

To use Vertex AI models, ensure you have a Google Cloud project and Application Default Credentials (ADC) set up:

gcloud auth application-default login

Features

Dynamic Models

The plugin automatically discovers available models from the API upon initialization. You can use any model name supported by the API (e.g., googleai/gemini-flash-latest, vertexai/gemini-2.5-pro).

Dynamic Configuration

New or experimental parameters can be passed flexibly using model_validate to bypass strict schema checks:

from genkit_google_genai import GeminiConfigSchema

config = GeminiConfigSchema.model_validate({
    'temperature': 1.0,
    'response_modalities': ['TEXT', 'IMAGE'],
})

Vertex AI Evaluators

Built-in evaluators for assessing model output quality. Evaluators are automatically registered when using the VertexAI plugin and are accessed via ai.evaluate():

from genkit import Genkit
from genkit.evaluator import BaseDataPoint
from genkit_google_genai import VertexAI

ai = Genkit(plugins=[VertexAI(project='my-project')])

# Prepare test dataset
dataset = [
    BaseDataPoint(
        input='Write about AI.',
        output='AI is transforming industries through intelligent automation.',
    ),
]

# Evaluate fluency (scores 1-5)
results = await ai.evaluate(
    evaluator='vertexai/fluency',
    dataset=dataset,
)

for result in results.root:
    print(f'Score: {result.evaluation.score}')

Supported evaluators:

Evaluator Description
vertexai/bleu Translation quality (compare to reference)
vertexai/rouge Summarization quality
vertexai/fluency Language mastery and readability
vertexai/safety Harmful/inappropriate content detection
vertexai/groundedness Hallucination detection
vertexai/summarization_quality Overall summarization ability

Examples

For comprehensive usage examples, see:

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

genkit_google_genai-0.10.0.tar.gz (80.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

genkit_google_genai-0.10.0-py3-none-any.whl (65.6 kB view details)

Uploaded Python 3

File details

Details for the file genkit_google_genai-0.10.0.tar.gz.

File metadata

  • Download URL: genkit_google_genai-0.10.0.tar.gz
  • Upload date:
  • Size: 80.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for genkit_google_genai-0.10.0.tar.gz
Algorithm Hash digest
SHA256 39309639354090e9bb5af59ef2bbbde0a502a68994eb46812f31d681384a6267
MD5 013f2db905ad55811985b49831c61642
BLAKE2b-256 41f54d3a543b797364f6c8c82d8052382a3bacdb6d57ddea1953473e5d60c7dd

See more details on using hashes here.

Provenance

The following attestation bundles were made for genkit_google_genai-0.10.0.tar.gz:

Publisher: publish_python.yml on genkit-ai/genkit

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file genkit_google_genai-0.10.0-py3-none-any.whl.

File metadata

File hashes

Hashes for genkit_google_genai-0.10.0-py3-none-any.whl
Algorithm Hash digest
SHA256 b8a5a1abca95282f7f83718d052dba8ae10876cd9860372f7dfeb73f3ad449fa
MD5 148395fa6bf6df720c124e6d73e3dc41
BLAKE2b-256 1119cc99c52ade66766200a21ddea5945246125312e62ebb4fb0a4b57a368a61

See more details on using hashes here.

Provenance

The following attestation bundles were made for genkit_google_genai-0.10.0-py3-none-any.whl:

Publisher: publish_python.yml on genkit-ai/genkit

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

0.11.0

2 files

This release

0.10.0 This release

2 files

0.9.0

2 files

0.8.1

2 files

0.8.0

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page