diff options
Diffstat (limited to 'lib/dispatch/adapter/response.rb')
| -rw-r--r-- | lib/dispatch/adapter/response.rb | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/lib/dispatch/adapter/response.rb b/lib/dispatch/adapter/response.rb new file mode 100644 index 0000000..d3e4789 --- /dev/null +++ b/lib/dispatch/adapter/response.rb @@ -0,0 +1,23 @@ +# frozen_string_literal: true + +module Dispatch + module Adapter + Response = Struct.new(:content, :tool_calls, :model, :stop_reason, :usage, keyword_init: true) do + def initialize(content: nil, tool_calls: [], model:, stop_reason:, usage:) + super(content:, tool_calls:, model:, stop_reason:, usage:) + end + end + + Usage = Struct.new(:input_tokens, :output_tokens, :cache_read_tokens, :cache_creation_tokens, keyword_init: true) do + def initialize(input_tokens:, output_tokens:, cache_read_tokens: 0, cache_creation_tokens: 0) + super(input_tokens:, output_tokens:, cache_read_tokens:, cache_creation_tokens:) + end + end + + StreamDelta = Struct.new(:type, :text, :tool_call_id, :tool_name, :argument_delta, keyword_init: true) do + def initialize(type:, text: nil, tool_call_id: nil, tool_name: nil, argument_delta: nil) + super(type:, text:, tool_call_id:, tool_name:, argument_delta:) + end + end + end +end |
