Add run steps

Author: Haegin

lib/openai.rb

@@ -11,6 +11,7 @@
 require_relative "openai/threads"
 require_relative "openai/messages"
 require_relative "openai/runs"
+require_relative "openai/run_steps"
 require_relative "openai/audio"
 require_relative "openai/version"
 

lib/openai/client.rb

@@ -69,6 +69,10 @@ def runs
       @runs ||= OpenAI::Runs.new(client: self)
     end
 
+    def run_steps
+      @run_steps ||= OpenAI::RunSteps.new(client: self)
+    end
+
     def moderations(parameters: {})
       json_post(path: "/moderations", parameters: parameters)
     end

lib/openai/run_steps.rb

@@ -0,0 +1,15 @@
+module OpenAI
+  class RunSteps
+    def initialize(client:)
+      @client = client.beta(assistants: "v1")
+    end
+
+    def list(thread_id:, run_id:)
+      @client.get(path: "/threads/#{thread_id}/runs/#{run_id}/steps")
+    end
+
+    def retrieve(thread_id:, run_id:, id:)
+      @client.get(path: "/threads/#{thread_id}/runs/#{run_id}/steps/#{id}")
+    end
+  end
+end

spec/fixtures/cassettes/run_steps_list.yml

@@ -0,0 +1,36 @@
+RSpec.describe OpenAI::Client do
+  describe "#run_steps" do
+    describe "#list" do
+      let(:cassette) { "run_steps list" }
+      let(:response) do
+        OpenAI::Client.new.run_steps.list(
+          thread_id: "thread_vd1d6cmJiUkTigpDbCMKBwry",
+          run_id: "run_kINaLRxQg4uZItMP0ExgGwAl"
+        )
+      end
+
+      it "succeeds" do
+        VCR.use_cassette(cassette) do
+          expect(response.dig("data", 0, "object")).to eq("thread.run.step")
+        end
+      end
+    end
+
+    describe "#retrieve" do
+      let(:cassette) { "run_steps retrieve" }
+      let(:response) do
+        OpenAI::Client.new.run_steps.retrieve(
+          thread_id: "thread_vd1d6cmJiUkTigpDbCMKBwry",
+          run_id: "run_kINaLRxQg4uZItMP0ExgGwAl",
+          id: "step_BM4yN3TSI1mm2dbAwHUD0ATS"
+        )
+      end
+
+      it "succeeds" do
+        VCR.use_cassette(cassette) do
+          expect(response["object"]).to eq("thread.run.step")
+        end
+      end
+    end
+  end
+end