omniai 3.2.3 → 3.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b3e745edbb32e4102309da753fb4ccdb4cec5b6ab69a4bebf7260a39d99b31b4
-  data.tar.gz: d8546d88f6c5cdc1c665d08bd7624f284c0e5baec04162f3a7ee29474f25beba
+  metadata.gz: 1428efe58af69bd2b3055f547b15234dd5a7d7f642705a2071df5880f894ae6a
+  data.tar.gz: 151da9550296302bd6b69f2e6a4d8c3b5a45451a0691ab7d689d571c92a71c18
 SHA512:
-  metadata.gz: 59798981414c5fb32fd3f59fcc2d159c21b37cc1b6ce979424b91e7154147d05ba36e6e048ef3ff002f70022be5b02af0c7e66eb1c2e6dab6a434cb1be9ee794
-  data.tar.gz: 45f18be41f25a010810901fc9fe001f31698549fdfff6c0a1b59d5eae612f53ad6cd4ff9a857cb11b2a77fe50a1e8f0d000a388f836604e7e26d1d8212635258
+  metadata.gz: 9bd92193965a8d1f3c310d1852fbc730e888963dc206ee53168f0290a633e6c82db4e7c4da7fdfef8cc7d1393a2e20ad6f824e1db1ab8fea425e8aab025efa74
+  data.tar.gz: 3d453e691a5f4a3fb89d5aef94e7cb436517ac328d9ae27ad284312879c8278e4dc857b0ce90d30ba36da034e2e7c99aa79a01778844fea6376a6016ecab7086

data/README.md

@@ -532,6 +532,39 @@ end
 client.chat('What is the weather in "London" in Celsius and "Paris" in Fahrenheit?', tools: [WeatherTool.new])
 ```
 
+#### Extended Thinking / Reasoning
+
+Some models support extended thinking or reasoning capabilities. OmniAI provides a unified `thinking:` option that works across all supported providers:
+
+```ruby
+# Enable thinking (provider uses sensible defaults)
+response = client.chat("What is 25 * 25?", thinking: true)
+
+# Access thinking content
+response.choices.first.message.contents.each do |content|
+  case content
+  when OmniAI::Chat::Thinking
+    puts "Thinking: #{content.thinking}"
+  when OmniAI::Chat::Text
+    puts "Response: #{content.text}"
+  end
+end
+```
+
+With streaming:
+
+```ruby
+client.chat("Solve this step by step: What is 123 * 456?", thinking: true, stream: $stdout)
+```
+
+**Provider Support:**
+
+| Provider | Option | Notes |
+|----------|--------|-------|
+| Anthropic | `thinking: true` or `thinking: { budget_tokens: N }` | Requires Claude 3.5+ models |
+| Google | `thinking: true` | Requires Gemini 2.0+ with thinking enabled |
+| OpenAI | `thinking: true` or `thinking: { effort: "high" }` | Requires o1/o3 models |
+
 ### 🎤 Speech to Text
 
 Clients that support transcribe (e.g. OpenAI w/ "Whisper") convert recordings to text via the following calls:

data/lib/omniai/chat/content.rb

@@ -37,6 +37,7 @@ module OmniAI
 
         case data["type"]
         when "text" then Text.deserialize(data, context:)
+        when "thinking" then Thinking.deserialize(data, context:)
         when /(.*)_url/ then URL.deserialize(data, context:)
         else raise ArgumentError, "unknown type=#{data['type'].inspect}"
         end

data/lib/omniai/chat/delta.rb

@@ -6,17 +6,28 @@ module OmniAI
     # back to the caller.
     class Delta
       # @!attribute [rw] text
-      #
+      #   @return [String, nil]
       attr_accessor :text
 
-      # @param text [String]
-      def initialize(text:)
+      # @!attribute [rw] thinking
+      #   @return [String, nil]
+      attr_accessor :thinking
+
+      # @param text [String, nil]
+      # @param thinking [String, nil]
+      def initialize(text: nil, thinking: nil)
         @text = text
+        @thinking = thinking
       end
 
       # @return [Boolean]
       def text?
-        !text.empty?
+        !@text.nil? && !@text.empty?
+      end
+
+      # @return [Boolean]
+      def thinking?
+        !@thinking.nil? && !@thinking.empty?
       end
     end
   end

data/lib/omniai/chat/message.rb

@@ -149,6 +149,19 @@ module OmniAI
         parts.map(&:text).join("\n") unless parts.empty?
       end
 
+      # @return [Boolean]
+      def thinking?
+        !thinking.nil?
+      end
+
+      # @return [String, nil]
+      def thinking
+        return if @content.nil?
+
+        parts = arrayify(@content).filter { |content| content.is_a?(Thinking) }
+        parts.map(&:thinking).join("\n") unless parts.empty?
+      end
+
       # @param object [Object]
       # @return [Array]
       def arrayify(object)

data/lib/omniai/chat/response.rb

@@ -83,6 +83,18 @@ module OmniAI
         messages.any?(&:text?)
       end
 
+      # @return [String, nil]
+      def thinking
+        return unless thinking?
+
+        messages.filter(&:thinking?).map(&:thinking).join("\n\n")
+      end
+
+      # @return [Boolean]
+      def thinking?
+        messages.any?(&:thinking?)
+      end
+
       # @return [ToolCallList, nil]
       def tool_call_list
         tool_call_lists = messages.filter(&:tool_call_list?).map(&:tool_call_list)

data/lib/omniai/chat/thinking.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    # Represents thinking/reasoning content from a model.
+    class Thinking < Content
+      # @return [String]
+      attr_accessor :thinking
+
+      # @return [Hash] Provider-specific metadata (e.g., Anthropic's signature)
+      attr_accessor :metadata
+
+      # @param thinking [String]
+      # @param metadata [Hash] Provider-specific data for round-tripping
+      def initialize(thinking = nil, metadata: {})
+        super()
+        @thinking = thinking
+        @metadata = metadata || {}
+      end
+
+      # @return [String]
+      def inspect
+        "#<#{self.class} thinking=#{@thinking.inspect}>"
+      end
+
+      # @return [String]
+      def summarize
+        @thinking
+      end
+
+      # @param data [Hash] required
+      # @param context [Context] optional
+      def self.deserialize(data, context: nil)
+        deserialize = context&.deserializer(:thinking)
+        return deserialize.call(data, context:) if deserialize
+
+        new(data["thinking"])
+      end
+
+      # @param context [Context] optional
+      # @param direction [String] optional - either "input" or "output"
+      #
+      # @return [Hash]
+      def serialize(context: nil, direction: nil) # rubocop:disable Lint/UnusedMethodArgument
+        serializer = context&.serializer(:thinking)
+        return serializer.call(self, context:) if serializer
+
+        { type: "thinking", thinking: @thinking }
+      end
+    end
+  end
+end

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = "3.2.3"
+  VERSION = "3.3.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 3.2.3
+  version: 3.3.3
 platform: ruby
 authors:
 - Kevin Sylvestre
@@ -107,6 +107,7 @@ files:
 - lib/omniai/chat/response.rb
 - lib/omniai/chat/stream.rb
 - lib/omniai/chat/text.rb
+- lib/omniai/chat/thinking.rb
 - lib/omniai/chat/tool_call.rb
 - lib/omniai/chat/tool_call_list.rb
 - lib/omniai/chat/tool_call_message.rb