ruby_llm_swarm 1.9.1 → 1.9.2

data/lib/ruby_llm/active_record/message_methods.rb

@@ -1,80 +1,82 @@
 # frozen_string_literal: true
 
-module RubyLLM
-  module ActiveRecord
-    # Methods mixed into message models.
-    module MessageMethods
-      extend ActiveSupport::Concern
+if defined?(ActiveRecord::Base)
+  module RubyLLM
+    module ActiveRecord
+      # Methods mixed into message models.
+      module MessageMethods
+        extend ActiveSupport::Concern
 
-      class_methods do
-        attr_reader :chat_class, :tool_call_class, :chat_foreign_key, :tool_call_foreign_key
-      end
+        class_methods do
+          attr_reader :chat_class, :tool_call_class, :chat_foreign_key, :tool_call_foreign_key
+        end
 
-      def to_llm
-        cached = has_attribute?(:cached_tokens) ? self[:cached_tokens] : nil
-        cache_creation = has_attribute?(:cache_creation_tokens) ? self[:cache_creation_tokens] : nil
+        def to_llm
+          cached = has_attribute?(:cached_tokens) ? self[:cached_tokens] : nil
+          cache_creation = has_attribute?(:cache_creation_tokens) ? self[:cache_creation_tokens] : 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,
-          cached_tokens: cached,
-          cache_creation_tokens: cache_creation,
-          model_id: model_association&.model_id
-        )
-      end
+          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,
+            cached_tokens: cached,
+            cache_creation_tokens: cache_creation,
+            model_id: model_association&.model_id
+          )
+        end
 
-      private
+        private
 
-      def extract_tool_calls
-        tool_calls_association.to_h do |tool_call|
-          [
-            tool_call.tool_call_id,
-            RubyLLM::ToolCall.new(
-              id: tool_call.tool_call_id,
-              name: tool_call.name,
-              arguments: tool_call.arguments
-            )
-          ]
+        def extract_tool_calls
+          tool_calls_association.to_h do |tool_call|
+            [
+              tool_call.tool_call_id,
+              RubyLLM::ToolCall.new(
+                id: tool_call.tool_call_id,
+                name: tool_call.name,
+                arguments: tool_call.arguments
+              )
+            ]
+          end
         end
-      end
 
-      def extract_tool_call_id
-        parent_tool_call&.tool_call_id
-      end
+        def extract_tool_call_id
+          parent_tool_call&.tool_call_id
+        end
 
-      def extract_content
-        return RubyLLM::Content::Raw.new(content_raw) if has_attribute?(:content_raw) && content_raw.present?
+        def extract_content
+          return RubyLLM::Content::Raw.new(content_raw) if has_attribute?(:content_raw) && content_raw.present?
 
-        content_value = self[:content]
+          content_value = self[:content]
 
-        return content_value unless respond_to?(:attachments) && attachments.attached?
+          return content_value unless respond_to?(:attachments) && attachments.attached?
 
-        RubyLLM::Content.new(content_value).tap do |content_obj|
-          @_tempfiles = []
+          RubyLLM::Content.new(content_value).tap do |content_obj|
+            @_tempfiles = []
 
-          attachments.each do |attachment|
-            tempfile = download_attachment(attachment)
-            content_obj.add_attachment(tempfile, filename: attachment.filename.to_s)
+            attachments.each do |attachment|
+              tempfile = download_attachment(attachment)
+              content_obj.add_attachment(tempfile, filename: attachment.filename.to_s)
+            end
           end
         end
-      end
 
-      def download_attachment(attachment)
-        ext = File.extname(attachment.filename.to_s)
-        basename = File.basename(attachment.filename.to_s, ext)
-        tempfile = Tempfile.new([basename, ext])
-        tempfile.binmode
+        def download_attachment(attachment)
+          ext = File.extname(attachment.filename.to_s)
+          basename = File.basename(attachment.filename.to_s, ext)
+          tempfile = Tempfile.new([basename, ext])
+          tempfile.binmode
 
-        attachment.download { |chunk| tempfile.write(chunk) }
+          attachment.download { |chunk| tempfile.write(chunk) }
 
-        tempfile.flush
-        tempfile.rewind
-        @_tempfiles << tempfile
-        tempfile
+          tempfile.flush
+          tempfile.rewind
+          @_tempfiles << tempfile
+          tempfile
+        end
       end
     end
   end

data/lib/ruby_llm/active_record/model_methods.rb

@@ -1,84 +1,86 @@
 # frozen_string_literal: true
 
-module RubyLLM
-  module ActiveRecord
-    # Methods mixed into model registry models.
-    module ModelMethods
-      extend ActiveSupport::Concern
+if defined?(ActiveRecord::Base)
+  module RubyLLM
+    module ActiveRecord
+      # Methods mixed into model registry models.
+      module ModelMethods
+        extend ActiveSupport::Concern
 
-      class_methods do # rubocop:disable Metrics/BlockLength
-        def refresh!
-          RubyLLM.models.refresh!
+        class_methods do # rubocop:disable Metrics/BlockLength
+          def refresh!
+            RubyLLM.models.refresh!
 
-          transaction do
-            RubyLLM.models.all.each do |model_info|
-              model = find_or_initialize_by(
-                model_id: model_info.id,
-                provider: model_info.provider
-              )
-              model.update!(from_llm_attributes(model_info))
+            transaction do
+              RubyLLM.models.all.each do |model_info|
+                model = find_or_initialize_by(
+                  model_id: model_info.id,
+                  provider: model_info.provider
+                )
+                model.update!(from_llm_attributes(model_info))
+              end
             end
           end
