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 |
---|---|---|---|
| Secret | Default: | Google AI Studio API key. |
| String |
| Name of the model to use. |
| [[Union[GenerationConfig, Dict "h-0":]] | Default: | The generation configuration to use. Can either be a |
| Dictionary of HarmCategory and HarmBlockThreshold | Default: | The safety settings to use. A dictionary with |
| List of tools | Default: | A list of Tool objects that you can use for Function calling. |
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.