@aws-sdk/client-bedrock 3.703.0 → 3.706.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, EvaluationInferenceConfig, InternalServerException, KnowledgeBaseConfig, RAGConfig, ResourceNotFoundException, RetrievalFilter, ServiceQuotaExceededException, ThrottlingException, TooManyTagsException, ValidationException, } from "../models/models_0";
+import { AccessDeniedException, ConflictException, EvaluationInferenceConfig, InternalServerException, KnowledgeBaseConfig, RAGConfig, ResourceNotFoundException, RetrievalFilter, ServiceQuotaExceededException, ServiceUnavailableException, ThrottlingException, TooManyTagsException, ValidationException, } from "../models/models_0";
 export const se_BatchDeleteEvaluationJobCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -95,6 +95,24 @@ export const se_CreateInferenceProfileCommand = async (input, context) => {
     b.m("POST").h(headers).b(body);
     return b.build();
 };
+export const se_CreateMarketplaceModelEndpointCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/marketplace-model/endpoints");
+    let body;
+    body = JSON.stringify(take(input, {
+        acceptEula: [],
+        clientRequestToken: [true, (_) => _ ?? generateIdempotencyToken()],
+        endpointConfig: (_) => _json(_),
+        endpointName: [],
+        modelSourceIdentifier: [],
+        tags: (_) => _json(_),
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_CreateModelCopyJobCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -125,6 +143,7 @@ export const se_CreateModelCustomizationJobCommand = async (input, context) => {
         customModelKmsKeyId: [],
         customModelName: [],
         customModelTags: (_) => _json(_),
+        customizationConfig: (_) => _json(_),
         customizationType: [],
         hyperParameters: (_) => _json(_),
         jobName: [],
@@ -237,6 +256,15 @@ export const se_DeleteInferenceProfileCommand = async (input, context) => {
     b.m("DELETE").h(headers).b(body);
     return b.build();
 };
+export const se_DeleteMarketplaceModelEndpointCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {};
+    b.bp("/marketplace-model/endpoints/{endpointArn}");
+    b.p("endpointArn", () => input.endpointArn, "{endpointArn}", false);
+    let body;
+    b.m("DELETE").h(headers).b(body);
+    return b.build();
+};
 export const se_DeleteModelInvocationLoggingConfigurationCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -254,6 +282,15 @@ export const se_DeleteProvisionedModelThroughputCommand = async (input, context)
     b.m("DELETE").h(headers).b(body);
     return b.build();
 };
+export const se_DeregisterMarketplaceModelEndpointCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {};
+    b.bp("/marketplace-model/endpoints/{endpointArn}/registration");
+    b.p("endpointArn", () => input.endpointArn, "{endpointArn}", false);
+    let body;
+    b.m("DELETE").h(headers).b(body);
+    return b.build();
+};
 export const se_GetCustomModelCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -311,6 +348,15 @@ export const se_GetInferenceProfileCommand = async (input, context) => {
     b.m("GET").h(headers).b(body);
     return b.build();
 };
+export const se_GetMarketplaceModelEndpointCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {};
+    b.bp("/marketplace-model/endpoints/{endpointArn}");
+    b.p("endpointArn", () => input.endpointArn, "{endpointArn}", false);
+    let body;
+    b.m("GET").h(headers).b(body);
+    return b.build();
+};
 export const se_GetModelCopyJobCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -355,6 +401,15 @@ export const se_GetModelInvocationLoggingConfigurationCommand = async (input, co
     b.m("GET").h(headers).b(body);
     return b.build();
 };
+export const se_GetPromptRouterCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {};
+    b.bp("/prompt-routers/{promptRouterArn}");
+    b.p("promptRouterArn", () => input.promptRouterArn, "{promptRouterArn}", false);
+    let body;
+    b.m("GET").h(headers).b(body);
+    return b.build();
+};
 export const se_GetProvisionedModelThroughputCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -460,6 +515,19 @@ export const se_ListInferenceProfilesCommand = async (input, context) => {
     b.m("GET").h(headers).q(query).b(body);
     return b.build();
 };
+export const se_ListMarketplaceModelEndpointsCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {};
+    b.bp("/marketplace-model/endpoints");
+    const query = map({
+        [_mR]: [() => input.maxResults !== void 0, () => input[_mR].toString()],
+        [_nT]: [, input[_nT]],
+        [_mSI]: [, input[_mSE]],
+    });
+    let body;
+    b.m("GET").h(headers).q(query).b(body);
+    return b.build();
+};
 export const se_ListModelCopyJobsCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -534,6 +602,18 @@ export const se_ListModelInvocationJobsCommand = async (input, context) => {
     b.m("GET").h(headers).q(query).b(body);
     return b.build();
 };
