ruby_llm 1.5.1 → 1.6.1

data/lib/ruby_llm/providers/perplexity/capabilities.rb

@@ -2,14 +2,11 @@
 
 module RubyLLM
   module Providers
-    module Perplexity
+    class Perplexity
       # Determines capabilities and pricing for Perplexity models
       module Capabilities
         module_function
 
-        # Returns the context window size for the given model ID
-        # @param model_id [String] the model identifier
-        # @return [Integer] the context window size in tokens
         def context_window_for(model_id)
           case model_id
           when /sonar-pro/ then 200_000
@@ -17,9 +14,6 @@ module RubyLLM
           end
         end
 
-        # Returns the maximum number of tokens that can be generated
-        # @param model_id [String] the model identifier
-        # @return [Integer] the maximum number of tokens
         def max_tokens_for(model_id)
           case model_id
           when /sonar-(?:pro|reasoning-pro)/ then 8_192
@@ -27,23 +21,14 @@ module RubyLLM
           end
         end
 
-        # Returns the price per million tokens for input
-        # @param model_id [String] the model identifier
-        # @return [Float] the price per million tokens in USD
         def input_price_for(model_id)
           PRICES.dig(model_family(model_id), :input) || 1.0
         end
 
-        # Returns the price per million tokens for output
-        # @param model_id [String] the model identifier
-        # @return [Float] the price per million tokens in USD
         def output_price_for(model_id)
           PRICES.dig(model_family(model_id), :output) || 1.0
         end
 
-        # Determines if the model supports vision capabilities
-        # @param model_id [String] the model identifier
-        # @return [Boolean] true if the model supports vision
         def supports_vision?(model_id)
           case model_id
           when /sonar-reasoning-pro/, /sonar-reasoning/, /sonar-pro/, /sonar/ then true
@@ -51,21 +36,14 @@ module RubyLLM
           end
         end
 
-        # Determines if the model supports function calling
-        # @param model_id [String] the model identifier
-        # @return [Boolean] true if the model supports functions
         def supports_functions?(_model_id)
           false
         end
 
-        # Determines if the model supports JSON mode
         def supports_json_mode?(_model_id)
           true
         end
 
-        # Formats the model ID into a human-readable display name
-        # @param model_id [String] the model identifier
-        # @return [String] the formatted display name
         def format_display_name(model_id)
           case model_id
           when 'sonar' then 'Sonar'
@@ -80,16 +58,10 @@ module RubyLLM
           end
         end
 
-        # Returns the model type
-        # @param model_id [String] the model identifier
-        # @return [String] the model type (e.g., 'chat')
         def model_type(_model_id)
           'chat'
         end
 
-        # Returns the model family
-        # @param model_id [String] the model identifier
-        # @return [Symbol] the model family
         def model_family(model_id)
           case model_id
           when 'sonar' then :sonar
@@ -123,7 +95,6 @@ module RubyLLM
             output_per_million: prices[:output]
           }
 
-          # Add special pricing if available
           standard_pricing[:citation_per_million] = prices[:citation] if prices[:citation]
           standard_pricing[:reasoning_per_million] = prices[:reasoning] if prices[:reasoning]
           standard_pricing[:search_per_thousand] = prices[:search_queries] if prices[:search_queries]

data/lib/ruby_llm/providers/perplexity/chat.rb

@@ -2,7 +2,7 @@
 
 module RubyLLM
   module Providers
-    module Perplexity
+    class Perplexity
       # Chat formatting for Perplexity provider
       module Chat
         module_function

data/lib/ruby_llm/providers/perplexity/models.rb

@@ -2,7 +2,7 @@
 
 module RubyLLM
   module Providers
-    module Perplexity
+    class Perplexity
       # Models methods of the Perplexity API integration
       module Models
         def list_models(**)

data/lib/ruby_llm/providers/perplexity.rb

@@ -3,34 +3,29 @@
 module RubyLLM
   module Providers
     # Perplexity API integration.
-    module Perplexity
-      extend OpenAI
-      extend Perplexity::Chat
-      extend Perplexity::Models
+    class Perplexity < OpenAI
+      include Perplexity::Chat
+      include Perplexity::Models
 
-      module_function
-
-      def api_base(_config)
+      def api_base
         'https://api.perplexity.ai'
       end
 
-      def headers(config)
+      def headers
         {
-          'Authorization' => "Bearer #{config.perplexity_api_key}",
+          'Authorization' => "Bearer #{@config.perplexity_api_key}",
           'Content-Type' => 'application/json'
         }
       end
 
-      def capabilities
-        Perplexity::Capabilities
-      end
-
-      def slug
-        'perplexity'
-      end
+      class << self
+        def capabilities
+          Perplexity::Capabilities
+        end
 
