BEDROCK-AGENT

AssociateAgentKnowledgeBase

enum_KnowledgeBaseState := [ "ENABLED", "DISABLED" ]

valid {
    input.Body.description == STRING
    input.Body.knowledgeBaseId == STRING
    input.Body.knowledgeBaseState == enum_KnowledgeBaseState[_]
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

CreateAgent

enum_CreationMode := [ "DEFAULT", "OVERRIDDEN" ]
enum_PromptState := [ "ENABLED", "DISABLED" ]
enum_PromptType := [ "PRE_PROCESSING", "ORCHESTRATION", "POST_PROCESSING", "KNOWLEDGE_BASE_RESPONSE_GENERATION" ]

valid {
    input.Body.agentName == STRING
    input.Body.agentResourceRoleArn == STRING
    input.Body.clientToken == STRING
    input.Body.customerEncryptionKeyArn == STRING
    input.Body.description == STRING
    input.Body.foundationModel == STRING
    input.Body.idleSessionTTLInSeconds == INTEGER
    input.Body.instruction == STRING
    input.Body.promptOverrideConfiguration.overrideLambda == STRING
    input.Body.promptOverrideConfiguration.promptConfigurations[_].basePromptTemplate == STRING
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.maximumLength == INTEGER
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.stopSequences[_] == STRING
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.temperature == FLOAT
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.topK == INTEGER
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.topP == FLOAT
    input.Body.promptOverrideConfiguration.promptConfigurations[_].parserMode == enum_CreationMode[_]
    input.Body.promptOverrideConfiguration.promptConfigurations[_].promptCreationMode == enum_CreationMode[_]
    input.Body.promptOverrideConfiguration.promptConfigurations[_].promptState == enum_PromptState[_]
    input.Body.promptOverrideConfiguration.promptConfigurations[_].promptType == enum_PromptType[_]
    input.Body.tags.STRING == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

CreateAgentActionGroup

enum_ActionGroupSignature := [ "AMAZON.UserInput" ]
enum_ActionGroupState := [ "ENABLED", "DISABLED" ]
enum_CustomControlMethod := [ "RETURN_CONTROL" ]
enum_Type := [ "string", "number", "integer", "boolean", "array" ]

valid {
    input.Body.actionGroupExecutor.customControl == enum_CustomControlMethod[_]
    input.Body.actionGroupExecutor.lambda == STRING
    input.Body.actionGroupName == STRING
    input.Body.actionGroupState == enum_ActionGroupState[_]
    input.Body.apiSchema.payload == STRING
    input.Body.apiSchema.s3.s3BucketName == STRING
    input.Body.apiSchema.s3.s3ObjectKey == STRING
    input.Body.clientToken == STRING
    input.Body.description == STRING
    input.Body.functionSchema.functions[_].description == STRING
    input.Body.functionSchema.functions[_].name == STRING
    input.Body.functionSchema.functions[_].parameters.STRING.description == STRING
    input.Body.functionSchema.functions[_].parameters.STRING.required == BOOLEAN
    input.Body.functionSchema.functions[_].parameters.STRING.type == enum_Type[_]
    input.Body.parentActionGroupSignature == enum_ActionGroupSignature[_]
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

CreateAgentAlias

valid {
    input.Body.agentAliasName == STRING
    input.Body.clientToken == STRING
    input.Body.description == STRING
    input.Body.routingConfiguration[_].agentVersion == STRING
    input.Body.tags.STRING == STRING
    input.ReqMap.agentId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

CreateDataSource

enum_ChunkingStrategy := [ "FIXED_SIZE", "NONE" ]
enum_DataDeletionPolicy := [ "RETAIN", "DELETE" ]
enum_DataSourceType := [ "S3" ]

valid {
    input.Body.clientToken == STRING
    input.Body.dataDeletionPolicy == enum_DataDeletionPolicy[_]
    input.Body.dataSourceConfiguration.s3Configuration.bucketArn == STRING
    input.Body.dataSourceConfiguration.s3Configuration.bucketOwnerAccountId == STRING
    input.Body.dataSourceConfiguration.s3Configuration.inclusionPrefixes[_] == STRING
    input.Body.dataSourceConfiguration.type == enum_DataSourceType[_]
    input.Body.description == STRING
    input.Body.name == STRING
    input.Body.serverSideEncryptionConfiguration.kmsKeyArn == STRING
    input.Body.vectorIngestionConfiguration.chunkingConfiguration.chunkingStrategy == enum_ChunkingStrategy[_]
    input.Body.vectorIngestionConfiguration.chunkingConfiguration.fixedSizeChunkingConfiguration.maxTokens == INTEGER
    input.Body.vectorIngestionConfiguration.chunkingConfiguration.fixedSizeChunkingConfiguration.overlapPercentage == INTEGER
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

CreateKnowledgeBase

enum_KnowledgeBaseStorageType := [ "OPENSEARCH_SERVERLESS", "PINECONE", "REDIS_ENTERPRISE_CLOUD", "RDS" ]
enum_KnowledgeBaseType := [ "VECTOR" ]

valid {
    input.Body.clientToken == STRING
    input.Body.description == STRING
    input.Body.knowledgeBaseConfiguration.type == enum_KnowledgeBaseType[_]
    input.Body.knowledgeBaseConfiguration.vectorKnowledgeBaseConfiguration.embeddingModelArn == STRING
    input.Body.name == STRING
    input.Body.roleArn == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.collectionArn == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.fieldMapping.metadataField == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.fieldMapping.textField == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.fieldMapping.vectorField == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.vectorIndexName == STRING
    input.Body.storageConfiguration.pineconeConfiguration.connectionString == STRING
    input.Body.storageConfiguration.pineconeConfiguration.credentialsSecretArn == STRING
    input.Body.storageConfiguration.pineconeConfiguration.fieldMapping.metadataField == STRING
    input.Body.storageConfiguration.pineconeConfiguration.fieldMapping.textField == STRING
    input.Body.storageConfiguration.pineconeConfiguration.namespace == STRING
    input.Body.storageConfiguration.rdsConfiguration.credentialsSecretArn == STRING
    input.Body.storageConfiguration.rdsConfiguration.databaseName == STRING
    input.Body.storageConfiguration.rdsConfiguration.fieldMapping.metadataField == STRING
    input.Body.storageConfiguration.rdsConfiguration.fieldMapping.primaryKeyField == STRING
    input.Body.storageConfiguration.rdsConfiguration.fieldMapping.textField == STRING
    input.Body.storageConfiguration.rdsConfiguration.fieldMapping.vectorField == STRING
    input.Body.storageConfiguration.rdsConfiguration.resourceArn == STRING
    input.Body.storageConfiguration.rdsConfiguration.tableName == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.credentialsSecretArn == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.endpoint == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.fieldMapping.metadataField == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.fieldMapping.textField == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.fieldMapping.vectorField == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.vectorIndexName == STRING
    input.Body.storageConfiguration.type == enum_KnowledgeBaseStorageType[_]
    input.Body.tags.STRING == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

DeleteAgent

valid {
    input.ReqMap.agentId == STRING
    input.Qs.skipResourceInUseCheck == BOOLEAN
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

DeleteAgentActionGroup

valid {
    input.ReqMap.actionGroupId == STRING
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.Qs.skipResourceInUseCheck == BOOLEAN
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

DeleteAgentAlias

valid {
    input.ReqMap.agentAliasId == STRING
    input.ReqMap.agentId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

DeleteAgentVersion

valid {
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.Qs.skipResourceInUseCheck == BOOLEAN
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

DeleteDataSource

valid {
    input.ReqMap.dataSourceId == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

DeleteKnowledgeBase

valid {
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

DisassociateAgentKnowledgeBase

valid {
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

GetAgent

valid {
    input.ReqMap.agentId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

GetAgentActionGroup

valid {
    input.ReqMap.actionGroupId == STRING
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

GetAgentAlias

valid {
    input.ReqMap.agentAliasId == STRING
    input.ReqMap.agentId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

GetAgentKnowledgeBase

valid {
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

GetAgentVersion

valid {
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

GetDataSource

valid {
    input.ReqMap.dataSourceId == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

GetIngestionJob

valid {
    input.ReqMap.dataSourceId == STRING
    input.ReqMap.ingestionJobId == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

GetKnowledgeBase

valid {
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

ListAgentActionGroups

valid {
    input.Body.maxResults == INTEGER
    input.Body.nextToken == STRING
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

ListAgentAliases

valid {
    input.Body.maxResults == INTEGER
    input.Body.nextToken == STRING
    input.ReqMap.agentId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

ListAgentKnowledgeBases

valid {
    input.Body.maxResults == INTEGER
    input.Body.nextToken == STRING
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

ListAgentVersions

valid {
    input.Body.maxResults == INTEGER
    input.Body.nextToken == STRING
    input.ReqMap.agentId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

ListAgents

valid {
    input.Body.maxResults == INTEGER
    input.Body.nextToken == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

ListDataSources

valid {
    input.Body.maxResults == INTEGER
    input.Body.nextToken == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

ListIngestionJobs

enum_IngestionJobFilterAttribute := [ "STATUS" ]
enum_IngestionJobFilterOperator := [ "EQ" ]
enum_IngestionJobSortByAttribute := [ "STATUS", "STARTED_AT" ]
enum_SortOrder := [ "ASCENDING", "DESCENDING" ]

valid {
    input.Body.filters[_].attribute == enum_IngestionJobFilterAttribute[_]
    input.Body.filters[_].operator == enum_IngestionJobFilterOperator[_]
    input.Body.filters[_].values[_] == STRING
    input.Body.maxResults == INTEGER
    input.Body.nextToken == STRING
    input.Body.sortBy.attribute == enum_IngestionJobSortByAttribute[_]
    input.Body.sortBy.order == enum_SortOrder[_]
    input.ReqMap.dataSourceId == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

ListKnowledgeBases

valid {
    input.Body.maxResults == INTEGER
    input.Body.nextToken == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

ListTagsForResource

valid {
    input.ReqMap.resourceArn == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

PrepareAgent

valid {
    input.ReqMap.agentId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

StartIngestionJob

valid {
    input.Body.clientToken == STRING
    input.Body.description == STRING
    input.ReqMap.dataSourceId == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

TagResource

valid {
    input.Body.tags.STRING == STRING
    input.ReqMap.resourceArn == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

UntagResource

valid {
    input.ReqMap.resourceArn == STRING
    input.Qs.tagKeys[_] == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

UpdateAgent

enum_CreationMode := [ "DEFAULT", "OVERRIDDEN" ]
enum_PromptState := [ "ENABLED", "DISABLED" ]
enum_PromptType := [ "PRE_PROCESSING", "ORCHESTRATION", "POST_PROCESSING", "KNOWLEDGE_BASE_RESPONSE_GENERATION" ]

valid {
    input.Body.agentName == STRING
    input.Body.agentResourceRoleArn == STRING
    input.Body.customerEncryptionKeyArn == STRING
    input.Body.description == STRING
    input.Body.foundationModel == STRING
    input.Body.idleSessionTTLInSeconds == INTEGER
    input.Body.instruction == STRING
    input.Body.promptOverrideConfiguration.overrideLambda == STRING
    input.Body.promptOverrideConfiguration.promptConfigurations[_].basePromptTemplate == STRING
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.maximumLength == INTEGER
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.stopSequences[_] == STRING
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.temperature == FLOAT
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.topK == INTEGER
    input.Body.promptOverrideConfiguration.promptConfigurations[_].inferenceConfiguration.topP == FLOAT
    input.Body.promptOverrideConfiguration.promptConfigurations[_].parserMode == enum_CreationMode[_]
    input.Body.promptOverrideConfiguration.promptConfigurations[_].promptCreationMode == enum_CreationMode[_]
    input.Body.promptOverrideConfiguration.promptConfigurations[_].promptState == enum_PromptState[_]
    input.Body.promptOverrideConfiguration.promptConfigurations[_].promptType == enum_PromptType[_]
    input.ReqMap.agentId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

UpdateAgentActionGroup

enum_ActionGroupSignature := [ "AMAZON.UserInput" ]
enum_ActionGroupState := [ "ENABLED", "DISABLED" ]
enum_CustomControlMethod := [ "RETURN_CONTROL" ]
enum_Type := [ "string", "number", "integer", "boolean", "array" ]

valid {
    input.Body.actionGroupExecutor.customControl == enum_CustomControlMethod[_]
    input.Body.actionGroupExecutor.lambda == STRING
    input.Body.actionGroupName == STRING
    input.Body.actionGroupState == enum_ActionGroupState[_]
    input.Body.apiSchema.payload == STRING
    input.Body.apiSchema.s3.s3BucketName == STRING
    input.Body.apiSchema.s3.s3ObjectKey == STRING
    input.Body.description == STRING
    input.Body.functionSchema.functions[_].description == STRING
    input.Body.functionSchema.functions[_].name == STRING
    input.Body.functionSchema.functions[_].parameters.STRING.description == STRING
    input.Body.functionSchema.functions[_].parameters.STRING.required == BOOLEAN
    input.Body.functionSchema.functions[_].parameters.STRING.type == enum_Type[_]
    input.Body.parentActionGroupSignature == enum_ActionGroupSignature[_]
    input.ReqMap.actionGroupId == STRING
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

UpdateAgentAlias

valid {
    input.Body.agentAliasName == STRING
    input.Body.description == STRING
    input.Body.routingConfiguration[_].agentVersion == STRING
    input.ReqMap.agentAliasId == STRING
    input.ReqMap.agentId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

UpdateAgentKnowledgeBase

enum_KnowledgeBaseState := [ "ENABLED", "DISABLED" ]

valid {
    input.Body.description == STRING
    input.Body.knowledgeBaseState == enum_KnowledgeBaseState[_]
    input.ReqMap.agentId == STRING
    input.ReqMap.agentVersion == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

UpdateDataSource

enum_ChunkingStrategy := [ "FIXED_SIZE", "NONE" ]
enum_DataDeletionPolicy := [ "RETAIN", "DELETE" ]
enum_DataSourceType := [ "S3" ]

valid {
    input.Body.dataDeletionPolicy == enum_DataDeletionPolicy[_]
    input.Body.dataSourceConfiguration.s3Configuration.bucketArn == STRING
    input.Body.dataSourceConfiguration.s3Configuration.bucketOwnerAccountId == STRING
    input.Body.dataSourceConfiguration.s3Configuration.inclusionPrefixes[_] == STRING
    input.Body.dataSourceConfiguration.type == enum_DataSourceType[_]
    input.Body.description == STRING
    input.Body.name == STRING
    input.Body.serverSideEncryptionConfiguration.kmsKeyArn == STRING
    input.Body.vectorIngestionConfiguration.chunkingConfiguration.chunkingStrategy == enum_ChunkingStrategy[_]
    input.Body.vectorIngestionConfiguration.chunkingConfiguration.fixedSizeChunkingConfiguration.maxTokens == INTEGER
    input.Body.vectorIngestionConfiguration.chunkingConfiguration.fixedSizeChunkingConfiguration.overlapPercentage == INTEGER
    input.ReqMap.dataSourceId == STRING
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}

UpdateKnowledgeBase

enum_KnowledgeBaseStorageType := [ "OPENSEARCH_SERVERLESS", "PINECONE", "REDIS_ENTERPRISE_CLOUD", "RDS" ]
enum_KnowledgeBaseType := [ "VECTOR" ]

valid {
    input.Body.description == STRING
    input.Body.knowledgeBaseConfiguration.type == enum_KnowledgeBaseType[_]
    input.Body.knowledgeBaseConfiguration.vectorKnowledgeBaseConfiguration.embeddingModelArn == STRING
    input.Body.name == STRING
    input.Body.roleArn == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.collectionArn == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.fieldMapping.metadataField == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.fieldMapping.textField == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.fieldMapping.vectorField == STRING
    input.Body.storageConfiguration.opensearchServerlessConfiguration.vectorIndexName == STRING
    input.Body.storageConfiguration.pineconeConfiguration.connectionString == STRING
    input.Body.storageConfiguration.pineconeConfiguration.credentialsSecretArn == STRING
    input.Body.storageConfiguration.pineconeConfiguration.fieldMapping.metadataField == STRING
    input.Body.storageConfiguration.pineconeConfiguration.fieldMapping.textField == STRING
    input.Body.storageConfiguration.pineconeConfiguration.namespace == STRING
    input.Body.storageConfiguration.rdsConfiguration.credentialsSecretArn == STRING
    input.Body.storageConfiguration.rdsConfiguration.databaseName == STRING
    input.Body.storageConfiguration.rdsConfiguration.fieldMapping.metadataField == STRING
    input.Body.storageConfiguration.rdsConfiguration.fieldMapping.primaryKeyField == STRING
    input.Body.storageConfiguration.rdsConfiguration.fieldMapping.textField == STRING
    input.Body.storageConfiguration.rdsConfiguration.fieldMapping.vectorField == STRING
    input.Body.storageConfiguration.rdsConfiguration.resourceArn == STRING
    input.Body.storageConfiguration.rdsConfiguration.tableName == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.credentialsSecretArn == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.endpoint == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.fieldMapping.metadataField == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.fieldMapping.textField == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.fieldMapping.vectorField == STRING
    input.Body.storageConfiguration.redisEnterpriseCloudConfiguration.vectorIndexName == STRING
    input.Body.storageConfiguration.type == enum_KnowledgeBaseStorageType[_]
    input.ReqMap.knowledgeBaseId == STRING
    input.ProviderMetadata.Account == STRING
    input.ProviderMetadata.AccessKeyId == STRING
    input.ProviderMetadata.Region == STRING
}