+export const se_ListPromptRoutersCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {};
+    b.bp("/prompt-routers");
+    const query = map({
+        [_mR]: [() => input.maxResults !== void 0, () => input[_mR].toString()],
+        [_nT]: [, input[_nT]],
+    });
+    let body;
+    b.m("GET").h(headers).q(query).b(body);
+    return b.build();
+};
 export const se_ListProvisionedModelThroughputsCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -579,6 +659,20 @@ export const se_PutModelInvocationLoggingConfigurationCommand = async (input, co
     b.m("PUT").h(headers).b(body);
     return b.build();
 };
+export const se_RegisterMarketplaceModelEndpointCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/marketplace-model/endpoints/{endpointIdentifier}/registration");
+    b.p("endpointIdentifier", () => input.endpointIdentifier, "{endpointIdentifier}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        modelSourceIdentifier: [],
+    }));
+    b.m("POST").h(headers).b(body);
+    return b.build();
+};
 export const se_StopEvaluationJobCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {};
@@ -657,6 +751,21 @@ export const se_UpdateGuardrailCommand = async (input, context) => {
     b.m("PUT").h(headers).b(body);
     return b.build();
 };
+export const se_UpdateMarketplaceModelEndpointCommand = async (input, context) => {
+    const b = rb(input, context);
+    const headers = {
+        "content-type": "application/json",
+    };
+    b.bp("/marketplace-model/endpoints/{endpointArn}");
+    b.p("endpointArn", () => input.endpointArn, "{endpointArn}", false);
+    let body;
+    body = JSON.stringify(take(input, {
+        clientRequestToken: [true, (_) => _ ?? generateIdempotencyToken()],
+        endpointConfig: (_) => _json(_),
+    }));
+    b.m("PATCH").h(headers).b(body);
+    return b.build();
+};
 export const se_UpdateProvisionedModelThroughputCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -748,6 +857,20 @@ export const de_CreateInferenceProfileCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_CreateMarketplaceModelEndpointCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        marketplaceModelEndpoint: (_) => de_MarketplaceModelEndpoint(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_CreateModelCopyJobCommand = async (output, context) => {
     if (output.statusCode !== 201 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -858,6 +981,16 @@ export const de_DeleteInferenceProfileCommand = async (output, context) => {
     await collectBody(output.body, context);
     return contents;
 };
+export const de_DeleteMarketplaceModelEndpointCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    await collectBody(output.body, context);
+    return contents;
+};
 export const de_DeleteModelInvocationLoggingConfigurationCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -878,6 +1011,16 @@ export const de_DeleteProvisionedModelThroughputCommand = async (output, context
     await collectBody(output.body, context);
     return contents;
 };
+export const de_DeregisterMarketplaceModelEndpointCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    await collectBody(output.body, context);
+    return contents;
+};
 export const de_GetCustomModelCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -889,6 +1032,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,
@@ -1021,6 +1165,20 @@ export const de_GetInferenceProfileCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_GetMarketplaceModelEndpointCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        marketplaceModelEndpoint: (_) => de_MarketplaceModelEndpoint(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_GetModelCopyJobCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1057,6 +1215,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,
@@ -1147,6 +1306,29 @@ export const de_GetModelInvocationLoggingConfigurationCommand = async (output, c
     Object.assign(contents, doc);
     return contents;
 };
+export const de_GetPromptRouterCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        description: __expectString,
+        fallbackModel: _json,
+        models: _json,
+        promptRouterArn: __expectString,
+        promptRouterName: __expectString,
+        routingCriteria: (_) => de_RoutingCriteria(_, context),
+        status: __expectString,
+        type: __expectString,
+        updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_GetProvisionedModelThroughputCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1262,6 +1444,21 @@ export const de_ListInferenceProfilesCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_ListMarketplaceModelEndpointsCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        marketplaceModelEndpoints: (_) => de_MarketplaceModelEndpointSummaries(_, context),
+        nextToken: __expectString,
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_ListModelCopyJobsCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1322,6 +1519,21 @@ export const de_ListModelInvocationJobsCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_ListPromptRoutersCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        nextToken: __expectString,
+        promptRouterSummaries: (_) => de_PromptRouterSummaries(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_ListProvisionedModelThroughputsCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1361,6 +1573,20 @@ export const de_PutModelInvocationLoggingConfigurationCommand = async (output, c
     await collectBody(output.body, context);
     return contents;
 };
+export const de_RegisterMarketplaceModelEndpointCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        marketplaceModelEndpoint: (_) => de_MarketplaceModelEndpoint(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_StopEvaluationJobCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1428,6 +1654,20 @@ export const de_UpdateGuardrailCommand = async (output, context) => {
     Object.assign(contents, doc);
     return contents;
 };
+export const de_UpdateMarketplaceModelEndpointCommand = async (output, context) => {
+    if (output.statusCode !== 200 && output.statusCode >= 300) {
+        return de_CommandError(output, context);
+    }
+    const contents = map({
+        $metadata: deserializeMetadata(output),
+    });
+    const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
+    const doc = take(data, {
+        marketplaceModelEndpoint: (_) => de_MarketplaceModelEndpoint(_, context),
+    });
+    Object.assign(contents, doc);
+    return contents;
+};
 export const de_UpdateProvisionedModelThroughputCommand = async (output, context) => {
     if (output.statusCode !== 200 && output.statusCode >= 300) {
         return de_CommandError(output, context);
@@ -1469,6 +1709,9 @@ const de_CommandError = async (output, context) => {
         case "TooManyTagsException":
         case "com.amazonaws.bedrock#TooManyTagsException":
             throw await de_TooManyTagsExceptionRes(parsedOutput, context);
+        case "ServiceUnavailableException":
+        case "com.amazonaws.bedrock#ServiceUnavailableException":
+            throw await de_ServiceUnavailableExceptionRes(parsedOutput, context);
         default:
             const parsedBody = parsedOutput.body;
             return throwDefaultError({
@@ -1544,6 +1787,19 @@ const de_ServiceQuotaExceededExceptionRes = async (parsedOutput, context) => {
     });
     return __decorateServiceException(exception, parsedOutput.body);
 };
+const de_ServiceUnavailableExceptionRes = async (parsedOutput, context) => {
+    const contents = map({});
+    const data = parsedOutput.body;
+    const doc = take(data, {
+        message: __expectString,
+    });
+    Object.assign(contents, doc);
+    const exception = new ServiceUnavailableException({
+        $metadata: deserializeMetadata(parsedOutput),
+        ...contents,
+    });
+    return __decorateServiceException(exception, parsedOutput.body);
+};
 const de_ThrottlingExceptionRes = async (parsedOutput, context) => {
     const contents = map({});
     const data = parsedOutput.body;
@@ -2004,6 +2260,37 @@ const de_KnowledgeBaseVectorSearchConfiguration = (output, context) => {
         overrideSearchType: __expectString,
     });
 };
+const de_MarketplaceModelEndpoint = (output, context) => {
+    return take(output, {
+        createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        endpointArn: __expectString,
+        endpointConfig: (_) => _json(__expectUnion(_)),
+        endpointStatus: __expectString,
+        endpointStatusMessage: __expectString,
+        modelSourceIdentifier: __expectString,
+        status: __expectString,
+        statusMessage: __expectString,
+        updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+    });
+};
+const de_MarketplaceModelEndpointSummaries = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_MarketplaceModelEndpointSummary(entry, context);
+    });
+    return retVal;
+};
+const de_MarketplaceModelEndpointSummary = (output, context) => {
+    return take(output, {
+        createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        endpointArn: __expectString,
+        modelSourceIdentifier: __expectString,
+        status: __expectString,
+        statusMessage: __expectString,
+        updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+    });
+};
 const de_ModelCopyJobSummaries = (output, context) => {
     const retVal = (output || [])
         .filter((e) => e != null)
@@ -2096,6 +2383,28 @@ const de_ModelInvocationJobSummary = (output, context) => {
         vpcConfig: _json,
     });
 };
+const de_PromptRouterSummaries = (output, context) => {
+    const retVal = (output || [])
+        .filter((e) => e != null)
+        .map((entry) => {
+        return de_PromptRouterSummary(entry, context);
+    });
+    return retVal;
+};
+const de_PromptRouterSummary = (output, context) => {
+    return take(output, {
+        createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+        description: __expectString,
+        fallbackModel: _json,
+        models: _json,
+        promptRouterArn: __expectString,
+        promptRouterName: __expectString,
+        routingCriteria: (_) => de_RoutingCriteria(_, context),
+        status: __expectString,
+        type: __expectString,
+        updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+    });
+};
 const de_ProvisionedModelSummaries = (output, context) => {
     const retVal = (output || [])
         .filter((e) => e != null)
@@ -2225,6 +2534,11 @@ const de_RetrieveConfig = (output, context) => {
         knowledgeBaseRetrievalConfiguration: (_) => de_KnowledgeBaseRetrievalConfiguration(_, context),
     });
 };
+const de_RoutingCriteria = (output, context) => {
+    return take(output, {
+        responseQualityDifference: __limitedParseDouble,
+    });
+};
 const de_TextInferenceConfig = (output, context) => {
     return take(output, {
         maxTokens: __expectInt32,
@@ -2272,6 +2586,8 @@ const _gV = "guardrailVersion";
 const _iO = "isOwned";
 const _mAE = "modelArnEquals";
 const _mR = "maxResults";
+const _mSE = "modelSourceEquals";
+const _mSI = "modelSourceIdentifier";
 const _nC = "nameContains";
 const _nT = "nextToken";
 const _oMNC = "outputModelNameContains";