fix handling ActiveStorage::Attached::Many

Author: manuelmeurer

lib/ruby_llm/active_record/acts_as_legacy.rb

@@ -302,10 +302,8 @@ def prepare_for_active_storage(attachments)
           case attachment
           when ActionDispatch::Http::UploadedFile, ActiveStorage::Blob
             attachment
-          when ActiveStorage::Attached::One
+          when ActiveStorage::Attached::One, ActiveStorage::Attachment
             attachment.blob
-          when ActiveStorage::Attached::Many
-            attachment.blobs
           when Hash
             attachment.values.map { |v| prepare_for_active_storage(v) }
           else

lib/ruby_llm/active_record/chat_methods.rb

@@ -307,10 +307,8 @@ def prepare_for_active_storage(attachments)
           case attachment
           when ActionDispatch::Http::UploadedFile, ActiveStorage::Blob
             attachment
-          when ActiveStorage::Attached::One
+          when ActiveStorage::Attached::One, ActiveStorage::Attachment
             attachment.blob
-          when ActiveStorage::Attached::Many
-            attachment.blobs
           when Hash
             attachment.values.map { |v| prepare_for_active_storage(v) }
           else

spec/ruby_llm/active_record/acts_as_attachment_spec.rb

@@ -65,29 +65,29 @@ def uploaded_file(path, type)
       expect(response.content).to be_present
     end
 
-    # it 'handles ActiveStorage::Attached::Many in ask method' do
-    #   chat = Chat.create!(model: model)
-
-    #   document = Document.create!(title: 'Test Document')
-    #   document.files.attach(
-    #     io: File.open(image_path),
-    #     filename: 'ruby.png',
-    #     content_type: 'image/png'
-    #   )
-    #   document.files.attach(
-    #     io: File.open(pdf_path),
-    #     filename: 'sample.pdf',
-    #     content_type: 'application/pdf'
-    #   )
-
-    #   response = chat.ask('Analyze these', with: document.files)
-
-    #   user_message = chat.messages.find_by(role: 'user')
-    #   expect(user_message.attachments.count).to eq(2)
-    #   filenames = user_message.attachments.map { |a| a.filename.to_s }.sort
-    #   expect(filenames).to eq(['ruby.png', 'sample.pdf'])
-    #   expect(response.content).to be_present
-    # end
+    it 'handles ActiveStorage::Attached::Many in ask method' do
+      chat = Chat.create!(model: model)
+
+      document = Document.create!(title: 'Test Document')
+      document.files.attach(
+        io: File.open(image_path),
+        filename: 'ruby.png',
+        content_type: 'image/png'
+      )
+      document.files.attach(
+        io: File.open(pdf_path),
+        filename: 'sample.pdf',
+        content_type: 'application/pdf'
+      )
+
+      response = chat.ask('Analyze these', with: document.files)
+
+      user_message = chat.messages.find_by(role: 'user')
+      expect(user_message.attachments.count).to eq(2)
+      filenames = user_message.attachments.map { |a| a.filename.to_s }.sort
+      expect(filenames).to eq(['ruby.png', 'sample.pdf'])
+      expect(response.content).to be_present
+    end
   end
 
   describe 'attachment types' do