class Riffer::Providers::Mock

# File lib/riffer/providers/mock.rb, line 14
def initialize(**options)
  @responses = options[:responses] || []
  @current_index = 0
  @calls = []
  @stubbed_responses = []
end

# File lib/riffer/providers/mock.rb, line 45
def clear_stubs
  @stubbed_responses = []
end

# File lib/riffer/providers/mock.rb, line 30
def stub_response(content, tool_calls: [], token_usage: nil)
  formatted_tool_calls = tool_calls.map.with_index do |tc, idx|
    Riffer::Messages::Assistant::ToolCall.new(
      id: tc[:id] || "mock_id_#{idx}",
      call_id: tc[:call_id] || tc[:id] || "mock_call_#{idx}",
      name: tc[:name],
      arguments: tc[:arguments].is_a?(String) ? tc[:arguments] : tc[:arguments].to_json
    )
  end
  @stubbed_responses << {role: "assistant", content: content, tool_calls: formatted_tool_calls, token_usage: token_usage}
end