ruby_llm 0.1.0.pre49 → 1.0.1

data/lib/ruby_llm/models.rb

@@ -12,22 +12,26 @@ module RubyLLM
   class Models
     include Enumerable
 
-    def self.instance
-      @instance ||= new
-    end
+    # Delegate class methods to the singleton instance
+    class << self
+      def instance
+        @instance ||= new
+      end
 
-    def self.provider_for(model)
-      Provider.for(model)
-    end
+      def provider_for(model)
+        Provider.for(model)
+      end
 
-    # Class method to refresh model data
-    def self.refresh!
-      models = RubyLLM.providers.flat_map(&:list_models).sort_by(&:id)
-      @instance = new(models)
-    end
+      def models_file
+        File.expand_path('models.json', __dir__)
+      end
+
+      # Class method to refresh model data
+      def refresh!
+        models = RubyLLM.providers.flat_map(&:list_models).sort_by(&:id)
+        @instance = new(models)
+      end
 
-    # Delegate class methods to the singleton instance
-    class << self
       def method_missing(method, ...)
         if instance.respond_to?(method)
           instance.send(method, ...)
@@ -48,12 +52,16 @@ module RubyLLM
 
     # Load models from the JSON file
     def load_models
-      data = JSON.parse(File.read(File.expand_path('models.json', __dir__)))
+      data = JSON.parse(File.read(self.class.models_file))
       data.map { |model| ModelInfo.new(model.transform_keys(&:to_sym)) }
     rescue Errno::ENOENT
       [] # Return empty array if file doesn't exist yet
     end
 
+    def save_models
+      File.write(self.class.models_file, JSON.pretty_generate(all.map(&:to_h)))
+    end
+
     # Return all models in the collection
     def all
       @models
@@ -103,8 +111,6 @@ module RubyLLM
     # Instance method to refresh models
     def refresh!
       self.class.refresh!
-      # Return self for method chaining
-      self
     end
   end
 end

data/lib/ruby_llm/provider.rb

@@ -8,8 +8,18 @@ module RubyLLM
     # Common functionality for all LLM providers. Implements the core provider
     # interface so specific providers only need to implement a few key methods.
     module Methods # rubocop:disable Metrics/ModuleLength
-      def complete(messages, tools:, temperature:, model:, &block)
-        payload = render_payload messages, tools: tools, temperature: temperature, model: model, stream: block_given?
+      def complete(messages, tools:, temperature:, model:, &block) # rubocop:disable Metrics/MethodLength
+        normalized_temperature = if capabilities.respond_to?(:normalize_temperature)
+                                   capabilities.normalize_temperature(temperature, model)
+                                 else
+                                   temperature
+                                 end
+
+        payload = render_payload(messages,
+                                 tools: tools,
+                                 temperature: normalized_temperature,
+                                 model: model,
+                                 stream: block_given?)
 
         if block_given?
           stream_response payload, &block

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

@@ -10,14 +10,6 @@ module RubyLLM
           'models'
         end
 
-        def list_models
-          response = connection.get("models?key=#{RubyLLM.config.gemini_api_key}") do |req|
-            req.headers.merge! headers
-          end
-
-          parse_list_models_response(response, slug, capabilities)
-        end
-
         private
 
         def parse_list_models_response(response, slug, capabilities) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength

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

@@ -50,15 +50,42 @@ module RubyLLM
         # 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)
-          model_id.match?(/gpt-4o|o1/) || model_id.match?(/gpt-4-(?!0314|0613)/)
+        def supports_vision?(model_id) # rubocop:disable Metrics/MethodLength
+          supporting_patterns = [
+            /^o1$/,
+            /^o1-(?!.*mini|.*preview).*$/,
+            /gpt-4\.5/,
+            /^gpt-4o$/,
+            /gpt-4o-2024/,
+            /gpt-4o-search/,
+            /^gpt-4o-mini$/,
+            /gpt-4o-mini-2024/,
+            /gpt-4o-mini-search/,
+            /chatgpt-4o/,
+            /gpt-4-turbo-2024/,
+            /computer-use-preview/,
+            /omni-moderation/
+          ]
+          supporting_patterns.any? { |regex| model_id.match?(regex) }
         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)
-          !model_id.include?('instruct')
+        def supports_functions?(model_id) # rubocop:disable Metrics/MethodLength
+          supporting_patterns = [
+            /^o1$/,
+            /gpt-4o/,
+            /gpt-4\.5/,
+            /chatgpt-4o/,
+            /gpt-4-turbo/,
+            /computer-use-preview/,
+            /o1-preview/,
+            /o1-\d{4}-\d{2}-\d{2}/,
+            /o1-pro/,
+            /o3-mini/
+          ]
+          supporting_patterns.any? { |regex| model_id.match?(regex) }
         end
 
         # Determines if the model supports audio input/output
