@aws-sdk/client-bedrock-agent 3.808.0 → 3.811.0

package/dist-es/protocols/Aws_restJson1.js

@@ -3,7 +3,8 @@ import { requestBuilder as rb } from "@smithy/core";
 import { _json, collectBody, decorateServiceException as __decorateServiceException, expectInt32 as __expectInt32, expectNonNull as __expectNonNull, expectObject as __expectObject, expectString as __expectString, limitedParseFloat32 as __limitedParseFloat32, map, parseRfc3339DateTimeWithOffset as __parseRfc3339DateTimeWithOffset, serializeFloat as __serializeFloat, take, withBaseException, } from "@smithy/smithy-client";
 import { v4 as generateIdempotencyToken } from "uuid";
 import { BedrockAgentServiceException as __BaseException } from "../models/BedrockAgentServiceException";
-import { AccessDeniedException, ConflictException, FlowNodeConfiguration, InternalServerException, PromptFlowNodeSourceConfiguration, PromptInferenceConfiguration, PromptTemplateConfiguration, ResourceNotFoundException, ServiceQuotaExceededException, ThrottlingException, Tool, ToolInputSchema, ValidationException, } from "../models/models_0";
+import { AccessDeniedException, ConflictException, InternalServerException, PromptFlowNodeSourceConfiguration, PromptInferenceConfiguration, PromptTemplateConfiguration, ResourceNotFoundException, ServiceQuotaExceededException, ThrottlingException, Tool, ToolInputSchema, ValidationException, } from "../models/models_0";
+import { FlowNodeConfiguration, } from "../models/models_1";
 export const se_AssociateAgentCollaboratorCommand = async (input, context) => {
     const b = rb(input, context);
     const headers = {
@@ -157,6 +158,7 @@ export const se_CreateFlowAliasCommand = async (input, context) => {
     let body;
     body = JSON.stringify(take(input, {
         clientToken: [true, (_) => _ ?? generateIdempotencyToken()],
+        concurrencyConfiguration: (_) => _json(_),
         description: [],
         name: [],
         routingConfiguration: (_) => _json(_),
@@ -987,6 +989,7 @@ export const se_UpdateFlowAliasCommand = async (input, context) => {
     b.p("aliasIdentifier", () => input.aliasIdentifier, "{aliasIdentifier}", false);
     let body;
     body = JSON.stringify(take(input, {
+        concurrencyConfiguration: (_) => _json(_),
         description: [],
         name: [],
         routingConfiguration: (_) => _json(_),
@@ -1161,6 +1164,7 @@ export const de_CreateFlowAliasCommand = async (output, context) => {
     const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
     const doc = take(data, {
         arn: __expectString,
+        concurrencyConfiguration: _json,
         createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
         description: __expectString,
         flowId: __expectString,
@@ -1569,6 +1573,7 @@ export const de_GetFlowAliasCommand = async (output, context) => {
     const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
     const doc = take(data, {
         arn: __expectString,
+        concurrencyConfiguration: _json,
         createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
         description: __expectString,
         flowId: __expectString,
@@ -2104,6 +2109,7 @@ export const de_UpdateFlowAliasCommand = async (output, context) => {
     const data = __expectNonNull(__expectObject(await parseBody(output.body, context)), "body");
     const doc = take(data, {
         arn: __expectString,
+        concurrencyConfiguration: _json,
         createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
         description: __expectString,
         flowId: __expectString,
@@ -2296,6 +2302,18 @@ 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, {
         data: context.base64Encoder,
@@ -2355,9 +2373,12 @@ const se_FlowNodeConfiguration = (input, context) => {
         inlineCode: (value) => ({ inlineCode: _json(value) }),
         input: (value) => ({ input: _json(value) }),
         iterator: (value) => ({ iterator: _json(value) }),
-        knowledgeBase: (value) => ({ knowledgeBase: _json(value) }),
+        knowledgeBase: (value) => ({ knowledgeBase: se_KnowledgeBaseFlowNodeConfiguration(value, context) }),
         lambdaFunction: (value) => ({ lambdaFunction: _json(value) }),
         lex: (value) => ({ lex: _json(value) }),
+        loop: (value) => ({ loop: se_LoopFlowNodeConfiguration(value, context) }),
+        loopController: (value) => ({ loopController: _json(value) }),
+        loopInput: (value) => ({ loopInput: _json(value) }),
         output: (value) => ({ output: _json(value) }),
         prompt: (value) => ({ prompt: se_PromptFlowNodeConfiguration(value, context) }),
         retrieval: (value) => ({ retrieval: _json(value) }),
@@ -2401,6 +2422,31 @@ const se_KnowledgeBaseDocuments = (input, context) => {
         return se_KnowledgeBaseDocument(entry, context);
     });
 };
+const se_KnowledgeBaseFlowNodeConfiguration = (input, context) => {
+    return take(input, {
+        guardrailConfiguration: _json,
+        inferenceConfiguration: (_) => se_PromptInferenceConfiguration(_, context),
+        knowledgeBaseId: [],
+        modelId: [],
+        numberOfResults: [],
+        orchestrationConfiguration: (_) => se_KnowledgeBaseOrchestrationConfiguration(_, context),
+        promptTemplate: _json,
+        rerankingConfiguration: (_) => se_VectorSearchRerankingConfiguration(_, context),
+    });
+};
+const se_KnowledgeBaseOrchestrationConfiguration = (input, context) => {
+    return take(input, {
+        additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+        inferenceConfig: (_) => se_PromptInferenceConfiguration(_, context),
+        performanceConfig: _json,
+        promptTemplate: _json,
+    });
+};
+const se_LoopFlowNodeConfiguration = (input, context) => {
+    return take(input, {
+        definition: (_) => se_FlowDefinition(_, context),
+    });
+};
 const se_MetadataAttribute = (input, context) => {
     return take(input, {
         key: [],
@@ -2543,6 +2589,25 @@ const se_ToolSpecification = (input, context) => {
         name: [],
     });
 };
+const se_VectorSearchBedrockRerankingConfiguration = (input, context) => {
+    return take(input, {
+        metadataConfiguration: _json,
+        modelConfiguration: (_) => se_VectorSearchBedrockRerankingModelConfiguration(_, context),
+        numberOfRerankedResults: [],
+    });
+};
+const se_VectorSearchBedrockRerankingModelConfiguration = (input, context) => {
+    return take(input, {
+        additionalModelRequestFields: (_) => se_AdditionalModelRequestFields(_, context),
+        modelArn: [],
+    });
+};
+const se_VectorSearchRerankingConfiguration = (input, context) => {
+    return take(input, {
+        bedrockRerankingConfiguration: (_) => se_VectorSearchBedrockRerankingConfiguration(_, context),
+        type: [],
+    });
+};
 const se_Document = (input, context) => {
     return input;
 };
@@ -2563,6 +2628,18 @@ const de_ActionGroupSummary = (output, context) => {
         updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
     });
 };
+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_Agent = (output, context) => {
     return take(output, {
         agentArn: __expectString,
@@ -2834,6 +2911,7 @@ const de_FlowAliasSummaries = (output, context) => {
 const de_FlowAliasSummary = (output, context) => {
     return take(output, {
         arn: __expectString,
+        concurrencyConfiguration: _json,
         createdAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
         description: __expectString,
         flowId: __expectString,
@@ -2891,7 +2969,7 @@ const de_FlowNodeConfiguration = (output, context) => {
     }
     if (output.knowledgeBase != null) {
         return {
-            knowledgeBase: _json(output.knowledgeBase),
+            knowledgeBase: de_KnowledgeBaseFlowNodeConfiguration(output.knowledgeBase, context),
         };
     }
     if (output.lambdaFunction != null) {
@@ -2904,6 +2982,21 @@ const de_FlowNodeConfiguration = (output, context) => {
             lex: _json(output.lex),
         };
     }
+    if (output.loop != null) {
+        return {
+            loop: de_LoopFlowNodeConfiguration(output.loop, context),
+        };
+    }
+    if (output.loopController != null) {
+        return {
+            loopController: _json(output.loopController),
+        };
+    }
+    if (output.loopInput != null) {
+        return {
+            loopInput: _json(output.loopInput),
+        };
+    }
     if (output.output != null) {
         return {
             output: _json(output.output),
@@ -3046,6 +3139,26 @@ const de_KnowledgeBaseDocumentDetails = (output, context) => {
     });
     return retVal;
 };
+const de_KnowledgeBaseFlowNodeConfiguration = (output, context) => {
+    return take(output, {
+        guardrailConfiguration: _json,
+        inferenceConfiguration: (_) => de_PromptInferenceConfiguration(__expectUnion(_), context),
+        knowledgeBaseId: __expectString,
+        modelId: __expectString,
+        numberOfResults: __expectInt32,
+        orchestrationConfiguration: (_) => de_KnowledgeBaseOrchestrationConfiguration(_, context),
+        promptTemplate: _json,
+        rerankingConfiguration: (_) => de_VectorSearchRerankingConfiguration(_, context),
+    });
+};
+const de_KnowledgeBaseOrchestrationConfiguration = (output, context) => {
+    return take(output, {
+        additionalModelRequestFields: (_) => de_AdditionalModelRequestFields(_, context),
+        inferenceConfig: (_) => de_PromptInferenceConfiguration(__expectUnion(_), context),
+        performanceConfig: _json,
+        promptTemplate: _json,
+    });
+};
 const de_KnowledgeBaseSummaries = (output, context) => {
     const retVal = (output || [])
         .filter((e) => e != null)
@@ -3063,6 +3176,11 @@ const de_KnowledgeBaseSummary = (output, context) => {
         updatedAt: (_) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
     });
 };
+const de_LoopFlowNodeConfiguration = (output, context) => {
+    return take(output, {
+        definition: (_) => de_FlowDefinition(_, context),
+    });
+};
 const de_PromptConfiguration = (output, context) => {
     return take(output, {
         additionalModelRequestFields: (_) => de_Document(_, context),
@@ -3227,6 +3345,25 @@ const de_ToolSpecification = (output, context) => {
         name: __expectString,
     });
 };
+const de_VectorSearchBedrockRerankingConfiguration = (output, context) => {
+    return take(output, {
+        metadataConfiguration: _json,
+        modelConfiguration: (_) => de_VectorSearchBedrockRerankingModelConfiguration(_, context),
+        numberOfRerankedResults: __expectInt32,
+    });
+};
+const de_VectorSearchBedrockRerankingModelConfiguration = (output, context) => {
+    return take(output, {
+        additionalModelRequestFields: (_) => de_AdditionalModelRequestFields(_, context),
+        modelArn: __expectString,
+    });
+};
+const de_VectorSearchRerankingConfiguration = (output, context) => {
+    return take(output, {
+        bedrockRerankingConfiguration: (_) => de_VectorSearchBedrockRerankingConfiguration(_, context),
+        type: __expectString,
+    });
+};
 const de_Document = (output, context) => {
     return output;
 };

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

@@ -42,6 +42,10 @@ declare const CreateFlowAliasCommand_base: {
  *       flowVersion: "STRING_VALUE",
  *     },
  *   ],
+ *   concurrencyConfiguration: { // FlowAliasConcurrencyConfiguration
+ *     type: "Automatic" || "Manual", // required
+ *     maxConcurrency: Number("int"),
+ *   },
  *   flowIdentifier: "STRING_VALUE", // required
  *   clientToken: "STRING_VALUE",
  *   tags: { // TagsMap
@@ -58,6 +62,10 @@ declare const CreateFlowAliasCommand_base: {
  * //       flowVersion: "STRING_VALUE",
  * //     },
  * //   ],
+ * //   concurrencyConfiguration: { // FlowAliasConcurrencyConfiguration
+ * //     type: "Automatic" || "Manual", // required
+ * //     maxConcurrency: Number("int"),
+ * //   },
  * //   flowId: "STRING_VALUE", // required
  * //   id: "STRING_VALUE", // required
  * //   arn: "STRING_VALUE", // required