feat(client-bedrock-agent): Add support for the prompt caching featur…

…e for Bedrock Prompt Management
aws · Jan 27, 2025 · e631752 · e631752
1 parent 3488ae9
commit e631752
Show file tree

Hide file tree

Showing 21 changed files with 669 additions and 152 deletions.
diff --git a/clients/client-bedrock-agent/src/commands/CreateAgentAliasCommand.ts b/clients/client-bedrock-agent/src/commands/CreateAgentAliasCommand.ts
@@ -80,7 +80,7 @@ export interface CreateAgentAliasCommandOutput extends CreateAgentAliasResponse,
  * //         startDate: new Date("TIMESTAMP"),
  * //       },
  * //     ],
- * //     agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required
+ * //     agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING" || "DISSOCIATED", // required
  * //     failureReasons: [ // FailureReasons
  * //       "STRING_VALUE",
  * //     ],

diff --git a/clients/client-bedrock-agent/src/commands/CreateFlowCommand.ts b/clients/client-bedrock-agent/src/commands/CreateFlowCommand.ts
@@ -83,6 +83,9 @@ export interface CreateFlowCommandOutput extends CreateFlowResponse, __MetadataB
  *                 templateConfiguration: { // PromptTemplateConfiguration Union: only one key present
  *                   text: { // TextPromptTemplateConfiguration
  *                     text: "STRING_VALUE", // required
+ *                     cachePoint: { // CachePointBlock
+ *                       type: "default", // required
+ *                     },
  *                     inputVariables: [ // PromptInputVariablesList
  *                       { // PromptInputVariable
  *                         name: "STRING_VALUE",
@@ -96,13 +99,17 @@ export interface CreateFlowCommandOutput extends CreateFlowResponse, __MetadataB
  *                         content: [ // ContentBlocks // required
  *                           { // ContentBlock Union: only one key present
  *                             text: "STRING_VALUE",
+ *                             cachePoint: {
+ *                               type: "default", // required
+ *                             },
  *                           },
  *                         ],
  *                       },
  *                     ],
  *                     system: [ // SystemContentBlocks
  *                       { // SystemContentBlock Union: only one key present
  *                         text: "STRING_VALUE",
+ *                         cachePoint: "<CachePointBlock>",
  *                       },
  *                     ],
  *                     inputVariables: [
@@ -120,6 +127,7 @@ export interface CreateFlowCommandOutput extends CreateFlowResponse, __MetadataB
  *                               json: "DOCUMENT_VALUE",
  *                             },
  *                           },
+ *                           cachePoint: "<CachePointBlock>",
  *                         },
  *                       ],
  *                       toolChoice: { // ToolChoice Union: only one key present
@@ -263,6 +271,9 @@ export interface CreateFlowCommandOutput extends CreateFlowResponse, __MetadataB
  * //                 templateConfiguration: { // PromptTemplateConfiguration Union: only one key present
  * //                   text: { // TextPromptTemplateConfiguration
  * //                     text: "STRING_VALUE", // required
+ * //                     cachePoint: { // CachePointBlock
+ * //                       type: "default", // required
+ * //                     },
  * //                     inputVariables: [ // PromptInputVariablesList
  * //                       { // PromptInputVariable
  * //                         name: "STRING_VALUE",
@@ -276,13 +287,17 @@ export interface CreateFlowCommandOutput extends CreateFlowResponse, __MetadataB
  * //                         content: [ // ContentBlocks // required
  * //                           { // ContentBlock Union: only one key present
  * //                             text: "STRING_VALUE",
+ * //                             cachePoint: {
+ * //                               type: "default", // required
+ * //                             },
  * //                           },
  * //                         ],
  * //                       },
  * //                     ],
  * //                     system: [ // SystemContentBlocks
  * //                       { // SystemContentBlock Union: only one key present
  * //                         text: "STRING_VALUE",
+ * //                         cachePoint: "<CachePointBlock>",
  * //                       },
  * //                     ],
  * //                     inputVariables: [
@@ -300,6 +315,7 @@ export interface CreateFlowCommandOutput extends CreateFlowResponse, __MetadataB
  * //                               json: "DOCUMENT_VALUE",
  * //                             },
  * //                           },
+ * //                           cachePoint: "<CachePointBlock>",
  * //                         },
  * //                       ],
  * //                       toolChoice: { // ToolChoice Union: only one key present

diff --git a/clients/client-bedrock-agent/src/commands/CreateFlowVersionCommand.ts b/clients/client-bedrock-agent/src/commands/CreateFlowVersionCommand.ts
@@ -94,6 +94,9 @@ export interface CreateFlowVersionCommandOutput extends CreateFlowVersionRespons
  * //                 templateConfiguration: { // PromptTemplateConfiguration Union: only one key present
  * //                   text: { // TextPromptTemplateConfiguration
  * //                     text: "STRING_VALUE", // required
+ * //                     cachePoint: { // CachePointBlock
+ * //                       type: "default", // required
+ * //                     },
  * //                     inputVariables: [ // PromptInputVariablesList
  * //                       { // PromptInputVariable
  * //                         name: "STRING_VALUE",
@@ -107,13 +110,17 @@ export interface CreateFlowVersionCommandOutput extends CreateFlowVersionRespons
  * //                         content: [ // ContentBlocks // required
  * //                           { // ContentBlock Union: only one key present
  * //                             text: "STRING_VALUE",
+ * //                             cachePoint: {
+ * //                               type: "default", // required
+ * //                             },
  * //                           },
  * //                         ],
  * //                       },
  * //                     ],
  * //                     system: [ // SystemContentBlocks
  * //                       { // SystemContentBlock Union: only one key present
  * //                         text: "STRING_VALUE",
+ * //                         cachePoint: "<CachePointBlock>",
  * //                       },
  * //                     ],
  * //                     inputVariables: [
@@ -131,6 +138,7 @@ export interface CreateFlowVersionCommandOutput extends CreateFlowVersionRespons
  * //                               json: "DOCUMENT_VALUE",
  * //                             },
  * //                           },
+ * //                           cachePoint: "<CachePointBlock>",
  * //                         },
  * //                       ],
  * //                       toolChoice: { // ToolChoice Union: only one key present

diff --git a/clients/client-bedrock-agent/src/commands/CreatePromptCommand.ts b/clients/client-bedrock-agent/src/commands/CreatePromptCommand.ts
@@ -52,6 +52,9 @@ export interface CreatePromptCommandOutput extends CreatePromptResponse, __Metad
  *       templateConfiguration: { // PromptTemplateConfiguration Union: only one key present
  *         text: { // TextPromptTemplateConfiguration
  *           text: "STRING_VALUE", // required
+ *           cachePoint: { // CachePointBlock
+ *             type: "default", // required
+ *           },
  *           inputVariables: [ // PromptInputVariablesList
  *             { // PromptInputVariable
  *               name: "STRING_VALUE",
@@ -65,13 +68,17 @@ export interface CreatePromptCommandOutput extends CreatePromptResponse, __Metad
  *               content: [ // ContentBlocks // required
  *                 { // ContentBlock Union: only one key present
  *                   text: "STRING_VALUE",
+ *                   cachePoint: {
+ *                     type: "default", // required
+ *                   },
  *                 },
  *               ],
  *             },
  *           ],
  *           system: [ // SystemContentBlocks
  *             { // SystemContentBlock Union: only one key present
  *               text: "STRING_VALUE",
+ *               cachePoint: "<CachePointBlock>",
  *             },
  *           ],
  *           inputVariables: [
@@ -89,6 +96,7 @@ export interface CreatePromptCommandOutput extends CreatePromptResponse, __Metad
  *                     json: "DOCUMENT_VALUE",
  *                   },
  *                 },
+ *                 cachePoint: "<CachePointBlock>",
  *               },
  *             ],
  *             toolChoice: { // ToolChoice Union: only one key present
@@ -145,6 +153,9 @@ export interface CreatePromptCommandOutput extends CreatePromptResponse, __Metad
  * //       templateConfiguration: { // PromptTemplateConfiguration Union: only one key present
  * //         text: { // TextPromptTemplateConfiguration
  * //           text: "STRING_VALUE", // required
+ * //           cachePoint: { // CachePointBlock
+ * //             type: "default", // required
+ * //           },
  * //           inputVariables: [ // PromptInputVariablesList
  * //             { // PromptInputVariable
  * //               name: "STRING_VALUE",
@@ -158,13 +169,17 @@ export interface CreatePromptCommandOutput extends CreatePromptResponse, __Metad
  * //               content: [ // ContentBlocks // required
  * //                 { // ContentBlock Union: only one key present
  * //                   text: "STRING_VALUE",
+ * //                   cachePoint: {
+ * //                     type: "default", // required
+ * //                   },
  * //                 },
  * //               ],
  * //             },
  * //           ],
  * //           system: [ // SystemContentBlocks
  * //             { // SystemContentBlock Union: only one key present
  * //               text: "STRING_VALUE",
+ * //               cachePoint: "<CachePointBlock>",
  * //             },
  * //           ],
  * //           inputVariables: [
@@ -182,6 +197,7 @@ export interface CreatePromptCommandOutput extends CreatePromptResponse, __Metad
  * //                     json: "DOCUMENT_VALUE",
  * //                   },
  * //                 },
+ * //                 cachePoint: "<CachePointBlock>",
  * //               },
  * //             ],
  * //             toolChoice: { // ToolChoice Union: only one key present

diff --git a/clients/client-bedrock-agent/src/commands/CreatePromptVersionCommand.ts b/clients/client-bedrock-agent/src/commands/CreatePromptVersionCommand.ts
@@ -61,6 +61,9 @@ export interface CreatePromptVersionCommandOutput extends CreatePromptVersionRes
  * //       templateConfiguration: { // PromptTemplateConfiguration Union: only one key present
  * //         text: { // TextPromptTemplateConfiguration
  * //           text: "STRING_VALUE", // required
+ * //           cachePoint: { // CachePointBlock
+ * //             type: "default", // required
+ * //           },
  * //           inputVariables: [ // PromptInputVariablesList
  * //             { // PromptInputVariable
  * //               name: "STRING_VALUE",
@@ -74,13 +77,17 @@ export interface CreatePromptVersionCommandOutput extends CreatePromptVersionRes
  * //               content: [ // ContentBlocks // required
  * //                 { // ContentBlock Union: only one key present
  * //                   text: "STRING_VALUE",
+ * //                   cachePoint: {
+ * //                     type: "default", // required
+ * //                   },
  * //                 },
  * //               ],
  * //             },
  * //           ],
  * //           system: [ // SystemContentBlocks
  * //             { // SystemContentBlock Union: only one key present
  * //               text: "STRING_VALUE",
+ * //               cachePoint: "<CachePointBlock>",
  * //             },
  * //           ],
  * //           inputVariables: [
@@ -98,6 +105,7 @@ export interface CreatePromptVersionCommandOutput extends CreatePromptVersionRes
  * //                     json: "DOCUMENT_VALUE",
  * //                   },
  * //                 },
+ * //                 cachePoint: "<CachePointBlock>",
  * //               },
  * //             ],
  * //             toolChoice: { // ToolChoice Union: only one key present

diff --git a/clients/client-bedrock-agent/src/commands/DeleteAgentAliasCommand.ts b/clients/client-bedrock-agent/src/commands/DeleteAgentAliasCommand.ts
@@ -44,7 +44,7 @@ export interface DeleteAgentAliasCommandOutput extends DeleteAgentAliasResponse,
  * // { // DeleteAgentAliasResponse
  * //   agentId: "STRING_VALUE", // required
  * //   agentAliasId: "STRING_VALUE", // required
- * //   agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required
+ * //   agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING" || "DISSOCIATED", // required
  * // };
  *
  * ```

diff --git a/clients/client-bedrock-agent/src/commands/DeleteKnowledgeBaseDocumentsCommand.ts b/clients/client-bedrock-agent/src/commands/DeleteKnowledgeBaseDocumentsCommand.ts
@@ -6,7 +6,8 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { BedrockAgentClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../BedrockAgentClient";
 import { commonParams } from "../endpoint/EndpointParameters";
-import { DeleteKnowledgeBaseDocumentsRequest, DeleteKnowledgeBaseDocumentsResponse } from "../models/models_0";
+import { DeleteKnowledgeBaseDocumentsRequest } from "../models/models_0";
+import { DeleteKnowledgeBaseDocumentsResponse } from "../models/models_1";
 import {
   de_DeleteKnowledgeBaseDocumentsCommand,
   se_DeleteKnowledgeBaseDocumentsCommand,

diff --git a/clients/client-bedrock-agent/src/commands/GetAgentAliasCommand.ts b/clients/client-bedrock-agent/src/commands/GetAgentAliasCommand.ts
@@ -69,7 +69,7 @@ export interface GetAgentAliasCommandOutput extends GetAgentAliasResponse, __Met
  * //         startDate: new Date("TIMESTAMP"),
  * //       },
  * //     ],
- * //     agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING", // required
+ * //     agentAliasStatus: "CREATING" || "PREPARED" || "FAILED" || "UPDATING" || "DELETING" || "DISSOCIATED", // required
  * //     failureReasons: [ // FailureReasons
  * //       "STRING_VALUE",
  * //     ],

diff --git a/clients/client-bedrock-agent/src/commands/GetFlowCommand.ts b/clients/client-bedrock-agent/src/commands/GetFlowCommand.ts
@@ -89,6 +89,9 @@ export interface GetFlowCommandOutput extends GetFlowResponse, __MetadataBearer
  * //                 templateConfiguration: { // PromptTemplateConfiguration Union: only one key present
  * //                   text: { // TextPromptTemplateConfiguration
  * //                     text: "STRING_VALUE", // required
+ * //                     cachePoint: { // CachePointBlock
+ * //                       type: "default", // required
+ * //                     },
  * //                     inputVariables: [ // PromptInputVariablesList
  * //                       { // PromptInputVariable
  * //                         name: "STRING_VALUE",
@@ -102,13 +105,17 @@ export interface GetFlowCommandOutput extends GetFlowResponse, __MetadataBearer
  * //                         content: [ // ContentBlocks // required
  * //                           { // ContentBlock Union: only one key present
  * //                             text: "STRING_VALUE",
+ * //                             cachePoint: {
+ * //                               type: "default", // required
+ * //                             },
  * //                           },
  * //                         ],
  * //                       },
  * //                     ],
  * //                     system: [ // SystemContentBlocks
  * //                       { // SystemContentBlock Union: only one key present
  * //                         text: "STRING_VALUE",
+ * //                         cachePoint: "<CachePointBlock>",
  * //                       },
  * //                     ],
  * //                     inputVariables: [
@@ -126,6 +133,7 @@ export interface GetFlowCommandOutput extends GetFlowResponse, __MetadataBearer
  * //                               json: "DOCUMENT_VALUE",
  * //                             },
  * //                           },
+ * //                           cachePoint: "<CachePointBlock>",
  * //                         },
  * //                       ],
  * //                       toolChoice: { // ToolChoice Union: only one key present
@@ -301,8 +309,16 @@ export interface GetFlowCommandOutput extends GetFlowResponse, __MetadataBearer
  * //           connection: "STRING_VALUE", // required
  * //         },
  * //         unspecified: {},
+ * //         unknownNodeInput: { // UnknownNodeInputFlowValidationDetails
+ * //           node: "STRING_VALUE", // required
+ * //           input: "STRING_VALUE", // required
+ * //         },
+ * //         unknownNodeOutput: { // UnknownNodeOutputFlowValidationDetails
+ * //           node: "STRING_VALUE", // required
+ * //           output: "STRING_VALUE", // required
+ * //         },
  * //       },
- * //       type: "CyclicConnection" || "DuplicateConnections" || "DuplicateConditionExpression" || "UnreachableNode" || "UnknownConnectionSource" || "UnknownConnectionSourceOutput" || "UnknownConnectionTarget" || "UnknownConnectionTargetInput" || "UnknownConnectionCondition" || "MalformedConditionExpression" || "MalformedNodeInputExpression" || "MismatchedNodeInputType" || "MismatchedNodeOutputType" || "IncompatibleConnectionDataType" || "MissingConnectionConfiguration" || "MissingDefaultCondition" || "MissingEndingNodes" || "MissingNodeConfiguration" || "MissingNodeInput" || "MissingNodeOutput" || "MissingStartingNodes" || "MultipleNodeInputConnections" || "UnfulfilledNodeInput" || "UnsatisfiedConnectionConditions" || "Unspecified",
+ * //       type: "CyclicConnection" || "DuplicateConnections" || "DuplicateConditionExpression" || "UnreachableNode" || "UnknownConnectionSource" || "UnknownConnectionSourceOutput" || "UnknownConnectionTarget" || "UnknownConnectionTargetInput" || "UnknownConnectionCondition" || "MalformedConditionExpression" || "MalformedNodeInputExpression" || "MismatchedNodeInputType" || "MismatchedNodeOutputType" || "IncompatibleConnectionDataType" || "MissingConnectionConfiguration" || "MissingDefaultCondition" || "MissingEndingNodes" || "MissingNodeConfiguration" || "MissingNodeInput" || "MissingNodeOutput" || "MissingStartingNodes" || "MultipleNodeInputConnections" || "UnfulfilledNodeInput" || "UnsatisfiedConnectionConditions" || "Unspecified" || "UnknownNodeInput" || "UnknownNodeOutput",
  * //     },
  * //   ],
  * // };

diff --git a/clients/client-bedrock-agent/src/commands/GetFlowVersionCommand.ts b/clients/client-bedrock-agent/src/commands/GetFlowVersionCommand.ts
@@ -93,6 +93,9 @@ export interface GetFlowVersionCommandOutput extends GetFlowVersionResponse, __M
  * //                 templateConfiguration: { // PromptTemplateConfiguration Union: only one key present
  * //                   text: { // TextPromptTemplateConfiguration
  * //                     text: "STRING_VALUE", // required
+ * //                     cachePoint: { // CachePointBlock
+ * //                       type: "default", // required
+ * //                     },
  * //                     inputVariables: [ // PromptInputVariablesList
  * //                       { // PromptInputVariable
  * //                         name: "STRING_VALUE",
@@ -106,13 +109,17 @@ export interface GetFlowVersionCommandOutput extends GetFlowVersionResponse, __M
  * //                         content: [ // ContentBlocks // required
  * //                           { // ContentBlock Union: only one key present
  * //                             text: "STRING_VALUE",
+ * //                             cachePoint: {
+ * //                               type: "default", // required
+ * //                             },
  * //                           },
  * //                         ],
  * //                       },
  * //                     ],
  * //                     system: [ // SystemContentBlocks
  * //                       { // SystemContentBlock Union: only one key present
  * //                         text: "STRING_VALUE",
+ * //                         cachePoint: "<CachePointBlock>",
  * //                       },
  * //                     ],
  * //                     inputVariables: [
@@ -130,6 +137,7 @@ export interface GetFlowVersionCommandOutput extends GetFlowVersionResponse, __M
  * //                               json: "DOCUMENT_VALUE",
  * //                             },
  * //                           },
+ * //                           cachePoint: "<CachePointBlock>",
  * //                         },
  * //                       ],
  * //                       toolChoice: { // ToolChoice Union: only one key present

diff --git a/clients/client-bedrock-agent/src/commands/GetKnowledgeBaseDocumentsCommand.ts b/clients/client-bedrock-agent/src/commands/GetKnowledgeBaseDocumentsCommand.ts
@@ -6,7 +6,7 @@ import { MetadataBearer as __MetadataBearer } from "@smithy/types";
 
 import { BedrockAgentClientResolvedConfig, ServiceInputTypes, ServiceOutputTypes } from "../BedrockAgentClient";
 import { commonParams } from "../endpoint/EndpointParameters";
-import { GetKnowledgeBaseDocumentsRequest, GetKnowledgeBaseDocumentsResponse } from "../models/models_0";
+import { GetKnowledgeBaseDocumentsRequest, GetKnowledgeBaseDocumentsResponse } from "../models/models_1";
 import { de_GetKnowledgeBaseDocumentsCommand, se_GetKnowledgeBaseDocumentsCommand } from "../protocols/Aws_restJson1";
 
 /**