Update FirebaseVertexAI/Sources/GenerateContentResponse.swift

paulb777 · andrewheard · web-flow · commit 678e3ca0e887 · 2025-04-22T16:30:22.000-07:00
Co-authored-by: Andrew Heard &lt;andrewheard@google.com&gt;
diff --git a/FirebaseVertexAI/Sources/GenerateContentResponse.swift b/FirebaseVertexAI/Sources/GenerateContentResponse.swift
@@ -97,15 +97,7 @@ public struct GenerateContentResponse: Sendable {
       )
       return []
     }
-    let inlineDataParts: [InlineDataPart] = candidate.content.parts.compactMap { part in
-      switch part {
-      case let inlineDataPart as InlineDataPart:
-        return inlineDataPart
-      default:
-        return nil
-      }
-    }
-    return inlineDataParts
+    return candidate.content.parts.compactMap { $0 as? InlineDataPart }
   }
 
   /// Initializer for SwiftUI previews or tests.

Original file line number	Diff line number	Diff line change
`@@ -97,15 +97,7 @@ public struct GenerateContentResponse: Sendable {`
`97`	`97`	`)`
`98`	`98`	`return []`
`99`	`99`	`}`
`100`		`- let inlineDataParts: [InlineDataPart] = candidate.content.parts.compactMap { part in`
`101`		`- switch part {`
`102`		`- case let inlineDataPart as InlineDataPart:`
`103`		`- return inlineDataPart`
`104`		`- default:`
`105`		`- return nil`
`106`		`- }`
`107`		`- }`
`108`		`- return inlineDataParts`
	`100`	`+ return candidate.content.parts.compactMap { $0 as? InlineDataPart }`
`109`	`101`	`}`
`110`	`102`
`111`	`103`	`/// Initializer for SwiftUI previews or tests.`