open_router_enhanced 2.1.0 → 2.2.1

data/lib/open_router/client.rb

@@ -4,17 +4,23 @@ require "active_support/core_ext/object/blank"
 require "active_support/core_ext/hash/indifferent_access"
 
 require_relative "http"
+require_relative "callbacks"
+require_relative "parameter_builder"
+require_relative "tool_serializer"
+require_relative "request_handler"
 
 module OpenRouter
   class ServerError < StandardError; end
 
-  # rubocop:disable Metrics/ClassLength
   class Client
     include OpenRouter::HTTP
+    include OpenRouter::Callbacks
+    include OpenRouter::ParameterBuilder
+    include OpenRouter::ToolSerializer
+    include OpenRouter::RequestHandler
 
     attr_reader :callbacks, :usage_tracker, :configuration
 
-    # Initializes the client with optional configurations.
     def initialize(access_token: nil, request_timeout: nil, uri_base: nil, extra_headers: {}, track_usage: true)
       # Build a per-instance configuration to avoid mutating the global singleton,
       # which would cause credential leakage across Client instances in concurrent use.
@@ -26,10 +32,8 @@ module OpenRouter
       @configuration.extra_headers = @configuration.extra_headers.merge(extra_headers) if extra_headers.any?
       yield(@configuration) if block_given?
 
-      # Instance-level tracking of capability warnings to avoid memory leaks
       @capability_warnings_shown = Set.new
 
-      # Initialize callback system
       @callbacks = {
         before_request: [],
         after_response: [],
@@ -39,57 +43,10 @@ module OpenRouter
         on_healing: []
       }
 
-      # Initialize usage tracking
       @track_usage = track_usage
       @usage_tracker = UsageTracker.new if @track_usage
     end
 
-    # Register a callback for a specific event
-    #
-    # @param event [Symbol] The event to register for (:before_request, :after_response, :on_tool_call, :on_error, :on_stream_chunk, :on_healing)
-    # @param block [Proc] The callback to execute
-    # @return [self] Returns self for method chaining
-    #
-    # @example
-    #   client.on(:after_response) do |response|
-    #     puts "Used #{response.total_tokens} tokens"
-    #   end
-    def on(event, &block)
-      unless @callbacks.key?(event)
-        raise ArgumentError, "Invalid event: #{event}. Valid events are: #{@callbacks.keys.join(", ")}"
-      end
-
-      @callbacks[event] << block
-      self
-    end
-
-    # Remove all callbacks for a specific event
-    #
-    # @param event [Symbol] The event to clear callbacks for
-    # @return [self] Returns self for method chaining
-    def clear_callbacks(event = nil)
-      if event
-        @callbacks[event] = [] if @callbacks.key?(event)
-      else
-        @callbacks.each_key { |key| @callbacks[key] = [] }
-      end
-      self
-    end
-
-    # Trigger callbacks for a specific event
-    #
-    # @param event [Symbol] The event to trigger
-    # @param data [Object] Data to pass to the callbacks
-    def trigger_callbacks(event, data = nil)
-      return unless @callbacks[event]
-
-      @callbacks[event].each do |callback|
-        callback.call(data)
-      rescue StandardError => e
-        warn "[OpenRouter] Callback error for #{event}: #{e.message}"
-      end
-    end
-
     # Performs a chat completion request to the OpenRouter API.
     #
     # @param messages [Array<Hash>] Array of message hashes with role and content
@@ -97,88 +54,52 @@ module OpenRouter
     # @param stream [Proc, nil] Optional callable object for streaming
     # @param kwargs [Hash] Additional options (merged with options parameter)
     # @return [Response] The completion response wrapped in a Response object
-    #
-    # @example Simple usage (unchanged)
-    #   client.complete(messages, model: "gpt-4")
-    #
-    # @example With CompletionOptions
-    #   opts = CompletionOptions.new(model: "gpt-4", temperature: 0.7, tools: my_tools)
-    #   client.complete(messages, opts)
-    #
-    # @example Hash options
-    #   client.complete(messages, { model: "gpt-4", temperature: 0.7 })
-    #
-    # @example Options with override
-    #   client.complete(messages, base_opts, temperature: 0.9)
     def complete(messages, options = nil, stream: nil, **kwargs)
       opts = normalize_options(options, kwargs)
       parameters = prepare_base_parameters(messages, opts, stream)
       forced_extraction = configure_tools_and_structured_outputs!(parameters, opts)
