omniai-anthropic 3.5.0 → 3.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fcf24d90210fce6c02f7be928c435515a00e75163590946e3222cf36d424c7c9
-  data.tar.gz: 0477ad1fdb3454dec2e84db5bcdf044b97eee780892f013ef85348460ac0fb70
+  metadata.gz: 70d172e3b288451b57742351a6db872672b84a7a9f0b139e2466299b17ef3a2e
+  data.tar.gz: b83fb955581badb868e95b99caee30d1b3e2aa9203a01bcd137b7b0517d14cf3
 SHA512:
-  metadata.gz: f91432693771d415d5b7375e00781c817e960711a74c66961808b9a2ebf68cc09a3c7ae455ed631666ef8b15643e9afa79962cef23e1334410eb8a352a233aa7
-  data.tar.gz: b02acec6c970f8e651f465e5fdc07512db9dca9600b787bb42eb2beab866236ec12ea3af626c268ef1734aec67e294892d8b9c2b5048f9c90d516ee7c28633ad
+  metadata.gz: 26ef3bb07ec0aae3544ec8008a745d754a0a2044645836b26507b1be5c148c341877763281826a8745e5a93df6534032454d38b95d4b062a926875fb32b6e263
+  data.tar.gz: 10efb98acf0f9ee712568d4c1a4cf7087c32bbaae0ec82b28dec5d046937803595483df3a6039dd042c28653a346a8cdfb1866329ef149eb6feec831cfde8c63

data/lib/omniai/anthropic/chat/stream.rb

@@ -81,10 +81,15 @@ module OmniAI
 
           input_tokens = data.dig("usage", "input_tokens")
           output_tokens = data.dig("usage", "output_tokens")
+          # The thinking breakdown arrives only on the final `message_delta`. Carrying it through keeps streamed
+          # responses reporting the same `thinking_tokens` as non-streamed ones; without it the key is dropped and
+          # every streamed response silently reports no reasoning.
+          output_tokens_details = data.dig("usage", "output_tokens_details")
 
           @data["usage"] ||= {}
           @data["usage"]["input_tokens"] = input_tokens if input_tokens
           @data["usage"]["output_tokens"] = output_tokens if output_tokens
+          @data["usage"]["output_tokens_details"] = output_tokens_details if output_tokens_details
         end
 
         # Handler for Type::MESSAGE_STOP

data/lib/omniai/anthropic/chat/usage_serializer.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module Anthropic
+    class Chat
+      # Overrides usage deserialize to read Anthropic's reasoning breakdown.
+      module UsageSerializer
+        # Anthropic already counts reasoning inside `output_tokens` and reports the breakdown alongside it at
+        # `output_tokens_details.thinking_tokens`, so the breakdown is read into `thinking_tokens` and the output
+        # count is left exactly as reported. Adding the two together would double count.
+        #
+        # When streaming, the breakdown arrives only on the final `message_delta`.
+        #
+        # @param data [Hash]
+        # @return [OmniAI::Chat::Usage]
+        def self.deserialize(data, *)
+          # Deserialize without a context so the generic flat parse runs rather than recursing into this method.
+          usage = OmniAI::Chat::Usage.deserialize(data)
+
+          # Only overwrite when the vendor container is actually present. A payload produced by `Usage#serialize`
+          # carries base's own `thinking_tokens` key and no `output_tokens_details`, so assigning unconditionally
+          # would clobber a correctly-parsed value with nil and break the round-trip. `unless nil?` rather than
+          # `||=`, so a reported zero from the wire still wins over base's nil.
+          thinking_tokens = data.dig("output_tokens_details", "thinking_tokens")
+          usage.thinking_tokens = thinking_tokens unless thinking_tokens.nil?
+
+          usage
+        end
+      end
+    end
+  end
+end

data/lib/omniai/anthropic/chat.rb

@@ -87,6 +87,8 @@ module OmniAI
 
         context.serializers[:thinking] = ThinkingSerializer.method(:serialize)
         context.deserializers[:thinking] = ThinkingSerializer.method(:deserialize)
+
+        context.deserializers[:usage] = UsageSerializer.method(:deserialize)
       end
 
       # @return [Hash]

data/lib/omniai/anthropic/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Anthropic
-    VERSION = "3.5.0"
+    VERSION = "3.6.0"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai-anthropic
 version: !ruby/object:Gem::Version
-  version: 3.5.0
+  version: 3.6.0
 platform: ruby
 authors:
 - Kevin Sylvestre
@@ -29,14 +29,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.7'
+        version: '3.8'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '3.7'
+        version: '3.8'
 - !ruby/object:Gem::Dependency
   name: openssl
   requirement: !ruby/object:Gem::Requirement
@@ -89,6 +89,7 @@ files:
 - lib/omniai/anthropic/chat/tool_call_serializer.rb
 - lib/omniai/anthropic/chat/tool_serializer.rb
 - lib/omniai/anthropic/chat/url_serializer.rb
+- lib/omniai/anthropic/chat/usage_serializer.rb
 - lib/omniai/anthropic/client.rb
 - lib/omniai/anthropic/computer.rb
 - lib/omniai/anthropic/config.rb