ruby_llm 1.14.1 → 1.16.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2e29291ded69046aa32b77f524a148e14ce1b1bfbcaddea1da99b5cee66f7b13
-  data.tar.gz: '08a7aaa292f898f0e25f69658f19289799eb5e520feabf8a736bb3737d05fb9d'
+  metadata.gz: ba1ac6556eb9321cc5c4840eb10c4f2253a35991298d3a4f9cf6dd85035ba069
+  data.tar.gz: f87c2f1e7c1898c31ae732db0bae3a96f8b9c9082d8f9abd9a728b2b9d295426
 SHA512:
-  metadata.gz: e9d4a4c8f284fc94669217c0979da4f6e6eccc95e0e47a8429b64c35c447ade3d994f278431c1593bc1fef804545e2bc0f5529775bc209b72dd728eeb4d99016
-  data.tar.gz: c6ed2232f289a940ad3b9ee29279005528bf74a5921b911fda19e8e2cc6ecb939405e46db76a7e1e83f580da69ed368fa197355b05efada524a537419ac37381
+  metadata.gz: f75855b2dc79679febdf4edb67130f5ee60d7445bd1a34d7f29c69f3cd363711b881ffe2b51eac8fa4f2a74b297e364e53263e12674333055c8daf702d528a3a
+  data.tar.gz: aceac99dd86c4e4db028411d1d58cd2602351f3ba0b3cb7fc2334d979459a97294f2617e10770d851b57f70aa26b49f9995dd239d1f52cff43814d1293419a75

data/README.md

@@ -5,9 +5,10 @@
   <img src="/docs/assets/images/logotype.svg" alt="RubyLLM" height="120" width="250">
 </picture>
 
-<strong>One *beautiful* Ruby API for GPT, Claude, Gemini, and more.</strong>
+<strong>A single, beautiful Ruby framework for all major AI providers. Easily build chatbots, AI agents, RAG applications, content generators, and every AI workflow you can think of.
+</strong>
 