@@ -230,6 +257,16 @@ module RubyLLM
             .gsub('Omni Moderation', 'Omni-Moderation')
             .gsub('Text Moderation', 'Text-Moderation')
         end
+
+        def normalize_temperature(temperature, model_id)
+          if model_id.match?(/o[13]/)
+            # O1/O3 models always use temperature 1.0
+            RubyLLM.logger.debug "Model #{model_id} requires temperature=1.0, ignoring provided value"
+            1.0
+          else
+            temperature
+          end
+        end
       end
     end
   end

data/lib/ruby_llm/tool.rb

@@ -18,14 +18,19 @@ module RubyLLM
   # interface for defining parameters and implementing tool behavior.
   #
   # Example:
-  #   class Calculator < RubyLLM::Tool
-  #     description "Performs arithmetic calculations"
-  #     param :expression, type: :string, desc: "Math expression to evaluate"
+  #    require 'tzinfo'
   #
-  #     def execute(expression:)
-  #       eval(expression).to_s
-  #     end
-  #   end
+  #    class TimeInfo < RubyLLM::Tool
+  #      description 'Gets the current time in various timezones'
+  #      param :timezone, desc: "Timezone name (e.g., 'UTC', 'America/New_York')"
+  #
+  #      def execute(timezone:)
+  #        time = TZInfo::Timezone.get(timezone).now.strftime('%Y-%m-%d %H:%M:%S')
+  #        "Current time in #{timezone}: #{time}"
+  #       rescue StandardError => e
+  #          { error: e.message }
+  #       end
+  #    end
   class Tool
     class << self
       def description(text = nil)
@@ -45,6 +50,9 @@ module RubyLLM
 
     def name
       self.class.name
+          .unicode_normalize(:nfkd)
+          .encode('ASCII', replace: '')
+          .gsub(/[^a-zA-Z0-9_-]/, '-')
           .gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
           .gsub(/([a-z\d])([A-Z])/, '\1_\2')
           .downcase

data/lib/ruby_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  VERSION = '0.1.0.pre49'
+  VERSION = '1.0.1'
 end

data/lib/ruby_llm.rb

@@ -17,6 +17,8 @@ loader.inflector.inflect(
   'api' => 'API',
   'deepseek' => 'DeepSeek'
 )
+loader.ignore("#{__dir__}/ruby_llm/railtie")
+loader.ignore("#{__dir__}/ruby_llm/active_record")
 loader.setup
 
 # A delightful Ruby interface to modern AI language models.

data/lib/tasks/models.rake

@@ -2,7 +2,9 @@
 
 require 'English'
 require 'faraday'
+require 'fileutils'
 require 'nokogiri'
