ruby_llm-sequel 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '0585a697d4f1e5f54bc6da6a34c9efb31a1f14089b0831f4fdd251393b13607b'
-  data.tar.gz: e21a843d782f843a0c07cc5101399598d0e80b7b7c523cd8b14d045afe130814
+  metadata.gz: 330da9ca4ee162f14bf3279e78c82ec8af719912565a2f89f381cb635ac9f30a
+  data.tar.gz: 26e46591377ad3412f1088089cbcd2f7ab46a0dd8c9633c69bb44d1a77b4ec74
 SHA512:
-  metadata.gz: 4bad9d895bf1efb29b79a893169d31053e13b4826d5a15a1961d88bd8580c0f3fef8c147b6b9562692e93fe19274bc4b1b06a88f9496d011601f55588085fde9
-  data.tar.gz: c76b4e7a17de940f381145e7ccd7fe981ea5895b18a2ff4e8c0883f54c4396330d504a2c70bffaaaa48fd8be569b6baf5d1a1487e09f0981516e9b51a9425097
+  metadata.gz: 3196bcf05083317ae9d13888ebf4849a8f3cc6f332061d75750b848969f571c6c564d55be559723a22dcdb4d0a0be212c69e01014a8972d37649b30491f96140
+  data.tar.gz: b35bae79b504befd0c64c5681653c427bbff4efafeb7cf154476f1197ef6ffc23073109155346c01a41733aae1bb740aa673b677d02750c2803eabd3f2bc9831

data/lib/ruby_llm/sequel/chat_methods.rb

@@ -105,6 +105,11 @@ module RubyLLM
         self
       end
 
+      def with_thinking(...)
+        to_llm.with_thinking(...)
+        self
+      end
+
       def on_new_message(&block)
         to_llm
 
@@ -244,6 +249,9 @@ module RubyLLM
           if @message.columns.include?(:cache_creation_tokens)
             attrs[:cache_creation_tokens] = message.cache_creation_tokens
           end
+          attrs[:thinking_text] = message.thinking&.text if @message.columns.include?(:thinking_text)
+          attrs[:thinking_signature] = message.thinking&.signature if @message.columns.include?(:thinking_signature)
+          attrs[:thinking_tokens] = message.thinking_tokens if @message.columns.include?(:thinking_tokens)
 
           attrs[self.class.model_association_name] = send(self.class.model_association_name)
 
@@ -262,9 +270,13 @@ module RubyLLM
       end
 
       def persist_tool_calls(tool_calls)
+        tool_call_klass = @message.class.tool_call_class.constantize
+        supports_thought_signature = tool_call_klass.columns.include?(:thought_signature)
+
         tool_calls.each_value do |tool_call|
           attributes = tool_call.to_h
           attributes[:tool_call_id] = attributes.delete(:id)
+          attributes.delete(:thought_signature) unless supports_thought_signature
           @message.send("add_#{@message.class.tool_calls_association_name.to_s.singularize}", **attributes)
         end
       end

data/lib/ruby_llm/sequel/message_methods.rb

@@ -6,16 +6,19 @@ module RubyLLM
       def to_llm
         cached = columns.include?(:cached_tokens) ? self[:cached_tokens] : nil
         cache_creation = columns.include?(:cache_creation_tokens) ? self[:cache_creation_tokens] : nil
+        thinking_tokens_val = columns.include?(:thinking_tokens) ? self[:thinking_tokens] : nil
 
         ::RubyLLM::Message.new(
           role: role.to_sym,
           content: extract_content,
+          thinking: extract_thinking,
           tool_calls: extract_tool_calls,
           tool_call_id: extract_tool_call_id,
           input_tokens: input_tokens,
           output_tokens: output_tokens,
           cached_tokens: cached,
           cache_creation_tokens: cache_creation,
+          thinking_tokens: thinking_tokens_val,
           model_id: model_association&.model_id
         )
       end
@@ -30,6 +33,13 @@ module RubyLLM
 
       private
 
+      def extract_thinking
+        text = columns.include?(:thinking_text) ? self[:thinking_text] : nil
+        signature = columns.include?(:thinking_signature) ? self[:thinking_signature] : nil
+
+        ::RubyLLM::Thinking.build(text: text, signature: signature)
+      end
+
       def extract_content
         return ::RubyLLM::Content::Raw.new(content_raw) if columns.include?(:content_raw) && !content_raw.nil?
 

data/lib/ruby_llm/sequel/tool_call_methods.rb

@@ -4,11 +4,14 @@ module RubyLLM
   module Sequel
     module ToolCallMethods
       def to_llm
-        ::RubyLLM::ToolCall.new(
+        attrs = {
           id: tool_call_id,
           name: name,
           arguments: parse_arguments
-        )
+        }
+        attrs[:thought_signature] = self[:thought_signature] if columns.include?(:thought_signature)
+
+        ::RubyLLM::ToolCall.new(**attrs)
       end
 
       def result_message

data/lib/ruby_llm/sequel/version.rb

@@ -2,6 +2,6 @@
 
 module RubyLLM
   module Sequel
-    VERSION = '0.1.3'
+    VERSION = '0.1.4'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-sequel
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Julian Pasquale
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.9'
+        version: '1.10'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.9'
+        version: '1.10'
 - !ruby/object:Gem::Dependency
   name: sequel
   requirement: !ruby/object:Gem::Requirement