feat(client-bedrock-agent-runtime): bedrock flow now support node action trace.

Author: awstools

clients/client-bedrock-agent-runtime/src/commands/InvokeAgentCommand.ts

@@ -44,25 +44,6 @@ export interface InvokeAgentCommandOutput extends InvokeAgentResponse, __Metadat
  *                <p>To activate trace enablement, turn <code>enableTrace</code> to <code>true</code>. Trace enablement helps you follow the agent's reasoning process that led it to the information it processed, the actions it took, and the final result it yielded. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/agents-test.html#trace-events">Trace enablement</a>.</p>
  *             </li>
  *             <li>
- *                <p>To stream agent responses, make sure that only orchestration prompt is enabled. Agent streaming is not supported for the following steps:
- *         </p>
- *                <ul>
- *                   <li>
- *                      <p>
- *                         <code>Pre-processing</code>
- *                      </p>
- *                   </li>
- *                   <li>
- *                      <p>
- *                         <code>Post-processing</code>
- *                      </p>
- *                   </li>
- *                   <li>
- *                      <p>Agent with 1 Knowledge base and <code>User Input</code> not enabled</p>
- *                   </li>
- *                </ul>
- *             </li>
- *             <li>
  *                <p>End a conversation by setting <code>endSession</code> to <code>true</code>.</p>
  *             </li>
  *             <li>

clients/client-bedrock-agent-runtime/src/commands/InvokeFlowCommand.ts

@@ -117,6 +117,13 @@ export interface InvokeFlowCommandOutput extends InvokeFlowResponse, __MetadataB
  * //             },
  * //           ],
  * //         },
+ * //         nodeActionTrace: { // FlowTraceNodeActionEvent
+ * //           nodeName: "STRING_VALUE", // required
+ * //           timestamp: new Date("TIMESTAMP"), // required
+ * //           requestId: "STRING_VALUE", // required
+ * //           serviceName: "STRING_VALUE", // required
+ * //           operationName: "STRING_VALUE", // required
+ * //         },
  * //       },
  * //     },
  * //     internalServerException: { // InternalServerException

clients/client-bedrock-agent-runtime/src/commands/UpdateSessionCommand.ts

@@ -10,7 +10,8 @@ import {
   ServiceOutputTypes,
 } from "../BedrockAgentRuntimeClient";
 import { commonParams } from "../endpoint/EndpointParameters";
-import { UpdateSessionRequest, UpdateSessionResponse } from "../models/models_0";
+import { UpdateSessionRequest } from "../models/models_0";
+import { UpdateSessionResponse } from "../models/models_1";
 import { de_UpdateSessionCommand, se_UpdateSessionCommand } from "../protocols/Aws_restJson1";
 
 /**

clients/client-bedrock-agent-runtime/src/models/models_0.ts

@@ -1774,6 +1774,43 @@ export interface FlowTraceConditionNodeResultEvent {
   satisfiedConditions: FlowTraceCondition[] | undefined;
 }
 
+/**
+ * <p>Contains information about an action (operation) called by a node in an Amazon Bedrock flow. The service generates action events for calls made by prompt nodes,
+ *             agent nodes, and Amazon Web Services Lambda nodes. </p>
+ * @public
+ */
+export interface FlowTraceNodeActionEvent {
+  /**
+   * <p>The name of the node that called the operation.</p>
+   * @public
+   */
+  nodeName: string | undefined;
+
+  /**
+   * <p>The date and time that the operation was called.</p>
+   * @public
+   */
+  timestamp: Date | undefined;
+
+  /**
+   * <p>The ID of the request that the node made to the operation.</p>
+   * @public
+   */
+  requestId: string | undefined;
+
+  /**
+   * <p>The name of the service that the node called.</p>
+   * @public
+   */
+  serviceName: string | undefined;
+
+  /**
+   * <p>The name of the operation that the node called.</p>
+   * @public
+   */
+  operationName: string | undefined;
+}
+
 /**
  * <p>Contains the content of the node input. For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/flows-trace.html">Track each step in your prompt flow by viewing its trace in Amazon Bedrock</a>.</p>
  * @public
@@ -1944,6 +1981,7 @@ export interface FlowTraceNodeOutputEvent {
  */
 export type FlowTrace =
   | FlowTrace.ConditionNodeResultTraceMember
+  | FlowTrace.NodeActionTraceMember
   | FlowTrace.NodeInputTraceMember
   | FlowTrace.NodeOutputTraceMember
   | FlowTrace.$UnknownMember;
@@ -1960,6 +1998,7 @@ export namespace FlowTrace {
     nodeInputTrace: FlowTraceNodeInputEvent;
     nodeOutputTrace?: never;
     conditionNodeResultTrace?: never;
+    nodeActionTrace?: never;
     $unknown?: never;
   }
 
