ruby_llm 0.1.0.pre14 → 0.1.0.pre15

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 634e727da013e56b973856876d2b456075841bd3cd1f9a5939d2028d96a47b38
-  data.tar.gz: af14dd677a8718883f0bc7c555272b5a894457d472edae1e0a4c44de8d9934ba
+  metadata.gz: ca489465046f71efd3a80717b46a68cdcbc754527f4b5ccade15442d2a95854e
+  data.tar.gz: b1091dd99c8b96266b767b1be69ade14a88cffc0c3f2cd07c1bf20d9efebe6e2
 SHA512:
-  metadata.gz: 9c1dcc9ac485aa9f85592c0dd12d2bfe3894bfbd7311f19a1168ed17289d21edc14c636532212b01e544365519477314e03be0b34d238f2f6390834ac0a280c7
-  data.tar.gz: 9e4db61a0e5e73d13d7b8a03149625e463f7eb52d181acfc8f3ce06db0a7294615cf17f0bdbcced5f5a2908aa503af3fbc6c0122541ad704f1ddb4d17ea3b494
+  metadata.gz: 8bd968d14ec94fb254ae21a3d253cbe8b952145d3626540bf92cc5ded13d660e8f45b651319d3591f6a6e6b137c02cb94540c2a00680c25c2d7d3ecf4d799bbf
+  data.tar.gz: d1a5e1ef82b95f5ebe59b27440d703e88d4332f47cb706da8c4f09487608d6de572e78a38756af938a0e0ed0dbf8dc0f4a45aa82ac1d32f444681c3172baf40a

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')
           include ChatMethods
 
           has_many :messages,
@@ -17,16 +17,13 @@ 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, to: :chat
         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
         end
       end
     end
@@ -37,42 +34,43 @@ module RubyLLM
       extend ActiveSupport::Concern
 
       def chat
-        @chat ||= begin
-          chat = RubyLLM.chat(model: model_id)
+        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) }
 
-          chat
-        end
+        chat
       end
 
-      private
+      def ask(message, &block)
+        message = { role: :user, content: message }
+        messages.create!(**message)
+        chat.complete(&block)
+      end
 
-      def persist_new_message(message)
-        return unless message
+      alias say ask
 
-        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
-        )
+      private
+
+      def persist_new_message
+        messages.create!
       end
 
       def persist_message_completion(message)
         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.pre15'
 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.pre15
 platform: ruby
 authors:
 - Carmine Paolino