Remove TemplateInput.Kind since it is equivalent to JSONValue

Author: andrewheard

FirebaseAI/Sources/TemplateInput.swift

@@ -15,41 +15,14 @@
 import Foundation
 
 public struct TemplateInput: Sendable {
-  let kind: Kind
+  let value: JSONValue
 
   public init(_ input: some TemplateInputRepresentable) {
-    self = .init(kind: input.templateInputRepresentation.kind)
+    self = .init(value: input.templateInputRepresentation.value)
   }
 
-  init(kind: Kind) {
-    self.kind = kind
-  }
-
-  enum Kind: Encodable, Sendable {
-    case string(String)
-    case int(Int)
-    case double(Double)
-    case bool(Bool)
-    case array([Kind])
-    case dictionary([String: Kind])
-
-    func encode(to encoder: Encoder) throws {
-      var container = encoder.singleValueContainer()
-      switch self {
-      case let .string(value):
-        try container.encode(value)
-      case let .int(value):
-        try container.encode(value)
-      case let .double(value):
-        try container.encode(value)
-      case let .bool(value):
-        try container.encode(value)
-      case let .array(value):
-        try container.encode(value)
-      case let .dictionary(value):
-        try container.encode(value)
-      }
-    }
+  init(value: JSONValue) {
+    self.value = value
   }
 }
 

FirebaseAI/Sources/Types/Public/TemplateInputRepresentable.swift

@@ -20,36 +20,38 @@ public protocol TemplateInputRepresentable: Encodable, Sendable {
 }
 
 extension String: TemplateInputRepresentable {
-  public var templateInputRepresentation: TemplateInput { TemplateInput(kind: .string(self)) }
+  public var templateInputRepresentation: TemplateInput { TemplateInput(value: .string(self)) }
 }
 
 extension Int: TemplateInputRepresentable {
-  public var templateInputRepresentation: TemplateInput { TemplateInput(kind: .int(self)) }
+  public var templateInputRepresentation: TemplateInput {
+    TemplateInput(value: .number(Double(self)))
+  }
 }
 
 extension Double: TemplateInputRepresentable {
-  public var templateInputRepresentation: TemplateInput { TemplateInput(kind: .double(self)) }
+  public var templateInputRepresentation: TemplateInput { TemplateInput(value: .number(self)) }
 }
 
 extension Float: TemplateInputRepresentable {
   public var templateInputRepresentation: TemplateInput {
-    TemplateInput(kind: .double(Double(self)))
+    TemplateInput(value: .number(Double(self)))
   }
 }
 
 extension Bool: TemplateInputRepresentable {
-  public var templateInputRepresentation: TemplateInput { TemplateInput(kind: .bool(self)) }
+  public var templateInputRepresentation: TemplateInput { TemplateInput(value: .bool(self)) }
 }
 
 extension Array: TemplateInputRepresentable where Element: TemplateInputRepresentable {
   public var templateInputRepresentation: TemplateInput {
-    TemplateInput(kind: .array(map { TemplateInput($0).kind }))
+    TemplateInput(value: .array(map { TemplateInput($0).value }))
   }
 }
 
 extension Dictionary: TemplateInputRepresentable
   where Key == String, Value: TemplateInputRepresentable {
   public var templateInputRepresentation: TemplateInput {
-    TemplateInput(kind: .dictionary(mapValues { TemplateInput($0).kind }))
+    TemplateInput(value: .object(mapValues { TemplateInput($0).value }))
   }
 }