Add and update reranking-related model helpers

Author: infinityrobot

lib/ruby_llm/model/info.rb

@@ -66,17 +66,16 @@ def output_price_per_million
         pricing.text_tokens.output
       end
 
-      def type # rubocop:disable Metrics/PerceivedComplexity
-        if modalities.output.include?('embeddings') && !modalities.output.include?('text')
-          'embedding'
-        elsif modalities.output.include?('image') && !modalities.output.include?('text')
-          'image'
-        elsif modalities.output.include?('audio') && !modalities.output.include?('text')
-          'audio'
-        elsif modalities.output.include?('moderation')
-          'moderation'
-        else
-          'chat'
+      def type
+        outputs = modalities.output
+
+        return 'moderation' if outputs.include?('moderation')
+        return 'chat' if outputs.include?('text')
+
+        case outputs.first
+        when 'embeddings' then 'embedding'
+        when 'rerank', 'image', 'audio' then outputs.first
+        else 'chat'
         end
       end
 

lib/ruby_llm/models.rb

@@ -184,6 +184,11 @@ def embedding_models
       self.class.new(all.select { |m| m.type == 'embedding' })
     end
 
+    # Filter to only rerank models
+    def rerank_models
+      self.class.new(all.select { |m| m.type == 'rerank' })
+    end
+
     # Filter to only audio models
     def audio_models
       self.class.new(all.select { |m| m.type == 'audio' })