feat(client-bedrock-agent-runtime): This release adds trace functiona…

…lity to Bedrock Prompt Flows
aws · Nov 8, 2024 · bc89971 · bc89971
1 parent 9933561
commit bc89971
Show file tree

Hide file tree

Showing 5 changed files with 904 additions and 77 deletions.
diff --git a/clients/client-bedrock-agent-runtime/src/commands/InvokeAgentCommand.ts b/clients/client-bedrock-agent-runtime/src/commands/InvokeAgentCommand.ts
@@ -374,6 +374,8 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //             traceId: "STRING_VALUE",
  * //             text: "STRING_VALUE",
  * //             type: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING",
+ * //             overrideLambda: "STRING_VALUE",
+ * //             promptCreationMode: "DEFAULT" || "OVERRIDDEN",
  * //             inferenceConfiguration: { // InferenceConfiguration
  * //               temperature: Number("float"),
  * //               topP: Number("float"),
@@ -383,8 +385,6 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //                 "STRING_VALUE",
  * //               ],
  * //             },
- * //             overrideLambda: "STRING_VALUE",
- * //             promptCreationMode: "DEFAULT" || "OVERRIDDEN",
  * //             parserMode: "DEFAULT" || "OVERRIDDEN",
  * //           },
  * //           modelInvocationOutput: { // PreProcessingModelInvocationOutput
@@ -505,6 +505,8 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //             traceId: "STRING_VALUE",
  * //             text: "STRING_VALUE",
  * //             type: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING",
+ * //             overrideLambda: "STRING_VALUE",
+ * //             promptCreationMode: "DEFAULT" || "OVERRIDDEN",
  * //             inferenceConfiguration: {
  * //               temperature: Number("float"),
  * //               topP: Number("float"),
@@ -514,8 +516,6 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //                 "STRING_VALUE",
  * //               ],
  * //             },
- * //             overrideLambda: "STRING_VALUE",
- * //             promptCreationMode: "DEFAULT" || "OVERRIDDEN",
  * //             parserMode: "DEFAULT" || "OVERRIDDEN",
  * //           },
  * //           modelInvocationOutput: { // OrchestrationModelInvocationOutput
@@ -536,6 +536,8 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //             traceId: "STRING_VALUE",
  * //             text: "STRING_VALUE",
  * //             type: "PRE_PROCESSING" || "ORCHESTRATION" || "KNOWLEDGE_BASE_RESPONSE_GENERATION" || "POST_PROCESSING",
+ * //             overrideLambda: "STRING_VALUE",
+ * //             promptCreationMode: "DEFAULT" || "OVERRIDDEN",
  * //             inferenceConfiguration: {
  * //               temperature: Number("float"),
  * //               topP: Number("float"),
@@ -545,8 +547,6 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  * //                 "STRING_VALUE",
  * //               ],
  * //             },
- * //             overrideLambda: "STRING_VALUE",
- * //             promptCreationMode: "DEFAULT" || "OVERRIDDEN",
  * //             parserMode: "DEFAULT" || "OVERRIDDEN",
  * //           },
  * //           modelInvocationOutput: { // PostProcessingModelInvocationOutput

diff --git a/clients/client-bedrock-agent-runtime/src/commands/InvokeFlowCommand.ts b/clients/client-bedrock-agent-runtime/src/commands/InvokeFlowCommand.ts
@@ -59,6 +59,7 @@ export interface InvokeFlowCommandOutput extends InvokeFlowResponse, __MetadataB
  *       },
  *     },
  *   ],
+ *   enableTrace: true || false,
  * };
  * const command = new InvokeFlowCommand(input);
  * const response = await client.send(command);
@@ -74,6 +75,43 @@ export interface InvokeFlowCommandOutput extends InvokeFlowResponse, __MetadataB
  * //     flowCompletionEvent: { // FlowCompletionEvent
  * //       completionReason: "SUCCESS", // required
  * //     },
+ * //     flowTraceEvent: { // FlowTraceEvent
+ * //       trace: { // FlowTrace Union: only one key present
+ * //         nodeInputTrace: { // FlowTraceNodeInputEvent
+ * //           nodeName: "STRING_VALUE", // required
+ * //           timestamp: new Date("TIMESTAMP"), // required
+ * //           fields: [ // FlowTraceNodeInputFields // required
+ * //             { // FlowTraceNodeInputField
+ * //               nodeInputName: "STRING_VALUE", // required
+ * //               content: { // FlowTraceNodeInputContent Union: only one key present
+ * //                 document: "DOCUMENT_VALUE",
+ * //               },
+ * //             },
+ * //           ],
+ * //         },
+ * //         nodeOutputTrace: { // FlowTraceNodeOutputEvent
+ * //           nodeName: "STRING_VALUE", // required
+ * //           timestamp: new Date("TIMESTAMP"), // required
+ * //           fields: [ // FlowTraceNodeOutputFields // required
+ * //             { // FlowTraceNodeOutputField
+ * //               nodeOutputName: "STRING_VALUE", // required
+ * //               content: { // FlowTraceNodeOutputContent Union: only one key present
+ * //                 document: "DOCUMENT_VALUE",
+ * //               },
+ * //             },
+ * //           ],
+ * //         },
+ * //         conditionNodeResultTrace: { // FlowTraceConditionNodeResultEvent
+ * //           nodeName: "STRING_VALUE", // required
+ * //           timestamp: new Date("TIMESTAMP"), // required
+ * //           satisfiedConditions: [ // FlowTraceConditions // required
+ * //             { // FlowTraceCondition
+ * //               conditionName: "STRING_VALUE", // required
+ * //             },
+ * //           ],
+ * //         },
+ * //       },
+ * //     },
  * //     internalServerException: { // InternalServerException
  * //       message: "STRING_VALUE",
  * //     },