AWS InvokeModel
CertifiedInvoke an Amazon Bedrock foundation model
AWS InvokeModel
Invoke an Amazon Bedrock foundation model
Sends a raw JSON payload to any Bedrock-supported foundation model and returns the parsed response body.
The caller is responsible for providing a body that matches the model's own request schema.
type: io.kestra.plugin.aws.bedrock.InvokeModelExamples
Generate text with Amazon Titan via InvokeModel
id: bedrock_invoke_model
namespace: company.team
inputs:
- id: prompt
type: STRING
defaults: "Summarize the key benefits of Apache Kafka in 3 bullet points."
tasks:
- id: invoke_titan
type: io.kestra.plugin.aws.bedrock.InvokeModel
region: "{{ secret('AWS_REGION') }}"
accessKeyId: "{{ secret('AWS_ACCESS_KEY_ID') }}"
secretKeyId: "{{ secret('AWS_SECRET_ACCESS_KEY') }}"
modelId: "amazon.titan-text-express-v1"
body:
inputText: "{{ inputs.prompt }}"
textGenerationConfig:
maxTokenCount: 512
temperature: 0.7
- id: log_response
type: io.kestra.plugin.core.log.Log
message: "{{ outputs.invoke_titan.body }}"
Properties
body *Requiredobject
Request body
Model-specific request payload as a map. The structure must match the target model's API schema.
modelId *Requiredstring
Model ID
The Bedrock model identifier, e.g. amazon.titan-text-express-v1 or anthropic.claude-3-5-sonnet-20241022-v2: 0.
accessKeyId string
Access Key Id in order to connect to AWS
If no credentials are defined, we will use the default credentials provider chain to fetch credentials.
compatibilityMode booleanstring
Enable compatibility mode
Use it to connect to S3 bucket with S3 compatible services that don't support the new transport client.
endpointOverride string
The endpoint with which the SDK should communicate
This property allows you to use a different S3 compatible storage backend.
forcePathStyle booleanstring
Force path style access
Must only be used when compatibilityMode is enabled.
pluginDefaultsRef Non-dynamicstring
Reference (ref) of the pluginDefaults to apply to this task.
region string
AWS region with which the SDK should communicate
secretKeyId string
Secret Key Id in order to connect to AWS
If no credentials are defined, we will use the default credentials provider chain to fetch credentials.
sessionToken string
AWS session token, retrieved from an AWS token service, used for authenticating that this user has received temporary permissions to access a given resource
If no credentials are defined, we will use the default credentials provider chain to fetch credentials.
stsEndpointOverride string
The AWS STS endpoint with which the SDKClient should communicate
stsRoleArn string
AWS STS Role
The Amazon Resource Name (ARN) of the role to assume. If set the task will use the StsAssumeRoleCredentialsProvider. If no credentials are defined, we will use the default credentials provider chain to fetch credentials.
stsRoleExternalId string
AWS STS External Id
A unique identifier that might be required when you assume a role in another account. This property is only used when an stsRoleArn is defined.
stsRoleSessionDuration string
PT15MAWS STS Session duration
The duration of the role session (default: 15 minutes, i.e., PT15M). This property is only used when an stsRoleArn is defined.
stsRoleSessionName string
AWS STS Session name
This property is only used when an stsRoleArn is defined.
Outputs
body object
Response body
Parsed JSON response body returned by the model.
modelId string
Model ID
The model that was invoked.