@@ -1971,6 +2010,7 @@ export namespace FlowTrace {
     nodeInputTrace?: never;
     nodeOutputTrace: FlowTraceNodeOutputEvent;
     conditionNodeResultTrace?: never;
+    nodeActionTrace?: never;
     $unknown?: never;
   }
 
@@ -1982,6 +2022,20 @@ export namespace FlowTrace {
     nodeInputTrace?: never;
     nodeOutputTrace?: never;
     conditionNodeResultTrace: FlowTraceConditionNodeResultEvent;
+    nodeActionTrace?: never;
+    $unknown?: never;
+  }
+
+  /**
+   * <p>Contains information about an action (operation) called by a node.
+   *             For more information, see <a href="https://docs.aws.amazon.com/bedrock/latest/userguide/flows-trace.html">Track each step in your prompt flow by viewing its trace in Amazon Bedrock</a>.</p>
+   * @public
+   */
+  export interface NodeActionTraceMember {
+    nodeInputTrace?: never;
+    nodeOutputTrace?: never;
+    conditionNodeResultTrace?: never;
+    nodeActionTrace: FlowTraceNodeActionEvent;
     $unknown?: never;
   }
 
@@ -1992,13 +2046,15 @@ export namespace FlowTrace {
     nodeInputTrace?: never;
     nodeOutputTrace?: never;
     conditionNodeResultTrace?: never;
+    nodeActionTrace?: never;
     $unknown: [string, any];
   }
 
   export interface Visitor<T> {
     nodeInputTrace: (value: FlowTraceNodeInputEvent) => T;
     nodeOutputTrace: (value: FlowTraceNodeOutputEvent) => T;
     conditionNodeResultTrace: (value: FlowTraceConditionNodeResultEvent) => T;
+    nodeActionTrace: (value: FlowTraceNodeActionEvent) => T;
     _: (name: string, value: any) => T;
   }
 
@@ -2007,6 +2063,7 @@ export namespace FlowTrace {
     if (value.nodeOutputTrace !== undefined) return visitor.nodeOutputTrace(value.nodeOutputTrace);
     if (value.conditionNodeResultTrace !== undefined)
       return visitor.conditionNodeResultTrace(value.conditionNodeResultTrace);
+    if (value.nodeActionTrace !== undefined) return visitor.nodeActionTrace(value.nodeActionTrace);
     return visitor._(value.$unknown[0], value.$unknown[1]);
   };
 }
