From 7587bb7f0607d9e06160e23a3fb462b7f978e56c Mon Sep 17 00:00:00 2001
From: Eric Shepherd <ericsh@amazon.com>
Date: Thu, 17 Oct 2024 15:34:38 +0000
Subject: [PATCH] Add missing SoS tag

---
 swift/example_code/sts/AssumeRole/Sources/entry.swift | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/swift/example_code/sts/AssumeRole/Sources/entry.swift b/swift/example_code/sts/AssumeRole/Sources/entry.swift
index 9f8dabb1d39..530f4c2cce0 100644
--- a/swift/example_code/sts/AssumeRole/Sources/entry.swift
+++ b/swift/example_code/sts/AssumeRole/Sources/entry.swift
@@ -245,20 +245,23 @@ func getIdentityResolver(accessKey: String?, secretKey: String?,
     
     if accessKey == nil || secretKey == nil {
         return nil
-    }
+    }  
 
     guard let accessKey = accessKey,
           let secretKey = secretKey else {
         return nil
     }
 
+    // snippet-start:[swift.sts.AssumeRole.create-static-resolver]
     let credentials = AWSCredentialIdentity(
         accessKey: accessKey,
         secret: secretKey,
         sessionToken: sessionToken
     )
 
-    return try StaticAWSCredentialIdentityResolver(credentials)
+    let identityResolver = try StaticAWSCredentialIdentityResolver(credentials)
+    // snippet-end:[swift.sts.AssumeRole.create-static-resolver]
+    return identityResolver
 }
 
 /// The program's asynchronous entry point.