diff options
Diffstat (limited to 'spec/dispatch/adapter/structs_spec.rb')
| -rw-r--r-- | spec/dispatch/adapter/structs_spec.rb | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/spec/dispatch/adapter/structs_spec.rb b/spec/dispatch/adapter/structs_spec.rb index 07c1198..ef8ec73 100644 --- a/spec/dispatch/adapter/structs_spec.rb +++ b/spec/dispatch/adapter/structs_spec.rb @@ -154,6 +154,32 @@ RSpec.describe Dispatch::Adapter do expect(info.supports_vision).to be(false) expect(info.supports_tool_use).to be(true) expect(info.supports_streaming).to be(true) + expect(info.premium_request_multiplier).to be_nil + end + + it "accepts premium_request_multiplier" do + info = Dispatch::Adapter::ModelInfo.new( + id: "o3", + name: "o3", + max_context_tokens: 200_000, + supports_vision: false, + supports_tool_use: true, + supports_streaming: true, + premium_request_multiplier: 30.0 + ) + expect(info.premium_request_multiplier).to eq(30.0) + end + + it "defaults premium_request_multiplier to nil" do + info = Dispatch::Adapter::ModelInfo.new( + id: "gpt-4.1-nano", + name: "GPT 4.1 Nano", + max_context_tokens: 1_047_576, + supports_vision: false, + supports_tool_use: true, + supports_streaming: true + ) + expect(info.premium_request_multiplier).to be_nil end end |
