use SwiftyJSONLines to handle json lines fetched from batche result…

…s api
fumito-ito · Oct 24, 2024 · 2353162 · 2353162
1 parent 3dc3cc2
commit 2353162
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 95 deletions.
diff --git a/.swiftpm/configuration/Package.resolved b/.swiftpm/configuration/Package.resolved
diff --git a/Package.resolved b/Package.resolved
@@ -32,26 +32,35 @@
       "kind" : "remoteSourceControl",
       "location" : "https://github.com/apple/swift-cmark.git",
       "state" : {
-        "revision" : "3bc2f3e25df0cecc5dc269f7ccae65d0f386f06a",
-        "version" : "0.4.0"
+        "revision" : "3ccff77b2dc5b96b77db3da0d68d28068593fa53",
+        "version" : "0.5.0"
       }
     },
     {
       "identity" : "swift-markdown",
       "kind" : "remoteSourceControl",
       "location" : "https://github.com/apple/swift-markdown.git",
       "state" : {
-        "revision" : "4aae40bf6fff5286e0e1672329d17824ce16e081",
-        "version" : "0.4.0"
+        "revision" : "8f79cb175981458a0a27e76cb42fee8e17b1a993",
+        "version" : "0.5.0"
       }
     },
     {
       "identity" : "swift-syntax",
       "kind" : "remoteSourceControl",
       "location" : "https://github.com/apple/swift-syntax.git",
       "state" : {
-        "revision" : "cb53fa1bd3219b0b23ded7dfdd3b2baff266fd25",
-        "version" : "600.0.0"
+        "revision" : "0687f71944021d616d34d922343dcef086855920",
+        "version" : "600.0.1"
+      }
+    },
+    {
+      "identity" : "swiftyjsonlines",
+      "kind" : "remoteSourceControl",
+      "location" : "https://github.com/fumito-ito/SwiftyJSONLines.git",
+      "state" : {
+        "revision" : "ae30455420897b2c45621a88b3ae3f052a8b5f3d",
+        "version" : "0.0.2"
       }
     }
   ],

diff --git a/Package.swift b/Package.swift
@@ -16,15 +16,17 @@ let package = Package(
             targets: ["AnthropicSwiftSDK"]),
     ],
     dependencies: [
-        .package(url: "https://github.com/fumito-ito/FunctionCalling", from: "0.4.0")
+        .package(url: "https://github.com/fumito-ito/FunctionCalling", from: "0.4.0"),
+        .package(url: "https://github.com/fumito-ito/SwiftyJSONLines.git", from: "0.0.2")
     ],
     targets: [
         // Targets are the basic building blocks of a package, defining a module or a test suite.
         // Targets can depend on other targets in this package and products from dependencies.
         .target(
             name: "AnthropicSwiftSDK",
             dependencies: [
-                .product(name: "FunctionCalling", package: "FunctionCalling")
+                .product(name: "FunctionCalling", package: "FunctionCalling"),
+                .product(name: "SwiftyJSONLines", package: "SwiftyJSONLines")
             ]
         ),
         .testTarget(

diff --git a/Sources/AnthropicSwiftSDK/MessageBatches.swift b/Sources/AnthropicSwiftSDK/MessageBatches.swift
@@ -7,6 +7,7 @@
 
 import Foundation
 import FunctionCalling
+import SwiftyJSONLines
 
 public struct MessageBatches {
     private let apiKey: String
@@ -113,7 +114,7 @@ public struct MessageBatches {
             throw AnthropicAPIError(fromHttpStatusCode: httpResponse.statusCode)
         }
 
-        return try anthropicJSONDecoder.decode([BatchResultResponse].self, from: data)
+        return try JSONLines(data: data).toObjects(with: anthropicJSONDecoder)
     }
 
     public func results(streamOf batchId: String) async throws -> AsyncThrowingStream<BatchResultResponse, Error> {