ruby_llm 1.5.1 → 1.6.0

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.0'
 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.0
 platform: ruby
 authors:
 - Carmine Paolino