Init numbers as Decimal before encoding

google-gemini · Feb 28, 2024 · 6ab6377 · 6ab6377
1 parent 04021e4
commit 6ab6377
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/Sources/GoogleAI/JSONValue.swift b/Sources/GoogleAI/JSONValue.swift
@@ -75,7 +75,7 @@ extension JSONValue: Encodable {
     case .null:
       try container.encodeNil()
     case let .number(numberValue):
-      try container.encode(numberValue)
+      try container.encode(Decimal(numberValue))
     case let .string(stringValue):
       try container.encode(stringValue)
     case let .bool(boolValue):

diff --git a/Tests/GoogleAITests/JSONValueTests.swift b/Tests/GoogleAITests/JSONValueTests.swift
@@ -21,6 +21,7 @@ final class JSONValueTests: XCTestCase {
 
   let numberKey = "pi"
   let numberValue = 3.14159
+  let numberValueEncoded = "3.14159"
   let stringKey = "hello"
   let stringValue = "Hello, world!"
 
@@ -127,7 +128,10 @@ final class JSONValueTests: XCTestCase {
     let jsonData = try encoder.encode(JSONValue.object(objectValue))
 
     let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
-    XCTAssertEqual(json, "{\"\(stringKey)\":\"\(stringValue)\",\"\(numberKey)\":\(numberValue)}")
+    XCTAssertEqual(
+      json,
+      "{\"\(stringKey)\":\"\(stringValue)\",\"\(numberKey)\":\(numberValueEncoded)}"
+    )
   }
 
   func testEncodeArray() throws {
@@ -136,6 +140,6 @@ final class JSONValueTests: XCTestCase {
     let jsonData = try encoder.encode(JSONValue.array(arrayValue))
 
     let json = try XCTUnwrap(String(data: jsonData, encoding: .utf8))
-    XCTAssertEqual(json, "[null,\(numberValue)]")
+    XCTAssertEqual(json, "[null,\(numberValueEncoded)]")
   }
 }