@@ -3114,8 +3171,7 @@ export interface VectorSearchRerankingConfiguration {
 }
 
 /**
- * <p>
- *             Configurations for streaming.</p>
+ * <p>Configurations for streaming.</p>
  * @public
  */
 export interface StreamingConfigurations {
@@ -3128,9 +3184,24 @@ export interface StreamingConfigurations {
   streamFinalResponse?: boolean | undefined;
 
   /**
-   * <p>
-   *             The guardrail interval to apply as response is generated.
-   *         </p>
+   * <p> The guardrail interval to apply as response is generated. By default, the guardrail
+   *             interval is set to 50 characters. If a larger interval is specified, the response will
+   *             be generated in larger chunks with fewer <code>ApplyGuardrail</code> calls. The
+   *             following examples show the response generated for <i>Hello, I am an
+   *                 agent</i> input string.</p>
+   *          <p>
+   *             <b>Example response in chunks: Interval set to 3 characters</b>
+   *          </p>
+   *          <p>
+   *             <code>'Hel', 'lo, ','I am', ' an', ' Age', 'nt'</code>
+   *          </p>
+   *          <p>Each chunk has at least 3 characters except for the last chunk</p>
+   *          <p>
+   *             <b>Example response in chunks: Interval set to 20 or more characters</b>
+   *          </p>
+   *          <p>
+   *             <code>Hello, I am an Agent</code>
+   *          </p>
    * @public
    */
   applyGuardrailInterval?: number | undefined;
@@ -9123,41 +9194,6 @@ export interface UpdateSessionRequest {
   sessionIdentifier: string | undefined;
 }
 
-/**
- * @public
- */
-export interface UpdateSessionResponse {
-  /**
-   * <p>The unique identifier of the session you updated.</p>
-   * @public
-   */
-  sessionId: string | undefined;
-
-  /**
-   * <p>The Amazon Resource Name (ARN) of the session that was updated.</p>
-   * @public
-   */
-  sessionArn: string | undefined;
-
-  /**
-   * <p>The status of the session you updated.</p>
-   * @public
-   */
-  sessionStatus: SessionStatus | undefined;
-
-  /**
-   * <p>The timestamp for when the session was created.</p>
-   * @public
-   */
-  createdAt: Date | undefined;
-
-  /**
-   * <p>The timestamp for when the session was last modified.</p>
-   * @public
-   */
-  lastUpdatedAt: Date | undefined;
-}
-
 /**
  * @internal
  */
@@ -9386,6 +9422,13 @@ export const FlowTraceConditionNodeResultEventFilterSensitiveLog = (obj: FlowTra
   ...(obj.satisfiedConditions && { satisfiedConditions: SENSITIVE_STRING }),
 });
 
+/**
+ * @internal
+ */
+export const FlowTraceNodeActionEventFilterSensitiveLog = (obj: FlowTraceNodeActionEvent): any => ({
+  ...obj,
+});
+
 /**
  * @internal
  */
@@ -9433,6 +9476,7 @@ export const FlowTraceFilterSensitiveLog = (obj: FlowTrace): any => {
   if (obj.nodeInputTrace !== undefined) return { nodeInputTrace: SENSITIVE_STRING };
   if (obj.nodeOutputTrace !== undefined) return { nodeOutputTrace: SENSITIVE_STRING };
   if (obj.conditionNodeResultTrace !== undefined) return { conditionNodeResultTrace: SENSITIVE_STRING };
+  if (obj.nodeActionTrace !== undefined) return { nodeActionTrace: SENSITIVE_STRING };
   if (obj.$unknown !== undefined) return { [obj.$unknown[0]]: "UNKNOWN" };
 };
 

clients/client-bedrock-agent-runtime/src/models/models_1.ts

@@ -34,11 +34,47 @@ import {
   RetrieveAndGenerateSessionConfiguration,
   RetrieveAndGenerateType,
   SearchType,
+  SessionStatus,
   StreamingConfigurations,
   VectorSearchRerankingConfiguration,
   VectorSearchRerankingConfigurationFilterSensitiveLog,
 } from "./models_0";
 
+/**
+ * @public
+ */
+export interface UpdateSessionResponse {
+  /**
+   * <p>The unique identifier of the session you updated.</p>
+   * @public
+   */
+  sessionId: string | undefined;
+
+  /**
+   * <p>The Amazon Resource Name (ARN) of the session that was updated.</p>
+   * @public
+   */
+  sessionArn: string | undefined;
+
+  /**
+   * <p>The status of the session you updated.</p>
+   * @public
+   */
+  sessionStatus: SessionStatus | undefined;
+
+  /**
+   * <p>The timestamp for when the session was created.</p>
+   * @public
+   */
+  createdAt: Date | undefined;
+
+  /**
+   * <p>The timestamp for when the session was last modified.</p>
+   * @public
+   */
+  lastUpdatedAt: Date | undefined;
+}
+
 /**
  * @public
  */

clients/client-bedrock-agent-runtime/src/protocols/Aws_restJson1.ts

@@ -117,6 +117,7 @@ import {
   FlowTrace,
   FlowTraceConditionNodeResultEvent,
   FlowTraceEvent,
+  FlowTraceNodeActionEvent,
   FlowTraceNodeInputContent,
   FlowTraceNodeInputEvent,
   FlowTraceNodeInputField,
@@ -3260,6 +3261,11 @@ const de_FlowTrace = (output: any, context: __SerdeContext): FlowTrace => {
       conditionNodeResultTrace: de_FlowTraceConditionNodeResultEvent(output.conditionNodeResultTrace, context),
     };
   }
+  if (output.nodeActionTrace != null) {
+    return {
+      nodeActionTrace: de_FlowTraceNodeActionEvent(output.nodeActionTrace, context),
+    };
+  }
   if (output.nodeInputTrace != null) {
     return {
       nodeInputTrace: de_FlowTraceNodeInputEvent(output.nodeInputTrace, context),
@@ -3300,6 +3306,19 @@ const de_FlowTraceEvent = (output: any, context: __SerdeContext): FlowTraceEvent
   }) as any;
 };
 
+/**
+ * deserializeAws_restJson1FlowTraceNodeActionEvent
+ */
+const de_FlowTraceNodeActionEvent = (output: any, context: __SerdeContext): FlowTraceNodeActionEvent => {
+  return take(output, {
+    nodeName: __expectString,
+    operationName: __expectString,
+    requestId: __expectString,
+    serviceName: __expectString,
+    timestamp: (_: any) => __expectNonNull(__parseRfc3339DateTimeWithOffset(_)),
+  }) as any;
+};
+
 /**
  * deserializeAws_restJson1FlowTraceNodeInputContent
  */