ruby_llm-sequel 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3422a45af2abd1286b9762fcbf6efd245af84375e6dde6ab067adb433a80c3e7
-  data.tar.gz: 54ec6f2324de674e4fe112280843115f0b866b5a8edc93776768d57b3b5b2f76
+  metadata.gz: 330da9ca4ee162f14bf3279e78c82ec8af719912565a2f89f381cb635ac9f30a
+  data.tar.gz: 26e46591377ad3412f1088089cbcd2f7ab46a0dd8c9633c69bb44d1a77b4ec74
 SHA512:
-  metadata.gz: 4fac8795a86055f72e7197042006de0f14cd6c2d7e9951ffe470721c0a9c8032d062a32b2b98be2ba7ff3ca3c9d98c35e779bc93dccbda057c8b43f3ec06f7e1
-  data.tar.gz: 7a662f9adb13c9570b564fb7268c8230c19e9b1c1bbbebef392a2d0a38f587b5499bc5b6cb04d4e399a870293c2e9c7c1fecd5bcb010878ee0fb3c02434fecf6
+  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.2'
+    VERSION = '0.1.4'
   end
 end

data/lib/sequel/plugins/ruby_llm.rb

@@ -174,8 +174,8 @@ module Sequel
 
           plugin :association_dependencies
 
-          many_to_one message, class: message_class, key: :"#{message}_id", eager_loader_key: nil
-          one_to_one result, class: result_class, key: :tool_call_id, eager_loader_key: nil
+          many_to_one message, class: message_class, key: :"#{message}_id"
+          one_to_one result, class: result_class, key: :tool_call_id
 
           add_association_dependencies result => :nullify
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-sequel
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  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