ruby_llm 1.2.0 → 1.3.0

data/lib/tasks/models_update.rake

@@ -0,0 +1,79 @@
+# frozen_string_literal: true
+
+require 'dotenv/load'
+require 'ruby_llm'
+
+task default: ['models:update']
+
+namespace :models do
+  desc 'Update available models from providers (API keys needed)'
+  task :update do
+    puts 'Configuring RubyLLM...'
+    configure_from_env
+
+    refresh_models
+    display_model_stats
+  end
+end
+
+def configure_from_env
+  RubyLLM.configure do |config|
+    config.openai_api_key = ENV.fetch('OPENAI_API_KEY', nil)
+    config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY', nil)
+    config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil)
+    config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil)
+    config.openrouter_api_key = ENV.fetch('OPENROUTER_API_KEY', nil)
+    configure_bedrock(config)
+    config.request_timeout = 30
+  end
+end
+
+def configure_bedrock(config)
+  config.bedrock_api_key = ENV.fetch('AWS_ACCESS_KEY_ID', nil)
+  config.bedrock_secret_key = ENV.fetch('AWS_SECRET_ACCESS_KEY', nil)
+  config.bedrock_region = ENV.fetch('AWS_REGION', nil)
+  config.bedrock_session_token = ENV.fetch('AWS_SESSION_TOKEN', nil)
+end
+
+def refresh_models
+  initial_count = RubyLLM.models.all.size
+  puts "Refreshing models (#{initial_count} cached)..."
+
+  models = RubyLLM.models.refresh!
+
+  if models.all.empty? && initial_count.zero?
+    puts 'Error: Failed to fetch models.'
+    exit(1)
+  elsif models.all.size == initial_count && initial_count.positive?
+    puts 'Warning: Model list unchanged.'
+  else
+    puts "Saving models.json (#{models.all.size} models)"
+    models.save_models
+  end
+
+  @models = models
+end
+
+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
+    count = provider_counts[sym.to_s] || 0
+    status = status(sym)
+    puts "  #{name}: #{count} models #{status}"
+  end
+
+  puts 'Refresh complete.'
+end
+
+def status(provider_sym)
+  if RubyLLM::Provider.providers[provider_sym].local?
+    ' (LOCAL - SKIP)'
+  elsif RubyLLM::Provider.providers[provider_sym].configured?
+    ' (OK)'
+  else
+    ' (NOT CONFIGURED)'
+  end
+end

data/lib/tasks/release.rake

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+namespace :release do
+  desc 'Verify cassettes are fresh enough for release'
+  task :verify_cassettes do
+    max_age_days = 1
+    cassette_dir = 'spec/fixtures/vcr_cassettes'
+    stale_cassettes = []
+
+    Dir.glob("#{cassette_dir}/**/*.yml").each do |cassette|
+      age_days = (Time.now - File.mtime(cassette)) / 86_400
+
+      next unless age_days > max_age_days
+
+      stale_cassettes << {
+        file: File.basename(cassette),
+        age: age_days.round(1)
+      }
+    end
+
+    if stale_cassettes.any?
+      puts "\n❌ Found stale cassettes (older than #{max_age_days} days):"
+      stale_cassettes.each do |c|
+        puts "   - #{c[:file]} (#{c[:age]} days old)"
+      end
+      puts "\nRun locally: bundle exec rspec"
+      exit 1
+    else
+      puts "✅ All cassettes are fresh (< #{max_age_days} days old)"
+    end
+  end
+end

data/lib/tasks/vcr.rake

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'dotenv/load'
+
 # Helper functions at the top level
 def record_all_cassettes(cassette_dir)
   # Re-record all cassettes
@@ -11,7 +13,7 @@ def record_all_cassettes(cassette_dir)
   puts 'Done recording. Please review the new cassettes.'
 end
 
-def record_for_providers(providers, cassette_dir) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
+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)
 
@@ -46,7 +48,7 @@ def record_for_providers(providers, cassette_dir) # rubocop:disable Metrics/AbcS
   puts 'Please review the updated cassettes for sensitive information.'
 end
 
-def find_matching_cassettes(dir, providers) # rubocop:disable Metrics/MethodLength
+def find_matching_cassettes(dir, providers)
   cassettes = []
 
   Dir.glob("#{dir}/**/*.yml").each do |file|

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 1.2.0
+  version: 1.3.0
 platform: ruby
 authors:
 - Carmine Paolino
-autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-04-17 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: base64
@@ -42,58 +41,72 @@ dependencies:
   name: faraday
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '2'
+        version: 1.10.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '2'
+        version: 1.10.0
 - !ruby/object:Gem::Dependency
   name: faraday-multipart
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '1'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
         version: '1'
 - !ruby/object:Gem::Dependency
   name: faraday-net_http
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '3'
+        version: '1'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '3'
+        version: '1'
 - !ruby/object:Gem::Dependency
   name: faraday-retry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: marcel
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2'
+        version: '1.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2'
+        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: zeitwerk
   requirement: !ruby/object:Gem::Requirement
@@ -108,11 +121,12 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2'
-description: A delightful Ruby way to work with AI. Chat in text, analyze and generate
-  images, understand audio, and use tools through a unified interface to OpenAI, Anthropic,
-  Google, AWS Bedrock Anthropic, and DeepSeek. Built for developer happiness with
-  automatic token counting, proper streaming, and Rails integration. No wrapping your
-  head around multiple APIs - just clean Ruby code that works.
+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.
 email:
 - carmine@paolino.me
 executables: []
@@ -125,15 +139,24 @@ files:
 - lib/ruby_llm/active_record/acts_as.rb
 - lib/ruby_llm/aliases.json
 - lib/ruby_llm/aliases.rb
+- lib/ruby_llm/attachment.rb
 - lib/ruby_llm/chat.rb
 - lib/ruby_llm/chunk.rb
 - lib/ruby_llm/configuration.rb
+- lib/ruby_llm/connection.rb
 - lib/ruby_llm/content.rb
+- lib/ruby_llm/context.rb
 - lib/ruby_llm/embedding.rb
 - lib/ruby_llm/error.rb
 - lib/ruby_llm/image.rb
 - lib/ruby_llm/message.rb
-- lib/ruby_llm/model_info.rb
+- lib/ruby_llm/mime_type.rb
+- lib/ruby_llm/model.rb
+- lib/ruby_llm/model/info.rb
+- lib/ruby_llm/model/modalities.rb
+- lib/ruby_llm/model/pricing.rb
+- lib/ruby_llm/model/pricing_category.rb
+- lib/ruby_llm/model/pricing_tier.rb
 - lib/ruby_llm/models.json
 - lib/ruby_llm/models.rb
 - lib/ruby_llm/provider.rb
@@ -148,6 +171,7 @@ files:
 - lib/ruby_llm/providers/bedrock.rb
 - lib/ruby_llm/providers/bedrock/capabilities.rb
 - lib/ruby_llm/providers/bedrock/chat.rb
+- lib/ruby_llm/providers/bedrock/media.rb
 - lib/ruby_llm/providers/bedrock/models.rb
 - lib/ruby_llm/providers/bedrock/signing.rb
 - lib/ruby_llm/providers/bedrock/streaming.rb
@@ -168,6 +192,9 @@ files:
 - lib/ruby_llm/providers/gemini/models.rb
 - lib/ruby_llm/providers/gemini/streaming.rb
 - lib/ruby_llm/providers/gemini/tools.rb
+- lib/ruby_llm/providers/ollama.rb
+- lib/ruby_llm/providers/ollama/chat.rb
+- lib/ruby_llm/providers/ollama/media.rb
 - lib/ruby_llm/providers/openai.rb
 - lib/ruby_llm/providers/openai/capabilities.rb
 - lib/ruby_llm/providers/openai/chat.rb
@@ -177,20 +204,19 @@ files:
 - lib/ruby_llm/providers/openai/models.rb
 - lib/ruby_llm/providers/openai/streaming.rb
 - lib/ruby_llm/providers/openai/tools.rb
+- lib/ruby_llm/providers/openrouter.rb
+- lib/ruby_llm/providers/openrouter/models.rb
 - lib/ruby_llm/railtie.rb
 - lib/ruby_llm/stream_accumulator.rb
 - lib/ruby_llm/streaming.rb
 - lib/ruby_llm/tool.rb
 - lib/ruby_llm/tool_call.rb
+- lib/ruby_llm/utils.rb
 - lib/ruby_llm/version.rb
-- lib/tasks/browser_helper.rb
-- lib/tasks/capability_generator.rb
-- lib/tasks/capability_scraper.rb
-- lib/tasks/cli_helper.rb
-- lib/tasks/code_validator.rb
-- lib/tasks/model_updater.rb
-- lib/tasks/models.rake
+- lib/tasks/aliases.rake
 - lib/tasks/models_docs.rake
+- lib/tasks/models_update.rake
+- lib/tasks/release.rake
 - lib/tasks/vcr.rake
 homepage: https://rubyllm.com
 licenses:
