Plugin AiCompletion

Classification ​Classification

yaml
type: "io.kestra.plugin.ai.completion.classification"
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
SubType string
Default {}
Possible Values
STOPLENGTHTOOL_EXECUTIONCONTENT_FILTEROTHER
Default https://api.deepseek.com/v1
Default TEXT
Possible Values
TEXTJSON
Default COHERE
Possible Values
COHERETITAN