-      configure_plugins!(parameters, opts.response_format, stream)
+      # Gate the response-healing plugin on what's actually on the wire — it requires
+      # response_format to be present, so keying off opts.response_format (intent) would
+      # attach it even when no response_format was sent and produce a 400.
+      configure_plugins!(parameters, parameters[:response_format], stream)
       validate_vision_support(opts.model, messages)
 
-      # Trigger before_request callbacks
       trigger_callbacks(:before_request, parameters)
 
       raw_response = execute_request(parameters)
       validate_response!(raw_response, stream)
 
-      response = build_response(raw_response, opts.response_format, forced_extraction)
+      response = build_response(raw_response, opts.response_format, forced_extraction, strict: resolve_strict(opts.strict))
 
-      # Track usage if enabled
       model_for_tracking = opts.model.is_a?(String) ? opts.model : opts.model.first
       @usage_tracker&.track(response, model: model_for_tracking)
 
-      # Trigger after_response callbacks
       trigger_callbacks(:after_response, response)
-
-      # Trigger on_tool_call callbacks if tool calls are present
       trigger_callbacks(:on_tool_call, response.tool_calls) if response.has_tool_calls?
 
       response
     end
 
     # Fetches the list of available models from the OpenRouter API.
-    # @return [Array<Hash>] The list of models.
     def models
       get(path: "/models")["data"]
     end
 
     # Queries the generation stats for a given id.
-    # @param generation_id [String] The generation id returned from a previous request.
-    # @return [Hash] The stats including token counts and cost.
     def query_generation_stats(generation_id)
       response = get(path: "/generation?id=#{generation_id}")
       response["data"]
     end
 
     # Performs a request to the Responses API Beta (/api/v1/responses)
-    # This is an OpenAI-compatible stateless API with support for reasoning.
     #
     # @param input [String, Array] The input text or structured message array
     # @param options [CompletionOptions, Hash, nil] Options object or hash with configuration
     # @param kwargs [Hash] Additional options (merged with options parameter)
     # @return [ResponsesResponse] The response wrapped in a ResponsesResponse object
-    #
-    # @example Basic usage
-    #   response = client.responses("What is 2+2?", model: "openai/o4-mini")
-    #   puts response.content
-    #
-    # @example With reasoning using CompletionOptions
-    #   opts = CompletionOptions.new(
-    #     model: "openai/o4-mini",
-    #     reasoning: { effort: "high" }
-    #   )
-    #   response = client.responses("Solve this step by step: What is 15% of 80?", opts)
-    #   puts response.reasoning_summary
-    #   puts response.content
-    #
-    # @example With kwargs (still works)
-    #   response = client.responses("Question", model: "openai/o4-mini", reasoning: { effort: "high" })
     def responses(input, options = nil, **kwargs)
       opts = normalize_options(options, kwargs)
 
-      # Model is required for Responses API
       if opts.model == "openrouter/auto"
         raise ArgumentError, "model is required for responses API (cannot use default 'openrouter/auto')"
       end
@@ -187,7 +108,6 @@ module OpenRouter
       parameters[:reasoning] = opts.reasoning if opts.reasoning
       parameters[:tools] = serialize_tools_for_responses(opts.tools) if opts.tools?
       parameters[:tool_choice] = opts.tool_choice if opts.tool_choice
-      # Prefer max_completion_tokens over max_tokens (consistent with complete() method)
       parameters[:max_output_tokens] = opts.max_completion_tokens || opts.max_tokens if opts.max_completion_tokens || opts.max_tokens
       parameters[:temperature] = opts.temperature if opts.temperature
       parameters[:top_p] = opts.top_p if opts.top_p
@@ -198,34 +118,14 @@ module OpenRouter
     end
 
     # Create a new ModelSelector for intelligent model selection
