ruby_llm 0.1.0.pre14 → 0.1.0.pre16

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 634e727da013e56b973856876d2b456075841bd3cd1f9a5939d2028d96a47b38
-  data.tar.gz: af14dd677a8718883f0bc7c555272b5a894457d472edae1e0a4c44de8d9934ba
+  metadata.gz: 85ade36a78c5038ed31c362682ee1e81cba3b952e278ce092c93d3695740e2a1
+  data.tar.gz: 572e88e7f493f9d125873c9a3ba07d014302c480246d3b7b0b326e1d44924f30
 SHA512:
-  metadata.gz: 9c1dcc9ac485aa9f85592c0dd12d2bfe3894bfbd7311f19a1168ed17289d21edc14c636532212b01e544365519477314e03be0b34d238f2f6390834ac0a280c7
-  data.tar.gz: 9e4db61a0e5e73d13d7b8a03149625e463f7eb52d181acfc8f3ce06db0a7294615cf17f0bdbcced5f5a2908aa503af3fbc6c0122541ad704f1ddb4d17ea3b494
+  metadata.gz: 370a6e32a748d7119a3af729e0a43cdb9f4b80d60e09d959e425a28b0c4ecad8ca9e3a1f3aec6ba279039e40b2129c49e56641e041622f3268c72624d17ebc9d
+  data.tar.gz: 2295e3a850d514eb60964d3bdc47ee9682ba8b630010661efe8374ca468a81edb10617a10484a70ba47df2db8547856246c470160e8d87c157bdaccae2c8b5c6

data/.overcommit.yml

@@ -1,6 +1,6 @@
 PreCommit:
   RuboCop:
-    enabled: false
+    enabled: true
     auto_correct: true
     on_warn: fail # Treat all warnings as failures
 

data/README.md

@@ -167,7 +167,7 @@ class CreateMessages < ActiveRecord::Migration[8.0]
       t.references :chat
       t.string :role
       t.text :content
-      t.json :tool_calls
+      t.json :tool_calls, default: {}
       t.string :tool_call_id
       t.integer :input_tokens
       t.integer :output_tokens
@@ -182,14 +182,14 @@ Then in your models:
 
 ```ruby
 class Chat < ApplicationRecord
-  acts_as_chat message_class: "Message"
+  acts_as_chat
 
   # Optional: Add Turbo Streams support
   broadcasts_to ->(chat) { "chat_#{chat.id}" }
 end
 
 class Message < ApplicationRecord
-  acts_as_message chat_class: "Chat"
+  acts_as_message
 end
 ```
 

data/lib/ruby_llm/active_record/acts_as.rb

@@ -9,7 +9,7 @@ module RubyLLM
       extend ActiveSupport::Concern
 
       class_methods do
-        def acts_as_chat(message_class:)
+        def acts_as_chat(message_class: 'Message') # rubocop:disable Metrics/MethodLength
           include ChatMethods
 
           has_many :messages,
@@ -17,16 +17,23 @@ module RubyLLM
                    class_name: message_class.to_s,
                    dependent: :destroy
 
-          # No more callback config - just expose the core chat functionality
-          delegate :ask, :say, :complete, to: :chat
+          delegate :complete,
+                   :with_tool,
+                   :with_tools,
+                   :with_model,
+                   :with_temperature,
+                   :on_new_message,
+                   :on_end_message,
+                   :add_message,
+                   to: :to_llm
         end
 
-        def acts_as_message(chat_class:)
+        def acts_as_message(chat_class: 'Chat')
           include MessageMethods
 
           belongs_to :chat, class_name: chat_class.to_s
 
-          serialize :tool_calls, coder: JSON
+          delegate :tool_call?, :tool_result?, :tool_results, to: :to_llm
         end
       end
     end
@@ -36,34 +43,33 @@ module RubyLLM
     module ChatMethods
       extend ActiveSupport::Concern
 
-      def chat
-        @chat ||= begin
-          chat = RubyLLM.chat(model: model_id)
+      def to_llm
+        chat = RubyLLM.chat(model: model_id)
 
-          # Load existing messages into chat
-          messages.each do |msg|
-            chat.add_message(msg.to_llm)
-          end
+        # Load existing messages into chat
+        messages.each do |msg|
+          chat.add_message(msg.to_llm)
+        end
 
-          # Set up message persistence
-          chat.on_new_message { |msg| persist_new_message(msg) }
-              .on_end_message { |msg| persist_message_completion(msg) }
+        # Set up message persistence
+        chat.on_new_message { persist_new_message }
+            .on_end_message { |msg| persist_message_completion(msg) }
+      end
 
