chore(codegen): run codegen for sts client

Author: siddsriv

clients/client-sts/src/defaultStsRoleAssumers.ts

@@ -84,17 +84,24 @@ export const getDefaultRoleAssumer = (
         logger: logger as any,
       });
     }
-    const { Credentials } = await stsClient.send(new AssumeRoleCommand(params));
+    const { Credentials, AssumedRoleUser } = await stsClient.send(new AssumeRoleCommand(params));
     if (!Credentials || !Credentials.AccessKeyId || !Credentials.SecretAccessKey) {
       throw new Error(`Invalid response from STS.assumeRole call with role ${params.RoleArn}`);
     }
+    let accountId;
+    try {
+      accountId = AssumedRoleUser.Arn.split(":")[4];
+    } catch (error) {
+      accountId = undefined;
+    }
     return {
       accessKeyId: Credentials.AccessKeyId,
       secretAccessKey: Credentials.SecretAccessKey,
       sessionToken: Credentials.SessionToken,
       expiration: Credentials.Expiration,
       // TODO(credentialScope): access normally when shape is updated.
       credentialScope: (Credentials as any).CredentialScope,
+      accountId,
     };
   };
 };
@@ -134,17 +141,24 @@ export const getDefaultRoleAssumerWithWebIdentity = (
         logger: logger as any,
       });
     }
-    const { Credentials } = await stsClient.send(new AssumeRoleWithWebIdentityCommand(params));
+    const { Credentials, AssumedRoleUser } = await stsClient.send(new AssumeRoleWithWebIdentityCommand(params));
     if (!Credentials || !Credentials.AccessKeyId || !Credentials.SecretAccessKey) {
       throw new Error(`Invalid response from STS.assumeRoleWithWebIdentity call with role ${params.RoleArn}`);
     }
+    let accountId;
+    try {
+      accountId = AssumedRoleUser.Arn.split(":")[4];
+    } catch (error) {
+      accountId = undefined;
+    }
     return {
       accessKeyId: Credentials.AccessKeyId,
       secretAccessKey: Credentials.SecretAccessKey,
       sessionToken: Credentials.SessionToken,
       expiration: Credentials.Expiration,
       // TODO(credentialScope): access normally when shape is updated.
       credentialScope: (Credentials as any).CredentialScope,
+      accountId,
     };
   };
 };