Google Models
Google AI model components provide access to Google's Gemini and other generative AI models through Google AI Studio and Vertex AI platforms.
Google Generative AI
This component generates text using Google's Generative AI models.
For more information, see the Google Generative AI documentation.
Inputs
Google API Key
Google API Key
Your Google API key to use for the Google Generative AI.
Model
Model
The name of the model to use, such as "gemini-pro".
Max Output Tokens
Max Output Tokens
The maximum number of tokens to generate.
Temperature
Temperature
Run inference with this temperature.
Top K
Top K
Consider the set of top K most probable tokens.
Top P
Top P
The maximum cumulative probability of tokens to consider when sampling.
N
N
Number of chat completions to generate for each prompt.
Outputs
model
LanguageModel
An instance of ChatGoogleGenerativeAI configured with the specified parameters.
VertexAI
This component generates text using Vertex AI LLMs.
For more information, see Google Vertex AI documentation.
Inputs
credentials
File
JSON credentials file. Leave empty to fallback to environment variables. File type: JSON.
model_name
String
The name of the Vertex AI model to use. Default: "gemini-1.5-pro".
project
String
The project ID (advanced).
location
String
The location for the Vertex AI API. Default: "us-central1" (advanced).
max_output_tokens
Integer
The maximum number of tokens to generate (advanced).
max_retries
Integer
Maximum number of retries for API calls. Default: 1 (advanced).
temperature
Float
Controls randomness in the output. Default: 0.0.
top_k
Integer
The number of highest probability vocabulary tokens to keep for top-k-filtering (advanced).
top_p
Float
The cumulative probability of parameter highest probability vocabulary tokens to keep for nucleus sampling. Default: 0.95 (advanced).
verbose
Boolean
Whether to print verbose output. Default: False (advanced).
Outputs
model
LanguageModel
An instance of ChatVertexAI configured with the specified parameters.
Usage Notes
Google Generative AI: Direct access to Google's AI Studio API with Gemini models
VertexAI: Enterprise-grade Google Cloud AI platform with additional models and enterprise features
VertexAI supports more advanced configurations and enterprise authentication methods
Both platforms provide access to state-of-the-art multimodal capabilities
Last updated