ruby_llm 0.1.0.pre45 → 0.1.0.pre47

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e122212234e51321ae20de6894bb476b46c83d745ca30bedd9ced246bdcff22f
-  data.tar.gz: c0619618ca5e09db904545594c613b87ade415a4d1ed38ebd2c8b5857933ea02
+  metadata.gz: 92de102c05319b9254741ad429d08bb6d3df6fca1037bb1acdb9a818fb3f4f2f
+  data.tar.gz: f8b1e3b1e10d3a666e1f8b7383b98cc832dd90fb7afed0958f9ec37d6206f180
 SHA512:
-  metadata.gz: e14f4c2ce93c2d6a4f8ba3909b78a2cad5fb336066b28999ace00135f6abb7a1c3a56389e5a01b41583c86a1b3521ff4296266be9eba831e06b591f558756045
-  data.tar.gz: 6b68676a11c6168cccb7237fd3e90624fbe624402b731ffd87925f3805bffec849a9358c807890dea142f83035a511964b097defbe52a9ef33c679f6cf9195a3
+  metadata.gz: 2d8d351c5b6597ac5501dff34f489866de04ea7230fb317fcb0762be39d9df1d33ecb040a1eaa056125597078ebf1e155de2dfd4ea5b996e4b20807ac8f7247c
+  data.tar.gz: a13dfa9b28566cb700380ab0ba7876e824b04dd61962158e9ccc3f9dff2f6524fb7e7692e48546ecbaf8c43d5d5cf16ce66fa3a940602032c7dcf59dbc9e5d07

data/.rspec_status

@@ -21,30 +21,30 @@ example_id                                         | status | run_time        |
 ./spec/ruby_llm/chat_streaming_spec.rb[1:1:2:1]    | passed | 0.50088 seconds |
 ./spec/ruby_llm/chat_streaming_spec.rb[1:1:3:1]    | passed | 5.69 seconds    |
 ./spec/ruby_llm/chat_streaming_spec.rb[1:1:4:1]    | passed | 1.22 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:1:1]        | passed | 3.75 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:1:2]        | passed | 6.1 seconds     |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:1:3]        | passed | 5.32 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:2:1]        | passed | 1.21 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:2:2]        | passed | 2.36 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:2:3]        | passed | 2.78 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:3:1]        | passed | 1.87 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:3:2]        | passed | 3.12 seconds    |
-./spec/ruby_llm/chat_tools_spec.rb[1:1:3:3]        | passed | 3.83 seconds    |
+./spec/ruby_llm/chat_tools_spec.rb[1:1:1]          | passed | 4.26 seconds    |
+./spec/ruby_llm/chat_tools_spec.rb[1:1:2]          | passed | 6.16 seconds    |
+./spec/ruby_llm/chat_tools_spec.rb[1:1:3]          | passed | 11.15 seconds   |
+./spec/ruby_llm/chat_tools_spec.rb[1:1:4]          | passed | 1.3 seconds     |
+./spec/ruby_llm/chat_tools_spec.rb[1:1:5]          | passed | 2.71 seconds    |
+./spec/ruby_llm/chat_tools_spec.rb[1:1:6]          | passed | 2.43 seconds    |
+./spec/ruby_llm/chat_tools_spec.rb[1:1:7]          | passed | 2.71 seconds    |
+./spec/ruby_llm/chat_tools_spec.rb[1:1:8]          | passed | 4 seconds       |
+./spec/ruby_llm/chat_tools_spec.rb[1:1:9]          | passed | 2.81 seconds    |
 ./spec/ruby_llm/embeddings_spec.rb[1:1:1:1]        | passed | 0.33357 seconds |
 ./spec/ruby_llm/embeddings_spec.rb[1:1:1:2]        | passed | 0.43632 seconds |
 ./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/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 |
+./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/active_record/acts_as.rb

@@ -96,7 +96,7 @@ module RubyLLM
       private
 
       def persist_new_message
-        messages.create!(
+        @message = messages.create!(
           role: :assistant,
           content: String.new
         )
@@ -110,7 +110,7 @@ module RubyLLM
         end
 
         transaction do
-          messages.last.update!(
+          @message.update!(
             role: message.role,
             content: message.content,
             model_id: message.model_id,
@@ -126,7 +126,7 @@ module RubyLLM
         tool_calls.each_value do |tool_call|
           attributes = tool_call.to_h
           attributes[:tool_call_id] = attributes.delete(:id)
-          messages.last.tool_calls.create!(**attributes)
+          @message.tool_calls.create!(**attributes)
         end
       end
     end

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

@@ -20,6 +20,7 @@ module RubyLLM
       Provider.for(model)
     end
 
+    # Class method to refresh model data
     def self.refresh!
       models = RubyLLM.providers.flat_map(&:list_models).sort_by(&:id)
       # Write to models.json
@@ -100,5 +101,12 @@ module RubyLLM
     def by_provider(provider)
       self.class.new(all.select { |m| m.provider == provider.to_s })
     end
+
+    # Instance method to refresh models
+    def refresh!
+      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.pre45'
+  VERSION = '0.1.0.pre47'
 end

data/lib/ruby_llm.rb

@@ -58,7 +58,7 @@ module RubyLLM
       @logger ||= Logger.new(
         $stdout,
         progname: 'RubyLLM',
-        level: ENV['RUBY_LLM_DEBUG'] ? Logger::DEBUG : Logger::INFO
+        level: ENV['RUBYLLM_DEBUG'] ? Logger::DEBUG : Logger::INFO
       )
     end
   end

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