Skip to content

Commit e0febca

Browse files
committed
Decode Float and Double more leniently
1 parent 5cafb19 commit e0febca

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

FirebaseAI/Sources/Types/Public/Generable/Generable.swift

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -118,11 +118,10 @@ extension Float: Generable {
118118
}
119119

120120
public init(_ content: ModelOutput) throws {
121-
// TODO: Determine if we need to use `exactly: ` or be more lenient.
122-
guard case let .number(value) = content.kind, let float = Float(exactly: value) else {
121+
guard case let .number(value) = content.kind else {
123122
throw Self.decodingFailure(content)
124123
}
125-
self = float
124+
self = Float(value)
126125
}
127126

128127
public var modelOutput: ModelOutput {
@@ -137,11 +136,10 @@ extension Double: Generable {
137136
}
138137

139138
public init(_ content: ModelOutput) throws {
140-
// TODO: Determine if we need to use `exactly: ` or be more lenient.
141-
guard case let .number(value) = content.kind, let double = Double(exactly: value) else {
139+
guard case let .number(value) = content.kind else {
142140
throw Self.decodingFailure(content)
143141
}
144-
self = double
142+
self = value
145143
}
146144

147145
public var modelOutput: ModelOutput {

0 commit comments

Comments
 (0)