@@ -202,7 +228,6 @@ metadata:
   documentation_uri: https://rubyllm.com
   bug_tracker_uri: https://github.com/crmne/ruby_llm/issues
   rubygems_mfa_required: 'true'
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -217,8 +242,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.22
-signing_key:
+rubygems_version: 3.6.7
 specification_version: 4
-summary: Beautiful Ruby interface to modern AI
+summary: A single delightful Ruby way to work with AI.
 test_files: []

data/lib/ruby_llm/model_info.rb

@@ -1,56 +0,0 @@
-# frozen_string_literal: true
-
-require 'time'
-
-module RubyLLM
-  # Information about an AI model's capabilities, pricing, and metadata.
-  # Used by the Models registry to help developers choose the right model
-  # for their needs.
-  #
-  # Example:
-  #   model = RubyLLM.models.find('gpt-4')
-  #   model.supports_vision?          # => true
-  #   model.supports_functions?       # => true
-  #   model.input_price_per_million   # => 30.0
-  class ModelInfo
-    attr_reader :id, :created_at, :display_name, :provider, :metadata,
-                :context_window, :max_tokens, :supports_vision, :supports_functions,
-                :supports_json_mode, :input_price_per_million, :output_price_per_million, :type, :family
-
-    def initialize(data) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
-      @id = data[:id]
-      @created_at = data[:created_at].is_a?(String) ? Time.parse(data[:created_at]) : data[:created_at]
-      @display_name = data[:display_name]
-      @provider = data[:provider]
-      @context_window = data[:context_window]
-      @max_tokens = data[:max_tokens]
-      @type = data[:type]
-      @family = data[:family]
-      @supports_vision = data[:supports_vision]
-      @supports_functions = data[:supports_functions]
-      @supports_json_mode = data[:supports_json_mode]
-      @input_price_per_million = data[:input_price_per_million]
-      @output_price_per_million = data[:output_price_per_million]
-      @metadata = data[:metadata] || {}
-    end
-
-    def to_h # rubocop:disable Metrics/MethodLength
-      {
-        id: id,
-        created_at: created_at&.iso8601,
-        display_name: display_name,
-        provider: provider,
-        context_window: context_window,
-        max_tokens: max_tokens,
-        type: type,
-        family: family,
-        supports_vision: supports_vision,
-        supports_functions: supports_functions,
-        supports_json_mode: supports_json_mode,
-        input_price_per_million: input_price_per_million,
-        output_price_per_million: output_price_per_million,
-        metadata: metadata
-      }
-    end
-  end
-end

data/lib/tasks/browser_helper.rb

@@ -1,97 +0,0 @@
-# frozen_string_literal: true
-
-require 'ferrum'
-require_relative 'cli_helper'
-
-class BrowserHelper # rubocop:disable Style/Documentation
-  REALISTIC_USER_AGENT = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36' # rubocop:disable Layout/LineLength
-
-  def initialize
-    @browser = create_browser
-  end
-
-  def goto(url)
-    @browser.goto(url)
-  end
-
-  def current_url
-    @browser.page.url
-  rescue StandardError
-    'N/A'
-  end
-
-  def get_page_content(context = 'current page') # rubocop:disable Metrics/MethodLength
-    puts "    Extracting HTML for #{context}..."
-
-    begin
-      sleep(1.0) # Small delay for page stability
-      html = @browser.body
-
-      if html && !html.empty?
-        puts "      Extracted ~#{html.length} chars of HTML"
-        puts '      WARNING: Challenge page detected' if html.match?(/challenge-platform|Checking site/)
-        html
-      else
-        puts '      Warning: Empty content returned'
-        ''
-      end
-    rescue StandardError => e
-      puts "      Error getting HTML: #{e.class} - #{e.message}"
-      ''
-    end
-  end
-
-  def wait_for_page_load
-    handle_cloudflare_challenge
-  end
-
-  def close
-    puts "\nClosing browser..."
-    @browser.quit
-  rescue StandardError => e
-    puts "  Warning: Error closing browser: #{e.message}"
-  end
-
-  private
-
-  def create_browser
-    puts '  Initializing browser for manual interaction...'
-
-    Ferrum::Browser.new(
-      window_size: [1366, 768],
-      headless: false,
-      browser_options: browser_options,
-      timeout: 120,
-      process_timeout: 120,
-      pending_connection_errors: false
-    )
-  end
-
-  def browser_options
-    {
-      'user-agent' => REALISTIC_USER_AGENT,
-      'disable-gpu' => nil,
-      'no-sandbox' => nil,
-      'disable-blink-features' => 'AutomationControlled',
-      'disable-infobars' => nil,
-      'start-maximized' => nil
-    }
-  end
-
-  def handle_cloudflare_challenge # rubocop:disable Metrics/MethodLength
-    puts "\nWaiting for Cloudflare challenge resolution..."
-    puts 'c: Challenge solved'
-    puts 'q: Quit/Skip'
-
-    choice = CliHelper.get_user_choice('Confirm when ready', %w[c q])
-    return false if choice == 'q'
-
-    begin
-      @browser.page.target_id
-      true
-    rescue StandardError
-      puts 'Browser check failed after challenge'
-      false
-    end
-  end
-end

