update to support claude

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