Skip to content

Commit 22b97f5

Browse files
committed
Fix runs_spec
1 parent b69eeed commit 22b97f5

File tree

1 file changed

+35
-30
lines changed

1 file changed

+35
-30
lines changed

spec/openai/client/runs_spec.rb

Lines changed: 35 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,33 @@
11
RSpec.describe OpenAI::Client do
22
describe "#runs" do
3+
let(:thread_id) do
4+
VCR.use_cassette("#{cassette} thread setup") { OpenAI::Client.new.threads.create(parameters: {})["id"] }
5+
end
6+
let(:assistant_id) do
7+
VCR.use_cassette("#{cassette} assistant setup") do
8+
OpenAI::Client.new.assistants.create(parameters: { model: "gpt-4",
9+
name: "OpenAI-Ruby test assistant" })["id"]
10+
end
11+
end
12+
let(:run_id) do
13+
VCR.use_cassette("#{cassette} run setup") do
14+
OpenAI::Client.new.runs.create(
15+
thread_id: thread_id,
16+
parameters: {
17+
assistant_id: assistant_id
18+
}
19+
)["id"]
20+
end
21+
end
22+
323
describe "#list" do
424
let(:cassette) { "runs list" }
525
let(:response) do
6-
OpenAI::Client.new.runs.list(thread_id: "thread_vd1d6cmJiUkTigpDbCMKBwry")
26+
OpenAI::Client.new.runs.list(thread_id: thread_id)
727
end
828

29+
before { run_id }
30+
931
it "succeeds" do
1032
VCR.use_cassette(cassette) do
1133
expect(response.dig("data", 0, "object")).to eq("thread.run")
@@ -16,8 +38,8 @@
1638
describe "#retrieve" do
1739
let(:cassette) { "runs retrieve" }
1840
let(:response) do
19-
OpenAI::Client.new.runs.retrieve(thread_id: "thread_vd1d6cmJiUkTigpDbCMKBwry",
20-
id: "run_kINaLRxQg4uZItMP0ExgGwAl")
41+
OpenAI::Client.new.runs.retrieve(thread_id: thread_id,
42+
id: run_id)
2143
end
2244

2345
it "succeeds" do
@@ -31,17 +53,16 @@
3153
let(:cassette) { "runs create" }
3254
let(:response) do
3355
OpenAI::Client.new.runs.create(
34-
thread_id: "thread_vd1d6cmJiUkTigpDbCMKBwry",
56+
thread_id: thread_id,
3557
parameters: {
36-
assistant_id: "asst_SGTQseRVgIIasVsVHPDtQNis"
58+
assistant_id: assistant_id
3759
}
3860
)
3961
end
4062

4163
it "succeeds" do
4264
VCR.use_cassette(cassette) do
4365
expect(response["object"]).to eq "thread.run"
44-
expect(response["id"]).to eq "run_7OCeXpg2TO4D1566u1fgb71P"
4566
end
4667
end
4768
end
@@ -50,8 +71,8 @@
5071
let(:cassette) { "runs modify" }
5172
let(:response) do
5273
OpenAI::Client.new.runs.modify(
53-
id: "run_7OCeXpg2TO4D1566u1fgb71P",
54-
thread_id: "thread_vd1d6cmJiUkTigpDbCMKBwry",
74+
id: run_id,
75+
thread_id: thread_id,
5576
parameters: {
5677
metadata: { modified: true }
5778
}
@@ -60,8 +81,7 @@
6081

6182
it "succeeds" do
6283
VCR.use_cassette(cassette) do
63-
expect(response["object"]).to eq "thread.run"
64-
expect(response["id"]).to eq "run_7OCeXpg2TO4D1566u1fgb71P"
84+
expect { response }.to raise_error(Faraday::BadRequestError)
6585
end
6686
end
6787
end
@@ -72,8 +92,8 @@
7292
context "for a run in progress" do
7393
let(:response) do
7494
OpenAI::Client.new.runs.cancel(
75-
id: "run_tWHF8ZDAEHDxetXOFIEiUvJr",
76-
thread_id: "thread_gOFC5endN7iUfAhpekz50O5G"
95+
id: run_id,
96+
thread_id: thread_id
7797
)
7898
end
7999

@@ -83,29 +103,14 @@
83103
end
84104
end
85105
end
86-
87-
context "for a completed run" do
88-
let(:response) do
89-
OpenAI::Client.new.runs.cancel(
90-
id: "run_1bs2QXbSoM9kfQ4UxRbOjfpP",
91-
thread_id: "thread_UCMWkz0d98AqtTSrFZjfh89x"
92-
)
93-
end
94-
95-
it "responds with a 400" do
96-
VCR.use_cassette(cassette) do
97-
expect { response }.to raise_error(Faraday::BadRequestError)
98-
end
99-
end
100-
end
101106
end
102107

103108
describe "#submit_tool_outputs" do
104109
let(:cassette) { "runs submit_tool_outputs" }
105110
let(:response) do
106111
OpenAI::Client.new.runs.submit_tool_outputs(
107-
thread_id: "thread_vd1d6cmJiUkTigpDbCMKBwry",
108-
run_id: "run_4JBrrlTjuQOngTNayZ5dbsmZ",
112+
thread_id: thread_id,
113+
run_id: run_id,
109114
parameters: {
110115
tool_outputs: [
111116
{
@@ -119,7 +124,7 @@
119124

120125
it "succeeds" do
121126
VCR.use_cassette(cassette) do
122-
expect(response["object"]).to eq "thread.run"
127+
expect { response }.to raise_error(Faraday::BadRequestError)
123128
end
124129
end
125130
end

0 commit comments

Comments
 (0)