Swift-format

Author: sebsto

Examples/retrieve/Package.swift

@@ -27,4 +27,4 @@ let package = Package(
             ]
         )
     ]
-)
+)

Examples/retrieve/Sources/Retrieve.swift

@@ -25,16 +25,16 @@ struct Main {
             print("Error:\n\(error)")
         }
     }
-    
+
     static func retrieve() async throws {
         var logger = Logger(label: "Retrieve")
         logger.logLevel = .debug
 
         let bedrock = try await BedrockService(
             region: .uswest2,
             logger: logger
-            // uncomment if you use SSO with AWS Identity Center
-            // authentication: .sso
+                // uncomment if you use SSO with AWS Identity Center
+                // authentication: .sso
         )
 
         let knowledgeBaseId = "EQ13XRVPLE"
@@ -54,15 +54,15 @@ struct Main {
         )
 
         print("Retrieved \(response.results?.count ?? 0) results:")
-        
+
         // Show best match using convenience function
         if let bestMatch = response.bestMatch() {
             print("\n--- Best Match (Score: \(bestMatch.score ?? 0)) ---")
             if let content = bestMatch.content?.text {
                 print("Content: \(content)")
             }
         }
-        
+
         // Show all results using convenience property
         // if let results = response.results {
         //     for (index, result) in results.enumerated() {
@@ -79,4 +79,4 @@ struct Main {
         //     }
         // }
     }
-}
+}

Sources/BedrockService/BedrockService+AgentRuntime.swift

@@ -48,10 +48,13 @@ extension BedrockService {
         retrievalQuery: String,
         numberOfResults: Int = 3
     ) async throws -> RetrieveResult {
-        logger.trace("Retrieving from knowledge base", metadata: [
-            "knowledgeBaseId": .string(knowledgeBaseId),
-            "numberOfResults": .stringConvertible(numberOfResults)
-        ])
+        logger.trace(
+            "Retrieving from knowledge base",
+            metadata: [
+                "knowledgeBaseId": .string(knowledgeBaseId),
+                "numberOfResults": .stringConvertible(numberOfResults),
+            ]
+        )
 
         let input = RetrieveInput(
             knowledgeBaseId: knowledgeBaseId,
@@ -66,4 +69,4 @@ extension BedrockService {
             try handleCommonError(error, context: "retrieving from knowledge base")
         }
     }
-}
+}

Sources/BedrockService/BedrockService.swift

@@ -14,8 +14,8 @@
 //===----------------------------------------------------------------------===//
 
 @preconcurrency import AWSBedrock
-@preconcurrency import AWSBedrockRuntime
 @preconcurrency import AWSBedrockAgentRuntime
+@preconcurrency import AWSBedrockRuntime
 import AWSClientRuntime
 import AWSSSOOIDC
 import AwsCommonRuntimeKit
@@ -112,7 +112,7 @@ public struct BedrockService: Sendable {
                 metadata: ["authentication type": "\(authentication)"]
             )
         }
-        
+
         self.logger.trace(
             "Initialized SwiftBedrock",
             metadata: ["region": .string(region.rawValue)]

Sources/BedrockService/Protocols/BedrockAgentRuntimeProtocol.swift

@@ -27,4 +27,4 @@ public protocol BedrockAgentRuntimeProtocol: Sendable {
     func retrieve(input: RetrieveInput) async throws -> RetrieveOutput
 }
 
-extension BedrockAgentRuntimeClient: @retroactive @unchecked Sendable, BedrockAgentRuntimeProtocol {}
+extension BedrockAgentRuntimeClient: @retroactive @unchecked Sendable, BedrockAgentRuntimeProtocol {}

Sources/BedrockService/Protocols/BedrockConfigProtocol.swift

@@ -14,8 +14,8 @@
 //===----------------------------------------------------------------------===//
 
 import AWSBedrock
-import AWSBedrockRuntime
 import AWSBedrockAgentRuntime
+import AWSBedrockRuntime
 import ClientRuntime
 import SmithyHTTPAuthAPI
 import SmithyIdentity

Sources/BedrockService/RetrieveResult.swift

@@ -31,31 +31,31 @@ struct SerializableResult: Codable {
 
 public struct RetrieveResult: Sendable {
     public let output: RetrieveOutput
-    
+
     public init(_ output: RetrieveOutput) {
         self.output = output
     }
-    
+
     public var results: [RAGRetrievalResult]? {
-        return output.retrievalResults
+        output.retrievalResults
     }
-    
+
     public func bestMatch() -> RAGRetrievalResult? {
-        return output.retrievalResults?.max { ($0.score ?? 0) < ($1.score ?? 0) }
+        output.retrievalResults?.max { ($0.score ?? 0) < ($1.score ?? 0) }
     }
-    
+
     public func toJSON() throws -> String {
         guard let results = output.retrievalResults else { return "[]" }
-        
+
         let serializableResults = results.map { result in
             SerializableResult(
                 content: result.content?.text,
                 score: result.score,
                 source: result.location?.s3Location?.uri
             )
         }
-        
+
         let jsonData = try JSONEncoder().encode(serializableResults)
         return String(data: jsonData, encoding: .utf8) ?? "[]"
     }
-}
+}