ruby_llm 0.1.0.pre34 → 0.1.0.pre35

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 77ec20e57439d352e965de0e435d0c43b3acd53de2d6035345f6ac6e716e7fef
-  data.tar.gz: 51d838bc1411303fd96c2f28d160ad99657f44846bd12ad2967ee329880e8922
+  metadata.gz: 586cc8473d06aebb4eb69ee646c8ad90250a522eac80d3471a22ef05e601b9de
+  data.tar.gz: 4f060d755a37373230d1364dda3fc8b74fabdf4128206eb4faf2b11fd02090ac
 SHA512:
-  metadata.gz: 9b0ca7a80113ed498125c9e23460879328926e6f3de1e6fe7e7a63453deb94fbdebbe7961d529e6cd50db7ad91776cb92ffeaed29f396aa523f931b12f835f7d
-  data.tar.gz: bcb30504074335dbdea39d171a3d810cfb14c7f5f9a3cc5b10e1a80970ce06f37775c7b1c5352a2e5fef0a14ea0d6603ae63197be6d2a866c7003b091036c842
+  metadata.gz: 8e10930fd06b2dbdbaca81aa93f45cde41e8d1199173597e29d94af7ea073db470a91be9869f89b622aebc28d915510baf070ac6655ddbbe1591a94fbef7b2f8
+  data.tar.gz: de99a1e6173f25b3c3bc7222910f6d1c8771b4c4f1062e8970f9472c7f551543c345cdffb1b3733ebd510e2351c27b5b274d483ba7047a32d90884b62577cb18

data/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",

data/lib/ruby_llm/models.rb

@@ -53,9 +53,7 @@ module RubyLLM
     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

data/lib/ruby_llm/provider.rb

@@ -23,7 +23,7 @@ module RubyLLM
           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 @@ module RubyLLM
       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)

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

@@ -11,7 +11,7 @@ module RubyLLM
           '/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'],

data/lib/ruby_llm/providers/anthropic.rb

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

data/lib/ruby_llm/providers/deepseek.rb

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

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

@@ -126,8 +126,6 @@ module RubyLLM
           end
         end
 
-        private
-
         def long_context_model?(model_id)
           model_id.match?(/gemini-1\.5-(?:pro|flash)/)
         end

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

@@ -7,12 +7,12 @@ module RubyLLM
       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

data/lib/ruby_llm/providers/gemini.rb

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

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

@@ -107,8 +107,6 @@ module RubyLLM
           end
         end
 
-        private
-
         PRICES = {
           o1: { input: 15.0, cached_input: 7.5, output: 60.0 },
           o1_mini: { input: 1.10, cached_input: 0.55, output: 4.40 },

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

@@ -11,7 +11,7 @@ module RubyLLM
           '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'],

data/lib/ruby_llm/providers/openai.rb

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

data/lib/ruby_llm/version.rb

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

data/lib/tasks/models.rake

@@ -12,7 +12,7 @@ PROVIDER_DOCS = {
   },
   gemini: {
     models: 'https://ai.google.dev/gemini-api/docs/models/gemini',
-    pricing: 'https://ai.google.dev/pricing'
+    pricing: 'https://ai.google.dev/gemini-api/docs/pricing'
   },
   deepseek: {
     models: 'https://api-docs.deepseek.com/quick_start/pricing/'
@@ -102,11 +102,12 @@ namespace :models do # rubocop:disable Metrics/BlockLength
       puts "Processing #{provider}..."
 
       # Initialize our AI assistants
+      #
       gemini = RubyLLM.chat(model: 'gemini-2.0-flash').with_temperature(0)
-      claude = RubyLLM.chat(model: 'claude-3-5-sonnet-20241022').with_temperature(0)
+      claude = RubyLLM.chat(model: 'claude-3-7-sonnet-20250219').with_temperature(0)
 
       # Read existing capabilities file if present
-      existing_file = "lib/ruby_llm/model_capabilities/#{provider}.rb"
+      existing_file = "lib/ruby_llm/providers/#{provider}/capabilities.rb"
       existing_code = File.read(existing_file) if File.exist?(existing_file)
 
       begin
@@ -155,18 +156,17 @@ namespace :models do # rubocop:disable Metrics/BlockLength
 
           #{model_info}
 
-          The module should go in lib/ruby_llm/model_capabilities/#{provider}.rb and follow these conventions:
+          The module should go in lib/ruby_llm/providers/#{provider}/capabilities.rb and follow these conventions:
 
-          1. Module name should be RubyLLM::ModelCapabilities::#{provider.to_s.capitalize}
-          2. Include methods for determining context windows, token limits, pricing, and capabilities
-          3. Use consistent naming with other providers
-          4. Include detailed pricing information in a PRICES constant
-          5. Follow the existing structure in the codebase
-          6. Use Ruby idioms and clean code practices
-          7. Include module_function to make methods callable at module level
-          8. Include all necessary method documentation
+          1. Include methods for determining context windows, token limits, pricing, and capabilities
+          2. Use consistent naming with other providers
+          3. Include detailed pricing information in a PRICES constant
+          4. Follow the existing structure in the codebase
+          5. Use Ruby idioms and clean code practices
+          6. Include module_function to make methods callable at module level
+          7. Include all necessary method documentation
 
-          Here's the existing implementation for reference (maintain similar structure):
+          Here's the existing implementation for reference (maintain similar structure and same method names):
 
           #{existing_code}
 
@@ -176,7 +176,7 @@ namespace :models do # rubocop:disable Metrics/BlockLength
         response = claude.ask(code_prompt)
 
         # Save the file
-        file_path = "lib/ruby_llm/model_capabilities/#{provider}.rb"
+        file_path = "lib/ruby_llm/providers/#{provider}/capabilities.rb"
         puts "  Writing #{file_path}..."
 
         FileUtils.mkdir_p(File.dirname(file_path))

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre34
+  version: 0.1.0.pre35
 platform: ruby
 authors:
 - Carmine Paolino
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2025-02-21 00:00:00.000000000 Z
+date: 2025-02-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser