[Firebase AI] Conform ModelOutput to ConvertibleToGeneratedContent

Author: andrewheard

FirebaseAI/Sources/Protocols/Public/StructuredOutput/ConvertibleFromModelOutput.swift

@@ -12,7 +12,23 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+#if canImport(FoundationModels)
+public import protocol FoundationModels.ConvertibleFromGeneratedContent
+public import struct FoundationModels.GeneratedContent
+#endif // canImport(FoundationModels)
+
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
 public protocol ConvertibleFromModelOutput {
   init(_ content: ModelOutput) throws
 }
+
+#if canImport(FoundationModels)
+@available(iOS 26.0, macOS 26.0, *)
+@available(tvOS, unavailable)
+@available(watchOS, unavailable)
+public extension ConvertibleFromModelOutput where Self: ConvertibleFromGeneratedContent {
+  init(_ content: ModelOutput) throws {
+    try self.init(try GeneratedContent(content))
+  }
+}
+#endif // canImport(FoundationModels)

FirebaseAI/Sources/Protocols/Public/StructuredOutput/ConvertibleToModelOutput.swift

@@ -12,6 +12,11 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+#if canImport(FoundationModels)
+public import protocol FoundationModels.ConvertibleToGeneratedContent
+public import struct FoundationModels.GeneratedContent
+#endif // canImport(FoundationModels)
+
 #if compiler(>=6.2)
   @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
   public protocol ConvertibleToModelOutput: SendableMetatype {
@@ -23,3 +28,14 @@
     var modelOutput: ModelOutput { get }
   }
 #endif // compiler(>=6.2)
+
+#if canImport(FoundationModels)
+@available(iOS 26.0, macOS 26.0, *)
+@available(tvOS, unavailable)
+@available(watchOS, unavailable)
+public extension ConvertibleToModelOutput where Self: ConvertibleToGeneratedContent {
+  var modelOutput: ModelOutput {
+    self.generatedContent.modelOutput
+  }
+}
+#endif // canImport(FoundationModels)

FirebaseAI/Sources/Types/Public/StructuredOutput/ModelOutput.swift

@@ -13,6 +13,11 @@
 // limitations under the License.
 
 import Foundation
+#if canImport(FoundationModels)
+public import protocol FoundationModels.ConvertibleToGeneratedContent
+public import struct FoundationModels.GenerationID
+public import struct FoundationModels.GeneratedContent
+#endif // canImport(FoundationModels)
 
 @available(iOS 15.0, macOS 12.0, macCatalyst 15.0, tvOS 15.0, watchOS 8.0, *)
 public struct ModelOutput: Sendable, CustomDebugStringConvertible, FirebaseGenerable {
@@ -24,8 +29,9 @@ public struct ModelOutput: Sendable, CustomDebugStringConvertible, FirebaseGener
 
   public var id: ResponseID?
 
-  init(kind: Kind) {
+  init(kind: Kind, id: ResponseID? = nil) {
     self.kind = kind
+    self.id = id
   }
 
   public var debugDescription: String {
@@ -173,3 +179,66 @@ public extension ModelOutput {
     }
   }
 }
+
+#if canImport(FoundationModels)
+@available(iOS 26.0, macOS 26.0, *)
+@available(tvOS, unavailable)
+@available(watchOS, unavailable)
+extension ModelOutput: ConvertibleToGeneratedContent {
+  public var generatedContent: GeneratedContent {
+    let generationID = id?.generationID
+
+    switch kind {
+    case .null:
+      return GeneratedContent(kind: .null, id: generationID)
+    case let .bool(value):
+      return GeneratedContent(kind: .bool(value), id: generationID)
+    case let .number(value):
+      return GeneratedContent(kind: .number(value), id: generationID)
+    case let .string(value):
+      return GeneratedContent(kind: .string(value), id: generationID)
+    case let .array(values):
+      return GeneratedContent(kind: .array(values.map { $0.generatedContent }), id: generationID)
+    case let .structure(properties: properties, orderedKeys: orderedKeys):
+      return GeneratedContent(
+        kind: .structure(
+          properties: properties.mapValues { $0.generatedContent }, orderedKeys: orderedKeys
+        ),
+        id: generationID
+      )
+    }
+  }
+}
+
+@available(iOS 26.0, macOS 26.0, *)
+@available(tvOS, unavailable)
+@available(watchOS, unavailable)
+extension GeneratedContent: ConvertibleToModelOutput {
+  public var modelOutput: ModelOutput {
+    let responseID = self.id.map { ResponseID(generationID: $0) }
+
+    switch self.kind {
+        case .null:
+      return ModelOutput(kind: .null, id: responseID)
+        case let .bool(value):
+      return ModelOutput(kind: .bool(value), id: responseID)
+        case let .number(value):
+      return ModelOutput(kind: .number(value), id: responseID)
+        case let .string(value):
+      return ModelOutput(kind: .string(value), id: responseID)
+        case let .array(values):
+      return ModelOutput(kind: .array(values.map { $0.modelOutput }), id: responseID)
+        case let .structure(properties: properties, orderedKeys: orderedKeys):
+      return ModelOutput(
+            kind: .structure(
+              properties: properties.mapValues { $0.modelOutput }, orderedKeys: orderedKeys
+            ),
+            id: responseID
+          )
+        @unknown default:
+          assertionFailure("Unknown `FoundationModels.GeneratedContent` kind: \(kind)")
+      return ModelOutput(kind: .null, id: responseID)
+        }
+  }
+}
+#endif // canImport(FoundationModels)

FirebaseAI/Sources/Types/Public/StructuredOutput/ResponseID.swift

@@ -50,6 +50,17 @@ public struct ResponseID: Sendable, Hashable {
     init(generationID: GenerationID) {
       identifier = .generationID(generationID)
     }
+
+    @available(iOS 26.0, macOS 26.0, *)
+    @available(tvOS, unavailable)
+    @available(watchOS, unavailable)
+    var generationID: GenerationID? {
+      guard case let .generationID(value) = identifier else {
+        return nil
+      }
+
+      return value as? GenerationID
+    }
   #endif // canImport(FoundationModels)
 }