Bump version to 0.1.0.pre35; refactor model response parsing and capabilities methods for providers

Author: crmne

lib/ruby_llm/models.json

@@ -155,6 +155,22 @@
     "output_price_per_million": 15.0,
     "metadata": {}
   },
+  {
+    "id": "claude-3-7-sonnet-20250219",
+    "created_at": "2025-02-19T00:00:00Z",
+    "display_name": "Claude 3.7 Sonnet",
+    "provider": "anthropic",
+    "context_window": 200000,
+    "max_tokens": 4096,
+    "type": "chat",
+    "family": "claude2",
+    "supports_vision": true,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 3.0,
+    "output_price_per_million": 15.0,
+    "metadata": {}
+  },
   {
     "id": "claude-3-haiku-20240307",
     "created_at": "2024-03-07T00:00:00Z",
@@ -830,6 +846,44 @@
       "owned_by": "google"
     }
   },
+  {
+    "id": "gemini-2.0-flash-lite",
+    "created_at": null,
+    "display_name": "Gemini 2.0 Flash Lite",
+    "provider": "gemini",
+    "context_window": 1048576,
+    "max_tokens": 8192,
+    "type": "chat",
+    "family": "gemini20_flash_lite",
+    "supports_vision": true,
+    "supports_functions": false,
+    "supports_json_mode": true,
+    "input_price_per_million": 0.075,
+    "output_price_per_million": 0.3,
+    "metadata": {
+      "object": "model",
+      "owned_by": "google"
+    }
+  },
+  {
+    "id": "gemini-2.0-flash-lite-001",
+    "created_at": null,
+    "display_name": "Gemini 2.0 Flash Lite 001",
+    "provider": "gemini",
+    "context_window": 1048576,
+    "max_tokens": 8192,
+    "type": "chat",
+    "family": "gemini20_flash_lite",
+    "supports_vision": true,
+    "supports_functions": false,
+    "supports_json_mode": true,
+    "input_price_per_million": 0.075,
+    "output_price_per_million": 0.3,
+    "metadata": {
+      "object": "model",
+      "owned_by": "google"
+    }
+  },
   {
     "id": "gemini-2.0-flash-lite-preview",
     "created_at": null,
@@ -868,6 +922,44 @@
       "owned_by": "google"
     }
   },
+  {
+    "id": "gemini-2.0-flash-mmgen-rev17",
+    "created_at": null,
+    "display_name": "Gemini 2.0 Flash Mmgen Rev17",
+    "provider": "gemini",
+    "context_window": 1048576,
+    "max_tokens": 8192,
+    "type": "chat",
+    "family": "gemini20_flash",
+    "supports_vision": true,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 0.1,
+    "output_price_per_million": 0.4,
+    "metadata": {
+      "object": "model",
+      "owned_by": "google"
+    }
+  },
+  {
+    "id": "gemini-2.0-flash-thinking-001",
+    "created_at": null,
+    "display_name": "Gemini 2.0 Flash Thinking 001",
+    "provider": "gemini",
+    "context_window": 1048576,
+    "max_tokens": 8192,
+    "type": "chat",
+    "family": "gemini20_flash",
+    "supports_vision": true,
+    "supports_functions": true,
+    "supports_json_mode": true,
+    "input_price_per_million": 0.1,
+    "output_price_per_million": 0.4,
+    "metadata": {
+      "object": "model",
+      "owned_by": "google"
+    }
+  },
   {
     "id": "gemini-2.0-flash-thinking-exp",
     "created_at": null,
@@ -1077,6 +1169,25 @@
       "owned_by": "google"
     }
   },
+  {
+    "id": "gemma-3-27b-it",
+    "created_at": null,
+    "display_name": "Gemma 3 27b It",
+    "provider": "gemini",
+    "context_window": 32768,
+    "max_tokens": 4096,
+    "type": "chat",
+    "family": "other",
+    "supports_vision": false,
+    "supports_functions": false,
+    "supports_json_mode": false,
+    "input_price_per_million": 0.075,
+    "output_price_per_million": 0.3,
+    "metadata": {
+      "object": "model",
+      "owned_by": "google"
+    }
+  },
   {
     "id": "gpt-3.5-turbo",
     "created_at": "2023-02-28T19:56:42+01:00",
@@ -2046,6 +2157,25 @@
       "owned_by": "system"
     }
   },