+require 'ruby_llm'
 
 # URLs to process
 PROVIDER_DOCS = {
@@ -64,8 +66,6 @@ end
 namespace :models do # rubocop:disable Metrics/BlockLength
   desc 'Update available models from providers'
   task :update do
-    require 'ruby_llm'
-
     # Configure API keys
     RubyLLM.configure do |config|
       config.openai_api_key = ENV.fetch('OPENAI_API_KEY')
@@ -74,15 +74,13 @@ namespace :models do # rubocop:disable Metrics/BlockLength
       config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY')
     end
 
-    # Refresh models (now returns self instead of models array)
-    models = RubyLLM.models.refresh!.all
-    # Write to models.json
-    File.write(File.expand_path('../ruby_llm/models.json', __dir__), JSON.pretty_generate(models.map(&:to_h)))
+    models = RubyLLM.models.refresh!
+    models.save_models
 
-    puts "Updated models.json with #{models.size} models:"
+    puts "Updated models.json with #{models.all.size} models:"
     RubyLLM::Provider.providers.each do |provider_sym, provider_module|
       provider_name = provider_module.to_s.split('::').last
-      provider_models = models.select { |m| m.provider == provider_sym.to_s }
+      provider_models = models.all.select { |m| m.provider == provider_sym.to_s }
       puts "#{provider_name} models: #{provider_models.size}"
     end
   end
@@ -91,8 +89,6 @@ namespace :models do # rubocop:disable Metrics/BlockLength
   task :update_capabilities do # rubocop:disable Metrics/BlockLength
     # Check if a specific provider was requested
     target_provider = ENV['PROVIDER']&.to_sym
-    require 'ruby_llm'
-    require 'fileutils'
 
     # Configure API keys
     RubyLLM.configure do |config|

data/lib/tasks/vcr.rake

@@ -0,0 +1,114 @@
+# frozen_string_literal: true
+
+# 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)
+
+  puts 'Recording cassettes for all providers...'
+  run_tests
+  puts 'Done recording. Please review the new cassettes.'
+end
+
+def record_for_providers(providers, cassette_dir) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
+  # 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(', ')}"
+    return
+  end
+
+  invalid_providers = providers - all_providers
+  if invalid_providers.any?
+    puts "Invalid providers: #{invalid_providers.join(', ')}"
+    puts "Available providers: #{all_providers.join(', ')}"
+    return
+  end
+
+  # Get URL patterns from the providers themselves
+  provider_patterns = get_provider_patterns(providers)
+
+  puts "Finding cassettes for providers: #{providers.join(', ')}"
+
+  # Find and delete matching cassettes
+  cassettes_to_delete = find_matching_cassettes(cassette_dir, provider_patterns)
+
+  if cassettes_to_delete.empty?
+    puts 'No cassettes found for the specified providers.'
+    puts 'Running tests to record new cassettes...'
+  else
+    delete_cassettes(cassettes_to_delete)
+    puts "\nRunning tests to record new cassettes..."
+  end
+
+  run_tests
+
+  puts "\nDone recording cassettes for #{providers.join(', ')}."
+  puts 'Please review the updated cassettes for sensitive information.'
+end
+
+def get_provider_patterns(providers) # rubocop:disable Metrics/MethodLength
+  provider_patterns = {}
+
+  providers.each do |provider_name|
+    provider_module = RubyLLM::Provider.providers[provider_name.to_sym]
+    next unless provider_module
+
+    # Extract the base URL from the provider's api_base method
+    api_base = provider_module.api_base.to_s
+
+    # Create a regex pattern from the domain
+    next unless api_base && !api_base.empty?
+
+    domain = URI.parse(api_base).host
+    pattern = Regexp.new(Regexp.escape(domain))
+    provider_patterns[provider_name] = pattern
+  end
+
+  provider_patterns
+end
+
+def find_matching_cassettes(dir, patterns)
+  cassettes = []
+
+  Dir.glob("#{dir}/**/*.yml").each do |file|
+    content = File.read(file)
+    cassettes << file if patterns.values.any? { |pattern| content.match?(pattern) }
+  end
+
+  cassettes
+end
+
+def delete_cassettes(cassettes)
+  puts "Deleting #{cassettes.size} cassettes for re-recording:"
+  cassettes.each do |file|
+    puts "  - #{File.basename(file)}"
+    File.delete(file)
+  end
+end
+
+def run_tests
+  system('bundle exec rspec') || abort('Tests failed')
+end
+
+namespace :vcr do
+  desc 'Record VCR cassettes (rake vcr:record[all] or vcr:record[openai,anthropic])'
+  task :record, [:providers] do |_, args|
+    require 'fileutils'
+    require 'ruby_llm'
+
+    providers = (args[:providers] || '').downcase.split(',')
+    cassette_dir = 'spec/fixtures/vcr_cassettes'
+    FileUtils.mkdir_p(cassette_dir)
+
+    if providers.include?('all')
+      record_all_cassettes(cassette_dir)
+    else
+      record_for_providers(providers, cassette_dir)
+    end
+  end
+end

data/ruby_llm.gemspec

@@ -35,6 +35,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ['lib']
 
   # Runtime dependencies
+  spec.add_dependency 'base64'
   spec.add_dependency 'event_stream_parser', '~> 1'
   spec.add_dependency 'faraday', '~> 2'
   spec.add_dependency 'faraday-multipart', '~> 1'

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre49
+  version: 1.0.1
 platform: ruby
 authors:
 - Carmine Paolino
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-02-28 00:00:00.000000000 Z
+date: 2025-03-23 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: base64
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
   requirement: !ruby/object:Gem::Requirement
@@ -96,9 +110,9 @@ files:
 - ".gitignore"
 - ".overcommit.yml"
 - ".rspec"
-- ".rspec_status"
 - ".rubocop.yml"
 - ".yardopts"
+- CONTRIBUTING.md
 - Gemfile
 - LICENSE
 - README.md
@@ -153,6 +167,7 @@ files:
 - lib/ruby_llm/tool_call.rb
 - lib/ruby_llm/version.rb
 - lib/tasks/models.rake
+- lib/tasks/vcr.rake
 - ruby_llm.gemspec
 homepage: https://rubyllm.com
 licenses:

data/.rspec_status

@@ -1,50 +0,0 @@
-example_id                                         | status | run_time        |
--------------------------------------------------- | ------ | --------------- |
-./spec/ruby_llm/active_record/acts_as_spec.rb[1:1] | passed | 3.38 seconds    |
-./spec/ruby_llm/active_record/acts_as_spec.rb[1:2] | passed | 2.48 seconds    |
-./spec/ruby_llm/chat_content_spec.rb[1:1:1]        | passed | 2.74 seconds    |
-./spec/ruby_llm/chat_content_spec.rb[1:1:2]        | passed | 1.29 seconds    |
-./spec/ruby_llm/chat_content_spec.rb[1:1:3]        | passed | 2.54 seconds    |
-./spec/ruby_llm/chat_content_spec.rb[1:2:1]        | passed | 2.77 seconds    |
-./spec/ruby_llm/chat_content_spec.rb[1:2:2]        | passed | 2.1 seconds     |
-./spec/ruby_llm/chat_pdf_spec.rb[1:1:1]            | passed | 7.75 seconds    |
-./spec/ruby_llm/chat_pdf_spec.rb[1:1:2]            | passed | 13.88 seconds   |
-./spec/ruby_llm/chat_spec.rb[1:1:1:1]              | passed | 1.02 seconds    |
-./spec/ruby_llm/chat_spec.rb[1:1:1:2]              | passed | 3.95 seconds    |
-./spec/ruby_llm/chat_spec.rb[1:1:2:1]              | passed | 0.4854 seconds  |
-./spec/ruby_llm/chat_spec.rb[1:1:2:2]              | passed | 1.37 seconds    |
-./spec/ruby_llm/chat_spec.rb[1:1:3:1]              | passed | 7.34 seconds    |
-./spec/ruby_llm/chat_spec.rb[1:1:3:2]              | passed | 19.22 seconds   |
-./spec/ruby_llm/chat_spec.rb[1:1:4:1]              | passed | 3.15 seconds    |
-./spec/ruby_llm/chat_spec.rb[1:1:4:2]              | passed | 2.51 seconds    |
-./spec/ruby_llm/chat_streaming_spec.rb[1:1:1:1]    | passed | 0.91374 seconds |
-./spec/ruby_llm/chat_streaming_spec.rb[1:1:2:1]    | passed | 0.50088 seconds |
-./spec/ruby_llm/chat_streaming_spec.rb[1:1:3:1]    | passed | 5.69 seconds    |
-./spec/ruby_llm/chat_streaming_spec.rb[1:1:4:1]    | passed | 1.22 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:1]          | passed | 4.26 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:2]          | passed | 6.16 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:3]          | passed | 11.15 seconds   |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:4]          | passed | 1.3 seconds     |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:5]          | passed | 2.71 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:6]          | passed | 2.43 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:7]          | passed | 2.71 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:8]          | passed | 4 seconds       |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:9]          | passed | 2.81 seconds    |
-./spec/ruby_llm/embeddings_spec.rb[1:1:1:1]        | passed | 0.33357 seconds |
-./spec/ruby_llm/embeddings_spec.rb[1:1:1:2]        | passed | 0.43632 seconds |
-./spec/ruby_llm/embeddings_spec.rb[1:1:2:1]        | passed | 0.65614 seconds |
-./spec/ruby_llm/embeddings_spec.rb[1:1:2:2]        | passed | 2.16 seconds    |
-./spec/ruby_llm/error_handling_spec.rb[1:1]        | passed | 0.29366 seconds |
-./spec/ruby_llm/image_generation_spec.rb[1:1:1]    | passed | 14.16 seconds   |
-./spec/ruby_llm/image_generation_spec.rb[1:1:2]    | passed | 16.22 seconds   |
-./spec/ruby_llm/image_generation_spec.rb[1:1:3]    | passed | 9.1 seconds     |
-./spec/ruby_llm/image_generation_spec.rb[1:1:4]    | passed | 0.00138 seconds |
-./spec/ruby_llm/models_spec.rb[1:1:1]              | passed | 0.01071 seconds |
-./spec/ruby_llm/models_spec.rb[1:1:2]              | passed | 0.00056 seconds |
-./spec/ruby_llm/models_spec.rb[1:1:3]              | passed | 0.00336 seconds |
-./spec/ruby_llm/models_spec.rb[1:2:1]              | passed | 0.00016 seconds |
-./spec/ruby_llm/models_spec.rb[1:2:2]              | passed | 0.00085 seconds |
-./spec/ruby_llm/models_spec.rb[1:3:1]              | passed | 1.44 seconds    |
-./spec/ruby_llm/models_spec.rb[1:3:2]              | passed | 1.23 seconds    |
-./spec/ruby_llm/models_spec.rb[1:4:1]              | passed | 0.0003 seconds  |
-./spec/ruby_llm/models_spec.rb[1:4:2]              | passed | 0.00175 seconds |