VertexAITextGenerator Parameters

Check the init parameters for VertexAIGeminiGenerator.

YAML Init Parameters

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

ParameterTypePossible valuesDescription
modelStringtext-bison
text-unicorn
text-bison-32k
Default: text-bison
Name of the model to use.
Required.
project_idStringID of the GCP project to use. You can find your project ID in the GCP resource manager, or locally by running gcloud projects list in your terminal. For information on using gcloud CLI, see the official documentation.
Required.
locationStringDefault: NoneThe default location to use when making API calls.
Optional.
kwargsDictionaryAdditional keyword arguments to pass to the model.

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.