Skip to content

Commit

Permalink
refactoring with latest guardrail change
Browse files Browse the repository at this point in the history
  • Loading branch information
shikha372 committed Jul 23, 2024
1 parent 9912b50 commit 0910098
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -275,14 +275,14 @@ export class BedrockInvokeModel extends sfn.TaskStateBase {
Output: this.props.output?.s3Location ? {
S3Uri: `s3://${this.props.output.s3Location.bucketName}/${this.props.output.s3Location.objectKey}`,
} : this.props.outputPath ? { S3Uri: this.props.outputPath }: undefined,
}),
GuardrailIdentifier: this.props.guardrail?.guardrailIdentifier,
GuardrailVersion: this.props.guardrail?.guardrailVersion,
Trace: this.props.traceEnabled === undefined
? undefined
: this.props.traceEnabled
? 'ENABLED'
: 'DISABLED',
}),
};
};
}
Expand Down

0 comments on commit 0910098

Please sign in to comment.