TextCompletion

yaml
type: "io.kestra.plugin.gcp.vertexai.TextCompletion"

Examples

yaml
id: gcp_vertexai_text_completion
namespace: company.team

tasks:
  - id: text_completion
    type: io.kestra.plugin.gcp.vertexai.TextCompletion
    region: us-central1
    projectId: my-project
    prompt: Please tell me a joke

Properties

prompt *string

region *string

impersonatedServiceAccount string

modelId string

Defaultgemini-pro

parameters

Default

{
  "temperature": 0.2,
  "maxOutputTokens": 128,
  "topK": 40,
  "topP": 0.95
}

Definitions

io.kestra.plugin.gcp.vertexai.AbstractGenerativeAi-ModelParameter

maxOutputTokensinteger

Default128

Minimum>= 1

Maximum<= 1024

temperaturenumber

Default0.2

Minimum>

Maximum<= 1

topKinteger

Default40

Minimum>= 1

Maximum<= 40

topPnumber

Default0.95

Minimum>

Maximum<= 1

projectId string

scopes array

SubTypestring

Default["https://www.googleapis.com/auth/cloud-platform"]

serviceAccount string

Outputs

predictions array

Definitions

io.kestra.plugin.gcp.vertexai.AbstractGenerativeAi-Prediction

citationMetadata

io.kestra.plugin.gcp.vertexai.AbstractGenerativeAi-CitationMetadata

citationsarray

io.kestra.plugin.gcp.vertexai.AbstractGenerativeAi-Citation

citationsarray

SubTypestring

contentstring

safetyAttributes

io.kestra.plugin.gcp.vertexai.AbstractGenerativeAi-SafetyAttributes

blockedboolean

categoriesarray

SubTypestring

scoresarray

SubTypenumber

Metrics

candidate.token.count counter

prompt.token.count counter

serialized.size counter

Unitbytes

total.token.count counter

Core Plugins and tasks

Tasks that provide Kestra's built-in orchestration, I/O, and observability capabilities.

CoreStorageScript

PostgreSQL

Tasks that connect to PostgreSQL via JDBC to run queries and loads.

Database

SurrealDB

Tasks that run SurrealQL queries against SurrealDB.

Database

TextCompletion

More Plugins in this Category

Core Plugins and tasks

PostgreSQL

SurrealDB

1.2.0