ruby_llm 2.0.0.rc2 → 2.0.0.rc3

data/lib/ruby_llm/protocols/bedrock/async_videos.rb

@@ -64,7 +64,8 @@ module RubyLLM
           prefix = job.raw.dig('outputDataConfig', 's3OutputDataConfig', 's3Uri')
           raise Error, 'Bedrock video job has no output S3 URI' unless prefix
 
-          uris = @provider.list_file_uris("#{prefix.sub(%r{/+\z}, '')}/")
+          prefix_length = (prefix.rindex(%r{[^/]}) || -1) + 1
+          uris = @provider.list_file_uris("#{prefix[0, prefix_length]}/")
           videos = uris.select { |uri| uri.downcase.end_with?('.mp4') }
           raise Error, 'Expected exactly one MP4 in the Bedrock video output' unless videos.one?
 

data/lib/ruby_llm/protocols/chat_completions/rerank.rb

@@ -38,10 +38,13 @@ module RubyLLM
         def parse_rerank_results(data, documents = [])
           Array(data['results']).map do |result|
             index = result['index']
+            unless valid_rerank_index?(index, documents)
+              raise Error, 'Rerank endpoint returned an invalid document index'
+            end
 
             RubyLLM::Rerank::Result.new(
               index: index,
-              document: rerank_document(result['document']) || (index && documents[index]),
+              document: rerank_document(result['document']) || documents[index],
               score: result['relevance_score']
             )
           end
@@ -50,6 +53,10 @@ module RubyLLM
         def rerank_document(document)
           document.is_a?(Hash) ? document['text'] : document
         end
+
+        def valid_rerank_index?(index, documents)
+          index.is_a?(Integer) && index.between?(0, documents.length - 1)
+        end
       end
     end
   end

data/lib/ruby_llm/protocols/cohere/rerank.rb

@@ -38,14 +38,21 @@ module RubyLLM
         def parse_rerank_results(data, documents)
           Array(data['results']).map do |result|
             index = result['index']
+            unless valid_rerank_index?(index, documents)
+              raise Error, 'Cohere reranking returned an invalid document index'
+            end
 
             RubyLLM::Rerank::Result.new(
               index: index,
-              document: result.dig('document', 'text') || (index && documents[index]),
+              document: result.dig('document', 'text') || documents[index],
               score: result['relevance_score']
             )
           end
         end
+
+        def valid_rerank_index?(index, documents)
+          index.is_a?(Integer) && index.between?(0, documents.length - 1)
+        end
       end
     end
   end

data/lib/ruby_llm/protocols/gemini/embedding_batches.rb

@@ -60,6 +60,11 @@ module RubyLLM
           metadata = responses.first.fetch('metadata')
           return if responses.size < metadata.fetch('embedding_count')
 
+          positions = responses.map { |inline| inline.dig('metadata', 'embedding_index') }
+          unless positions.sort == (0...responses.size).to_a
+            return [index, nil, batch_failure(key, 'Invalid or duplicate embedding record positions')]
+          end
+
           vectors = embedding_batch_vectors(responses)
           return [index, nil, batch_failure(key, 'Gemini returned no embedding')] unless vectors
 

data/lib/ruby_llm/protocols/interactions/tools.rb

@@ -31,7 +31,9 @@ module RubyLLM
         end
 
         def parse_interaction_arguments(arguments)
-          arguments.is_a?(String) ? JSON.parse(arguments) : arguments || {}
+          return {} if arguments.nil? || (arguments.is_a?(String) && arguments.empty?)
+
+          arguments.is_a?(String) ? JSON.parse(arguments) : arguments
         rescue JSON::ParserError => e
           raise ToolCallParseError.new(finish_reason: :tool_calls), cause: e
         end

data/lib/ruby_llm/providers/deepseek/responses.rb

@@ -8,7 +8,6 @@ module RubyLLM
       # instead of summaries, both in responses and in the stream.
       class Responses < Protocols::Responses
         SERVER_TOOL_ALIASES = {
-          web_search: { tool: { type: 'web_search' } },
           apply_patch: { tool: { type: 'custom', name: 'apply_patch' } }
         }.freeze
 

data/lib/ruby_llm/providers/mistral/ocr.rb

@@ -22,7 +22,7 @@ module RubyLLM
         end
 
         def ocr_document_part(attachment)
-          reference = attachment.url? ? attachment.source.to_s : attachment.for_llm
+          reference = attachment.url? ? attachment.source.to_s : ocr_data_uri(attachment)
 
           if attachment.image?
             { type: 'image_url', image_url: reference }
@@ -31,6 +31,10 @@ module RubyLLM
           end
         end
 
+        def ocr_data_uri(attachment)
+          "data:#{attachment.mime_type};base64,#{attachment.encoded}"
+        end
+
         def parse_ocr_response(response, model:)
           data = response.body
 

data/lib/ruby_llm/version.rb

@@ -2,5 +2,5 @@
 
 module RubyLLM
   # The version of the ruby_llm gem, as a string.
-  VERSION = '2.0.0.rc2'
+  VERSION = '2.0.0.rc3'
 end

data/lib/tasks/ruby_llm.rake

@@ -7,7 +7,7 @@ namespace :ruby_llm do
       task action => :environment do
         require 'generators/ruby_llm/upgrade/upgrade_migration'
 
-        RubyLLM::Generators::UpgradeMigration.new.public_send(action)
+        RubyLLM::Generators::UpgradeMigration.for.public_send(action)
         puts "RubyLLM copy upgrade: #{action} completed"
       end
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 2.0.0.rc2
+  version: 2.0.0.rc3
 platform: ruby
 authors:
 - Carmine Paolino
@@ -275,6 +275,11 @@ files:
 - lib/generators/ruby_llm/tool/templates/tool_call.html.erb.tt
 - lib/generators/ruby_llm/tool/templates/tool_result.html.erb.tt
 - lib/generators/ruby_llm/tool/tool_generator.rb
+- lib/generators/ruby_llm/upgrade/legacy_content_sql.rb
+- lib/generators/ruby_llm/upgrade/online_copy_migration.rb
+- lib/generators/ruby_llm/upgrade/online_copy_migration/data.rb
+- lib/generators/ruby_llm/upgrade/online_copy_migration/journal.rb
+- lib/generators/ruby_llm/upgrade/online_copy_migration/verification.rb
 - lib/generators/ruby_llm/upgrade/templates/backfill_v2_data.rb.tt
 - lib/generators/ruby_llm/upgrade/templates/cleanup_v2_upgrade.rb.tt
 - lib/generators/ruby_llm/upgrade/templates/finish_v2_upgrade.rb.tt
@@ -612,7 +617,7 @@ metadata:
   funding_uri: https://github.com/sponsors/crmne
   rubygems_mfa_required: 'true'
 post_install_message: |
-  RubyLLM 2.0.0.rc2
+  RubyLLM 2.0.0.rc3
 
     2.0 renames several APIs and changes what message content returns. Coming
     from 1.x? Read the upgrade guide before you boot: