ruby_llm 0.1.0.pre44 → 0.1.0.pre46

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 519b87700f2ba3d5aeb8ea3a87d0881f4ef58f974991647b7af4c9f138cbf3d2
-  data.tar.gz: 553b6441ad59fe5efe6d51374103690101c2cb8d072f91bf2a850d84b4934601
+  metadata.gz: 047c20a013a29b030be1c93064a7d1545c570f56845e32c864d20a3fe501a964
+  data.tar.gz: f5bfc01bc51e19a6a2d197ae4e9ac50e0fb36757850552a028466a4fcdb35dce
 SHA512:
-  metadata.gz: e52cb96479a0f4443521bf3b5b2e98c882fb1c404e3d1527c74f638137bf56c2cb9442cc6e30ef197157fbdaa7dc189b7f6331a31e6f31ddfa28a926035d6578
-  data.tar.gz: b1d7948851e6c1ffa5257e214f01e6d6efaf64d32d98dcc66ec8f5d6776df232a92b3b26ff1de78889f4195ff27e41dc7375a4b2530dc1b97edc8fd57bb2c835
+  metadata.gz: 9888bfab59a6a366dd8b4139586f1fa01ed5fb05a93b0256102b2a5ac3bba3bfb912f48f911f050c2e23a0d432dcc1eaa5d5a3044d1e5affe8f18fd7ac21618e
+  data.tar.gz: dd8970bc6f942e5a523b545f9bc09d82e98d65228d081589273509f8deb7cd95ba3adda33cece93498bcf2e045b69b6decfb8aebd42e56ed52f6ca69057c3d59

data/.rspec_status

@@ -35,7 +35,16 @@ example_id                                         | status | run_time        |
 ./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 | 24.16 seconds   |
-./spec/ruby_llm/image_generation_spec.rb[1:1:2]    | passed | 14.81 seconds   |
-./spec/ruby_llm/image_generation_spec.rb[1:1:3]    | passed | 9.17 seconds    |
-./spec/ruby_llm/image_generation_spec.rb[1:1:4]    | passed | 0.00083 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 |

data/lib/ruby_llm/models.json

@@ -168,7 +168,7 @@
   },
   {
     "id": "claude-3-7-sonnet-20250219",
-    "created_at": "2025-02-19T00:00:00Z",
+    "created_at": "2025-02-24T00:00:00Z",
     "display_name": "Claude 3.7 Sonnet",
     "provider": "anthropic",
     "context_window": 200000,
@@ -883,6 +883,56 @@
       ]
     }
   },
+  {
+    "id": "gemini-2.0-flash-lite-preview",
+    "created_at": null,
+    "display_name": "Gemini 2.0 Flash-Lite Preview",
+    "provider": "gemini",
+    "context_window": 1048576,
+    "max_tokens": 8192,
+    "type": "chat",
+    "family": "gemini20_flash_lite",
+    "supports_vision": true,
+    "supports_functions": false,
+    "supports_json_mode": false,
+    "input_price_per_million": 0.075,
+    "output_price_per_million": 0.3,
+    "metadata": {
+      "version": "preview-02-05",
+      "description": "Preview release (February 5th, 2025) of Gemini 2.0 Flash Lite",
+      "input_token_limit": 1048576,
+      "output_token_limit": 8192,
+      "supported_generation_methods": [
+        "generateContent",
+        "countTokens"
+      ]
+    }
+  },
+  {
+    "id": "gemini-2.0-flash-lite-preview-02-05",
+    "created_at": null,
+    "display_name": "Gemini 2.0 Flash-Lite Preview 02-05",
+    "provider": "gemini",
+    "context_window": 1048576,
+    "max_tokens": 8192,
+    "type": "chat",
+    "family": "gemini20_flash_lite",
+    "supports_vision": true,
+    "supports_functions": false,
+    "supports_json_mode": false,
+    "input_price_per_million": 0.075,
+    "output_price_per_million": 0.3,
+    "metadata": {
+      "version": "preview-02-05",
+      "description": "Preview release (February 5th, 2025) of Gemini 2.0 Flash Lite",
+      "input_token_limit": 1048576,
+      "output_token_limit": 8192,
+      "supported_generation_methods": [
+        "generateContent",
+        "countTokens"
+      ]
+    }
+  },
   {
     "id": "gemini-2.0-flash-thinking-exp",
     "created_at": null,
@@ -1305,6 +1355,44 @@
       "owned_by": "system"
     }
   },