-        end
 
-        def save_to_database
-          transaction do
-            RubyLLM.models.all.each do |model_info|
-              model = find_or_initialize_by(
-                model_id: model_info.id,
-                provider: model_info.provider
-              )
-              model.update!(from_llm_attributes(model_info))
+          def save_to_database
+            transaction do
+              RubyLLM.models.all.each do |model_info|
+                model = find_or_initialize_by(
+                  model_id: model_info.id,
+                  provider: model_info.provider
+                )
+                model.update!(from_llm_attributes(model_info))
+              end
             end
           end
-        end
 
-        def from_llm(model_info)
-          new(from_llm_attributes(model_info))
-        end
+          def from_llm(model_info)
+            new(from_llm_attributes(model_info))
+          end
 
-        private
+          private
 
-        def from_llm_attributes(model_info)
-          {
-            model_id: model_info.id,
-            name: model_info.name,
-            provider: model_info.provider,
-            family: model_info.family,
-            model_created_at: model_info.created_at,
-            context_window: model_info.context_window,
-            max_output_tokens: model_info.max_output_tokens,
-            knowledge_cutoff: model_info.knowledge_cutoff,
-            modalities: model_info.modalities.to_h,
-            capabilities: model_info.capabilities,
-            pricing: model_info.pricing.to_h,
-            metadata: model_info.metadata
-          }
+          def from_llm_attributes(model_info)
+            {
+              model_id: model_info.id,
+              name: model_info.name,
+              provider: model_info.provider,
+              family: model_info.family,
+              model_created_at: model_info.created_at,
+              context_window: model_info.context_window,
+              max_output_tokens: model_info.max_output_tokens,
+              knowledge_cutoff: model_info.knowledge_cutoff,
+              modalities: model_info.modalities.to_h,
+              capabilities: model_info.capabilities,
+              pricing: model_info.pricing.to_h,
+              metadata: model_info.metadata
+            }
+          end
         end
-      end
 
-      def to_llm
-        RubyLLM::Model::Info.new(
-          id: model_id,
-          name: name,
-          provider: provider,
-          family: family,
-          created_at: model_created_at,
-          context_window: context_window,
-          max_output_tokens: max_output_tokens,
-          knowledge_cutoff: knowledge_cutoff,
-          modalities: modalities&.deep_symbolize_keys || {},
-          capabilities: capabilities,
-          pricing: pricing&.deep_symbolize_keys || {},
-          metadata: metadata&.deep_symbolize_keys || {}
-        )
-      end
+        def to_llm
+          RubyLLM::Model::Info.new(
+            id: model_id,
+            name: name,
+            provider: provider,
+            family: family,
+            created_at: model_created_at,
+            context_window: context_window,
+            max_output_tokens: max_output_tokens,
+            knowledge_cutoff: knowledge_cutoff,
+            modalities: modalities&.deep_symbolize_keys || {},
+            capabilities: capabilities,
+            pricing: pricing&.deep_symbolize_keys || {},
+            metadata: metadata&.deep_symbolize_keys || {}
+          )
+        end
 
-      delegate :supports?, :supports_vision?, :supports_functions?, :type,
-               :input_price_per_million, :output_price_per_million,
-               :function_calling?, :structured_output?, :batch?,
-               :reasoning?, :citations?, :streaming?, :provider_class,
-               to: :to_llm
+        delegate :supports?, :supports_vision?, :supports_functions?, :type,
+                 :input_price_per_million, :output_price_per_million,
+                 :function_calling?, :structured_output?, :batch?,
+                 :reasoning?, :citations?, :streaming?, :provider_class,
+                 to: :to_llm
+      end
     end
   end
 end

data/lib/ruby_llm/railtie.rb

@@ -12,6 +12,10 @@ if defined?(Rails::Railtie)
 
       initializer 'ruby_llm.active_record' do
         ActiveSupport.on_load :active_record do
+          require 'ruby_llm/active_record/chat_methods'
+          require 'ruby_llm/active_record/message_methods'
+          require 'ruby_llm/active_record/model_methods'
+
           if RubyLLM.config.use_new_acts_as
             require 'ruby_llm/active_record/acts_as'
             ::ActiveRecord::Base.include RubyLLM::ActiveRecord::ActsAs

data/lib/ruby_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  VERSION = '1.9.1'
+  VERSION = '1.9.2'
 end

data/lib/ruby_llm.rb

@@ -32,6 +32,7 @@ loader.inflector.inflect(
 loader.ignore("#{__dir__}/tasks")
 loader.ignore("#{__dir__}/generators")
 loader.ignore("#{__dir__}/ruby_llm/railtie.rb")
+loader.ignore("#{__dir__}/ruby_llm/active_record")
 loader.setup
 
 # A delightful Ruby interface to modern AI language models.
@@ -136,5 +137,4 @@ require_relative 'ruby_llm/tool_executors'
 
 if defined?(Rails::Railtie)
   require 'ruby_llm/railtie'
-  require 'ruby_llm/active_record/acts_as'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm_swarm
 version: !ruby/object:Gem::Version
-  version: 1.9.1
+  version: 1.9.2
 platform: ruby
 authors:
 - Paulo Arruda