ruby_llm 1.6.1 → 1.6.3

data/lib/tasks/models_docs.rake

@@ -6,12 +6,8 @@ require 'fileutils'
 namespace :models do
   desc 'Generate available models documentation'
   task :docs do
-    FileUtils.mkdir_p('docs') # ensure output directory exists
-
-    # Generate markdown content
+    FileUtils.mkdir_p('docs')
     output = generate_models_markdown
-
-    # Write the output
     File.write('docs/_reference/available-models.md', output)
     puts 'Generated docs/_reference/available-models.md'
   end
@@ -86,7 +82,7 @@ def generate_models_markdown
 end
 
 def generate_provider_sections
-  RubyLLM::Provider.providers.map do |provider, provider_class|
+  RubyLLM::Provider.providers.filter_map do |provider, provider_class|
     models = RubyLLM.models.by_provider(provider)
     next if models.none?
 
@@ -95,7 +91,7 @@ def generate_provider_sections
 
       #{models_table(models)}
     PROVIDER
-  end.compact.join("\n\n")
+  end.join("\n\n")
 end
 
 def generate_capability_sections
@@ -107,7 +103,7 @@ def generate_capability_sections
     'Batch Processing' => RubyLLM.models.select { |m| m.capabilities.include?('batch') }
   }
 
-  capabilities.map do |capability, models|
+  capabilities.filter_map do |capability, models|
     next if models.none?
 
     <<~CAPABILITY
@@ -115,13 +111,12 @@ def generate_capability_sections
 
       #{models_table(models)}
     CAPABILITY
-  end.compact.join("\n\n")
+  end.join("\n\n")
 end
 
 def generate_modality_sections # rubocop:disable Metrics/PerceivedComplexity
   sections = []
 
-  # Models that support vision/images
   vision_models = RubyLLM.models.select { |m| (m.modalities.input || []).include?('image') }
   if vision_models.any?
     sections << <<~SECTION
@@ -133,7 +128,6 @@ def generate_modality_sections # rubocop:disable Metrics/PerceivedComplexity
     SECTION
   end
 
-  # Models that support audio
   audio_models = RubyLLM.models.select { |m| (m.modalities.input || []).include?('audio') }
   if audio_models.any?
     sections << <<~SECTION
@@ -145,7 +139,6 @@ def generate_modality_sections # rubocop:disable Metrics/PerceivedComplexity
     SECTION
   end
 
-  # Models that support PDFs
   pdf_models = RubyLLM.models.select { |m| (m.modalities.input || []).include?('pdf') }
   if pdf_models.any?
     sections << <<~SECTION
@@ -157,7 +150,6 @@ def generate_modality_sections # rubocop:disable Metrics/PerceivedComplexity
     SECTION
   end
 
-  # Models for embeddings
   embedding_models = RubyLLM.models.select { |m| (m.modalities.output || []).include?('embeddings') }
   if embedding_models.any?
     sections << <<~SECTION
@@ -179,7 +171,6 @@ def models_table(models)
   alignment = [':--', ':--', '--:', '--:', ':--']
 
   rows = models.sort_by { |m| [m.provider, m.name] }.map do |model|
-    # Format pricing information
     pricing = standard_pricing_display(model)
 
     [
@@ -203,7 +194,6 @@ def models_table(models)
 end
 
 def standard_pricing_display(model)
-  # Access pricing data using to_h to get the raw hash
   pricing_data = model.pricing.to_h[:text_tokens]&.dig(:standard) || {}
 
   if pricing_data.any?

data/lib/tasks/models_update.rake

@@ -61,7 +61,7 @@ def refresh_models
 end
 
 def validate_models!(models)
-  schema_path = File.expand_path('../ruby_llm/models_schema.json', __dir__)
+  schema_path = RubyLLM::Models.schema_file
   models_data = models.all.map(&:to_h)
 
   validation_errors = JSON::Validator.fully_validate(schema_path, models_data)

data/lib/tasks/vcr.rake

@@ -2,9 +2,7 @@
 
 require 'dotenv/load'
 
-# Helper functions at the top level
 def record_all_cassettes(cassette_dir)
-  # Re-record all cassettes
   FileUtils.rm_rf(cassette_dir)
   FileUtils.mkdir_p(cassette_dir)
 
@@ -14,10 +12,8 @@ def record_all_cassettes(cassette_dir)
 end
 
 def record_for_providers(providers, cassette_dir)
-  # Get the list of available providers from RubyLLM itself
   all_providers = RubyLLM::Provider.providers.keys.map(&:to_s)
 
-  # Check for valid providers
   if providers.empty?
     puts "Please specify providers or 'all'. Example: rake vcr:record[openai,anthropic]"
     puts "Available providers: #{all_providers.join(', ')}"
@@ -31,7 +27,6 @@ def record_for_providers(providers, cassette_dir)
     return
   end
 
-  # Find and delete matching cassettes
   cassettes_to_delete = find_matching_cassettes(cassette_dir, providers)
 
   if cassettes_to_delete.empty?
@@ -54,9 +49,7 @@ def find_matching_cassettes(dir, providers)
   Dir.glob("#{dir}/**/*.yml").each do |file|
     basename = File.basename(file)
 
-    # Precise matching to avoid cross-provider confusion
     providers.each do |provider|
-      # Match only exact provider prefixes
       next unless basename =~ /^[^_]*_#{provider}_/ || # For first section like "chat_openai_"
                   basename =~ /_#{provider}_[^_]+_/    # For middle sections like "_openai_gpt4_"
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 1.6.1
+  version: 1.6.3
 platform: ruby
 authors:
 - Carmine Paolino