ruby_llm 0.1.0.pre44 → 0.1.0.pre45

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 519b87700f2ba3d5aeb8ea3a87d0881f4ef58f974991647b7af4c9f138cbf3d2
-  data.tar.gz: 553b6441ad59fe5efe6d51374103690101c2cb8d072f91bf2a850d84b4934601
+  metadata.gz: e122212234e51321ae20de6894bb476b46c83d745ca30bedd9ced246bdcff22f
+  data.tar.gz: c0619618ca5e09db904545594c613b87ade415a4d1ed38ebd2c8b5857933ea02
 SHA512:
-  metadata.gz: e52cb96479a0f4443521bf3b5b2e98c882fb1c404e3d1527c74f638137bf56c2cb9442cc6e30ef197157fbdaa7dc189b7f6331a31e6f31ddfa28a926035d6578
-  data.tar.gz: b1d7948851e6c1ffa5257e214f01e6d6efaf64d32d98dcc66ec8f5d6776df232a92b3b26ff1de78889f4195ff27e41dc7375a4b2530dc1b97edc8fd57bb2c835
+  metadata.gz: e14f4c2ce93c2d6a4f8ba3909b78a2cad5fb336066b28999ace00135f6abb7a1c3a56389e5a01b41583c86a1b3521ff4296266be9eba831e06b591f558756045
+  data.tar.gz: 6b68676a11c6168cccb7237fd3e90624fbe624402b731ffd87925f3805bffec849a9358c807890dea142f83035a511964b097defbe52a9ef33c679f6cf9195a3

data/.rspec_status

@@ -39,3 +39,12 @@ example_id                                         | status | run_time        |
 ./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/models_spec.rb[1:1:1]              | passed | 0.00337 seconds |
+./spec/ruby_llm/models_spec.rb[1:2:1]              | passed | 0.00435 seconds |
+./spec/ruby_llm/models_spec.rb[1:2:2]              | passed | 0.00491 seconds |
+./spec/ruby_llm/models_spec.rb[1:2:3]              | passed | 0.00028 seconds |
+./spec/ruby_llm/models_spec.rb[1:3:1]              | passed | 0.00007 seconds |
+./spec/ruby_llm/models_spec.rb[1:3:2]              | passed | 0.00036 seconds |
+./spec/ruby_llm/models_spec.rb[1:3:3]              | passed | 0.00252 seconds |
+./spec/ruby_llm/models_spec.rb[1:4:1]              | passed | 0.0016 seconds  |
+./spec/ruby_llm/models_spec.rb[1:4:2]              | passed | 0.00156 seconds |

data/lib/ruby_llm/models.rb

@@ -5,55 +5,100 @@ 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)) }
+    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 }
-    end
-
-    def default_model
-      'gpt-4o-mini'
+      self.class.new(all.select { |m| m.family == family.to_s })
     end
 
-    def refresh!
-      @all = RubyLLM.providers.flat_map(&:list_models).sort_by(&:id)
+    # Filter models by provider
+    def by_provider(provider)
+      self.class.new(all.select { |m| m.provider == provider.to_s })
     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.pre45'
 end

data/lib/ruby_llm.rb

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

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