+  {
+    "id": "veo-2.0-generate-001",
+    "created_at": null,
+    "display_name": "Veo 2.0 Generate 001",
+    "provider": "gemini",
+    "context_window": 32768,
+    "max_tokens": 4096,
+    "type": "chat",
+    "family": "other",
+    "supports_vision": false,
+    "supports_functions": false,
+    "supports_json_mode": false,
+    "input_price_per_million": 0.075,
+    "output_price_per_million": 0.3,
+    "metadata": {
+      "object": "model",
+      "owned_by": "google"
+    }
+  },
   {
     "id": "whisper-1",
     "created_at": "2023-02-27T22:13:04+01:00",

lib/ruby_llm/models.rb

@@ -53,9 +53,7 @@ def default_model
     end
 
     def refresh!
-      @all = RubyLLM.providers.flat_map do |provider|
-        provider.new.list_models
-      end.sort_by(&:id)
+      @all = RubyLLM.providers.flat_map(&:list_models).sort_by(&:id)
     end
   end
 end

lib/ruby_llm/provider.rb

@@ -23,7 +23,7 @@ def list_models
           req.headers.merge! headers
         end
 
-        parse_list_models_response response
+        parse_list_models_response response, slug, capabilities
       end
 
       def embed(text, model:)
@@ -150,15 +150,6 @@ def parse_error(response)
       body.is_a?(Hash) ? body.dig('error', 'message') : body
     end
 
-    def capabilities
-      provider_name = self.class.name.split('::').last
-      provider_name::Capabilities
-    end
-
-    def slug
-      self.class.name.split('::').last.downcase
-    end
-
     class << self
       def extended(base)
         base.extend(Methods)

lib/ruby_llm/providers/anthropic.rb

@@ -24,6 +24,14 @@ def headers
           'anthropic-version' => '2023-06-01'
         }
       end
+
+      def capabilities
+        Anthropic::Capabilities
+      end
+
+      def slug
+        'anthropic'
+      end
     end
   end
 end

lib/ruby_llm/providers/anthropic/models.rb

@@ -11,7 +11,7 @@ def models_url
           '/v1/models'
         end
 
-        def parse_list_models_response(response) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
+        def parse_list_models_response(response, slug, capabilities) # rubocop:disable Metrics/AbcSize,Metrics/MethodLength
           (response.body['data'] || []).map do |model|
             ModelInfo.new(
               id: model['id'],

lib/ruby_llm/providers/deepseek.rb

@@ -17,6 +17,14 @@ def headers
           'Authorization' => "Bearer #{RubyLLM.config.deepseek_api_key}"
         }
       end
+
+      def capabilities
+        DeepSeek::Capabilities
+      end
+
+      def slug
+        'deepseek'
+      end
     end
   end
 end

lib/ruby_llm/providers/gemini.rb

@@ -18,6 +18,14 @@ def headers
           'Authorization' => "Bearer #{RubyLLM.config.gemini_api_key}"
         }
       end
+
+      def capabilities
+        Gemini::Capabilities
+      end
+
+      def slug
+        'gemini'
+      end
     end
   end
 end

lib/ruby_llm/providers/gemini/capabilities.rb

@@ -126,8 +126,6 @@ def pricing_family(model_id) # rubocop:disable Metrics/CyclomaticComplexity
           end
         end
 
-        private
-
         def long_context_model?(model_id)
           model_id.match?(/gemini-1\.5-(?:pro|flash)/)
         end

lib/ruby_llm/providers/gemini/models.rb

@@ -7,12 +7,12 @@ module Gemini
       module Models
         module_function
 
-        def parse_list_models_response(response)
+        def parse_list_models_response(response, slug, capabilities)
           response.body['data']&.each do |model|
             model['id'] = model['id'].delete_prefix('models/')
           end
 
-          OpenAI::Models.parse_list_models_response(response)
+          OpenAI::Models.parse_list_models_response(response, slug, capabilities)
         end
       end
     end

lib/ruby_llm/providers/openai.rb

@@ -37,6 +37,14 @@ def headers
           'Authorization' => "Bearer #{RubyLLM.config.openai_api_key}"
         }
       end
+
+      def capabilities
+        OpenAI::Capabilities
+      end
+
+      def slug
+        'openai'
+      end
     end
   end
 end