Agents for Amazon Bedrock Runtime Update: Changes for enhanced metadata in trace

Author: AWS

.changes/next-release/feature-AgentsforAmazonBedrockRuntime-4e89831.json

@@ -0,0 +1,6 @@
+{
+    "type": "feature",
+    "category": "Agents for Amazon Bedrock Runtime",
+    "contributor": "",
+    "description": "Changes for enhanced metadata in trace"
+}

services/bedrockagentruntime/src/main/resources/codegen-resources/service-2.json

@@ -613,6 +613,10 @@
     "ActionGroupInvocationOutput":{
       "type":"structure",
       "members":{
+        "metadata":{
+          "shape":"Metadata",
+          "documentation":"<p>Contains information about the action group output.</p>"
+        },
         "text":{
           "shape":"ActionGroupOutputString",
           "documentation":"<p>The JSON-formatted string returned by the API invoked by the action group.</p>"
@@ -783,6 +787,10 @@
           "shape":"String",
           "documentation":"<p>The output's agent collaborator name.</p>"
         },
+        "metadata":{
+          "shape":"Metadata",
+          "documentation":"<p>Contains information about the output from the agent collaborator.</p>"
+        },
         "output":{
           "shape":"AgentCollaboratorOutputPayload",
           "documentation":"<p>The output's output.</p>"
@@ -1215,6 +1223,10 @@
         "files":{
           "shape":"Files",
           "documentation":"<p>Contains output files, if generated by code execution.</p>"
+        },
+        "metadata":{
+          "shape":"Metadata",
+          "documentation":"<p>Contains information about the output from the code interpreter.</p>"
         }
       },
       "documentation":"<p>Contains the JSON-formatted string returned by the API invoked by the code interpreter.</p>"
@@ -1735,10 +1747,18 @@
     "FailureTrace":{
       "type":"structure",
       "members":{
+        "failureCode":{
+          "shape":"Integer",
+          "documentation":"<p>The failure code for the trace.</p>"
+        },
         "failureReason":{
           "shape":"FailureReasonString",
           "documentation":"<p>The reason the interaction failed.</p>"
         },
+        "metadata":{
+          "shape":"Metadata",
+          "documentation":"<p>Information about the failure that occurred.</p>"
+        },
         "traceId":{
           "shape":"TraceId",
           "documentation":"<p>The unique identifier of the trace.</p>"
@@ -1856,6 +1876,10 @@
     "FinalResponse":{
       "type":"structure",
       "members":{
+        "metadata":{
+          "shape":"Metadata",
+          "documentation":"<p>Contains information about the invoke agent operation.</p>"
+        },
         "text":{
           "shape":"FinalResponseString",
           "documentation":"<p>The text in the response to the user.</p>"
@@ -3059,6 +3083,10 @@
           "shape":"GuardrailAssessmentList",
           "documentation":"<p>The details of the input assessments used in the Guardrail Trace.</p>"
         },
+        "metadata":{
+          "shape":"Metadata",
+          "documentation":"<p>Contains information about the Guardrail output.</p>"
+        },
         "outputAssessments":{
           "shape":"GuardrailAssessmentList",
           "documentation":"<p>The details of the output assessments used in the Guardrail Trace.</p>"
@@ -4035,6 +4063,10 @@
     "KnowledgeBaseLookupOutput":{
       "type":"structure",
       "members":{
+        "metadata":{
+          "shape":"Metadata",
+          "documentation":"<p>Contains information about the knowledge base output.</p>"
+        },
         "retrievedReferences":{
           "shape":"RetrievedReferences",
           "documentation":"<p>Contains metadata about the sources cited for the generated response.</p>"
@@ -4304,6 +4336,10 @@
         }
       }
     },
+    "Long":{
+      "type":"long",
+      "box":true
+    },
     "MaxResults":{
       "type":"integer",
       "documentation":"<p>Max Results.</p>",
@@ -4399,12 +4435,32 @@
     "Metadata":{
       "type":"structure",
       "members":{
+        "clientRequestId":{
+          "shape":"String",
+          "documentation":"<p>A unique identifier associated with the downstream invocation. This ID can be used for tracing, debugging, and identifying specific invocations in customer logs or systems.</p>"
+        },
+        "endTime":{
+          "shape":"SyntheticTimestamp_date_time",
+          "documentation":"<p>In the final response, <code>endTime</code> is the end time time of the agent invocation operation.</p>"
+        },
+        "operationTotalTimeMs":{
+          "shape":"Long",
+          "documentation":"<p>The total time it took for the agent to complete execution. This field is only set for the final response.</p>"
+        },
+        "startTime":{
+          "shape":"SyntheticTimestamp_date_time",
+          "documentation":"<p>In the final response, <code>startTime</code> is the start time of the agent invocation operation.</p>"
+        },
+        "totalTimeMs":{
+          "shape":"Long",
+          "documentation":"<p> The total execution time for the specific invocation being processed (model, knowledge base, guardrail, agent collaborator, or code interpreter). It represents how long the individual invocation took.</p>"
+        },
         "usage":{
           "shape":"Usage",
-          "documentation":"<p>Contains details of the foundation model usage.</p>"
+          "documentation":"<p>Specific to model invocation and contains details about the usage of a foundation model.</p>"
         }
       },
-      "documentation":"<p>Provides details of the foundation model.</p>",
+      "documentation":"<p>Provides information about the execution process for different types of invocations, such as model invocation, knowledge base invocation, agent collaborator invocation, guardrail invocation, and code interpreter Invocation.</p>",
       "sensitive":true
     },
     "MetadataAttributeSchema":{