ruby_llm 2.1.0 → 2.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2f0478372c3c7c58cd1452546f21068161f553949b064377738c28e74b67da9d
-  data.tar.gz: 2f8f63b90a885326904724a95074934877251b589b41ba00d9366dd72d5b3d69
+  metadata.gz: 8000141c827d6b8f899bcf2ec5a23153ecf83f8196fcd35cee7af8de2f9e58e4
+  data.tar.gz: '069b322ad8df77cd46ee2dd9b72ec02578281ff9e6c96b349e46aff911ee85b8'
 SHA512:
-  metadata.gz: 4028eea4e3515ffa779ada1d1464ab5ed4fa4c21418c1ebaa8fe09bb0da11fd6998f6b0132dde1cd59f876b54a43c03f89507be73e7d3e39b3b3794fbb33f3c5
-  data.tar.gz: e57218438a1479c2d4974926d4afdf509fa18f2c57ffe0524e116feea919d1fb337d80f883dca30205a442ed754faafafb076bcc780eceb2f86b4c6372a47321
+  metadata.gz: 6d651244192242fc5f06cf8126932659b46dec1228ed97e3a9876dd7ee581f8972d42fbaca209698b1c42e1704ebb766ca07b450cfd7651d9192f8d46d3f8c42
+  data.tar.gz: 9511e7a1c10a8b73cebbefb59a26be1fd7dcec10c369f57d60b7a348d180a8037700826400d9da41b630d190b940aa73ba7c5cbdb7fbd0f4b6cffbcbe563d732

data/README.md