-    #
-    # @return [ModelSelector] A new ModelSelector instance
-    # @example
-    #   client = OpenRouter::Client.new
-    #   model = client.select_model.optimize_for(:cost).require(:function_calling).choose
     def select_model
       ModelSelector.new
     end
 
     # Smart completion that automatically selects the best model based on requirements
-    #
-    # @param messages [Array<Hash>] Array of message hashes
-    # @param requirements [Hash] Model selection requirements
-    # @param optimization [Symbol] Optimization strategy (:cost, :performance, :latest, :context)
-    # @param extras [Hash] Additional parameters for the completion request
-    # @return [Response] The completion response
-    # @raise [ModelSelectionError] If no suitable model is found
-    #
-    # @example
-    #   response = client.smart_complete(
-    #     messages: [{ role: "user", content: "Analyze this data" }],
-    #     requirements: { capabilities: [:function_calling], max_input_cost: 0.01 },
-    #     optimization: :cost
-    #   )
     def smart_complete(messages, requirements: {}, optimization: :cost, **extras)
       selector = ModelSelector.new.optimize_for(optimization)
 
-      # Apply requirements using fluent interface
       selector = selector.require(*requirements[:capabilities]) if requirements[:capabilities]
 
       if requirements[:max_cost] || requirements[:max_input_cost]
@@ -241,43 +141,23 @@ module OpenRouter
         case requirements[:providers]
         when Hash
           selector = selector.prefer_providers(*requirements[:providers][:prefer]) if requirements[:providers][:prefer]
-          if requirements[:providers][:require]
-            selector = selector.require_providers(*requirements[:providers][:require])
-          end
+          selector = selector.require_providers(*requirements[:providers][:require]) if requirements[:providers][:require]
           selector = selector.avoid_providers(*requirements[:providers][:avoid]) if requirements[:providers][:avoid]
         when Array
           selector = selector.prefer_providers(*requirements[:providers])
         end
       end
 
-      # Select the best model
       model = selector.choose
       raise ModelSelectionError, "No model found matching requirements: #{requirements}" unless model
 
-      # Perform the completion with the selected model
       complete(messages, model:, **extras)
     end
 
     # Smart completion with automatic fallback to alternative models
-    #
-    # @param messages [Array<Hash>] Array of message hashes
-    # @param requirements [Hash] Model selection requirements
-    # @param optimization [Symbol] Optimization strategy
-    # @param max_retries [Integer] Maximum number of fallback attempts
-    # @param extras [Hash] Additional parameters for the completion request
-    # @return [Response] The completion response
-    # @raise [ModelSelectionError] If all fallback attempts fail
-    #
-    # @example
-    #   response = client.smart_complete_with_fallback(
-    #     messages: [{ role: "user", content: "Hello" }],
-    #     requirements: { capabilities: [:function_calling] },
-    #     max_retries: 3
-    #   )
     def smart_complete_with_fallback(messages, requirements: {}, optimization: :cost, max_retries: 3, **extras)
       selector = ModelSelector.new.optimize_for(optimization)
 
-      # Apply requirements (same logic as smart_complete)
       selector = selector.require(*requirements[:capabilities]) if requirements[:capabilities]
 
       if requirements[:max_cost] || requirements[:max_input_cost]
@@ -293,16 +173,13 @@ module OpenRouter
         case requirements[:providers]
         when Hash
           selector = selector.prefer_providers(*requirements[:providers][:prefer]) if requirements[:providers][:prefer]
-          if requirements[:providers][:require]
-            selector = selector.require_providers(*requirements[:providers][:require])
-          end
+          selector = selector.require_providers(*requirements[:providers][:require]) if requirements[:providers][:require]
           selector = selector.avoid_providers(*requirements[:providers][:avoid]) if requirements[:providers][:avoid]
         when Array
           selector = selector.prefer_providers(*requirements[:providers])
         end
       end
 
-      # Get fallback models
       fallback_models = selector.choose_with_fallbacks(limit: max_retries + 1)
       raise ModelSelectionError, "No models found matching requirements: #{requirements}" if fallback_models.empty?
 
@@ -312,26 +189,25 @@ module OpenRouter
         return complete(messages, model:, **extras)
       rescue StandardError => e
         last_error = e
-        # Continue to next model in fallback list
       end
 
