Class VertexAI (1.2.0)

public class VertexAI implements AutoCloseable

This class holds default settings and credentials to make Vertex AI API calls.

Note: The VertexAI instance will start a service client when the first API call is made. Please close the VertexAI instance after making any API calls so that clients get closed as well.


 try (VertexAI vertexAi = new VertexAI("my-project", "us-central1"); ) {
   GenerativeModel model = new GenerativeModel("gemini-pro", vertexAi)
   // Do something with the model.
 }
 

Inheritance

java.lang.Object > VertexAI

Implements

AutoCloseable

Constructors

VertexAI(String projectId, String location)

public VertexAI(String projectId, String location)

Constructs a VertexAI instance.

Parameters
Name Description
projectId String

the default project to use when making API calls

location String

the default location to use when making API calls

Methods

close()

public void close()

Closes the VertexAI instance together with all its instantiated clients.

getApiEndpoint()

public String getApiEndpoint()

Returns the default endpoint to use when making API calls.

Returns
Type Description
String

API endpoint in string format.

getCredentials()

public Credentials getCredentials()

Returns the default credentials to use when making API calls.

Returns
Type Description
com.google.auth.Credentials

Credentials to use when making API calls.

Exceptions
Type Description
IOException

getLlmUtilityClient()

public LlmUtilityServiceClient getLlmUtilityClient()

Returns the LlmUtilityServiceClient with GRPC or REST, based on the Transport type. The client will be instantiated when the first API call is made.

Returns
Type Description
LlmUtilityServiceClient

LlmUtilityServiceClient that makes calls to the backing service through method calls that map to the API methods.

getLocation()

public String getLocation()

Returns the default location to use when making API calls.

Returns
Type Description
String

Location in string format.

getPredictionServiceClient()

public PredictionServiceClient getPredictionServiceClient()

Returns the PredictionServiceClient with GRPC or REST, based on the Transport type. The client will be instantiated when the first prediction API call is made.

Returns
Type Description
PredictionServiceClient

PredictionServiceClient that send requests to the backing service through method calls that map to the API methods.

getProjectId()

public String getProjectId()

Returns the default project to use when making API calls.

Returns
Type Description
String

Project ID in string format.

getTransport()

public Transport getTransport()

Returns the default Transport layer to use to send API requests.

Returns
Type Description
Transport

Transport layer used when sending API requests.