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[str, Any]]] | 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.