BEDROCK-AGENT-RUNTIME

InvokeAgent

valid {
    input.Body.sessionState.sessionAttributes.STRING == STRING
    input.Body.sessionState.promptSessionAttributes.STRING == STRING
    input.Body.endSession == BOOLEAN
    input.Body.enableTrace == BOOLEAN
    input.Body.inputText == STRING
    input.ReqMap.agentId == STRING
    input.ReqMap.agentAliasId == STRING
    input.ReqMap.sessionId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

Retrieve

enum_SearchType := [ "HYBRID", "SEMANTIC" ]

valid {
    input.Body.retrievalQuery.text == STRING
    input.Body.retrievalConfiguration.vectorSearchConfiguration.numberOfResults == INTEGER
    input.Body.retrievalConfiguration.vectorSearchConfiguration.overrideSearchType == enum_SearchType[_]
    input.Body.nextToken == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

RetrieveAndGenerate

enum_RetrieveAndGenerateType := [ "KNOWLEDGE_BASE" ]
enum_SearchType := [ "HYBRID", "SEMANTIC" ]

valid {
    input.Body.sessionId == STRING
    input.Body.input.text == STRING
    input.Body.retrieveAndGenerateConfiguration.type == enum_RetrieveAndGenerateType[_]
    input.Body.retrieveAndGenerateConfiguration.knowledgeBaseConfiguration.knowledgeBaseId == STRING
    input.Body.retrieveAndGenerateConfiguration.knowledgeBaseConfiguration.modelArn == STRING
    input.Body.retrieveAndGenerateConfiguration.knowledgeBaseConfiguration.retrievalConfiguration.vectorSearchConfiguration.numberOfResults == INTEGER
    input.Body.retrieveAndGenerateConfiguration.knowledgeBaseConfiguration.retrievalConfiguration.vectorSearchConfiguration.overrideSearchType == enum_SearchType[_]
    input.Body.sessionConfiguration.kmsKeyArn == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}