@aws-sdk/client-bedrock 3.699.0 → 3.705.0

package/dist-es/protocols/Aws_restJson1.js

@@ -3,7 +3,7 @@ import { requestBuilder as rb } from "@smithy/core";
 import { _json, collectBody, decorateServiceException as __decorateServiceException, expectBoolean as __expectBoolean, expectInt32 as __expectInt32, expectNonNull as __expectNonNull, expectObject as __expectObject, expectString as __expectString, limitedParseDouble as __limitedParseDouble, limitedParseFloat32 as __limitedParseFloat32, map, parseRfc3339DateTimeWithOffset as __parseRfc3339DateTimeWithOffset, serializeDateTime as __serializeDateTime, serializeFloat as __serializeFloat, take, withBaseException, } from "@smithy/smithy-client";
 import { v4 as generateIdempotencyToken } from "uuid";
 import { BedrockServiceException as __BaseException } from "../models/BedrockServiceException";
-import { AccessDeniedException, ConflictException, InternalServerException, ResourceNotFoundException, ServiceQuotaExceededException, ThrottlingException, TooManyTagsException, ValidationException, } from "../models/models_0";
+import { AccessDeniedException, ConflictException, EvaluationInferenceConfig, InternalServerException, KnowledgeBaseConfig, RAGConfig, ResourceNotFoundException, RetrievalFilter, ServiceQuotaExceededException, ThrottlingException, TooManyTagsException, ValidationException, } from "../models/models_0";
 export const se_BatchDeleteEvaluationJobCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -25,10 +25,11 @@ export const se_CreateEvaluationJobCommand = async (input, context) => {
     b.bp("/evaluation-jobs");
     let body;
     body = JSON.stringify(take(input, {
+        applicationType: [],
         clientRequestToken: [true, (_) => _ ?? generateIdempotencyToken()],
         customerEncryptionKeyId: [],
         evaluationConfig: (_) => _json(_),
-        inferenceConfig: (_) => _json(_),
+        inferenceConfig: (_) => se_EvaluationInferenceConfig(_, context),
         jobDescription: [],
         jobName: [],
         jobTags: (_) => _json(_),
@@ -124,6 +125,7 @@ export const se_CreateModelCustomizationJobCommand = async (input, context) => {
         customModelKmsKeyId: [],
         customModelName: [],
         customModelTags: (_) => _json(_),
+        customizationConfig: (_) => _json(_),
         customizationType: [],
         hyperParameters: (_) => _json(_),
         jobName: [],
@@ -391,6 +393,7 @@ export const se_ListEvaluationJobsCommand = async (input, context) => {
         [_cTA]: [() => input.creationTimeAfter !== void 0, () => __serializeDateTime(input[_cTA]).toString()],
         [_cTB]: [() => input.creationTimeBefore !== void 0, () => __serializeDateTime(input[_cTB]).toString()],
         [_sE]: [, input[_sE]],
+        [_aTE]: [, input[_aTE]],
         [_nC]: [, input[_nC]],
         [_mR]: [() => input.maxResults !== void 0, () => input[_mR].toString()],
         [_nT]: [, input[_nT]],
@@ -887,6 +890,7 @@ export const de_GetCustomModelCommand = async (output, context) => {
     const doc = take(data, {
         baseModelArn: __expectString,
         creationTime: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        customizationConfig: (_) => _json(__expectUnion(_)),
         customizationType: __expectString,
         hyperParameters: _json,
         jobArn: __expectString,
@@ -912,11 +916,12 @@ export const de_GetEvaluationJobCommand = async (output, context) => {
     });
     const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
     const doc = take(data, {
+        applicationType: __expectString,
         creationTime: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
         customerEncryptionKeyId: __expectString,
         evaluationConfig: (_) => _json(__expectUnion(_)),
         failureMessages: _json,
-        inferenceConfig: (_) => _json(__expectUnion(_)),
+        inferenceConfig: (_) => de_EvaluationInferenceConfig(__expectUnion(_), context),
         jobArn: __expectString,
         jobDescription: __expectString,
         jobName: __expectString,
@@ -1054,6 +1059,7 @@ export const de_GetModelCustomizationJobCommand = async (output, context) => {
         baseModelArn: __expectString,
         clientRequestToken: __expectString,
         creationTime: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        customizationConfig: (_) => _json(__expectUnion(_)),
         customizationType: __expectString,
         endTime: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
         failureMessage: __expectString,
@@ -1581,6 +1587,78 @@ const de_ValidationExceptionRes = async (parsedOutput, context) => {
     });
     return __decorateServiceException(exception, parsedOutput.body);
 };
+const se_AdditionalModelRequestFields = (input, context) => {
+    return Object.entries(input).reduce((acc, [key, value]) => {
+        if (value === null) {
+            return acc;
+        }
+        acc[key] = se_AdditionalModelRequestFieldsValue(value, context);
+        return acc;
+    }, {});
+};
+const se_AdditionalModelRequestFieldsValue = (input, context) => {
+    return input;
+};
+const se_ByteContentDoc = (input, context) => {
+    return take(input, {
+        contentType: [],
+        data: context.base64Encoder,
+        identifier: [],
+    });
+};
+const se_EvaluationInferenceConfig = (input, context) => {
+    return EvaluationInferenceConfig.visit(input, {
+        models: (value) => ({ models: _json(value) }),
+        ragConfigs: (value) => ({ ragConfigs: se_RagConfigs(value, context) }),
+        _: (name, value) => ({ name: value }),
+    });
+};
+const se_ExternalSource = (input, context) => {
+    return take(input, {
+        byteContent: (_) => se_ByteContentDoc(_, context),
+        s3Location: _json,
+        sourceType: [],
+    });
+};
+const se_ExternalSources = (input, context) => {
+    return input
+        .filter((e) => e != null)
+        .map((entry) => {
+        return se_ExternalSource(entry, context);
+    });
+};
+const se_ExternalSourcesGenerationConfiguration = (input, context) => {
+    return take(input, {
+        additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+        guardrailConfiguration: _json,
+        kbInferenceConfig: (_) => se_KbInferenceConfig(_, context),
+        promptTemplate: _json,
+    });
+};
+const se_ExternalSourcesRetrieveAndGenerateConfiguration = (input, context) => {
+    return take(input, {
+        generationConfiguration: (_) => se_ExternalSourcesGenerationConfiguration(_, context),
+        modelArn: [],
+        sources: (_) => se_ExternalSources(_, context),
+    });
+};
+const se_FilterAttribute = (input, context) => {
+    return take(input, {
+        key: [],
+        value: (_) => se_FilterValue(_, context),
+    });
+};
+const se_FilterValue = (input, context) => {
+    return input;
+};
+const se_GenerationConfiguration = (input, context) => {
+    return take(input, {
+        additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+        guardrailConfiguration: _json,
+        kbInferenceConfig: (_) => se_KbInferenceConfig(_, context),
+        promptTemplate: _json,
+    });
+};
 const se_GuardrailContextualGroundingFilterConfig = (input, context) => {
     return take(input, {
         threshold: __serializeFloat,
@@ -1599,6 +1677,119 @@ const se_GuardrailContextualGroundingPolicyConfig = (input, context) => {
         filtersConfig: (_) => se_GuardrailContextualGroundingFiltersConfig(_, context),
     });
 };
+const se_KbInferenceConfig = (input, context) => {
+    return take(input, {
+        textInferenceConfig: (_) => se_TextInferenceConfig(_, context),
+    });
+};
+const se_KnowledgeBaseConfig = (input, context) => {
+    return KnowledgeBaseConfig.visit(input, {
+        retrieveAndGenerateConfig: (value) => ({
+            retrieveAndGenerateConfig: se_RetrieveAndGenerateConfiguration(value, context),
+        }),
+        retrieveConfig: (value) => ({ retrieveConfig: se_RetrieveConfig(value, context) }),
+        _: (name, value) => ({ name: value }),
+    });
+};
+const se_KnowledgeBaseRetrievalConfiguration = (input, context) => {
+    return take(input, {
+        vectorSearchConfiguration: (_) => se_KnowledgeBaseVectorSearchConfiguration(_, context),
+    });
+};
+const se_KnowledgeBaseRetrieveAndGenerateConfiguration = (input, context) => {
+    return take(input, {
+        generationConfiguration: (_) => se_GenerationConfiguration(_, context),
+        knowledgeBaseId: [],
+        modelArn: [],
+        orchestrationConfiguration: _json,
+        retrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context),
+    });
+};
+const se_KnowledgeBaseVectorSearchConfiguration = (input, context) => {
+    return take(input, {
+        filter: (_) => se_RetrievalFilter(_, context),
+        numberOfResults: [],
+        overrideSearchType: [],
+    });
+};
+const se_RAGConfig = (input, context) => {
+    return RAGConfig.visit(input, {
+        knowledgeBaseConfig: (value) => ({ knowledgeBaseConfig: se_KnowledgeBaseConfig(value, context) }),
+        _: (name, value) => ({ name: value }),
+    });
+};
+const se_RagConfigs = (input, context) => {
+    return input
+        .filter((e) => e != null)
+        .map((entry) => {
+        return se_RAGConfig(entry, context);
+    });
+};
+const se_RetrievalFilter = (input, context) => {
+    return RetrievalFilter.visit(input, {
+        andAll: (value) => ({ andAll: se_RetrievalFilterList(value, context) }),
+        equals: (value) => ({ equals: se_FilterAttribute(value, context) }),
+        greaterThan: (value) => ({ greaterThan: se_FilterAttribute(value, context) }),
+        greaterThanOrEquals: (value) => ({ greaterThanOrEquals: se_FilterAttribute(value, context) }),
+        in: (value) => ({ in: se_FilterAttribute(value, context) }),
+        lessThan: (value) => ({ lessThan: se_FilterAttribute(value, context) }),
+        lessThanOrEquals: (value) => ({ lessThanOrEquals: se_FilterAttribute(value, context) }),
+        listContains: (value) => ({ listContains: se_FilterAttribute(value, context) }),
+        notEquals: (value) => ({ notEquals: se_FilterAttribute(value, context) }),
+        notIn: (value) => ({ notIn: se_FilterAttribute(value, context) }),
+        orAll: (value) => ({ orAll: se_RetrievalFilterList(value, context) }),
+        startsWith: (value) => ({ startsWith: se_FilterAttribute(value, context) }),
+        stringContains: (value) => ({ stringContains: se_FilterAttribute(value, context) }),
+        _: (name, value) => ({ name: value }),
+    });
+};
+const se_RetrievalFilterList = (input, context) => {
+    return input
+        .filter((e) => e != null)
+        .map((entry) => {
+        return se_RetrievalFilter(entry, context);
+    });
+};
+const se_RetrieveAndGenerateConfiguration = (input, context) => {
+    return take(input, {
+        externalSourcesConfiguration: (_) => se_ExternalSourcesRetrieveAndGenerateConfiguration(_, context),
+        knowledgeBaseConfiguration: (_) => se_KnowledgeBaseRetrieveAndGenerateConfiguration(_, context),
+        type: [],
+    });
+};
+const se_RetrieveConfig = (input, context) => {
+    return take(input, {
+        knowledgeBaseId: [],
+        knowledgeBaseRetrievalConfiguration: (_) => se_KnowledgeBaseRetrievalConfiguration(_, context),
+    });
+};
+const se_TextInferenceConfig = (input, context) => {
+    return take(input, {
+        maxTokens: [],
+        stopSequences: _json,
+        temperature: __serializeFloat,
+        topP: __serializeFloat,
+    });
+};
+const de_AdditionalModelRequestFields = (output, context) => {
+    return Object.entries(output).reduce((acc, [key, value]) => {
+        if (value === null) {
+            return acc;
+        }
+        acc[key] = de_AdditionalModelRequestFieldsValue(value, context);
+        return acc;
+    }, {});
+};
+const de_AdditionalModelRequestFieldsValue = (output, context) => {
+    return output;
+};
+const de_ByteContentDoc = (output, context) => {
+    return take(output, {
+        contentType: __expectString,
+        data: context.base64Decoder,
+        identifier: __expectString,
+    });
+};
 const de_CustomModelSummary = (output, context) => {
     return take(output, {
         baseModelArn: __expectString,
@@ -1618,6 +1809,19 @@ const de_CustomModelSummaryList = (output, context) => {
     });
     return retVal;
 };
+const de_EvaluationInferenceConfig = (output, context) => {
+    if (output.models != null) {
+        return {
+            models: _json(output.models),
+        };
+    }
+    if (output.ragConfigs != null) {
+        return {
+            ragConfigs: de_RagConfigs(output.ragConfigs, context),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
 const de_EvaluationSummaries = (output, context) => {
     const retVal = (output || [])
         .filter((e) => e != null)
@@ -1628,15 +1832,65 @@ const de_EvaluationSummaries = (output, context) => {
 };
 const de_EvaluationSummary = (output, context) => {
     return take(output, {
+        applicationType: __expectString,
         creationTime: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
         evaluationTaskTypes: _json,
+        evaluatorModelIdentifiers: _json,
         jobArn: __expectString,
         jobName: __expectString,
         jobType: __expectString,
         modelIdentifiers: _json,
+        ragIdentifiers: _json,
         status: __expectString,
     });
 };
+const de_ExternalSource = (output, context) => {
+    return take(output, {
+        byteContent: (_) => de_ByteContentDoc(_, context),
+        s3Location: _json,
+        sourceType: __expectString,
+    });
+};
+const de_ExternalSources = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_ExternalSource(entry, context);
+    });
+    return retVal;
+};
+const de_ExternalSourcesGenerationConfiguration = (output, context) => {
+    return take(output, {
+        additionalModelRequestFields: (_) => de_AdditionalModelRequestFields(_, context),
+        guardrailConfiguration: _json,
+        kbInferenceConfig: (_) => de_KbInferenceConfig(_, context),
+        promptTemplate: _json,
+    });
+};
+const de_ExternalSourcesRetrieveAndGenerateConfiguration = (output, context) => {
+    return take(output, {
+        generationConfiguration: (_) => de_ExternalSourcesGenerationConfiguration(_, context),
+        modelArn: __expectString,
+        sources: (_) => de_ExternalSources(_, context),
+    });
+};
+const de_FilterAttribute = (output, context) => {
+    return take(output, {
+        key: __expectString,
+        value: (_) => de_FilterValue(_, context),
+    });
+};
+const de_FilterValue = (output, context) => {
+    return output;
+};
+const de_GenerationConfiguration = (output, context) => {
+    return take(output, {
+        additionalModelRequestFields: (_) => de_AdditionalModelRequestFields(_, context),
+        guardrailConfiguration: _json,
+        kbInferenceConfig: (_) => de_KbInferenceConfig(_, context),
+        promptTemplate: _json,
+    });
+};
 const de_GuardrailContextualGroundingFilter = (output, context) => {
     return take(output, {
         threshold: __limitedParseDouble,
@@ -1714,6 +1968,45 @@ const de_InferenceProfileSummary = (output, context) => {
         updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
     });
 };
+const de_KbInferenceConfig = (output, context) => {
+    return take(output, {
+        textInferenceConfig: (_) => de_TextInferenceConfig(_, context),
+    });
+};
+const de_KnowledgeBaseConfig = (output, context) => {
+    if (output.retrieveAndGenerateConfig != null) {
+        return {
+            retrieveAndGenerateConfig: de_RetrieveAndGenerateConfiguration(output.retrieveAndGenerateConfig, context),
+        };
+    }
+    if (output.retrieveConfig != null) {
+        return {
+            retrieveConfig: de_RetrieveConfig(output.retrieveConfig, context),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
+const de_KnowledgeBaseRetrievalConfiguration = (output, context) => {
+    return take(output, {
+        vectorSearchConfiguration: (_) => de_KnowledgeBaseVectorSearchConfiguration(_, context),
+    });
+};
+const de_KnowledgeBaseRetrieveAndGenerateConfiguration = (output, context) => {
+    return take(output, {
+        generationConfiguration: (_) => de_GenerationConfiguration(_, context),
+        knowledgeBaseId: __expectString,
+        modelArn: __expectString,
+        orchestrationConfiguration: _json,
+        retrievalConfiguration: (_) => de_KnowledgeBaseRetrievalConfiguration(_, context),
+    });
+};
+const de_KnowledgeBaseVectorSearchConfiguration = (output, context) => {
+    return take(output, {
+        filter: (_) => de_RetrievalFilter(__expectUnion(_), context),
+        numberOfResults: __expectInt32,
+        overrideSearchType: __expectString,
+    });
+};
 const de_ModelCopyJobSummaries = (output, context) => {
     const retVal = (output || [])
         .filter((e) => e != null)
@@ -1830,6 +2123,119 @@ const de_ProvisionedModelSummary = (output, context) => {
         status: __expectString,
     });
 };
+const de_RAGConfig = (output, context) => {
+    if (output.knowledgeBaseConfig != null) {
+        return {
+            knowledgeBaseConfig: de_KnowledgeBaseConfig(__expectUnion(output.knowledgeBaseConfig), context),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
+const de_RagConfigs = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_RAGConfig(__expectUnion(entry), context);
+    });
+    return retVal;
+};
+const de_RetrievalFilter = (output, context) => {
+    if (output.andAll != null) {
+        return {
+            andAll: de_RetrievalFilterList(output.andAll, context),
+        };
+    }
+    if (output.equals != null) {
+        return {
+            equals: de_FilterAttribute(output.equals, context),
+        };
+    }
+    if (output.greaterThan != null) {
+        return {
+            greaterThan: de_FilterAttribute(output.greaterThan, context),
+        };
+    }
+    if (output.greaterThanOrEquals != null) {
+        return {
+            greaterThanOrEquals: de_FilterAttribute(output.greaterThanOrEquals, context),
+        };
+    }
+    if (output.in != null) {
+        return {
+            in: de_FilterAttribute(output.in, context),
+        };
+    }
+    if (output.lessThan != null) {
+        return {
+            lessThan: de_FilterAttribute(output.lessThan, context),
+        };
+    }
+    if (output.lessThanOrEquals != null) {
+        return {
+            lessThanOrEquals: de_FilterAttribute(output.lessThanOrEquals, context),
+        };
+    }
+    if (output.listContains != null) {
+        return {
+            listContains: de_FilterAttribute(output.listContains, context),
+        };
+    }
+    if (output.notEquals != null) {
+        return {
+            notEquals: de_FilterAttribute(output.notEquals, context),
+        };
+    }
+    if (output.notIn != null) {
+        return {
+            notIn: de_FilterAttribute(output.notIn, context),
+        };
+    }
+    if (output.orAll != null) {
+        return {
+            orAll: de_RetrievalFilterList(output.orAll, context),
+        };
+    }
+    if (output.startsWith != null) {
+        return {
+            startsWith: de_FilterAttribute(output.startsWith, context),
+        };
+    }
+    if (output.stringContains != null) {
+        return {
+            stringContains: de_FilterAttribute(output.stringContains, context),
+        };
+    }
+    return { $unknown: Object.entries(output)[0] };
+};
+const de_RetrievalFilterList = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_RetrievalFilter(__expectUnion(entry), context);
+    });
+    return retVal;
+};
+const de_RetrieveAndGenerateConfiguration = (output, context) => {
+    return take(output, {
+        externalSourcesConfiguration: (_) => de_ExternalSourcesRetrieveAndGenerateConfiguration(_, context),
+        knowledgeBaseConfiguration: (_) => de_KnowledgeBaseRetrieveAndGenerateConfiguration(_, context),
+        type: __expectString,
+    });
+};
+const de_RetrieveConfig = (output, context) => {
+    return take(output, {
+        knowledgeBaseId: __expectString,
+        knowledgeBaseRetrievalConfiguration: (_) => de_KnowledgeBaseRetrievalConfiguration(_, context),
+    });
+};
+const de_TextInferenceConfig = (output, context) => {
+    return take(output, {
+        maxTokens: __expectInt32,
+        stopSequences: _json,
+        temperature: __limitedParseFloat32,
+        topP: __limitedParseFloat32,
+    });
+};
 const de_TrainingMetrics = (output, context) => {
     return take(output, {
         trainingLoss: __limitedParseFloat32,
@@ -1855,6 +2261,7 @@ const deserializeMetadata = (output) => ({
     cfId: output.headers["x-amz-cf-id"],
 });
 const collectBodyString = (streamBody, context) => collectBody(streamBody, context).then((body) => context.utf8Encoder(body));
+const _aTE = "applicationTypeEquals";
 const _bCT = "byCustomizationType";
 const _bIT = "byInferenceType";
 const _bMAE = "baseModelArnEquals";

package/dist-types/commands/BatchDeleteEvaluationJobCommand.d.ts

@@ -27,7 +27,9 @@ declare const BatchDeleteEvaluationJobCommand_base: {
     getEndpointParameterInstructions(): import("@smithy/middleware-endpoint").EndpointParameterInstructions;
 };
 /**
- * <p>Creates a batch deletion job. A model evaluation job can only be deleted if it has following status <code>FAILED</code>, <code>COMPLETED</code>, and <code>STOPPED</code>. You can request up to 25 model evaluation jobs be deleted in a single request.</p>
+ * <p>Deletes a batch of evaluation jobs. An evaluation job can only be deleted if it has
+ *          following status <code>FAILED</code>, <code>COMPLETED</code>, and <code>STOPPED</code>.
+ *          You can request up to 25 model evaluation jobs be deleted in a single request.</p>
  * @example
  * Use a bare-bones client and the command you need to make an API call.
  * ```javascript