Support pagination parameters for batches.list

Author: marckohlbrugge

lib/openai/batches.rb

@@ -4,8 +4,8 @@ def initialize(client:)
       @client = client.beta(assistants: OpenAI::Assistants::BETA_VERSION)
     end
 
-    def list
-      @client.get(path: "/batches")
+    def list(parameters: {})
+      @client.get(path: "/batches", parameters: parameters)
     end
 
     def retrieve(id:)

spec/openai/client/batches_spec.rb

@@ -24,7 +24,17 @@
     describe "#list", :vcr do
       let(:input_cassette) { "batches list input" }
       let(:cassette) { "batches list" }
+      let(:limit) { 10 }
       let(:response) { OpenAI::Client.new.batches.list }
+      let(:existing_batch) do
+        OpenAI::Client.new.batches.create(
+          parameters: {
+            input_file_id: input_file_id,
+            endpoint: "/v1/chat/completions",
+            completion_window: "24h"
+          }
+        )
+      end
 
       before { batch_id }
 
@@ -33,6 +43,18 @@
           expect(response.dig("data", 0, "object")).to eq("batch")
         end
       end
+
+      it "supports after and limit parameters" do
+        VCR.use_cassette(cassette) do
+          response = OpenAI::Client.new.batches.list(parameters: {
+                                                       after: existing_batch["id"],
+                                                       limit: limit
+                                                     })
+
+          expect(response.dig("data", 0, "object")).to eq("batch")
+          expect(response["data"].length).to be <= limit
+        end
+      end
     end
 
     describe "#retrieve" do