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:

ParameterTypePossible valuesDescription
api_keySecretDefault: Secret.from_env_var("GOOGLE_API_KEY")Google AI Studio API key.
Required.
modelStringgemini-pro
gemini-pro-vision
gemini-ultra
Default: gemini-pro-vision
Name of the model to use.
Required.
generation_config[[Union[GenerationConfig, Dict[str, Any]]]Default: NoneThe generation configuration to use. Can either be a GenerationConfig object or a dictionary of parameters. For available parameters, see GenerationConfig API reference. Optional.
safety_settingsDictionary of HarmCategory and HarmBlockThresholdDefault: NoneThe safety settings to use. A dictionary with HarmCategory as keys and HarmBlockThreshold as values. For more information, see API reference.
Optional.
toolsList of toolsDefault: NoneA 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.