+  {
+    "id": "gpt-4.5-preview",
+    "created_at": "2025-02-27T03:24:19+01:00",
+    "display_name": "GPT-4.5 Preview",
+    "provider": "openai",
+    "context_window": 4096,
+    "max_tokens": 4096,
+    "type": "chat",
+    "family": "gpt4",
+    "supports_vision": false,
+    "supports_functions": true,
+    "supports_json_mode": false,
+    "input_price_per_million": 0.5,
+    "output_price_per_million": 1.5,
+    "metadata": {
+      "object": "model",
+      "owned_by": "system"
+    }
+  },
+  {
+    "id": "gpt-4.5-preview-2025-02-27",
+    "created_at": "2025-02-27T03:28:24+01:00",
+    "display_name": "GPT-4.5 Preview 20250227",
+    "provider": "openai",
+    "context_window": 4096,
+    "max_tokens": 4096,
+    "type": "chat",
+    "family": "gpt4",
+    "supports_vision": false,
+    "supports_functions": true,
+    "supports_json_mode": false,
+    "input_price_per_million": 0.5,
+    "output_price_per_million": 1.5,
+    "metadata": {
+      "object": "model",
+      "owned_by": "system"
+    }
+  },
   {
     "id": "gpt-4o",
     "created_at": "2024-05-10T20:50:49+02:00",

data/lib/ruby_llm/models.rb

@@ -5,55 +5,108 @@ module RubyLLM
   # to discover and work with models from different providers.
   #
   # Example:
-  #   RubyLLM.models.all                  # All available models
-  #   RubyLLM.models.chat_models          # Models that support chat
-  #   RubyLLM.models.find('claude-3')     # Get info about a specific model
-  module Models
-    module_function
+  #   RubyLLM.models.all                                  # All available models
+  #   RubyLLM.models.chat_models                          # Models that support chat
+  #   RubyLLM.models.by_provider('openai').chat_models    # OpenAI chat models
+  #   RubyLLM.models.find('claude-3')                     # Get info about a specific model
+  class Models
+    include Enumerable
 
-    def provider_for(model)
+    def self.instance
+      @instance ||= new
+    end
+
+    def self.provider_for(model)
       Provider.for(model)
     end
 
-    def all
-      @all ||= begin
-        data = JSON.parse(File.read(File.expand_path('models.json', __dir__)))
-        data.map { |model| ModelInfo.new(model.transform_keys(&:to_sym)) }
+    # Class method to refresh model data
+    def self.refresh!
+      models = RubyLLM.providers.flat_map(&:list_models).sort_by(&:id)
+      # Write to models.json
+      File.write(File.expand_path('models.json', __dir__), JSON.pretty_generate(models.map(&:to_h)))
+      @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, ...)
+        else
+          super
+        end
+      end
+
+      def respond_to_missing?(method, include_private = false)
+        instance.respond_to?(method, include_private) || super
       end
+    end
+
+    # Initialize with optional pre-filtered models
+    def initialize(models = nil)
+      @models = models || load_models
+    end
+
+    # Load models from the JSON file
+    def load_models
+      data = JSON.parse(File.read(File.expand_path('models.json', __dir__)))
+      data.map { |model| ModelInfo.new(model.transform_keys(&:to_sym)) }
     rescue Errno::ENOENT
       [] # Return empty array if file doesn't exist yet
     end
 
+    # Return all models in the collection
+    def all
+      @models
+    end
+
+    # Allow enumeration over all models
+    def each(&)
+      all.each(&)
+    end
+
+    # Find a specific model by ID
     def find(model_id)
-      all.find { |m| m.id == model_id } or raise ModelNotFoundError, "Unknown model: #{model_id}"
+      all.find { |m| m.id == model_id } or
+        raise ModelNotFoundError, "Unknown model: #{model_id}"
     end
 
+    # Filter to only chat models
     def chat_models
-      all.select { |m| m.type == 'chat' }
+      self.class.new(all.select { |m| m.type == 'chat' })
     end
 
+    # Filter to only embedding models
     def embedding_models
-      all.select { |m| m.type == 'embedding' }
+      self.class.new(all.select { |m| m.type == 'embedding' })
     end
 
+    # Filter to only audio models
     def audio_models
-      all.select { |m| m.type == 'audio' }
+      self.class.new(all.select { |m| m.type == 'audio' })
     end
 
+    # Filter to only image models
     def image_models
-      all.select { |m| m.type == 'image' }
+      self.class.new(all.select { |m| m.type == 'image' })
     end
 
+    # Filter models by family
     def by_family(family)
-      all.select { |m| m.family == family }
+      self.class.new(all.select { |m| m.family == family.to_s })
     end
 
-    def default_model
-      'gpt-4o-mini'
+    # Filter models by provider
+    def by_provider(provider)
+      self.class.new(all.select { |m| m.provider == provider.to_s })
     end
 
+    # Instance method to refresh models
     def refresh!
-      @all = RubyLLM.providers.flat_map(&:list_models).sort_by(&:id)
+      self.class.refresh!
+      # Return self for method chaining
+      self
     end
   end
 end

data/lib/ruby_llm/version.rb

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

data/lib/ruby_llm.rb

@@ -39,7 +39,7 @@ module RubyLLM
     end
 
     def models
-      Models
+      Models.instance
     end
 
     def providers

data/lib/tasks/models.rake

@@ -70,21 +70,18 @@ namespace :models do # rubocop:disable Metrics/BlockLength
     RubyLLM.configure do |config|
       config.openai_api_key = ENV.fetch('OPENAI_API_KEY')
       config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY')
-      config.gemini_api_key = ENV.fetch('GEMINI_API_KEY', nil)
-      config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY', nil)
+      config.gemini_api_key = ENV.fetch('GEMINI_API_KEY')
+      config.deepseek_api_key = ENV.fetch('DEEPSEEK_API_KEY')
     end
 
-    # Get all models
-    models = RubyLLM.models.refresh!
-
-    # Write to models.json
-    models_file = File.expand_path('../../lib/ruby_llm/models.json', __dir__)
-    File.write(models_file, JSON.pretty_generate(models.map(&:to_h)))
+    # Refresh models (now returns self instead of models array)
+    models = RubyLLM.models.refresh!.all
 
     puts "Updated models.json with #{models.size} models:"
     RubyLLM::Provider.providers.each do |provider_sym, provider_module|
       provider_name = provider_module.to_s.split('::').last
-      puts "#{provider_name} models: #{models.count { |m| m.provider == provider_sym.to_s }}"
+      provider_models = models.select { |m| m.provider == provider_sym.to_s }
+      puts "#{provider_name} models: #{provider_models.size}"
     end
   end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre44
+  version: 0.1.0.pre46
 platform: ruby
 authors:
 - Carmine Paolino