lex-llm 0.6.15 → 0.6.16

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 07f6e605230a187e17667abdf947db738dab8b823783cde124766278159c49f1
-  data.tar.gz: e1858e04a213ed78175c8647c74f05a5c300db4a8a54caadaba523cb3a1e71ef
+  metadata.gz: 3ba576c21739b35e60594706468f7a05769871b25e5acd466ae01cbc451eec70
+  data.tar.gz: e0ac7077fdc03270059abe385f202b681928e62d667f0cefad10a0a7d1b79760
 SHA512:
-  metadata.gz: 10216387247f307ee05fe4e06a199cbf6cc762f6869636b16043adb18e1fdae355127f99e433e67614580e3c85b3d18b9d6199291d2703167184ee93e8810745
-  data.tar.gz: f010c914eddd807fc13593692a6fc20353ea1c12487838eed859493088b908a64a6459ee1866d5ffaaa9e4b3924e521b8ed3e79fce9e94fe27364283fb7be62e
+  metadata.gz: 997f024896ce8d0eb54d8574615312c0280efea81ecae75859b3bd702b8dd65aae0fbd5f615801e3e43b9d50cf6079ce4638d9a39488edb7e3a86887ee94a977
+  data.tar.gz: 3a33857e99d99a8f3358e781d46903560cb6932026d6889be2a922e23b865f3c2dacef4e38bd6a34e52af35474eab7f7e935d4c68360a8c56e2758836648d589

data/CHANGELOG.md

@@ -1,5 +1,10 @@
 # Changelog
 
+## 0.6.16 - 2026-08-04
+
+### Fixed
+- **Interleaved streaming tool-call fragments are correlated by provider index.** `ToolCall` now carries the provider wire index, and `StreamAccumulator` maps continuation fragments back to the call opened at that index. Parallel calls no longer send every id-less fragment to the most recently opened call, which previously lost one call's arguments and contaminated another's.
+
 ## 0.6.15 - 2026-07-31
 
 ### Fixed

data/lib/legion/extensions/llm/stream_accumulator.rb

@@ -26,6 +26,7 @@ module Legion
           @untagged_preamble_pending = true
           @untagged_preamble_buffer = +''
           @latest_tool_call_id = nil
+          @index_to_id = {}
         end
 
         def add(chunk)
@@ -158,13 +159,15 @@ module Legion
         end
 
         def start_tool_call(tool_call)
-          @tool_calls[tool_call.id] = ToolCall.new(
-            id: tool_call.id.empty? ? SecureRandom.uuid : tool_call.id,
+          resolved_id = tool_call.id.empty? ? SecureRandom.uuid : tool_call.id
+          @tool_calls[resolved_id] = ToolCall.new(
+            id: resolved_id,
             name: tool_call.name,
             arguments: mutable_tool_arguments(tool_call.arguments),
             thought_signature: tool_call.thought_signature
           )
-          @latest_tool_call_id = tool_call.id
+          @latest_tool_call_id = resolved_id
+          @index_to_id[tool_call.index] = resolved_id unless tool_call.index.nil?
         end
 
         def mutable_tool_arguments(arguments)
@@ -186,10 +189,20 @@ module Legion
             thought_signature: tool_call.thought_signature
           )
           @latest_tool_call_id = generated_id
+          @index_to_id[tool_call.index] = generated_id unless tool_call.index.nil?
         end
 
+        # Correlate a continuation fragment (id=nil, name=nil) to its call by the
+        # provider's authoritative wire index. Interleaved parallel calls
+        # (opener0, opener1, frag0, frag1, ...) only reassemble correctly when
+        # each fragment lands on the call at ITS index; recency (@latest_tool_call_id)
+        # sends every fragment to the last-opened call, losing one call's arguments
+        # and contaminating the other. Recency is retained only as the fallback for
+        # providers that emit no index.
         def append_tool_call_fragment(tool_call)
-          existing = @tool_calls[@latest_tool_call_id]
+          target_id = @index_to_id[tool_call.index] if tool_call.index
+          target_id ||= @latest_tool_call_id
+          existing = @tool_calls[target_id]
           return unless existing
 
           existing.arguments << tool_call.arguments.to_s

data/lib/legion/extensions/llm/tool_call.rb

@@ -5,14 +5,19 @@ module Legion
     module Llm
       # Represents a function call from an AI model to a Tool.
       class ToolCall
-        attr_reader :id, :name, :arguments
+        attr_reader :id, :name, :arguments, :index
         attr_accessor :thought_signature
 
-        def initialize(id:, name:, arguments: {}, thought_signature: nil)
+        # index carries the provider's authoritative wire position
+        # (delta.tool_calls[N].index) so the StreamAccumulator can correlate
+        # interleaved argument fragments to the right parallel call by index
+        # rather than by recency. nil for non-streaming/single-call paths.
+        def initialize(id:, name:, arguments: {}, thought_signature: nil, index: nil)
           @id = id
           @name = name
           @arguments = arguments
           @thought_signature = thought_signature
+          @index = index
         end
 
         def to_h

data/lib/legion/extensions/llm/version.rb

@@ -3,7 +3,7 @@
 module Legion
   module Extensions
     module Llm
-      VERSION = '0.6.15'
+      VERSION = '0.6.16'
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-llm
 version: !ruby/object:Gem::Version
-  version: 0.6.15
+  version: 0.6.16
 platform: ruby
 authors:
 - LegionIO