-      def configuration_requirements
-        %i[perplexity_api_key]
+        def configuration_requirements
+          %i[perplexity_api_key]
+        end
       end
 
       def parse_error(response)

data/lib/ruby_llm/stream_accumulator.rb

@@ -16,7 +16,7 @@ module RubyLLM
     end
 
     def add(chunk)
-      RubyLLM.logger.debug chunk.inspect
+      RubyLLM.logger.debug chunk.inspect if RubyLLM.config.log_stream_debug
       @model_id ||= chunk.model_id
 
       if chunk.tool_call?
@@ -26,7 +26,7 @@ module RubyLLM
       end
 
       count_tokens chunk
-      RubyLLM.logger.debug inspect
+      RubyLLM.logger.debug inspect if RubyLLM.config.log_stream_debug
     end
 
     def to_message(response)
@@ -62,7 +62,7 @@ module RubyLLM
     end
 
     def accumulate_tool_calls(new_tool_calls)
-      RubyLLM.logger.debug "Accumulating tool calls: #{new_tool_calls}"
+      RubyLLM.logger.debug "Accumulating tool calls: #{new_tool_calls}" if RubyLLM.config.log_stream_debug
       new_tool_calls.each_value do |tool_call|
         if tool_call.id
           tool_call_id = tool_call.id.empty? ? SecureRandom.uuid : tool_call.id

data/lib/ruby_llm/streaming.rb

@@ -8,10 +8,12 @@ module RubyLLM
   module Streaming
     module_function
 
-    def stream_response(connection, payload, &block)
+    def stream_response(connection, payload, additional_headers = {}, &block)
       accumulator = StreamAccumulator.new
 
       response = connection.post stream_url, payload do |req|
+        # Merge additional headers, with existing headers taking precedence
+        req.headers = additional_headers.merge(req.headers) unless additional_headers.empty?
         if req.options.respond_to?(:on_data)
           # Handle Faraday 2.x streaming with on_data method
           req.options.on_data = handle_stream do |chunk|
@@ -27,7 +29,9 @@ module RubyLLM
         end
       end
 
-      accumulator.to_message(response)
+      message = accumulator.to_message(response)
+      RubyLLM.logger.debug "Stream completed: #{message.inspect}"
+      message
     end
 
     def handle_stream(&block)
@@ -56,7 +60,7 @@ module RubyLLM
     end
 
     def process_stream_chunk(chunk, parser, env, &)
-      RubyLLM.logger.debug "Received chunk: #{chunk}"
+      RubyLLM.logger.debug "Received chunk: #{chunk}" if RubyLLM.config.log_stream_debug
 
       if error_chunk?(chunk)
         handle_error_chunk(chunk, env)
@@ -145,5 +149,14 @@ module RubyLLM
     rescue JSON::ParserError => e
       RubyLLM.logger.debug "Failed to parse error event: #{e.message}"
     end
+
+    # Default implementation - providers should override this method
+    def parse_streaming_error(data)
+      error_data = JSON.parse(data)
+      [500, error_data['message'] || 'Unknown streaming error']
+    rescue JSON::ParserError => e
+      RubyLLM.logger.debug "Failed to parse streaming error: #{e.message}"
+      [500, "Failed to parse error: #{data}"]
+    end
   end
 end

data/lib/ruby_llm/tool.rb

@@ -32,6 +32,19 @@ module RubyLLM
   #       end
   #    end
   class Tool
+    # Stops conversation continuation after tool execution
+    class Halt
+      attr_reader :content
+
+      def initialize(content)
+        @content = content
+      end
+
+      def to_s
+        @content.to_s
+      end
+    end
+
     class << self
       def description(text = nil)
         return @description unless text
@@ -77,5 +90,11 @@ module RubyLLM
     def execute(...)
       raise NotImplementedError, 'Subclasses must implement #execute'
     end
+
+    protected
+
+    def halt(message)
+      Halt.new(message)
+    end
   end
 end

data/lib/ruby_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  VERSION = '1.5.1'
+  VERSION = '1.6.1'
 end

data/lib/tasks/models_docs.rake

@@ -12,8 +12,8 @@ namespace :models do
     output = generate_models_markdown
 
     # Write the output
-    File.write('docs/available-models.md', output)
-    puts 'Generated docs/available-models.md'
+    File.write('docs/_reference/available-models.md', output)
+    puts 'Generated docs/_reference/available-models.md'
   end
 end
 
@@ -22,15 +22,16 @@ def generate_models_markdown
     ---
     layout: default
     title: Available Models
