Add video file support to attachments

Author: arnodirlam

lib/ruby_llm/attachment.rb

@@ -76,6 +76,7 @@ def for_llm
 
     def type
       return :image if image?
+      return :video if video?
       return :audio if audio?
       return :pdf if pdf?
       return :text if text?
@@ -87,6 +88,10 @@ def image?
       RubyLLM::MimeType.image? mime_type
     end
 
+    def video?
+      RubyLLM::MimeType.video? mime_type
+    end
+
     def audio?
       RubyLLM::MimeType.audio? mime_type
     end

lib/ruby_llm/mime_type.rb

@@ -15,6 +15,10 @@ def image?(type)
       type.start_with?('image/')
     end
 
+    def video?(type)
+      type.start_with?('video/')
+    end
+
     def audio?(type)
       type.start_with?('audio/')
     end

spec/fixtures/ruby.mp4

@@ -89,6 +89,22 @@ def uploaded_file(path, type)
       expect(attachment.type).to eq(:image)
     end
 
+    it 'handles videos' do # rubocop:disable RSpec/ExampleLength
+      video_path = File.expand_path('../../fixtures/ruby.mp4', __dir__)
+      chat = Chat.create!(model_id: model)
+      message = chat.messages.create!(role: 'user', content: 'Video test')
+
+      message.attachments.attach(
+        io: File.open(video_path),
+        filename: 'test.mp4',
+        content_type: 'video/mp4'
+      )
+
+      llm_message = message.to_llm
+      attachment = llm_message.content.attachments.first
+      expect(attachment.type).to eq(:video)
+    end
+
     it 'handles PDFs' do
       chat = Chat.create!(model: model)
       message = chat.messages.create!(role: 'user', content: 'PDF test')