data/lib/tasks/capability_generator.rb

@@ -1,123 +0,0 @@
-# frozen_string_literal: true
-
-require 'ruby_llm'
-require 'json'
-require_relative 'code_validator'
-
-class CapabilityGenerator # rubocop:disable Style/Documentation
-  def initialize(provider_name, docs_html)
-    @provider_name = provider_name
-    @docs_html = docs_html
-    @processed_html = process_html(docs_html)
-  end
-
-  def generate_and_save # rubocop:disable Metrics/MethodLength
-    puts "  Starting code generation for #{@provider_name}..."
-
-    existing_path = File.expand_path("../../lib/ruby_llm/providers/#{@provider_name}/capabilities.rb", __dir__)
-    unless File.exist?(existing_path)
-      puts "  Skipping: No file at #{existing_path}"
-      return
-    end
-
-    existing_code = File.read(existing_path)
-    puts '  Read existing code'
-
-    generated_code = generate_capabilities(existing_code)
-
-    if generated_code
-      puts "  Writing updated code to #{existing_path}..."
-      File.write(existing_path, generated_code)
-      puts "  Updated #{@provider_name}"
-
-      verify_code_with_models_update
-    else
-      puts "  Failed to generate valid code for #{@provider_name}"
-    end
-  end
-
-  private
-
-  def process_html(docs_html) # rubocop:disable Metrics/MethodLength
-    docs_html.transform_values do |html|
-      next '' if html.nil? || html.empty?
-
-      # Extract just the main content areas, skip scripts, styles, etc
-      main_content = html.scan(%r{<main.*?>.*?</main>}m).first ||
-                     html.scan(%r{<article.*?>.*?</article>}m).first ||
-                     html.scan(%r{<div class="content.*?>.*?</div>}m).first
-
-      if main_content
-        # Further clean up the content
-        main_content.gsub(%r{<script.*?>.*?</script>}m, '')
-                    .gsub(%r{<style.*?>.*?</style>}m, '')
-                    .gsub(/<!--.*?-->/m, '')
-                    .gsub(/\s+/, ' ')
-                    .strip
-      else
-        ''
-      end
-    end
-  end
-
-  def generate_capabilities(existing_code) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
-    max_retries = 3
-    retries = 0
-
-    loop do # rubocop:disable Metrics/BlockLength
-      puts "  Attempt #{retries + 1}/#{max_retries}..."
-
-      begin
-        gemini = RubyLLM.chat(model: 'gemini-1.5-pro-latest')
-                        .with_temperature(0.1)
-
-        docs_json = JSON.pretty_generate(@processed_html)
-
-        prompt = <<~PROMPT
-          Update RubyLLM::Providers::#{@provider_name.capitalize}::Capabilities module.
-          Only use the provided HTML content and existing code structure.
-          Focus on updating values while preserving the module structure.
-
-          Existing code to maintain structure:
-          ```ruby
-          #{existing_code}
-          ```
-
-          HTML content to extract new values from:
-          ```json
-          #{docs_json}
-          ```
-
-          Return ONLY the complete Ruby code within ```ruby ``` tags.
-        PROMPT
-
-        response = gemini.ask(prompt)
-        generated_code = CodeValidator.extract_code_from_response(response.content)
-
-        return generated_code if generated_code && CodeValidator.validate_syntax(generated_code)
-      rescue RubyLLM::BadRequestError => e
-        puts "  Error: #{e.message}"
-        # Try with even less content if we hit token limits
-        @processed_html = @processed_html.transform_values { |html| html[0..10_000] }
-      rescue StandardError => e
-        puts "  Error: #{e.class} - #{e.message}"
-      end
-
-      retries += 1
-      break if retries >= max_retries
-    end
-
-    nil
-  end
-
-  def verify_code_with_models_update
-    puts '  Verifying with models:update...'
-    begin
-      Rake::Task['models:update'].reenable
-      Rake::Task['models:update'].invoke
-      puts '  Verification successful'
-    rescue StandardError => e
-      puts "  Verification failed: #{e.message}"
-    end
-  end
-end