Class GoogleVertexAI

Enables calls to the Google Cloud's Vertex AI API to access Large Language Models.

This entrypoint and class are intended to be used in web environments like Edge functions where you do not have access to the file system. It supports passing service account credentials directly as a "GOOGLE_VERTEX_AI_WEB_CREDENTIALS" environment variable or directly as "authOptions.credentials".

Example

const model = new GoogleVertexAI({
  temperature: 0.7,
});
const stream = await model.stream(
  "What would be a good company name for a company that makes colorful socks?",
);
for await (const chunk of stream) {
  console.log(chunk);
}

Hierarchy

  • BaseGoogleVertexAI<WebGoogleAuthOptions>
    • GoogleVertexAI

Constructors

constructor

Properties

maxOutputTokens model temperature topK topP connection streamedConnection

Methods

extractGenerationFromPrediction extractPredictionFromResponse formatInstance formatInstanceCode formatInstanceText formatParameters

Constructors

constructor

Properties

maxOutputTokens

maxOutputTokens: number = 1024

model

model: string = "text-bison"

temperature

temperature: number = 0.7

topK

topK: number = 40

topP

topP: number = 0.8

Protected connection

connection: GoogleVertexAILLMConnection<BaseLanguageModelCallOptions, GoogleVertexAILLMInstance, TextPrediction, WebGoogleAuthOptions>

Protected streamedConnection

streamedConnection: GoogleVertexAILLMConnection<BaseLanguageModelCallOptions, GoogleVertexAILLMInstance, TextPrediction, WebGoogleAuthOptions>

Methods

extractGenerationFromPrediction

  • Parameters

    • prediction: TextPrediction

    Returns Generation

extractPredictionFromResponse

  • Extracts the prediction from the API response.

    Parameters

    • result: GoogleVertexAILLMResponse<TextPrediction>

      The API response from which to extract the prediction.

    Returns TextPrediction

    A TextPrediction object representing the extracted prediction.

formatInstance

  • Formats the input instance for the Google Vertex AI model based on the model type (text or code).

    Parameters

    • prompt: string

      Prompt to be formatted as an instance.

    Returns GoogleVertexAILLMInstance

    A GoogleVertexAILLMInstance object representing the formatted instance.

formatInstanceCode

  • Formats the input instance as a code instance for the Google Vertex AI model.

    Parameters

    • prompt: string

      Prompt to be formatted as a code instance.

    Returns GoogleVertexAILLMInstance

    A GoogleVertexAILLMInstance object representing the formatted code instance.

formatInstanceText

  • Formats the input instance as a text instance for the Google Vertex AI model.

    Parameters

    • prompt: string

      Prompt to be formatted as a text instance.

    Returns GoogleVertexAILLMInstance

    A GoogleVertexAILLMInstance object representing the formatted text instance.

formatParameters

  • Returns GoogleVertexAIModelParams

Settings

Member Visibility

Theme

On This Page

Generated using TypeDoc