Merge pull request #25698 from hashicorp/f-aws_transcribe_language_model

New Resource: aws_transcribe_language_model

.changelog/25698.txt

@@ -0,0 +1,3 @@
+```release-note:new-resource
+ aws_transcribe_language_model
+```

internal/provider/provider.go

@@ -2054,6 +2054,7 @@ func New(_ context.Context) (*schema.Provider, error) {
 			"aws_timestreamwrite_database": timestreamwrite.ResourceDatabase(),
 			"aws_timestreamwrite_table":    timestreamwrite.ResourceTable(),
 
+			"aws_transcribe_language_model":     transcribe.ResourceLanguageModel(),
 			"aws_transcribe_medical_vocabulary": transcribe.ResourceMedicalVocabulary(),
 			"aws_transcribe_vocabulary":         transcribe.ResourceVocabulary(),
 			"aws_transcribe_vocabulary_filter":  transcribe.ResourceVocabularyFilter(),

internal/service/transcribe/language_model.go

@@ -0,0 +1,337 @@
+package transcribe
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"log"
+	"strings"
+	"time"
+
+	"github.com/aws/aws-sdk-go-v2/aws"
+	"github.com/aws/aws-sdk-go-v2/aws/arn"
+	"github.com/aws/aws-sdk-go-v2/service/transcribe"
+	"github.com/aws/aws-sdk-go-v2/service/transcribe/types"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
+	"github.com/hashicorp/terraform-provider-aws/internal/conns"
+	"github.com/hashicorp/terraform-provider-aws/internal/enum"
+	tftags "github.com/hashicorp/terraform-provider-aws/internal/tags"
+	"github.com/hashicorp/terraform-provider-aws/internal/tfresource"
+	"github.com/hashicorp/terraform-provider-aws/internal/verify"
+	"github.com/hashicorp/terraform-provider-aws/names"
+)
+
+func ResourceLanguageModel() *schema.Resource {
+	return &schema.Resource{
+		CreateWithoutTimeout: resourceLanguageModelCreate,
+		ReadWithoutTimeout:   resourceLanguageModelRead,
+		UpdateWithoutTimeout: resourceLanguageModelUpdate,
+		DeleteWithoutTimeout: resourceLanguageModelDelete,
+
+		Importer: &schema.ResourceImporter{
+			StateContext: schema.ImportStatePassthroughContext,
+		},
+
+		Timeouts: &schema.ResourceTimeout{
+			Create: schema.DefaultTimeout(600 * time.Minute),
+		},
+
+		Schema: map[string]*schema.Schema{
+			"arn": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"base_model_name": {
+				Type:             schema.TypeString,
+				Required:         true,
+				ForceNew:         true,
+				ValidateDiagFunc: enum.Validate[types.BaseModelName](),
+			},
+			"input_data_config": {
+				Type:     schema.TypeList,
+				Required: true,
+				ForceNew: true,
+				MaxItems: 1,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"data_access_role_arn": {
+							Type:             schema.TypeString,
+							Required:         true,
+							ForceNew:         true,
+							ValidateDiagFunc: validation.ToDiagFunc(verify.ValidARN),
+						},
+						"s3_uri": {
+							Type:     schema.TypeString,
+							Required: true,
+							ForceNew: true,
+						},
+						"tuning_data_s3_uri": {
+							Type:     schema.TypeString,
+							Optional: true,
+							ForceNew: true,
+							Computed: true,
+						},
+					},
+				},
+			},
+			"language_code": {
+				Type:             schema.TypeString,
+				Required:         true,
+				ForceNew:         true,
+				ValidateDiagFunc: enum.Validate[types.LanguageCode](),
+			},
+			"model_name": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+			"tags":     tftags.TagsSchema(),
+			"tags_all": tftags.TagsSchemaComputed(),
+		},
+
+		CustomizeDiff: verify.SetTagsDiff,
+	}
+}
+
+const (
+	ResNameLanguageModel = "Language Model"
+
+	propagationTimeout = 2 * time.Minute
+)
+
+func resourceLanguageModelCreate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	conn := meta.(*conns.AWSClient).TranscribeConn
+
+	in := &transcribe.CreateLanguageModelInput{
+		BaseModelName: types.BaseModelName(d.Get("base_model_name").(string)),
+		LanguageCode:  types.CLMLanguageCode(d.Get("language_code").(string)),
+		ModelName:     aws.String(d.Get("model_name").(string)),
+	}
+
+	if v, ok := d.GetOk("input_data_config"); ok && len(v.([]interface{})) > 0 {
+		in.InputDataConfig = expandInputDataConfig(v.([]interface{}))
+	}
+
+	defaultTagsConfig := meta.(*conns.AWSClient).DefaultTagsConfig
+	tags := defaultTagsConfig.MergeTags(tftags.New(d.Get("tags").(map[string]interface{})))
+
+	if len(tags) > 0 {
+		in.Tags = Tags(tags.IgnoreAWS())
+	}
+
+	outputRaw, err := tfresource.RetryWhen(propagationTimeout,
+		func() (interface{}, error) {
+			return conn.CreateLanguageModel(ctx, in)
+		},
+		func(err error) (bool, error) {
+			var bre *types.BadRequestException
+			if errors.As(err, &bre) {
+				return strings.Contains(bre.ErrorMessage(), "Make sure that you have read permission"), err
+			}
+			return false, err
+		},
+	)
+
+	if err != nil {
+		return names.DiagError(names.Transcribe, names.ErrActionCreating, ResNameLanguageModel, d.Get("model_name").(string), err)
+	}
+
+	d.SetId(aws.ToString(outputRaw.(*transcribe.CreateLanguageModelOutput).ModelName))
+
+	if _, err := waitLanguageModelCreated(ctx, conn, d.Id(), d.Timeout(schema.TimeoutCreate)); err != nil {
+		return names.DiagError(names.Transcribe, names.ErrActionWaitingForCreation, ResNameLanguageModel, d.Get("model_name").(string), err)
+	}
+
+	return resourceLanguageModelRead(ctx, d, meta)
+}
+
+func resourceLanguageModelRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	conn := meta.(*conns.AWSClient).TranscribeConn
+
+	out, err := FindLanguageModelByName(ctx, conn, d.Id())
+
+	if !d.IsNewResource() && tfresource.NotFound(err) {
+		log.Printf("[WARN] Transcribe LanguageModel (%s) not found, removing from state", d.Id())
+		d.SetId("")
+		return nil
+	}
+
+	if err != nil {
+		return names.DiagError(names.Transcribe, names.ErrActionReading, ResNameLanguageModel, d.Id(), err)
+	}
+
+	arn := arn.ARN{
+		AccountID: meta.(*conns.AWSClient).AccountID,
+		Partition: meta.(*conns.AWSClient).Partition,
+		Service:   "transcribe",
+		Region:    meta.(*conns.AWSClient).Region,
+		Resource:  fmt.Sprintf("language-model/%s", d.Id()),
+	}.String()
+
+	d.Set("arn", arn)
+	d.Set("base_model_name", out.BaseModelName)
+	d.Set("language_code", out.LanguageCode)
+	d.Set("model_name", out.ModelName)
+
+	if err := d.Set("input_data_config", flattenInputDataConfig(out.InputDataConfig)); err != nil {
+		return names.DiagError(names.Transcribe, names.ErrActionSetting, ResNameLanguageModel, d.Id(), err)
+	}
+
+	tags, err := ListTags(ctx, conn, arn)
+	if err != nil {
+		return names.DiagError(names.Transcribe, names.ErrActionReading, ResNameLanguageModel, d.Id(), err)
+	}
+
+	defaultTagsConfig := meta.(*conns.AWSClient).DefaultTagsConfig
+	ignoreTagsConfig := meta.(*conns.AWSClient).IgnoreTagsConfig
+	tags = tags.IgnoreAWS().IgnoreConfig(ignoreTagsConfig)
+
+	//lintignore:AWSR002
+	if err := d.Set("tags", tags.RemoveDefaultConfig(defaultTagsConfig).Map()); err != nil {
+		return names.DiagError(names.Transcribe, names.ErrActionSetting, ResNameLanguageModel, d.Id(), err)
+	}
+
+	if err := d.Set("tags_all", tags.Map()); err != nil {
+		return names.DiagError(names.Transcribe, names.ErrActionSetting, ResNameLanguageModel, d.Id(), err)
+	}
+
+	return nil
+}
+
+func resourceLanguageModelUpdate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	conn := meta.(*conns.AWSClient).TranscribeConn
+
+	if d.HasChange("tags_all") {
+		o, n := d.GetChange("tags_all")
+
+		if err := UpdateTags(ctx, conn, d.Get("arn").(string), o, n); err != nil {
+			return names.DiagError(names.Transcribe, names.ErrActionUpdating, ResNameLanguageModel, d.Id(), err)
+		}
+	}
+
+	return resourceLanguageModelRead(ctx, d, meta)
+}
+
+func resourceLanguageModelDelete(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
+	conn := meta.(*conns.AWSClient).TranscribeConn
+
+	log.Printf("[INFO] Deleting Transcribe LanguageModel %s", d.Id())
+
+	_, err := conn.DeleteLanguageModel(ctx, &transcribe.DeleteLanguageModelInput{
+		ModelName: aws.String(d.Id()),
+	})
+
+	var resourceNotFoundException *types.NotFoundException
+	if errors.As(err, &resourceNotFoundException) {
+		return nil
+	}
+
+	if err != nil {
+		return names.DiagError(names.Transcribe, names.ErrActionDeleting, ResNameLanguageModel, d.Id(), err)
+	}
+
+	return nil
+}
+
+func waitLanguageModelCreated(ctx context.Context, conn *transcribe.Client, id string, timeout time.Duration) (*types.LanguageModel, error) {
+	stateConf := &resource.StateChangeConf{
+		Pending:                   enum.Slice(types.ModelStatusInProgress),
+		Target:                    enum.Slice(types.ModelStatusCompleted),
+		Refresh:                   statusLanguageModel(ctx, conn, id),
+		Timeout:                   timeout,
+		NotFoundChecks:            20,
+		ContinuousTargetOccurence: 2,
+	}
+
+	outputRaw, err := stateConf.WaitForStateContext(ctx)
+	if out, ok := outputRaw.(*types.LanguageModel); ok {
+		return out, err
+	}
+
+	return nil, err
+}
+
+func statusLanguageModel(ctx context.Context, conn *transcribe.Client, name string) resource.StateRefreshFunc {
+	return func() (interface{}, string, error) {
+		out, err := FindLanguageModelByName(ctx, conn, name)
+		if tfresource.NotFound(err) {
+			return nil, "", nil
+		}
+
+		if err != nil {
+			return nil, "", err
+		}
+
+		return out, string(out.ModelStatus), nil
+	}
+}
+
+func FindLanguageModelByName(ctx context.Context, conn *transcribe.Client, id string) (*types.LanguageModel, error) {
+	in := &transcribe.DescribeLanguageModelInput{
+		ModelName: aws.String(id),
+	}
+
+	out, err := conn.DescribeLanguageModel(ctx, in)
+
+	var bre *types.BadRequestException
+	if errors.As(err, &bre) {
+		return nil, &resource.NotFoundError{
+			LastError:   err,
+			LastRequest: in,
+		}
+	}
+
+	if err != nil {
+		return nil, err
+	}
+
+	if out == nil || out.LanguageModel == nil {
+		return nil, tfresource.NewEmptyResultError(in)
+	}
+
+	return out.LanguageModel, nil
+}
+
+func flattenInputDataConfig(apiObjects *types.InputDataConfig) []interface{} {
+	if apiObjects == nil {
+		return nil
+	}
+
+	m := map[string]interface{}{
+		"data_access_role_arn": apiObjects.DataAccessRoleArn,
+		"s3_uri":               apiObjects.S3Uri,
+	}
+
+	if aws.ToString(apiObjects.TuningDataS3Uri) != "" {
+		m["tuning_data_s3_uri"] = apiObjects.TuningDataS3Uri
+	}
+
+	return []interface{}{m}
+}
+
+func expandInputDataConfig(tfList []interface{}) *types.InputDataConfig {
+	if len(tfList) == 0 || tfList[0] == nil {
+		return nil
+	}
+
+	s := &types.InputDataConfig{}
+
+	i := tfList[0].(map[string]interface{})
+
+	if val, ok := i["data_access_role_arn"]; ok {
+		s.DataAccessRoleArn = aws.String(val.(string))
+	}
+
+	if val, ok := i["s3_uri"]; ok {
+		s.S3Uri = aws.String(val.(string))
+	}
+
+	if val, ok := i["tuning_data_s3_uri"]; ok && val != "" {
+		s.TuningDataS3Uri = aws.String(val.(string))
+	}
+
+	return s
+}