com.pulumi.awsnative.bedrock.kotlin.outputs.AgentPromptConfiguration.kt Maven / Gradle / Ivy

Go to download
@file:Suppress("NAME_SHADOWING", "DEPRECATION")

package com.pulumi.awsnative.bedrock.kotlin.outputs

import com.pulumi.awsnative.bedrock.kotlin.enums.AgentCreationMode
import com.pulumi.awsnative.bedrock.kotlin.enums.AgentPromptState
import com.pulumi.awsnative.bedrock.kotlin.enums.AgentPromptType
import kotlin.String
import kotlin.Suppress

/**
 * BasePromptConfiguration per Prompt Type.
 * @property basePromptTemplate Base Prompt Template.
 * @property inferenceConfiguration Contains inference parameters to use when the agent invokes a foundation model in the part of the agent sequence defined by the `promptType` . For more information, see [Inference parameters for foundation models](https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters.html) .
 * @property parserMode Specifies whether to override the default parser Lambda function when parsing the raw foundation model output in the part of the agent sequence defined by the `promptType` . If you set the field as `OVERRIDEN` , the `overrideLambda` field in the [PromptOverrideConfiguration](https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_PromptOverrideConfiguration.html) must be specified with the ARN of a Lambda function.
 * @property promptCreationMode Specifies whether to override the default prompt template for this `promptType` . Set this value to `OVERRIDDEN` to use the prompt that you provide in the `basePromptTemplate` . If you leave it as `DEFAULT` , the agent uses a default prompt template.
 * @property promptState Specifies whether to allow the agent to carry out the step specified in the `promptType` . If you set this value to `DISABLED` , the agent skips that step. The default state for each `promptType` is as follows.
 * - `PRE_PROCESSING` – `ENABLED`
 * - `ORCHESTRATION` – `ENABLED`
 * - `KNOWLEDGE_BASE_RESPONSE_GENERATION` – `ENABLED`
 * - `POST_PROCESSING` – `DISABLED`
 * @property promptType The step in the agent sequence that this prompt configuration applies to.
 */
public data class AgentPromptConfiguration(
    public val basePromptTemplate: String? = null,
    public val inferenceConfiguration: AgentInferenceConfiguration? = null,
    public val parserMode: AgentCreationMode? = null,
    public val promptCreationMode: AgentCreationMode? = null,
    public val promptState: AgentPromptState? = null,
    public val promptType: AgentPromptType? = null,
) {
    public companion object {
        public fun toKotlin(javaType: com.pulumi.awsnative.bedrock.outputs.AgentPromptConfiguration): AgentPromptConfiguration = AgentPromptConfiguration(
            basePromptTemplate = javaType.basePromptTemplate().map({ args0 -> args0 }).orElse(null),
            inferenceConfiguration = javaType.inferenceConfiguration().map({ args0 ->
                args0.let({ args0 ->
                    com.pulumi.awsnative.bedrock.kotlin.outputs.AgentInferenceConfiguration.Companion.toKotlin(args0)
                })
            }).orElse(null),
            parserMode = javaType.parserMode().map({ args0 ->
                args0.let({ args0 ->
                    com.pulumi.awsnative.bedrock.kotlin.enums.AgentCreationMode.Companion.toKotlin(args0)
                })
            }).orElse(null),
            promptCreationMode = javaType.promptCreationMode().map({ args0 ->
                args0.let({ args0 ->
                    com.pulumi.awsnative.bedrock.kotlin.enums.AgentCreationMode.Companion.toKotlin(args0)
                })
            }).orElse(null),
            promptState = javaType.promptState().map({ args0 ->
                args0.let({ args0 ->
                    com.pulumi.awsnative.bedrock.kotlin.enums.AgentPromptState.Companion.toKotlin(args0)
                })
            }).orElse(null),
            promptType = javaType.promptType().map({ args0 ->
                args0.let({ args0 ->
                    com.pulumi.awsnative.bedrock.kotlin.enums.AgentPromptType.Companion.toKotlin(args0)
                })
            }).orElse(null),
        )
    }
}