swift-format

Author: sebsto

Sources/InvokeModel/BedrockService+InvokeModelImage.swift

@@ -160,31 +160,31 @@ extension BedrockService {
 
     /// Sends the request to invoke the model and returns the generated image(s)
     private func sendRequest(request: InvokeModelRequest, model: BedrockModel) async throws -> ImageGenerationOutput {
-            let input: InvokeModelInput = try request.getInvokeModelInput(forRegion: self.region)
+        let input: InvokeModelInput = try request.getInvokeModelInput(forRegion: self.region)
+        logger.trace(
+            "Sending request to invokeModel",
+            metadata: [
+                "model": .string(model.id), "request": .string(String(describing: input)),
+            ]
+        )
+        let response = try await self.bedrockRuntimeClient.invokeModel(input: input)
+        guard let responseBody = response.body else {
             logger.trace(
-                "Sending request to invokeModel",
+                "Invalid response",
                 metadata: [
-                    "model": .string(model.id), "request": .string(String(describing: input)),
+                    "response": .string(String(describing: response)),
+                    "hasBody": .stringConvertible(response.body != nil),
                 ]
             )
-            let response = try await self.bedrockRuntimeClient.invokeModel(input: input)
-            guard let responseBody = response.body else {
-                logger.trace(
-                    "Invalid response",
-                    metadata: [
-                        "response": .string(String(describing: response)),
-                        "hasBody": .stringConvertible(response.body != nil),
-                    ]
-                )
-                throw BedrockLibraryError.invalidSDKResponse(
-                    "Something went wrong while extracting body from response."
-                )
-            }
-            let invokemodelResponse: InvokeModelResponse = try InvokeModelResponse.createImageResponse(
-                body: responseBody,
-                model: model
+            throw BedrockLibraryError.invalidSDKResponse(
+                "Something went wrong while extracting body from response."
             )
-            return try invokemodelResponse.getGeneratedImage()
+        }
+        let invokemodelResponse: InvokeModelResponse = try InvokeModelResponse.createImageResponse(
+            body: responseBody,
+            model: model
+        )
+        return try invokemodelResponse.getGeneratedImage()
     }
 
     /// Generates 1 to 5 image variation(s) from reference images and a text prompt using a specific model

Sources/InvokeModel/InvokeModelRequest.swift

@@ -206,7 +206,7 @@ struct InvokeModelRequest {
     /// - Returns: A configured InvokeModelInput containing the model ID, content type, and encoded request body
     /// - Throws: BedrockLibraryError.encodingError if the request body cannot be encoded to JSON
     public func getInvokeModelInput(forRegion region: Region) throws -> InvokeModelInput {
-        do {        
+        do {
             let jsonData: Data = try JSONEncoder().encode(self.body)
             return InvokeModelInput(
                 accept: self.accept.headerValue,

Sources/Modalities/CrossRegionInference.swift

@@ -15,10 +15,10 @@
 
 public protocol CrossRegionInferenceModality: Sendable {}
 extension CrossRegionInferenceModality {
-	public func crossRegionPrefix(forRegion region: Region) -> String {
-		if region.isEURegion() { return "eu." }
-		if region.isUSRegion() { return "us." }
-		if region.isAPRegion() { return "ap." }
-		return ""
-	}
+    public func crossRegionPrefix(forRegion region: Region) -> String {
+        if region.isEURegion() { return "eu." }
+        if region.isUSRegion() { return "us." }
+        if region.isAPRegion() { return "ap." }
+        return ""
+    }
 }

Sources/Models/Mistral/MistralBedrockModels.swift

@@ -21,7 +21,7 @@ import Foundation
 
 typealias MistralConverse = StandardConverse
 
-// TODO: define a new struct to support Pixtral Large 25 + cross region inference 
+// TODO: define a new struct to support Pixtral Large 25 + cross region inference
 
 extension BedrockModel {
     public static let mistral_large_2402 = BedrockModel(

Sources/Region.swift

@@ -309,12 +309,12 @@ extension Region {
 // https://docs.aws.amazon.com/bedrock/latest/userguide/inference-profiles-support.html
 extension Region {
     public func isUSRegion() -> Bool {
-        return self.rawValue.hasPrefix("us-") 
+        self.rawValue.hasPrefix("us-")
     }
     public func isEURegion() -> Bool {
-        return self.rawValue.hasPrefix("eu-") 
+        self.rawValue.hasPrefix("eu-")
     }
     public func isAPRegion() -> Bool {
-        return self.rawValue.hasPrefix("ap-") 
+        self.rawValue.hasPrefix("ap-")
     }
-}
+}

Tests/Mock/MockBedrockRuntimeClient.swift

@@ -14,11 +14,9 @@
 //===----------------------------------------------------------------------===//
 
 @preconcurrency import AWSBedrockRuntime
-
-import Testing 
-
 import BedrockService
 import Foundation
+import Testing
 
 public struct MockBedrockRuntimeClient: BedrockRuntimeClientProtocol {
     public init() {}