ruby_llm 2.0.0.rc1 → 2.0.0.rc2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3e6a6455ecb1c6bf276d863d9e1f181f1a1d3ea87092800a01522c1fb9be35fc
-  data.tar.gz: 39554eb45edfde45e43e0a9eb9afe2a5e07cf6601b9b8076e19065493e62e0fd
+  metadata.gz: ef3c927159add14551e31ce03609ce42e150f9f161af9b68602f9c0e080fc31d
+  data.tar.gz: b894b99f6d6e541c643d76810b9542364f9d0beef4d00efd7b87b80d36c071af
 SHA512:
-  metadata.gz: c92879f31f983745665f2a08db2e9b01a293d40ebd2a0e68ee899e1af4b77f7d4c6767c700bcbdecc85088a43a0fae58a11b617c4409e1b04b73d44a7f59b4da
-  data.tar.gz: 853e3cdde61afafbfdff5ff84cdc3c9878002f4da1b81e45e1b105b0caf759587647a03cb05d85d806a5e3c91c48ec426d24e356615914802860cb1cdec9c6cb
+  metadata.gz: 7dacd51baf534c7b226e1db52f09421b72d9939ec17af6cfd4b18db8bd7d9fbe2a22b488a8571d64f2194d46ed82ba045a726a309465b448f3a31a077a1f98c1
+  data.tar.gz: 90f866305245b759ae821cd8932d518c8648f5a3afce588028b820ced1f8aa7b05ac5de7c55a6b0f89e5aac80052d348fbc3fec8cf3b9efc83dec5effa32b260

data/README.md

@@ -38,7 +38,7 @@ In Rails, the API works on your own Chat and Message records, with Active Storag
 
 ## Show me the code
 
-These examples use **2.0.0.rc1**. Follow [Getting Started](https://rubyllm.com/next/getting-started/) to install it and configure the providers you want to try.
+These examples use **2.0.0.rc2**. Follow [Getting Started](https://rubyllm.com/next/getting-started/) to install it and configure the providers you want to try.
 
 ```ruby
 # Just ask questions
@@ -195,7 +195,7 @@ response.parsed
 Install the 2.0 release candidate:
 
 ```bash
-bundle add ruby_llm --version 2.0.0.rc1
+bundle add ruby_llm --version 2.0.0.rc2
 ```
 
 Configure a provider in your script, or in `config/initializers/ruby_llm.rb` in Rails:

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

@@ -21,7 +21,5 @@ class <%= create_migration_class_name(message_table_name) %> < ActiveRecord::Mig
       t.string :finish_reason
       t.timestamps
     end
-
-    add_index :<%= message_table_name %>, :role
   end
 end

data/lib/generators/ruby_llm/provider/templates/gem/gemfile.erb

@@ -5,9 +5,7 @@ source 'https://rubygems.org'
 gemspec
 
 group :development do
-  if RUBY_ENGINE == 'ruby' && Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('3.2')
-    gem 'archspec', github: 'crmne/archspec', tag: 'v1.1.0.rc1'
-  end
+  gem 'archspec' if RUBY_ENGINE == 'ruby' && Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('3.2')
   gem 'bundler', '>= 2.0'
   gem 'dotenv'
   gem 'flay'

data/lib/generators/ruby_llm/provider/templates/gem/gemspec.erb

@@ -24,5 +24,5 @@ Gem::Specification.new do |spec|
                %w[.flayignore .overcommit.yml .rspec .rubocop.yml Archspec.rb LICENSE README.md]
   spec.require_paths = ['lib']
 
-  spec.add_dependency 'ruby_llm', '>= 2.0'
+  spec.add_dependency 'ruby_llm', '>= 2.0.0.rc1'
 end

data/lib/generators/ruby_llm/upgrade/templates/cleanup_v2_upgrade.rb.tt

@@ -46,6 +46,7 @@ class <%= cleanup_migration_class_name %> < ActiveRecord::Migration<%= migration
 
   def remove_legacy_message_columns
     table = :<%= message_table_name %>
+    remove_matching_indexes(table, :role)
     reference_columns = %i[<%= v1_model_foreign_key %> <%= v1_tool_call_foreign_key %>]
     reference_columns.each do |column|
       remove_column_foreign_keys(table, column)

data/lib/ruby_llm/active_record/attachment_helpers.rb

@@ -96,11 +96,17 @@ module RubyLLM
 
       def attachment_sources
         change = pending_attachment_change
-        return attachments.map { |attachment| [attachment, nil] } unless pending_attachment_change?(change)
+        return attachments_with_blobs.map { |attachment| [attachment, nil] } unless pending_attachment_change?(change)
 
         change.attachments.zip(change.attachables)
       end
 
+      def attachments_with_blobs
+        records = attachments.to_a
+        ::ActiveRecord::Associations::Preloader.new(records: records, associations: :blob).call
+        records
+      end
+
       def pending_attachment_change
         attachment_changes['attachments'] if respond_to?(:attachment_changes)
       end

data/lib/ruby_llm/active_record/chat_methods.rb

@@ -777,11 +777,16 @@ module RubyLLM
         return messages unless assoc.respond_to?(:klass)
 
         associations = %i[ruby_llm_tool_calls ruby_llm_parent_tool_call ruby_llm_usages]
+        associations << { attachments_attachments: :blob } if attachment_association?(assoc.klass)
 
         ::ActiveRecord::Associations::Preloader.new(records: messages, associations: associations).call
         messages
       end
 
+      def attachment_association?(message_class)
+        message_class.reflect_on_association(:attachments_attachments).present?
+      end
+
       def build_llm_chat
         chat = (context || RubyLLM).chat(
           model: model_id,