-          chat
-        end
+      def ask(message, &block)
+        message = { role: :user, content: message }
+        messages.create!(**message)
+        chat.complete(&block)
       end
 
-      private
+      alias say ask
 
-      def persist_new_message(message)
-        return unless message
+      private
 
+      def persist_new_message
         messages.create!(
-          role: message.role,
-          content: message.content,
-          tool_calls: message.tool_calls,
-          tool_call_id: message.tool_call_id,
-          model_id: message.model_id
+          role: :assistant,
+          content: String.new
         )
       end
 
@@ -71,8 +77,11 @@ module RubyLLM
         return unless message
 
         messages.last.update!(
+          role: message.role,
           content: message.content,
+          model_id: message.model_id,
           tool_calls: message.tool_calls,
+          tool_call_id: message.tool_call_id,
           input_tokens: message.input_tokens,
           output_tokens: message.output_tokens
         )

data/lib/ruby_llm/chat.rb

@@ -87,6 +87,12 @@ module RubyLLM
       end
     end
 
+    def add_message(message_or_attributes)
+      message = message_or_attributes.is_a?(Message) ? message_or_attributes : Message.new(message_or_attributes)
+      messages << message
+      message
+    end
+
     private
 
     def handle_tool_calls(response, &block)
@@ -104,13 +110,6 @@ module RubyLLM
       tool.call(args)
     end
 
-    def add_message(message_or_attributes)
-      message = message_or_attributes.is_a?(Message) ? message_or_attributes : Message.new(message_or_attributes)
-      # TODO: callback
-      messages << message
-      message
-    end
-
     def add_tool_result(tool_use_id, result)
       add_message(
         role: :tool,

data/lib/ruby_llm/railtie.rb

@@ -3,9 +3,9 @@
 module RubyLLM
   # Rails integration for RubyLLM
   class Railtie < Rails::Railtie
-    initializer 'ruby_llm.initialize' do
-      ActiveSupport.on_load(:active_record) do
-        extend RubyLLM::ActiveRecord::ActsAs
+    initializer 'ruby_llm.active_record' do
+      ActiveSupport.on_load :active_record do
+        include RubyLLM::ActiveRecord::ActsAs
       end
     end
   end

data/lib/ruby_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  VERSION = '0.1.0.pre14'
+  VERSION = '0.1.0.pre16'
 end

data/lib/ruby_llm.rb

@@ -7,6 +7,15 @@ require 'logger'
 require 'event_stream_parser'
 require 'securerandom'
 
+loader = Zeitwerk::Loader.for_gem
+loader.inflector.inflect(
+  'ruby_llm' => 'RubyLLM',
+  'llm' => 'LLM',
+  'openai' => 'OpenAI',
+  'api' => 'API'
+)
+loader.setup
+
 # A delightful Ruby interface to modern AI language models.
 # Provides a unified way to interact with models from OpenAI, Anthropic and others
 # with a focus on developer happiness and convention over configuration.
@@ -34,40 +43,16 @@ module RubyLLM
       @logger ||= Logger.new(
         $stdout,
         progname: 'RubyLLM',
-        level: ENV['RUBY_LLM_DEBUG'] == 'true' ? Logger::DEBUG : Logger::INFO
+        level: ENV['RUBY_LLM_DEBUG'] ? Logger::DEBUG : Logger::INFO
       )
     end
   end
 end
 
-loader = Zeitwerk::Loader.for_gem
-
-# Add lib directory to the load path
-loader.push_dir(File.expand_path('..', __dir__))
-
-# Configure custom inflections
-loader.inflector.inflect(
-  'ruby_llm' => 'RubyLLM',
-  'llm' => 'LLM',
-  'openai' => 'OpenAI',
-  'api' => 'API'
-)
-
-# Ignore Rails-specific files and specs
-loader.ignore("#{__dir__}/ruby_llm/railtie.rb")
-loader.ignore("#{__dir__}/ruby_llm/active_record")
-loader.ignore(File.expand_path('../spec', __dir__).to_s)
-
-loader.enable_reloading if ENV['RUBY_LLM_DEBUG']
-
-loader.setup
-loader.eager_load if ENV['RUBY_LLM_DEBUG']
-
 RubyLLM::Provider.register :openai, RubyLLM::Providers::OpenAI
 RubyLLM::Provider.register :anthropic, RubyLLM::Providers::Anthropic
 
-# Load Rails integration if Rails is defined
-if defined?(Rails)
+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
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre14
+  version: 0.1.0.pre16
 platform: ruby
 authors:
 - Carmine Paolino