ConverseStream

id: bedrock_converse_stream
namespace: company.team

inputs:
  - id: user_message
    type: STRING

tasks:
  - id: stream_chat
    type: io.kestra.plugin.aws.bedrock.ConverseStream
    region: "{{ secret('AWS_REGION') }}"
    accessKeyId: "{{ secret('AWS_ACCESS_KEY_ID') }}"
    secretKeyId: "{{ secret('AWS_SECRET_ACCESS_KEY') }}"
    modelId: "anthropic.claude-3-5-sonnet-20241022-v2:0"
    messages:
      - role: user
        content: "{{ inputs.user_message }}"
    system: "You are a concise data engineering assistant."
    inferenceConfig:
      maxTokens: 2048

  - id: log_reply
    type: io.kestra.plugin.core.log.Log
    message: "{{ outputs.stream_chat.content }}"