Classification

yaml
type: "io.kestra.plugin.ai.completion.Classification"

Examples

yaml
id: text_categorization
namespace: company.ai

tasks:
  - id: categorize
    type: io.kestra.plugin.ai.completion.Classification
    prompt: "Categorize the sentiment of: I love this product!"
    classes:
      - positive
      - negative
      - neutral
    provider:
      type: io.kestra.plugin.ai.provider.GoogleGemini
      apiKey: "{{ kv('GEMINI_API_KEY') }}"
      modelName: gemini-2.5-flash

Properties

classes *array

SubTypestring

prompt *string

provider *

Definitions

Amazon Bedrock Model Provider

accessKeyId*string

modelName*string

secretAccessKey*string

type*object

baseUrlstring

caPemstring

clientPemstring

modelTypestring

DefaultCOHERE

Possible Values

COHERETITAN

Anthropic AI Model Provider

apiKey*string

modelName*string

type*object

baseUrlstring

caPemstring

clientPemstring

maxTokensintegerstring

Azure OpenAI Model Provider

endpoint*string

modelName*string

type*object

apiKeystring

baseUrlstring

caPemstring

clientIdstring

clientPemstring

clientSecretstring

serviceVersionstring

tenantIdstring

DashScope (Qwen) Model Provider from Alibaba Cloud

apiKey*string

modelName*string

type*object

baseUrlstring

Defaulthttps://dashscope-intl.aliyuncs.com/api/v1

caPemstring

clientPemstring

enableSearchbooleanstring

maxTokensintegerstring

repetitionPenaltynumberstring

Deepseek Model Provider

apiKey*string

modelName*string

type*object

baseUrlstring

Defaulthttps://api.deepseek.com/v1

caPemstring

clientPemstring

GitHub Models AI Model Provider

gitHubToken*string

modelName*string

type*object

baseUrlstring

caPemstring

clientPemstring

Google Gemini Model Provider

apiKey*string

modelName*string

type*object

baseUrlstring

caPemstring

clientPemstring

embeddingModelConfiguration

io.kestra.plugin.ai.provider.GoogleGemini-EmbeddingModelConfiguration

maxRetriesintegerstring

outputDimensionalityintegerstring

taskTypestring

Possible Values

RETRIEVAL_QUERYRETRIEVAL_DOCUMENTSEMANTIC_SIMILARITYCLASSIFICATIONCLUSTERINGQUESTION_ANSWERINGFACT_VERIFICATION

timeoutstring

Formatduration

titleMetadataKeystring

Google VertexAI Model Provider

endpoint*string

location*string

modelName*string

project*string

type*object

baseUrlstring

caPemstring

clientPemstring

HuggingFace Model Provider

apiKey*string

modelName*string

type*object

baseUrlstring

Defaulthttps://router.huggingface.co/v1

caPemstring

clientPemstring

LocalAI Model Provider

baseUrl*string

modelName*string

type*object

caPemstring

clientPemstring

Mistral AI Model Provider

apiKey*string

modelName*string

type*object

baseUrlstring

caPemstring

clientPemstring

OciGenAI Model Provider

compartmentId*string

modelName*string

region*string

type*object

authProviderstring

baseUrlstring

caPemstring

clientPemstring

Ollama Model Provider

endpoint*string

modelName*string

type*object

baseUrlstring

caPemstring

clientPemstring

OpenAI Model Provider

apiKey*string

modelName*string

type*object

baseUrlstring

Defaulthttps://api.openai.com/v1

caPemstring

clientPemstring

OpenRouter Model Provider

apiKey*string

modelName*string

type*object

baseUrlstring

caPemstring

clientPemstring

Watsonx AI Model Provider

apiKey*string

modelName*string

projectId*string

type*object

baseUrlstring

caPemstring

clientPemstring

WorkersAI Model Provider

accountId*string

apiKey*string

modelName*string

type*object

baseUrlstring

caPemstring

clientPemstring

ZhiPu AI Model Provider

apiKey*string

modelName*string

type*object

baseUrlstring

Defaulthttps://open.bigmodel.cn/

caPemstring

clientPemstring

maxRetriesintegerstring

maxTokenintegerstring

stopsarray

SubTypestring

configuration

Default{}

Definitions

io.kestra.plugin.ai.domain.ChatConfiguration

logRequestsbooleanstring

logResponsesbooleanstring

maxTokenintegerstring

responseFormat

io.kestra.plugin.ai.domain.ChatConfiguration-ResponseFormat

jsonSchemaobject

jsonSchemaDescriptionstring

typestring

DefaultTEXT

Possible Values

TEXTJSON

returnThinkingbooleanstring

seedintegerstring

temperaturenumberstring

thinkingBudgetTokensintegerstring

thinkingEnabledbooleanstring

topKintegerstring

topPnumberstring

systemMessage string

DefaultRespond by only one of the following classes by typing just the exact class name: {{ classes }}

Outputs

classification string

finishReason string

Possible Values

STOPLENGTHTOOL_EXECUTIONCONTENT_FILTEROTHER

tokenUsage

Definitions

io.kestra.plugin.ai.domain.TokenUsage

inputTokenCountinteger

outputTokenCountinteger

totalTokenCountinteger

Metrics

input.token.count counter

Unittoken

output.token.count counter

Unittoken

total.token.count counter

Unittoken

PostgreSQL

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

Database

SurrealDB

Tasks that run SurrealQL queries against SurrealDB.

Database

Debezium Microsoft SQL Server

Tasks that stream Microsoft SQL Server change data capture events into Kestra with Debezium.

Database

Classification

More Plugins in this Category

PostgreSQL

SurrealDB

Debezium Microsoft SQL Server

1.5.0