-Battle tested at [<picture><source media="(prefers-color-scheme: dark)" srcset="https://chatwithwork.com/logotype-dark.svg"><img src="https://chatwithwork.com/logotype.svg" alt="Chat with Work" height="30" align="absmiddle"></picture>](https://chatwithwork.com) — *Your AI coworker*
+Battle tested at [<picture><source media="(prefers-color-scheme: dark)" srcset="https://chatwithwork.com/logotype-dark.svg"><img src="https://chatwithwork.com/logotype.svg" alt="Chat with Work" height="30" align="absmiddle"></picture>](https://chatwithwork.com) - *Fully private work AI*
 
 [![Gem Version](https://badge.fury.io/rb/ruby_llm.svg)](https://badge.fury.io/rb/ruby_llm)
 [![Ruby Style Guide](https://img.shields.io/badge/code_style-rubocop-brightgreen.svg)](https://github.com/rubocop/rubocop)
@@ -24,7 +25,7 @@ Battle tested at [<picture><source media="(prefers-color-scheme: dark)" srcset="
 
 Build chatbots, AI agents, RAG applications. Works with OpenAI, xAI, Anthropic, Google, AWS, local models, and any OpenAI-compatible API.
 
-## From zero to AI chat app in under two minutes
+## Build a working Ruby AI chat in two minutes
 
 https://github.com/user-attachments/assets/65422091-9338-47da-a303-92b918bd1345
 
@@ -32,7 +33,7 @@ https://github.com/user-attachments/assets/65422091-9338-47da-a303-92b918bd1345
 
 Every AI provider ships their own bloated client. Different APIs. Different response formats. Different conventions. It's exhausting.
 
-RubyLLM gives you one beautiful API for all of them. Same interface whether you're using GPT, Claude, or your local Ollama. Just three dependencies: Faraday, Zeitwerk, and Marcel. That's it.
+RubyLLM gives you one beautiful framework for all of them. Same interface whether you're using GPT, Claude, or your local Ollama. Just three dependencies: Faraday, Zeitwerk, and Marcel. That's it.
 
 ## Show me the code
 
@@ -86,9 +87,7 @@ RubyLLM.moderate "Check if this text is safe"
 ```ruby
 # Let AI use your code
 class Weather < RubyLLM::Tool
-  description "Get current weather"
-  param :latitude
-  param :longitude
+  desc "Get current weather"
 
   def execute(latitude:, longitude:)
     url = "https://api.open-meteo.com/v1/forecast?latitude=#{latitude}&longitude=#{longitude}&current=temperature_2m,wind_speed_10m"

data/lib/generators/ruby_llm/generator_helpers.rb

@@ -87,6 +87,7 @@ module RubyLLM
 
         add_association_params(params, :message, message_table_name, message_model_name,
                                owner_table: tool_call_table_name, owner_model_name: tool_call_model_name)
+        add_result_foreign_key_param(params)
 
         "acts_as_tool_call#{" #{params.join(', ')}" if params.any?}"
       end
@@ -178,6 +179,13 @@ module RubyLLM
         "#{owner_model_name.demodulize.underscore}_id"
       end
 
+      def add_result_foreign_key_param(params)
+        foreign_key = "#{tool_call_table_name.singularize}_id"
+        default_foreign_key = "#{tool_call_model_name.demodulize.underscore}_id"
+
+        params << "result_foreign_key: :#{foreign_key}" if foreign_key != default_foreign_key
+      end
+
       # Convert namespaced model names to proper table names
       # e.g., "Assistant::Chat" -> "assistant_chats" (not "assistant/chats")
       def table_name_for(model_name)

data/lib/generators/ruby_llm/install/templates/initializer.rb.tt

@@ -2,7 +2,7 @@ RubyLLM.configure do |config|
   config.openai_api_key = ENV.fetch("OPENAI_API_KEY", Rails.application.credentials.dig(:openai_api_key))
   # config.default_model = "gpt-5-nano"
 
-  # Use the new association-based acts_as API (recommended)
+  # Use the association-based acts_as API (recommended)
   config.use_new_acts_as = true
 <% if model_model_name != 'Model' -%>
 

data/lib/generators/ruby_llm/tool/templates/tool.rb.tt

@@ -1,5 +1,5 @@
 class <%= class_name %>Tool < RubyLLM::Tool
-  description "TODO: describe what this tool does"
+  desc "TODO: describe what this tool does"
 
   def execute
     # TODO: return something to the LLM.

data/lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb

@@ -6,7 +6,7 @@ require_relative '../generator_helpers'
 
 module RubyLLM
   module Generators
-    # Generator to upgrade existing RubyLLM apps to v1.7 with new Rails-like API
+    # Generator to upgrade existing RubyLLM apps to v1.7 with association-based Rails API
     class UpgradeToV17Generator < Rails::Generators::Base
       include Rails::Generators::Migration
       include RubyLLM::Generators::GeneratorHelpers
@@ -24,7 +24,7 @@ module RubyLLM
 
       argument :model_mappings, type: :array, default: [], banner: 'chat:ChatName message:MessageName ...'
 
-      desc 'Upgrades existing RubyLLM apps to v1.7 with new Rails-like API\n' \
+      desc 'Upgrades existing RubyLLM apps to v1.7 with association-based Rails API\n' \
            'Usage: bin/rails g ruby_llm:upgrade_to_v1_7 [chat:ChatName] [message:MessageName] ...'
 
       def self.next_migration_number(dirname)
@@ -78,7 +78,7 @@ module RubyLLM
         return if initializer_content.include?('config.use_new_acts_as')
 
         inject_into_file initializer_path, before: /^end/ do
-          lines = ["\n  # Enable the new Rails-like API", '  config.use_new_acts_as = true']
+          lines = ["\n  # Enable the association-based Rails API", '  config.use_new_acts_as = true']
           lines << "  config.model_registry_class = \"#{model_model_name}\"" if model_model_name != 'Model'
           lines << "\n"
           lines.join("\n")

data/lib/ruby_llm/active_record/acts_as.rb

@@ -1,39 +1,17 @@
 # frozen_string_literal: true
 
+require 'active_support/concern'
+require 'active_support/inflector'
+
 module RubyLLM
   module ActiveRecord
     # Adds chat and message persistence capabilities to ActiveRecord models.
     module ActsAs
       extend ActiveSupport::Concern
 
-      # When ActsAs is included, ensure models are loaded from database
       def self.included(base)
         super
-        # Monkey-patch Models to use database when ActsAs is active
-        RubyLLM::Models.class_eval do
-          def self.load_models
-            database_models = read_from_database
-            return database_models if database_models.any?
-
-            RubyLLM.logger.debug { 'Model registry is empty in database, falling back to JSON registry' }
-            read_from_json
-          rescue StandardError => e
-            RubyLLM.logger.debug { "Failed to load models from database: #{e.message}, falling back to JSON" }
-            read_from_json
-          end
-
-          def self.read_from_database
-            model_class = RubyLLM.config.model_registry_class
-            model_class = model_class.constantize if model_class.is_a?(String)
-            return [] unless model_class.table_exists?
-
-            model_class.all.map(&:to_llm)
-          end
-
-          def load_from_database!
-            @models = self.class.read_from_database
-          end
-        end
+        RubyLLM.config.model_registry_source ||= RubyLLM::ModelRegistry::ActiveRecordSource.new
       end
 
       class_methods do # rubocop:disable Metrics/BlockLength

data/lib/ruby_llm/active_record/acts_as_legacy.rb

@@ -1,5 +1,8 @@
 # frozen_string_literal: true
 
+require 'active_support/concern'
+require 'active_support/inflector'
+
 module RubyLLM
   module ActiveRecord
     # Adds chat and message persistence capabilities to ActiveRecord models.
@@ -160,27 +163,33 @@ module RubyLLM
         self
       end
 
-      def on_new_message(&block)
-        to_llm
+      def on_new_message(&)
+        to_llm.on_new_message(&)
+        self
+      end
 
-        existing_callback = @chat.instance_variable_get(:@on)[:new_message]
+      def on_end_message(&)
+        to_llm.on_end_message(&)
+        self
+      end
 
-        @chat.on_new_message do
-          existing_callback&.call
-          block&.call
-        end
+      def before_message(...)
+        to_llm.before_message(...)
         self
       end
 
-      def on_end_message(&block)
-        to_llm
+      def after_message(...)
+        to_llm.after_message(...)
+        self
+      end
 
-        existing_callback = @chat.instance_variable_get(:@on)[:end_message]
+      def before_tool_call(...)
+        to_llm.before_tool_call(...)
+        self
+      end
 
-        @chat.on_end_message do |msg|
-          existing_callback&.call(msg)
-          block&.call(msg)
-        end
+      def after_tool_result(...)
+        to_llm.after_tool_result(...)
         self
       end
 
@@ -213,7 +222,7 @@ module RubyLLM
       end
 
       def create_user_message(content, with: nil)
-        RubyLLM.logger.warn(
+        RubyLLM.deprecator.warn(
           '`create_user_message` is deprecated and will be removed in RubyLLM 2.0. ' \
           'Use `add_message(role: :user, content: ...)` instead.'
         )
@@ -319,8 +328,8 @@ module RubyLLM
       def setup_persistence_callbacks
         return @chat if @chat.instance_variable_get(:@_persistence_callbacks_setup)
 
-        @chat.on_new_message { persist_new_message }
-        @chat.on_end_message { |msg| persist_message_completion(msg) }
+        @chat.before_message { persist_new_message }
+        @chat.after_message { |msg| persist_message_completion(msg) }
 
         @chat.instance_variable_set(:@_persistence_callbacks_setup, true)
         @chat
@@ -383,8 +392,8 @@ module RubyLLM
           case attachment
           when ActionDispatch::Http::UploadedFile, ActiveStorage::Blob
             attachment
-          when ActiveStorage::Attached::One, ActiveStorage::Attached::Many
-            attachment.blobs
+          when ActiveStorage::Attachment, ActiveStorage::Attached::One, ActiveStorage::Attached::Many
+            active_storage_blobs(attachment)
           when Hash
             attachment.values.map { |v| prepare_for_active_storage(v) }
           else
@@ -398,16 +407,28 @@ module RubyLLM
 
         attachment = source.is_a?(RubyLLM::Attachment) ? source : RubyLLM::Attachment.new(source)
 
-        {
-          io: StringIO.new(attachment.content),
-          filename: attachment.filename,
-          content_type: attachment.mime_type
-        }
+        if attachment.active_storage?
+          active_storage_blobs(attachment.source)
+        else
+          {
+            io: StringIO.new(attachment.content),
+            filename: attachment.filename,
+            content_type: attachment.mime_type
+          }
+        end
       rescue StandardError => e
         RubyLLM.logger.warn "Failed to process attachment #{source}: #{e.message}"
         nil
       end
 
+      def active_storage_blobs(attachment)
+        case attachment
+        when ActiveStorage::Blob then attachment
+        when ActiveStorage::Attachment, ActiveStorage::Attached::One then attachment.blob
+        when ActiveStorage::Attached::Many then attachment.blobs
+        end
+      end
+
       def build_content(message, attachments)
         return message if content_like?(message)
 
@@ -473,18 +494,91 @@ module RubyLLM
       end
 
       def extract_content
-        return content unless respond_to?(:attachments) && attachments.attached?
+        text_content = if content.respond_to?(:to_plain_text)
+                         content.to_plain_text
+                       else
+                         content
+                       end
+
+        return text_content unless respond_to?(:attachments) && attachments.attached?
 
-        RubyLLM::Content.new(content).tap do |content_obj|
+        RubyLLM::Content.new(text_content).tap do |content_obj|
           @_tempfiles = []
 
-          attachments.each do |attachment|
-            tempfile = download_attachment(attachment)
-            content_obj.add_attachment(tempfile, filename: attachment.filename.to_s)
+          attachment_sources.each do |attachment, attachable|
+            add_attachment_to_content(content_obj, attachment, attachable)
           end
         end
       end
 
+      def attachment_sources
+        change = pending_attachment_change
+        return attachments.map { |attachment| [attachment, nil] } unless pending_attachment_change?(change)
+
+        change.attachments.zip(change.attachables)
+      end
+
+      def pending_attachment_change
+        attachment_changes['attachments'] if respond_to?(:attachment_changes)
+      end
+
+      def pending_attachment_change?(change)
+        change.respond_to?(:attachments) && change.respond_to?(:attachables)
+      end
+
+      def add_attachment_to_content(content_obj, attachment, attachable)
+        if pending_upload_attachable?(attachable)
+          add_pending_upload_attachment(content_obj, attachable)
+        else
+          tempfile = download_attachment(attachment)
+          content_obj.add_attachment(tempfile, filename: attachment.filename.to_s)
+        end
+      end
+
+      def pending_upload_attachable?(attachable)
+        return false if attachable.nil? || attachable.is_a?(String)
+        return false if instance_of_class?(attachable, 'ActiveStorage::Blob')
+
+        uploaded_file?(attachable) || active_storage_upload_hash?(attachable) ||
+          attachable.is_a?(File) || pathname?(attachable)
+      end
+
+      def uploaded_file?(attachable)
+        instance_of_class?(attachable, 'ActionDispatch::Http::UploadedFile') ||
+          instance_of_class?(attachable, 'Rack::Test::UploadedFile')
+      end
+
+      def active_storage_upload_hash?(attachable)
+        attachable.is_a?(Hash) && attachment_hash_io(attachable).present?
+      end
+
+      def pathname?(attachable)
+        defined?(Pathname) && attachable.is_a?(Pathname)
+      end
+
+      def add_pending_upload_attachment(content_obj, attachable)
+        if attachable.is_a?(Hash)
+          content_obj.add_attachment(attachment_hash_io(attachable), filename: attachment_hash_filename(attachable))
+        else
+          content_obj.add_attachment(attachable)
+        end
+      end
+
+      def attachment_hash_io(attachable)
+        attachable[:io] || attachable['io']
+      end
+
+      def attachment_hash_filename(attachable)
+        filename = attachable[:filename] || attachable['filename']
+        filename&.to_s
+      end
+
+      def instance_of_class?(object, class_name)
+        Object.const_get(class_name).then { |klass| object.is_a?(klass) }
+      rescue NameError
+        false
+      end
+
       def download_attachment(attachment)
         ext = File.extname(attachment.filename.to_s)
         basename = File.basename(attachment.filename.to_s, ext)

data/lib/ruby_llm/active_record/chat_methods.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'active_support/concern'
+
 module RubyLLM
   module ActiveRecord
     # Methods mixed into chat models.
@@ -63,8 +65,8 @@ module RubyLLM
           m.context_window = model_info.context_window
           m.max_output_tokens = model_info.max_output_tokens
           m.capabilities = model_info.capabilities || []
-          m.modalities = model_info.modalities || {}
-          m.pricing = model_info.pricing || {}
+          m.modalities = model_info.modalities.to_h
+          m.pricing = model_info.pricing.to_h
           m.metadata = model_info.metadata || {}
         end
 
@@ -154,27 +156,33 @@ module RubyLLM
         self
       end
 
-      def on_new_message(&block)
-        to_llm
+      def on_new_message(&)
+        to_llm.on_new_message(&)
+        self
+      end
 
-        existing_callback = @chat.instance_variable_get(:@on)[:new_message]
+      def on_end_message(&)
+        to_llm.on_end_message(&)
+        self
+      end
 
-        @chat.on_new_message do
-          existing_callback&.call
-          block&.call
-        end
+      def before_message(...)
+        to_llm.before_message(...)
         self
       end
 
-      def on_end_message(&block)
-        to_llm
+      def after_message(...)
+        to_llm.after_message(...)
+        self
+      end
 
-        existing_callback = @chat.instance_variable_get(:@on)[:end_message]
+      def before_tool_call(...)
+        to_llm.before_tool_call(...)
+        self
+      end
 
-        @chat.on_end_message do |msg|
-          existing_callback&.call(msg)
-          block&.call(msg)
-        end
+      def after_tool_result(...)
+        to_llm.after_tool_result(...)
         self
       end
 
@@ -208,6 +216,10 @@ module RubyLLM
         message_record
       end
 
+      def cost
+        RubyLLM::Cost.aggregate(messages_association.map(&:cost))
+      end
+
       def create_user_message(content, with: nil)
         add_message(role: :user, content: build_content(content, with))
       end
@@ -258,8 +270,8 @@ module RubyLLM
       def setup_persistence_callbacks
         return @chat if @chat.instance_variable_get(:@_persistence_callbacks_setup)
 
-        @chat.on_new_message { persist_new_message }
-        @chat.on_end_message { |msg| persist_message_completion(msg) }
+        @chat.before_message { persist_new_message }
+        @chat.after_message { |msg| persist_message_completion(msg) }
 
         @chat.instance_variable_set(:@_persistence_callbacks_setup, true)
         @chat
@@ -402,8 +414,8 @@ module RubyLLM
           case attachment
           when ActionDispatch::Http::UploadedFile, ActiveStorage::Blob
             attachment
-          when ActiveStorage::Attached::One, ActiveStorage::Attached::Many
-            attachment.blobs
+          when ActiveStorage::Attachment, ActiveStorage::Attached::One, ActiveStorage::Attached::Many
+            active_storage_blobs(attachment)
           when Hash
             attachment.values.map { |v| prepare_for_active_storage(v) }
           else
@@ -418,10 +430,7 @@ module RubyLLM
         attachment = source.is_a?(RubyLLM::Attachment) ? source : RubyLLM::Attachment.new(source)
 
         if attachment.active_storage?
-          case attachment.source
-          when ActiveStorage::Blob then attachment.source
-          when ActiveStorage::Attached::One, ActiveStorage::Attached::Many then attachment.source.blobs
-          end
+          active_storage_blobs(attachment.source)
         else
           {
             io: StringIO.new(attachment.content),
@@ -434,6 +443,14 @@ module RubyLLM
         nil
       end
 
+      def active_storage_blobs(attachment)
+        case attachment
+        when ActiveStorage::Blob then attachment
+        when ActiveStorage::Attachment, ActiveStorage::Attached::One then attachment.blob
+        when ActiveStorage::Attached::Many then attachment.blobs
+        end
+      end
+
       def build_content(message, attachments)
         return message if content_like?(message)
 

data/lib/ruby_llm/active_record/message_methods.rb

@@ -1,5 +1,8 @@
 # frozen_string_literal: true
 
+require 'active_support/concern'
+require 'ruby_llm/active_record/payload_helpers'
+
 module RubyLLM
   module ActiveRecord
     # Methods mixed into message models.
@@ -40,6 +43,18 @@ module RubyLLM
         )
       end
 
+      def cost
+        RubyLLM::Cost.new(tokens:, model: model_association)
+      end
+
+      def cache_read_tokens
+        cached_value
+      end
+
+      def cache_write_tokens
+        cache_creation_value
+      end
+
       def to_partial_path
         partial_prefix = self.class.name.underscore.pluralize
         role_partial = if to_llm.tool_call?
@@ -99,20 +114,88 @@ module RubyLLM
       def extract_content
         return RubyLLM::Content::Raw.new(content_raw) if has_attribute?(:content_raw) && content_raw.present?
 
-        content_value = self[:content]
+        content_value = content
+        content_value = content_value.to_plain_text if content_value.respond_to?(:to_plain_text)
 
         return content_value unless respond_to?(:attachments) && attachments.attached?
 
         RubyLLM::Content.new(content_value).tap do |content_obj|
           @_tempfiles = []
 
-          attachments.each do |attachment|
-            tempfile = download_attachment(attachment)
-            content_obj.add_attachment(tempfile, filename: attachment.filename.to_s)
+          attachment_sources.each do |attachment, attachable|
+            add_attachment_to_content(content_obj, attachment, attachable)
           end
         end
       end
 
+      def attachment_sources
+        change = pending_attachment_change
+        return attachments.map { |attachment| [attachment, nil] } unless pending_attachment_change?(change)
+
+        change.attachments.zip(change.attachables)
+      end
+
+      def pending_attachment_change
+        attachment_changes['attachments'] if respond_to?(:attachment_changes)
+      end
+
+      def pending_attachment_change?(change)
+        change.respond_to?(:attachments) && change.respond_to?(:attachables)
+      end
+
+      def add_attachment_to_content(content_obj, attachment, attachable)
+        if pending_upload_attachable?(attachable)
+          add_pending_upload_attachment(content_obj, attachable)
+        else
+          tempfile = download_attachment(attachment)
+          content_obj.add_attachment(tempfile, filename: attachment.filename.to_s)
+        end
+      end
+
+      def pending_upload_attachable?(attachable)
+        return false if attachable.nil? || attachable.is_a?(String)
+        return false if instance_of_class?(attachable, 'ActiveStorage::Blob')
+
+        uploaded_file?(attachable) || active_storage_upload_hash?(attachable) ||
+          attachable.is_a?(File) || pathname?(attachable)
+      end
+
+      def uploaded_file?(attachable)
+        instance_of_class?(attachable, 'ActionDispatch::Http::UploadedFile') ||
+          instance_of_class?(attachable, 'Rack::Test::UploadedFile')
+      end
+
+      def active_storage_upload_hash?(attachable)
+        attachable.is_a?(Hash) && attachment_hash_io(attachable).present?
+      end
+
+      def pathname?(attachable)
+        defined?(Pathname) && attachable.is_a?(Pathname)
+      end
+
+      def add_pending_upload_attachment(content_obj, attachable)
+        if attachable.is_a?(Hash)
+          content_obj.add_attachment(attachment_hash_io(attachable), filename: attachment_hash_filename(attachable))
+        else
+          content_obj.add_attachment(attachable)
+        end
+      end
+
+      def attachment_hash_io(attachable)
+        attachable[:io] || attachable['io']
+      end
+
+      def attachment_hash_filename(attachable)
+        filename = attachable[:filename] || attachable['filename']
+        filename&.to_s
+      end
+
+      def instance_of_class?(object, class_name)
+        Object.const_get(class_name).then { |klass| object.is_a?(klass) }
+      rescue NameError
+        false
+      end
+
       def download_attachment(attachment)
         ext = File.extname(attachment.filename.to_s)
         basename = File.basename(attachment.filename.to_s, ext)

data/lib/ruby_llm/active_record/model_methods.rb

@@ -1,5 +1,8 @@
 # frozen_string_literal: true
 
+require 'active_support/concern'
+require 'active_support/core_ext/module/delegation'
+
 module RubyLLM
   module ActiveRecord
     # Methods mixed into model registry models.
@@ -10,15 +13,7 @@ module RubyLLM
         def refresh!
           RubyLLM.models.refresh!
 
-          transaction do
-            RubyLLM.models.all.each do |model_info|
-              model = find_or_initialize_by(
-                model_id: model_info.id,
-                provider: model_info.provider
-              )
-              model.update!(from_llm_attributes(model_info))
-            end
-          end
+          save_to_database
         end
 
         def save_to_database
@@ -76,8 +71,11 @@ module RubyLLM
 
       delegate :supports?, :supports_vision?, :supports_functions?, :type,
                :input_price_per_million, :output_price_per_million,
+               :cache_read_input_price_per_million, :cache_write_input_price_per_million,
+               :cached_input_price_per_million, :cache_creation_input_price_per_million,
                :function_calling?, :structured_output?, :batch?,
                :reasoning?, :citations?, :streaming?, :provider_class, :label,
+               :cost_for,
                to: :to_llm
     end
   end

data/lib/ruby_llm/active_record/payload_helpers.rb

@@ -1,5 +1,8 @@
 # frozen_string_literal: true
 
+require 'active_support/core_ext/object/blank'
+require 'json'
+
 module RubyLLM
   module ActiveRecord
     # Shared helpers for parsing serialized payloads on ActiveRecord-backed models.