com.google.cloud.vertexai.generativeai

Class GenerativeModel.Builder

java.lang.Object

com.google.cloud.vertexai.generativeai.GenerativeModel.Builder

Enclosing class:

GenerativeModel

public static class GenerativeModel.Builder
extends Object

Builder class for GenerativeModel.

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
GenerativeModel	build()
GenerativeModel.Builder	setGenerationConfig(GenerationConfig generationConfig) Sets GenerationConfig that will be used by default to interact with the generative model.
GenerativeModel.Builder	setModelName(String modelName) Sets the name of the generative model.
GenerativeModel.Builder	setSafetySettings(List<SafetySetting> safetySettings) Sets a list of SafetySetting that will be used by default to interact with the generative model.
GenerativeModel.Builder	setSystemInstruction(Content systemInstruction) Sets a system instruction that will be used by default to interact with the generative model.
GenerativeModel.Builder	setToolConfig(ToolConfig toolConfig) Sets a ToolConfig that will be used by default to interact with the generative model.
GenerativeModel.Builder	setTools(List<Tool> tools) Sets a list of Tool that will be used by default to interact with the generative model.
GenerativeModel.Builder	setVertexAi(VertexAI vertexAi) Sets VertexAI that contains the default configs for the generative model.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Builder

public Builder()

Method Detail

build

public GenerativeModel build()

setModelName

@CanIgnoreReturnValue
public GenerativeModel.Builder setModelName(String modelName)

Sets the name of the generative model. This is required for building a GenerativeModel instance. Supported format: "gemini-pro", "models/gemini-pro", "publishers/google/models/gemini-pro", where "gemini-pro" is the model name. Valid model names can be found in the Gemini models documentation: https://cloud.google.com/vertex-ai/docs/generative-ai/learn/models#gemini-models

setVertexAi

@CanIgnoreReturnValue
public GenerativeModel.Builder setVertexAi(VertexAI vertexAi)

Sets VertexAI that contains the default configs for the generative model. This is required for building a GenerativeModel instance.

setGenerationConfig

@CanIgnoreReturnValue
public GenerativeModel.Builder setGenerationConfig(GenerationConfig generationConfig)

Sets GenerationConfig that will be used by default to interact with the generative model.

setSafetySettings

@CanIgnoreReturnValue
public GenerativeModel.Builder setSafetySettings(List<SafetySetting> safetySettings)

Sets a list of SafetySetting that will be used by default to interact with the generative model.

setTools

@CanIgnoreReturnValue
public GenerativeModel.Builder setTools(List<Tool> tools)

Sets a list of Tool that will be used by default to interact with the generative model.

setToolConfig

@CanIgnoreReturnValue
public GenerativeModel.Builder setToolConfig(ToolConfig toolConfig)

Sets a ToolConfig that will be used by default to interact with the generative model.

setSystemInstruction

@CanIgnoreReturnValue
public GenerativeModel.Builder setSystemInstruction(Content systemInstruction)

Sets a system instruction that will be used by default to interact with the generative model.