@@ -21,6 +21,8 @@ Battle tested at [<picture><source media="(prefers-color-scheme: dark)" srcset="
 <a href="https://trendshift.io/repositories/13640" target="_blank"><img src="https://trendshift.io/api/badge/repositories/13640" alt="crmne%2Fruby_llm | Trendshift" style="width: 250px; height: 55px;" width="250" height="55"/></a>
 </div>
 
+https://github.com/user-attachments/assets/519fcb20-ab8c-4af3-9185-0953b3ba7b81
+
 > [!NOTE]
 > Using RubyLLM? [Share your story](https://tally.so/r/3Na02p)! Takes 5 minutes.
 
@@ -34,7 +36,7 @@ Build with the models you want. Move between hosted and local providers without
 
 ## Start with one line. Add files, tools, and agents
 
-These examples use **2.1.0**. Follow [Getting Started](https://rubyllm.com/getting-started/) to install it and configure the providers you want to try. For 1.x, use the [1.x docs](https://rubyllm.com/v1/).
+These examples use **2.1.1**. Follow [Getting Started](https://rubyllm.com/getting-started/) to install it and configure the providers you want to try. For 1.x, use the [1.x docs](https://rubyllm.com/v1/).
 
 ```ruby
 # Just ask
@@ -249,7 +251,7 @@ Agents, evaluations, workflows, RAG, images, audio, and video. Built in, with us
 Install RubyLLM 2.1:
 
 ```bash
-bundle add ruby_llm --version 2.1.0
+bundle add ruby_llm --version 2.1.1
 ```
 
 Configure a provider in your script, or in `config/initializers/ruby_llm.rb` in Rails:

data/lib/ruby_llm/active_record/chat_methods.rb

@@ -741,10 +741,14 @@ module RubyLLM
       end
 
       def link_usage_entries(message)
-        message.ruby_llm_usage_entries.each do |entry|
-          record = usage_records_by_entry[entry]
-          record&.update!(message: @message)
+        Usage.transaction(requires_new: true) do
+          message.ruby_llm_usage_entries.each do |entry|
+            record = usage_records_by_entry[entry]
+            record&.update!(message: @message)
+          end
         end
+      rescue StandardError => e
+        RubyLLM.logger.warn("RubyLLM could not link usage to message #{@message.id}: #{e.class}: #{e.message}")
       end
 
       def usage_records_by_entry

data/lib/ruby_llm/chat.rb

@@ -1471,7 +1471,11 @@ module RubyLLM
     end
 
     def discard_missing_uploads(error)
-      @provider.discard_missing_uploads(messages, error, model: @model, protocol: @protocol)
+      upload_messages = messages.map do |message|
+        replacements = message.attachments.filter_map { |attachment| @attachment_replacements[attachment] }
+        message.with_attachments(message.attachments + replacements)
+      end
+      @provider.discard_missing_uploads(upload_messages, error, model: @model, protocol: @protocol)
     end
 
     def record_usage_entry(entry)

data/lib/ruby_llm/mcp/client.rb

@@ -6,8 +6,10 @@ module RubyLLM
     # 2026-07-28 revision and falls back to the initialize handshake for
     # servers that predate it, declaring only its extensions so those
     # servers never call back. A server that rejects 2026-07-28 while
-    # listing it gets the request once more; one that answers the handshake
-    # with a revision missing from LEGACY_VERSIONS is disconnected.
+    # listing it gets the request once more; one that rejects it while
+    # listing a revision from LEGACY_VERSIONS, or without listing any,
+    # gets the handshake instead; one that answers the handshake with a
+    # revision missing from LEGACY_VERSIONS is disconnected.
     #
     # Servers that predate 2026-07-28 may announce that their tools,
     # prompts, or resources changed on the stream of any request. Those
@@ -103,7 +105,7 @@ module RubyLLM
         result = call('server/discover', timeout: DISCOVERY_TIMEOUT)
         result if Array(result['supportedVersions']).include?(VERSION)
       rescue Error => e
-        raise if MODERN_ERRORS.include?(e.code)
+        raise if MODERN_ERRORS.include?(e.code) && !predates_revision?(e)
       end
 
       def handshake
@@ -188,8 +190,19 @@ module RubyLLM
       end
 
       def offers_version?(error)
+        modern? && error.code == UNSUPPORTED_VERSION && supported_versions(error).include?(VERSION)
+      end
+
+      def predates_revision?(error)
+        return false unless error.code == UNSUPPORTED_VERSION
+
+        supported = supported_versions(error)
+        supported.empty? || supported.intersect?(LEGACY_VERSIONS)
+      end
+
+      def supported_versions(error)
         supported = error.data['supported'] if error.data.is_a?(Hash)
-        modern? && error.code == UNSUPPORTED_VERSION && Array(supported).include?(VERSION)
+        Array(supported)
       end
 
       def message(method, params = {}, id: nil)

data/lib/ruby_llm/protocol.rb

@@ -753,12 +753,16 @@ module RubyLLM
     end
 
     def missing_uploads(messages, error, scope)
-      uploads = messages.flat_map(&:attachments).filter_map do |attachment|
+      uploads = attachment_uploads(messages, scope)
+      named = uploads.select { |_, upload| names_file?(error, upload.id) }
+      named.empty? && error.response&.status == 404 ? uploads : named
+    end
+
+    def attachment_uploads(messages, scope)
+      messages.flat_map(&:attachments).uniq(&:object_id).filter_map do |attachment|
         upload = attachment.provider_uploads[scope]
         [attachment, upload] if upload
       end
-      named = uploads.select { |_, upload| names_file?(error, upload.id) }
-      named.empty? && error.response&.status == 404 ? uploads : named
     end
 
     def names_file?(error, id)

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

@@ -29,6 +29,27 @@ module RubyLLM
         def parse_batch_completion_response(body)
           parse_embedding_response(Response.new(body), model: body['model'], text: nil)
         end
+
+        def parse_batch_result(line)
+          body = line.dig('response', 'body')
+          return super unless body.is_a?(Hash) && body['data'].is_a?(Array)
+          unless valid_embedding_positions?(body['data'])
+            return [batch_result_index(line['custom_id']), nil,
+                    batch_failure(line['custom_id'], 'Invalid or duplicate embedding record positions')]
+          end
+
+          ordered = body.merge('data' => body['data'].sort_by { |row| row['index'] })
+          super(line.merge('response' => line['response'].merge('body' => ordered)))
+        end
+
+        # A row's index names the input its vector belongs to, so rows are put
+        # back in that order and positions that are not exactly 0...N fail the
+        # request rather than pairing vectors with the wrong texts.
+        def valid_embedding_positions?(rows)
+          positions = rows.map { |row| row.is_a?(Hash) ? row['index'] : nil }
+
+          positions.all?(Integer) && positions.sort == (0...positions.size).to_a
+        end
       end
     end
   end

data/lib/ruby_llm/providers/bedrock/models.rb

@@ -185,6 +185,7 @@ module RubyLLM
         def resolve_registry_id(model_id, models, config)
           region = config.bedrock_region.to_s
           return model_id if region.empty?
+          return model_id if region_prefixed?(model_id)
           return model_id if mantle_model_id?(model_id)
 
           candidate = registered_profile_candidate(model_id, models, region)
@@ -197,8 +198,6 @@ module RubyLLM
         def registered_profile_candidate(model_id, models, region)
           region_prefix_candidates(region).each do |prefix|
             prefixed = prefixed_with(model_id, prefix)
-            next if prefixed == model_id
-
             candidate = models.all_including_unlisted.find { |m| m.provider == 'bedrock' && m.id == prefixed }
             return candidate if candidate
           end

data/lib/ruby_llm/providers/mistral/chat_completions/batches.rb

@@ -99,9 +99,13 @@ module RubyLLM
             custom_id, shape = line['custom_id'].split(':', 2)
             index = batch_result_index(custom_id)
             response = line['response']
+            body = response && response['body']
 
-            if response && response['body']
-              body = response['body']
+            if body && body['data'].is_a?(Array) && !embedding_positions?(body['data'])
+              return [index, nil, batch_failure(line['custom_id'], 'Invalid or duplicate embedding record positions')]
+            end
+
+            if body
               [index, parse_mistral_batch_body(body, shape:)]
             else
               [index, nil, batch_failure(line['custom_id'], batch_error_message(line))]
@@ -111,7 +115,17 @@ module RubyLLM
           def parse_mistral_batch_body(body, shape:)
             return parse_completion_body(body, raw: body) unless body['data'].is_a?(Array)
 
-            parse_embedding_response(Response.new(body), model: body['model'], text: shape == 'array' ? [] : nil)
+            ordered = body.merge('data' => body['data'].sort_by { |row| row['index'] })
+            parse_embedding_response(Response.new(ordered), model: body['model'], text: shape == 'array' ? [] : nil)
+          end
+
+          # A row's index names the input its vector belongs to, so rows are put
+          # back in that order and positions that are not exactly 0...N fail the
+          # request rather than pairing vectors with the wrong inputs.
+          def embedding_positions?(rows)
+            positions = rows.map { |row| row.is_a?(Hash) ? row['index'] : nil }
+
+            positions.all?(Integer) && positions.sort == (0...positions.size).to_a
           end
         end
       end

data/lib/ruby_llm/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 2.1.0
+  version: 2.1.1
 platform: ruby
 authors:
 - Carmine Paolino
@@ -30,6 +30,9 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 1.10.0
+    - - "!="
+      - !ruby/object:Gem::Version
+        version: 2.0.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -37,6 +40,9 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 1.10.0
+    - - "!="
+      - !ruby/object:Gem::Version
+        version: 2.0.0
 - !ruby/object:Gem::Dependency
   name: faraday-multipart
   requirement: !ruby/object:Gem::Requirement
@@ -679,7 +685,7 @@ metadata:
   funding_uri: https://github.com/sponsors/crmne
   rubygems_mfa_required: 'true'
 post_install_message: |
-  RubyLLM 2.1.0
+  RubyLLM 2.1.1
 
     Upgrading? Read the upgrade guide before you boot: