summaryrefslogtreecommitdiffhomepage
path: root/lib/dispatch/adapter/interface/model_info.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/dispatch/adapter/interface/model_info.rb')
-rw-r--r--lib/dispatch/adapter/interface/model_info.rb14
1 files changed, 13 insertions, 1 deletions
diff --git a/lib/dispatch/adapter/interface/model_info.rb b/lib/dispatch/adapter/interface/model_info.rb
index 8ba2977..29228a8 100644
--- a/lib/dispatch/adapter/interface/model_info.rb
+++ b/lib/dispatch/adapter/interface/model_info.rb
@@ -2,14 +2,26 @@
module Dispatch
module Adapter
+ ModelPricing = Struct.new(
+ :input_per_mtok, :output_per_mtok,
+ :cache_read_per_mtok, :cache_write_per_mtok,
+ keyword_init: true
+ ) do
+ def initialize(input_per_mtok:, output_per_mtok:,
+ cache_read_per_mtok: 0.0, cache_write_per_mtok: 0.0)
+ super
+ end
+ end
+
ModelInfo = Struct.new(
:id, :name, :max_context_tokens,
:supports_vision, :supports_tool_use, :supports_streaming,
:premium_request_multiplier,
+ :pricing,
keyword_init: true
) do
def initialize(id:, name:, max_context_tokens:, supports_vision:, supports_tool_use:, supports_streaming:,
- premium_request_multiplier: nil)
+ premium_request_multiplier: nil, pricing: nil)
super
end
end