-      # If we get here, all models failed
       raise ModelSelectionError, "All fallback models failed. Last error: #{last_error&.message}"
     end
 
     private
 
-    # Normalize options from various input formats into CompletionOptions
-    #
-    # @param options [CompletionOptions, Hash, nil] Options object or hash
-    # @param kwargs [Hash] Additional keyword arguments
-    # @return [CompletionOptions] Normalized options object
+    # Per-call `strict:` wins; otherwise fall back to the configured default.
+    def resolve_strict(strict)
+      return strict unless strict.nil?
+
+      configuration.structured_output_strict
+    end
+
     def normalize_options(options, kwargs)
       case options
       when CompletionOptions
         kwargs.empty? ? options : options.merge(**kwargs)
       when Hash
-        # Symbolize keys to handle both string and symbol key hashes
         symbolized = options.transform_keys(&:to_sym)
         CompletionOptions.new(**symbolized.merge(kwargs))
       when nil
@@ -340,437 +216,5 @@ module OpenRouter
         raise ArgumentError, "options must be CompletionOptions, Hash, or nil"
       end
     end
-
-    # Prepare the base parameters for the API request
-    #
-    # @param messages [Array<Hash>] Array of message hashes
-    # @param opts [CompletionOptions] Normalized options object
-    # @param stream [Proc, nil] Optional streaming handler
-    # @return [Hash] Parameters hash for the API request
-    def prepare_base_parameters(messages, opts, stream)
-      parameters = { messages: messages.dup }
-
-      configure_model_parameter!(parameters, opts.model)
-      configure_provider_parameter!(parameters, opts)
-      configure_transforms_parameter!(parameters, opts.transforms)
-      configure_plugins_parameter!(parameters, opts.plugins)
-      configure_prediction_parameter!(parameters, opts.prediction)
-      configure_stream_parameter!(parameters, stream)
-      configure_sampling_parameters!(parameters, opts)
-      configure_output_parameters!(parameters, opts)
-      configure_routing_parameters!(parameters, opts)
-
-      # Merge any extras last (allows overriding anything)
-      parameters.merge!(opts.extras || {})
-      parameters
-    end
-
-    # Configure the model parameter (single model or fallback array)
-    def configure_model_parameter!(parameters, model)
-      if model.is_a?(String)
-        parameters[:model] = model
-      elsif model.is_a?(Array)
-        parameters[:models] = model
-        parameters[:route] = "fallback"
-      end
-    end
-
-    # Configure the provider parameter from options
-    #
-    # @param parameters [Hash] Request parameters hash
-    # @param opts [CompletionOptions] Options object
-    def configure_provider_parameter!(parameters, opts)
-      # Full provider config takes precedence over simple providers array
-      if opts.provider && !opts.provider.empty?
-        parameters[:provider] = opts.provider
-      elsif opts.providers.any?
-        parameters[:provider] = { order: opts.providers }
-      end
-
-      # Route parameter for fallback models
-      parameters[:route] = opts.route if opts.route
-    end
-
-    # Configure sampling parameters (temperature, top_p, etc.)
-    #
-    # @param parameters [Hash] Request parameters hash
-    # @param opts [CompletionOptions] Options object
-    def configure_sampling_parameters!(parameters, opts)
-      parameters[:temperature] = opts.temperature if opts.temperature
-      parameters[:top_p] = opts.top_p if opts.top_p
-      parameters[:top_k] = opts.top_k if opts.top_k
-      parameters[:frequency_penalty] = opts.frequency_penalty if opts.frequency_penalty
-      parameters[:presence_penalty] = opts.presence_penalty if opts.presence_penalty
-      parameters[:repetition_penalty] = opts.repetition_penalty if opts.repetition_penalty
-      parameters[:min_p] = opts.min_p if opts.min_p
-      parameters[:top_a] = opts.top_a if opts.top_a
-      parameters[:seed] = opts.seed if opts.seed
-    end
-
-    # Configure output control parameters
-    #
-    # @param parameters [Hash] Request parameters hash
-    # @param opts [CompletionOptions] Options object
-    def configure_output_parameters!(parameters, opts)
-      # Prefer max_completion_tokens over max_tokens if both are set
-      if opts.max_completion_tokens
-        parameters[:max_completion_tokens] = opts.max_completion_tokens
-      elsif opts.max_tokens
-        parameters[:max_tokens] = opts.max_tokens
-      end
-
-      parameters[:stop] = opts.stop if opts.stop
-      parameters[:logprobs] = opts.logprobs unless opts.logprobs.nil?
-      parameters[:top_logprobs] = opts.top_logprobs if opts.top_logprobs
-      parameters[:logit_bias] = opts.logit_bias if opts.logit_bias && !opts.logit_bias.empty?
-      parameters[:parallel_tool_calls] = opts.parallel_tool_calls unless opts.parallel_tool_calls.nil?
-      parameters[:verbosity] = opts.verbosity if opts.verbosity
-    end
-
-    # Configure OpenRouter-specific routing parameters
-    #
-    # @param parameters [Hash] Request parameters hash
-    # @param opts [CompletionOptions] Options object
-    def configure_routing_parameters!(parameters, opts)
-      parameters[:metadata] = opts.metadata if opts.metadata && !opts.metadata.empty?
-      parameters[:user] = opts.user if opts.user
-      parameters[:session_id] = opts.session_id if opts.session_id
-    end
-
-    # Configure the transforms parameter if transforms are specified
-    def configure_transforms_parameter!(parameters, transforms)
-      parameters[:transforms] = transforms if transforms.any?
-    end
-
-    # Configure the plugins parameter if plugins are specified
-    def configure_plugins_parameter!(parameters, plugins)
-      parameters[:plugins] = plugins.dup if plugins.any?
-    end
-
-    # Configure the prediction parameter for latency optimization
-    def configure_prediction_parameter!(parameters, prediction)
-      parameters[:prediction] = prediction if prediction
-    end
-
-    # Configure the stream parameter if streaming is enabled
-    def configure_stream_parameter!(parameters, stream)
-      parameters[:stream] = stream if stream
-    end
-
-    # Auto-add response-healing plugin when using structured outputs (non-streaming only)
-    # This leverages OpenRouter's native JSON healing for better reliability
-    def configure_plugins!(parameters, response_format, stream)
-      return unless should_auto_add_healing?(response_format, stream)
-
-      parameters[:plugins] ||= []
-
-      # Don't duplicate if user already specified response-healing
-      return if parameters[:plugins].any? { |p| p[:id] == "response-healing" || p["id"] == "response-healing" }
-
-      parameters[:plugins] << { id: "response-healing" }
-    end
-
-    # Determine if we should auto-add the response-healing plugin
-    def should_auto_add_healing?(response_format, stream)
-      return false unless configuration.auto_native_healing
-      return false if stream # Response healing doesn't work with streaming
-      return false unless response_format
-
-      # Check if response_format is a structured output type
-      case response_format
-      when OpenRouter::Schema
-        true
-      when Hash
-        type = response_format[:type] || response_format["type"]
-        %w[json_schema json_object].include?(type.to_s)
-      else
-        false
-      end
-    end
-
-    # Configure tools and structured outputs, returning forced_extraction flag
-    #
-    # @param parameters [Hash] Request parameters hash
-    # @param opts [CompletionOptions] Options object
-    # @return [Boolean] Whether forced extraction mode is being used
-    def configure_tools_and_structured_outputs!(parameters, opts)
-      configure_tool_calling!(parameters, opts)
-      configure_structured_outputs!(parameters, opts)
-    end
-
-    # Configure tool calling support
-    #
-    # @param parameters [Hash] Request parameters hash
-    # @param opts [CompletionOptions] Options object
-    def configure_tool_calling!(parameters, opts)
-      return unless opts.tools?
-
-      warn_if_unsupported(opts.model, :function_calling, "tool calling")
-      parameters[:tools] = serialize_tools(opts.tools)
-      parameters[:tool_choice] = opts.tool_choice if opts.tool_choice
-    end
-
-    # Configure structured output support and return forced_extraction flag
-    #
-    # @param parameters [Hash] Request parameters hash
-    # @param opts [CompletionOptions] Options object
-    # @return [Boolean] Whether forced extraction mode is being used
-    def configure_structured_outputs!(parameters, opts)
-      return false unless opts.response_format?
-
-      force_extraction = determine_forced_extraction_mode(opts.model, opts.force_structured_output)
-
-      if force_extraction
-        handle_forced_structured_output!(parameters, opts.model, opts.response_format)
-        true
-      else
-        handle_native_structured_output!(parameters, opts.model, opts.response_format)
-        false
-      end
-    end
-
-    # Determine whether to use forced extraction mode
-    def determine_forced_extraction_mode(model, force_structured_output)
-      return force_structured_output unless force_structured_output.nil?
-
-      if model.is_a?(String) &&
-         model != "openrouter/auto" &&
-         !ModelRegistry.has_capability?(model, :structured_outputs) &&
-         configuration.auto_force_on_unsupported_models
-        warn "[OpenRouter] Model '#{model}' doesn't support native structured outputs. Automatically using forced extraction mode."
-        true
-      else
-        false
-      end
-    end
-
-    # Handle forced structured output mode
-    def handle_forced_structured_output!(parameters, model, response_format)
-      # In strict mode, still validate to ensure user is aware of capability limits
-      warn_if_unsupported(model, :structured_outputs, "structured outputs") if configuration.strict_mode
-      inject_schema_instructions!(parameters[:messages], response_format)
-    end
-
-    # Handle native structured output mode
-    def handle_native_structured_output!(parameters, model, response_format)
-      warn_if_unsupported(model, :structured_outputs, "structured outputs")
-      parameters[:response_format] = serialize_response_format(response_format)
-    end
-
-    # Validate vision support if messages contain images
-    def validate_vision_support(model, messages)
-      warn_if_unsupported(model, :vision, "vision/image processing") if messages_contain_images?(messages)
-    end
-
-    # Execute the HTTP request with comprehensive error handling
-    def execute_request(parameters)
-      post(path: "/chat/completions", parameters: parameters)
-    rescue ConfigurationError => e
-      trigger_callbacks(:on_error, e)
-      raise ServerError, e.message
-    rescue Faraday::Error => e
-      trigger_callbacks(:on_error, e)
-      handle_faraday_error(e)
-    end
-
-    # Handle Faraday errors with specific error message extraction
-    def handle_faraday_error(error)
-      case error
-      when Faraday::UnauthorizedError
-        raise error
-      when Faraday::BadRequestError
-        error_message = extract_error_message(error)
-        raise ServerError, "Bad Request: #{error_message}"
-      when Faraday::ServerError
-        raise ServerError, "Server Error: #{error.message}"
-      else
-        raise ServerError, "Network Error: #{error.message}"
-      end
-    end
-
-    # Extract error message from Faraday error response
-    def extract_error_message(error)
-      return error.message unless error.response&.dig(:body)
-
-      body = error.response[:body]
-
-      if body.is_a?(Hash)
-        body.dig("error", "message") || error.message
-      elsif body.is_a?(String)
-        extract_error_from_json_string(body) || error.message
-      else
-        error.message
-      end
-    end
-
-    # Extract error message from JSON string response
-    def extract_error_from_json_string(json_string)
-      parsed_body = JSON.parse(json_string)
-      parsed_body.dig("error", "message")
-    rescue JSON::ParserError
-      nil
-    end
-
-    # Validate the API response for errors
-    def validate_response!(raw_response, stream)
-      raise ServerError, raw_response.dig("error", "message") if raw_response.presence&.dig("error", "message").present?
-
-      return unless stream.blank? && raw_response.blank?
-
-      raise ServerError, "Empty response from OpenRouter. Might be worth retrying once or twice."
-    end
-
-    # Build and configure the Response object
-    def build_response(raw_response, response_format, forced_extraction)
-      response = Response.new(raw_response, response_format: response_format, forced_extraction: forced_extraction)
-      response.client = self
-      response
-    end
-
-    # Warn if a model is being used with an unsupported capability
-    def warn_if_unsupported(model, capability, feature_name)
-      # Skip warnings for array models (fallbacks) or auto-selection
-      return if model.is_a?(Array) || model == "openrouter/auto"
-
-      return if ModelRegistry.has_capability?(model, capability)
-
-      if configuration.strict_mode
-        raise CapabilityError,
-              "Model '#{model}' does not support #{feature_name} (missing :#{capability} capability). Enable non-strict mode to allow this request."
-      end
-
-      warning_key = "#{model}:#{capability}"
-      return if @capability_warnings_shown.include?(warning_key)
-
-      warn "[OpenRouter Warning] Model '#{model}' may not support #{feature_name} (missing :#{capability} capability). The request will still be attempted."
-      @capability_warnings_shown << warning_key
-    end
-
-    # Check if messages contain image content
-    def messages_contain_images?(messages)
-      messages.any? do |msg|
-        content = msg[:content] || msg["content"]
-        if content.is_a?(Array)
-          content.any? { |part| part.is_a?(Hash) && (part[:type] == "image_url" || part["type"] == "image_url") }
-        else
-          false
-        end
-      end
-    end
-
-    # Serialize tools to the format expected by OpenRouter Chat Completions API
-    # Format: { type: "function", function: { name: ..., parameters: ... } }
-    def serialize_tools(tools)
-      tools.map do |tool|
-        case tool
-        when Tool
-          tool.to_h
-        when Hash
-          tool
-        else
-          raise ArgumentError, "Tools must be Tool objects or hashes"
-        end
-      end
-    end
-
-    # Serialize tools to the flat format expected by Responses API
-    # Format: { type: "function", name: ..., parameters: ... }
-    def serialize_tools_for_responses(tools)
-      tools.map do |tool|
-        tool_hash = case tool
-                    when Tool
-                      tool.to_h
-                    when Hash
-                      tool.transform_keys(&:to_sym)
-                    else
-                      raise ArgumentError, "Tools must be Tool objects or hashes"
-                    end
-
-        # Flatten the nested function structure if present
-        if tool_hash[:function]
-          {
-            type: "function",
-            name: tool_hash[:function][:name],
-            description: tool_hash[:function][:description],
-            parameters: tool_hash[:function][:parameters]
-          }.compact
-        else
-          # Already in flat format
-          tool_hash
-        end
-      end
-    end
-
-    # Serialize response format to the format expected by OpenRouter API
-    def serialize_response_format(response_format)
-      case response_format
-      when Hash
-        if response_format[:json_schema].is_a?(Schema)
-          response_format.merge(json_schema: response_format[:json_schema].to_h)
-        else
-          response_format
-        end
-      when Schema
-        {
-          type: "json_schema",
-          json_schema: response_format.to_h
-        }
-      else
-        response_format
-      end
-    end
-
-    # Inject schema instructions into messages for forced structured output
-    def inject_schema_instructions!(messages, response_format)
-      schema = extract_schema(response_format)
-      return unless schema
-
-      instruction_content = if schema.respond_to?(:get_format_instructions)
-                              schema.get_format_instructions
-                            else
-                              build_schema_instruction(schema)
-                            end
-
-      # Add as system message
-      messages << { role: "system", content: instruction_content }
-    end
-
-    # Extract schema from response_format
-    def extract_schema(response_format)
-      case response_format
-      when Schema
-        response_format
-      when Hash
-        # Handle both Schema objects and raw hash schemas
-        if response_format[:json_schema].is_a?(Schema)
-          response_format[:json_schema]
-        elsif response_format[:json_schema].is_a?(Hash)
-          response_format[:json_schema]
-        else
-          response_format
-        end
-      end
-    end
-
-    # Build schema instruction when schema doesn't have get_format_instructions
-    def build_schema_instruction(schema)
-      schema_json = schema.respond_to?(:to_h) ? schema.to_h.to_json : schema.to_json
-
-      <<~INSTRUCTION
-        You must respond with valid JSON matching this exact schema:
-
-        ```json
-        #{schema_json}
-        ```
-
-        Rules:
-        - Return ONLY the JSON object, no other text
-        - Ensure all required fields are present
-        - Match the exact data types specified
-        - Follow any format constraints (email, date, etc.)
-        - Do not include trailing commas or comments
-      INSTRUCTION
-    end
   end
-  # rubocop:enable Metrics/ClassLength
 end