-    nav_order: 5
-    permalink: /available-models
+    nav_order: 1
     description: Browse hundreds of AI models from every major provider. Always up-to-date, automatically generated.
+    redirect_from:
+      - /guides/available-models
     ---
 
-    # Available Models
+    # {{ page.title }}
     {: .no_toc }
 
-    Every model, every provider, always current. Your complete AI model reference.
+    {{ page.description }}
     {: .fs-6 .fw-300 }
 
     ## Table of contents
@@ -41,6 +42,13 @@ def generate_models_markdown
 
     ---
 
+    After reading this guide, you will know:
+
+    * How RubyLLM's model registry works and where data comes from
+    * How to find models by provider, capability, or purpose
+    * What information is available for each model
+    * How to use model aliases for simpler configuration
+
     ## How Model Data Works
 
     RubyLLM's model registry combines data from multiple sources:
@@ -78,12 +86,12 @@ def generate_models_markdown
 end
 
 def generate_provider_sections
-  RubyLLM::Provider.providers.keys.map do |provider|
+  RubyLLM::Provider.providers.map do |provider, provider_class|
     models = RubyLLM.models.by_provider(provider)
     next if models.none?
 
     <<~PROVIDER
-      ### #{provider.to_s.capitalize} (#{models.count})
+      ### #{provider_class.name} (#{models.count})
 
       #{models_table(models)}
     PROVIDER
@@ -167,15 +175,14 @@ end
 def models_table(models)
   return '*No models found*' if models.none?
 
-  headers = ['Model', 'ID', 'Provider', 'Context', 'Max Output', 'Standard Pricing (per 1M tokens)']
-  alignment = [':--', ':--', ':--', '--:', '--:', ':--']
+  headers = ['Model', 'Provider', 'Context', 'Max Output', 'Standard Pricing (per 1M tokens)']
+  alignment = [':--', ':--', '--:', '--:', ':--']
 
   rows = models.sort_by { |m| [m.provider, m.name] }.map do |model|
     # Format pricing information
     pricing = standard_pricing_display(model)
 
     [
-      model.name,
       model.id,
       model.provider,
       model.context_window || '-',

data/lib/tasks/models_update.rake

@@ -86,8 +86,8 @@ def display_model_stats
   puts "\nModel count:"
   provider_counts = @models.all.group_by(&:provider).transform_values(&:count)
 
-  RubyLLM::Provider.providers.each_key do |sym|
-    name = sym.to_s.capitalize
+  RubyLLM::Provider.providers.each do |sym, provider_class|
+    name = provider_class.name
     count = provider_counts[sym.to_s] || 0
     status = status(sym)
     puts "  #{name}: #{count} models #{status}"
@@ -97,9 +97,10 @@ def display_model_stats
 end
 
 def status(provider_sym)
-  if RubyLLM::Provider.providers[provider_sym].local?
+  provider_class = RubyLLM::Provider.providers[provider_sym]
+  if provider_class.local?
     ' (LOCAL - SKIP)'
-  elsif RubyLLM::Provider.providers[provider_sym].configured?
+  elsif provider_class.configured?(RubyLLM.config)
     ' (OK)'
   else
     ' (NOT CONFIGURED)'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 1.5.1
+  version: 1.6.1
 platform: ruby
 authors:
 - Carmine Paolino
@@ -121,12 +121,13 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2'
-description: A delightful Ruby way to work with AI. Tired of juggling different SDKs?
-  RubyLLM provides one beautiful, Ruby-like interface for OpenAI, Anthropic, Gemini,
-  Bedrock, OpenRouter, DeepSeek, Ollama, and any OpenAI-compatible API. Chat (with
-  text, images, audio, PDFs), generate images, create embeddings, use tools (function
-  calling), stream responses, and integrate with Rails effortlessly. Minimal dependencies,
-  maximum developer happiness - just clean Ruby code that works.
+description: One beautiful Ruby API for GPT, Claude, Gemini, and more. Easily build
+  chatbots, AI agents, RAG applications, and content generators. Features chat (text,
+  images, audio, PDFs), image generation, embeddings, tools (function calling), structured
+  output, Rails integration, and streaming. Works with OpenAI, Anthropic, Google Gemini,
+  AWS Bedrock, DeepSeek, Mistral, Ollama (local models), OpenRouter, Perplexity, GPUStack,
+  and any OpenAI-compatible API. Minimal dependencies - just Faraday, Zeitwerk, and
+  Marcel.
 email:
 - carmine@paolino.me
 executables: []
@@ -266,5 +267,5 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 rubygems_version: 3.6.9
 specification_version: 4
-summary: A single delightful Ruby way to work with AI.
+summary: One beautiful Ruby API for GPT, Claude, Gemini, and more.
 test_files: []