ruby_llm-responses_api 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b83e9dbc3e924cceb9a1468e0da1271950bdbeb20d3bc0f3f2f859e3d17da565
-  data.tar.gz: 6ffbb8e1792cc333fb375b35a588f11d4d65d54a74182b95a0a3482ea716f39f
+  metadata.gz: ba5d18029ff9e07a0577fa0fc92b5946fe0b43768e495b5550dc6e3b396520b5
+  data.tar.gz: 67f47ef8f9ca80c54ad4765f0ade6f84a2dbdc04dfc9c9e42136c5c5aa8d1fe3
 SHA512:
-  metadata.gz: 1178f162885b3fa9e7f084f183568789a3d08b3be0e27550f2aa8f7e069376a9a2165fc62f5a26ce8a8ec72ba4aebc4a5957ed9f83ab19ba9edc0b29eae0f82e
-  data.tar.gz: 7f922f72564bc21cee14d11e7a6d220a79519eb09906e6e1e17321d6e5db9bf424c9b18b1b8219d6a85eee096be9ec498c4da04d251bf10cfb190b1b5cc25fcb
+  metadata.gz: d943d82abf6f1afe6c268fbdbc8b9c0179321f67f188c3c1c25058387591673e91356616b1fafdf868da79ee1084aca90ad0117d0835fcd3dd0a238fb96cb712
+  data.tar.gz: 6b5906f1a604543b7a7cd34e25eb21faf500b2dd14558b174c6165da60cad656576b12af5e72cc49cd6960e9ad5e1210a6836bcc706be92e0b9ac77fea45c3ca

data/lib/ruby_llm/providers/openai_responses/active_record_extension.rb

@@ -9,8 +9,9 @@ module RubyLLM
       # This is automatically applied when Rails loads ActiveRecord.
       # Just add a migration: add_column :messages, :response_id, :string
       #
+
+      # Extension for the NEW MessageMethods (RubyLLM 2.0+)
       module MessageMethodsExtension
-        # Override to_llm to include response_id for Responses API support
         def to_llm
           cached = has_attribute?(:cached_tokens) ? self[:cached_tokens] : nil
           cache_creation = has_attribute?(:cache_creation_tokens) ? self[:cache_creation_tokens] : nil
@@ -31,13 +32,52 @@ module RubyLLM
         end
       end
 
-      # Extends RubyLLM's ActiveRecord ChatMethods to persist response_id
+      # Extension for the NEW ChatMethods (RubyLLM 2.0+)
       module ChatMethodsExtension
-        # Override persist_message_completion to also save response_id
         def persist_message_completion(message)
           super
+          save_response_id(message)
+        end
+
+        private
 
-          # After the parent saves, update response_id if the column exists and message has one
+        def save_response_id(message)
+          return unless message
+          return unless message.respond_to?(:response_id) && message.response_id
+          return unless @message.has_attribute?(:response_id)
+
+          @message.update_column(:response_id, message.response_id)
+        end
+      end
+
+      # Extension for LEGACY MessageLegacyMethods (RubyLLM 1.x)
+      module MessageLegacyMethodsExtension
+        def to_llm
+          resp_id = has_attribute?(:response_id) ? self[:response_id] : nil
+
+          RubyLLM::Message.new(
+            role: role.to_sym,
+            content: extract_content,
+            tool_calls: extract_tool_calls,
+            tool_call_id: extract_tool_call_id,
+            input_tokens: input_tokens,
+            output_tokens: output_tokens,
+            model_id: model_id,
+            response_id: resp_id
+          )
+        end
+      end
+
+      # Extension for LEGACY ChatLegacyMethods (RubyLLM 1.x)
+      module ChatLegacyMethodsExtension
+        def persist_message_completion(message)
+          super
+          save_response_id_legacy(message)
+        end
+
+        private
+
+        def save_response_id_legacy(message)
           return unless message
           return unless message.respond_to?(:response_id) && message.response_id
           return unless @message.has_attribute?(:response_id)
@@ -53,16 +93,37 @@ module RubyLLM
       def self.apply_active_record_extensions!
         return if @active_record_extensions_applied
 
-        require 'ruby_llm/active_record/message_methods'
-        require 'ruby_llm/active_record/chat_methods'
+        applied = false
 
-        RubyLLM::ActiveRecord::MessageMethods.prepend(MessageMethodsExtension)
-        RubyLLM::ActiveRecord::ChatMethods.prepend(ChatMethodsExtension)
+        # Try to apply to NEW modules (RubyLLM 2.0+)
+        begin
+          require 'ruby_llm/active_record/message_methods'
+          require 'ruby_llm/active_record/chat_methods'
+
+          RubyLLM::ActiveRecord::MessageMethods.prepend(MessageMethodsExtension)
+          RubyLLM::ActiveRecord::ChatMethods.prepend(ChatMethodsExtension)
+          applied = true
+        rescue LoadError, NameError
+          # New modules not available
+        end
+
+        # Try to apply to LEGACY modules (RubyLLM 1.x)
+        begin
+          require 'ruby_llm/active_record/acts_as_legacy'
+
+          if defined?(RubyLLM::ActiveRecord::MessageLegacyMethods)
+            RubyLLM::ActiveRecord::MessageLegacyMethods.prepend(MessageLegacyMethodsExtension)
+          end
+
+          if defined?(RubyLLM::ActiveRecord::ChatLegacyMethods)
+            RubyLLM::ActiveRecord::ChatLegacyMethods.prepend(ChatLegacyMethodsExtension)
+          end
+          applied = true
+        rescue LoadError, NameError
+          # Legacy modules not available
+        end
 
-        @active_record_extensions_applied = true
-      rescue LoadError, NameError
-        # RubyLLM ActiveRecord support not available, skip silently
-        nil
+        @active_record_extensions_applied = applied
       end
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-responses_api
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Chris Hasinski