GoogleAIGeminiGenerator Parameters

Check the init and runtime parameters you can use to customize GoogleAIGeminiGenerator.

YAML Init Parameters

These are the parameters you can pass to this component in the pipeline YAML configuration:

Parameter

Type

Possible values

Description

api_key

Secret

Default: Secret.from_env_var("GOOGLE_API_KEY")

Google AI Studio API key.
Required.

model

String

gemini-pro
gemini-pro-vision
gemini-ultra
Default: gemini-pro-vision

Name of the model to use.
Required.

generation_config

[[Union[GenerationConfig, Dict "h-0":]]

Default: None

The generation configuration to use. Can either be a GenerationConfig object or a dictionary of parameters. For available parameters, see GenerationConfig API reference. Optional.

safety_settings

Dictionary of HarmCategory and HarmBlockThreshold

Default: None

The safety settings to use. A dictionary with HarmCategory as keys and HarmBlockThreshold as values. For more information, see API reference.
Optional.

tools

List of tools

Default: None

A list of Tool objects that you can use for Function calling.
Optional.



REST API Runtime Parameters

There are no runtime parameters you can pass to this component when making a request to the Search REST API endpoint.