ruby_llm_community 1.3.0 → 1.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b347c49cf17f54616f1a3dae46674509ab2a5809bc36522cf6d2275da8a3605d
-  data.tar.gz: 7d80d568b98caeb73b4f36e6c6e610e2dac201b455cc270ea9001cbec5e1a839
+  metadata.gz: 729b5ec7255b0df4e787401cbdae2003e3be157cdf6d59ef1dbc94fdd6333a48
+  data.tar.gz: 0fd2092a50fe41c167d17b2b9c2fa9a06388336c6041faf92648c836f279756d
 SHA512:
-  metadata.gz: 320df6f2d9250ea9df1fbea5db215927fe9ae5665344a3065fd2f185ae8b53f38e8ab4c7f174258cd290a6d8b5cabb8277a8c2a91b3d6d5f4e3094437a508fcd
-  data.tar.gz: 92caeac605d923fd1c1eef4f5d8cc39f57e0972a33a57c09c518d3694da5202b0df7de5023135e9183931085bdfa5fadc76d8a354697ff11cbaa9255cc2b673c
+  metadata.gz: ec904259bdfdd81b996efd80054f65ecab1c3cb32d998c2b6f3bb9fccadce98fde58f3b7c9a3fbcb0ff26b71de5f918d35dbb6a9559a985471c4fb824d36196e
+  data.tar.gz: 835fed78e030f1f510251aed5a566545eaf8fc34b855fbe430ad7679ac349d6537742fc813aa648e6717593660dad1b3d2dde80a4c11e69bab8ea6b0c1386439

data/README.md

@@ -36,8 +36,7 @@ Use this gem to get early access to features currently in PR at the main gem.
 
 Some examples --
 
-- Prompt caching for Anthropic
-- Image editing for Gemini and OpenAI
+- Image editing for OpenAI
 - Responses API for OpenAI
 - xAI provider that supports Grok 2, Grok 3, Grok 4, and Grok Code
 - Red Candle provider

data/lib/generators/ruby_llm/generator_helpers.rb

@@ -52,8 +52,10 @@ module RubyLLM
       def acts_as_chat_declaration
         params = []
 
-        add_association_params(params, :messages, message_table_name, message_model_name, plural: true)
-        add_association_params(params, :model, model_table_name, model_model_name)
+        add_association_params(params, :messages, message_table_name, message_model_name,
+                               owner_table: chat_table_name, plural: true)
+        add_association_params(params, :model, model_table_name, model_model_name,
+                               owner_table: chat_table_name)
 
         "acts_as_chat#{" #{params.join(', ')}" if params.any?}"
       end
@@ -61,9 +63,12 @@ module RubyLLM
       def acts_as_message_declaration
         params = []
 
-        add_association_params(params, :chat, chat_table_name, chat_model_name)
-        add_association_params(params, :tool_calls, tool_call_table_name, tool_call_model_name, plural: true)
-        add_association_params(params, :model, model_table_name, model_model_name)
+        add_association_params(params, :chat, chat_table_name, chat_model_name,
+                               owner_table: message_table_name)
+        add_association_params(params, :tool_calls, tool_call_table_name, tool_call_model_name,
+                               owner_table: message_table_name, plural: true)
+        add_association_params(params, :model, model_table_name, model_model_name,
+                               owner_table: message_table_name)
 
         "acts_as_message#{" #{params.join(', ')}" if params.any?}"
       end
@@ -71,7 +76,8 @@ module RubyLLM
       def acts_as_model_declaration
         params = []
 
-        add_association_params(params, :chats, chat_table_name, chat_model_name, plural: true)
+        add_association_params(params, :chats, chat_table_name, chat_model_name,
+                               owner_table: model_table_name, plural: true)
 
         "acts_as_model#{" #{params.join(', ')}" if params.any?}"
       end
@@ -79,7 +85,8 @@ module RubyLLM
       def acts_as_tool_call_declaration
         params = []
 
-        add_association_params(params, :message, message_table_name, message_model_name)
+        add_association_params(params, :message, message_table_name, message_model_name,
+                               owner_table: tool_call_table_name)
 
         "acts_as_tool_call#{" #{params.join(', ')}" if params.any?}"
       end
@@ -120,6 +127,12 @@ module RubyLLM
         false
       end
 
+      def mysql?
+        ::ActiveRecord::Base.connection.adapter_name.downcase.include?('mysql')
+      rescue StandardError
+        false
+      end
+
       def table_exists?(table_name)
         ::ActiveRecord::Base.connection.table_exists?(table_name)
       rescue StandardError
@@ -128,13 +141,21 @@ module RubyLLM
 
       private
 
-      def add_association_params(params, default_assoc, table_name, model_name, plural: false)
+      def add_association_params(params, default_assoc, table_name, model_name, owner_table:, plural: false) # rubocop:disable Metrics/ParameterLists
         assoc = plural ? table_name.to_sym : table_name.singularize.to_sym
 
-        return if assoc == default_assoc
+        default_foreign_key = "#{default_assoc}_id"
+        # has_many/has_one: foreign key is on the associated table pointing back to owner
+        # belongs_to:       foreign key is on the owner table pointing to associated table
+        foreign_key = if plural || default_assoc.to_s.pluralize == default_assoc.to_s # has_many or has_one
+                        "#{owner_table.singularize}_id"
+                      else # belongs_to
+                        "#{table_name.singularize}_id"
+                      end
 
-        params << "#{default_assoc}: :#{assoc}"
+        params << "#{default_assoc}: :#{assoc}" if assoc != default_assoc
         params << "#{default_assoc.to_s.singularize}_class: '#{model_name}'" if model_name != assoc.to_s.classify
+        params << "#{default_assoc}_foreign_key: :#{foreign_key}" if foreign_key != default_foreign_key
       end
 
       # Convert namespaced model names to proper table names

data/lib/generators/ruby_llm/install/templates/create_models_migration.rb.tt

@@ -14,6 +14,11 @@ class Create<%= model_model_name.gsub('::', '').pluralize %> < ActiveRecord::Mig
       t.jsonb :capabilities, default: []
       t.jsonb :pricing, default: {}
       t.jsonb :metadata, default: {}
+<% elsif mysql? %>
+      t.json :modalities
+      t.json :capabilities
+      t.json :pricing
+      t.json :metadata
 <% else %>
       t.json :modalities, default: {}
       t.json :capabilities, default: []

data/lib/generators/ruby_llm/install/templates/create_tool_calls_migration.rb.tt

@@ -4,7 +4,13 @@ class Create<%= tool_call_model_name.gsub('::', '').pluralize %> < ActiveRecord:
     create_table :<%= tool_call_table_name %> do |t|
       t.string :tool_call_id, null: false
       t.string :name, null: false
-      t.<%= postgresql? ? 'jsonb' : 'json' %> :arguments, default: {}
+<% if postgresql? %>
+      t.jsonb :arguments, default: {}
+<% elsif mysql? %>
+      t.json :arguments
+<% else %>
+      t.json :arguments, default: {}
+<% end %>
       t.timestamps
     end
 

data/lib/ruby_llm/active_record/acts_as.rb

@@ -31,8 +31,8 @@ module RubyLLM
       end
 
       class_methods do # rubocop:disable Metrics/BlockLength
-        def acts_as_chat(messages: :messages, message_class: nil,
-                         model: :model, model_class: nil)
+        def acts_as_chat(messages: :messages, message_class: nil, messages_foreign_key: nil, # rubocop:disable Metrics/ParameterLists
+                         model: :model, model_class: nil, model_foreign_key: nil)
           include RubyLLM::ActiveRecord::ChatMethods
 
           class_attribute :messages_association_name, :model_association_name, :message_class, :model_class
@@ -45,12 +45,12 @@ module RubyLLM
           has_many messages,
                    -> { order(created_at: :asc) },
                    class_name: self.message_class,
-                   foreign_key: ActiveSupport::Inflector.foreign_key(table_name.singularize),
+                   foreign_key: messages_foreign_key,
                    dependent: :destroy
 
           belongs_to model,
                      class_name: self.model_class,
-                     foreign_key: ActiveSupport::Inflector.foreign_key(model.to_s.singularize),
+                     foreign_key: model_foreign_key,
                      optional: true
 
           delegate :add_message, to: :to_llm
@@ -68,7 +68,7 @@ module RubyLLM
           end
         end
 
-        def acts_as_model(chats: :chats, chat_class: nil)
+        def acts_as_model(chats: :chats, chat_class: nil, chats_foreign_key: nil)
           include RubyLLM::ActiveRecord::ModelMethods
 
           class_attribute :chats_association_name, :chat_class
@@ -80,18 +80,16 @@ module RubyLLM
           validates :provider, presence: true
           validates :name, presence: true
 
-          has_many chats,
-                   class_name: self.chat_class,
-                   foreign_key: ActiveSupport::Inflector.foreign_key(table_name.singularize)
+          has_many chats, class_name: self.chat_class, foreign_key: chats_foreign_key
 
           define_method :chats_association do
             send(chats_association_name)
           end
         end
 
-        def acts_as_message(chat: :chat, chat_class: nil, touch_chat: false, # rubocop:disable Metrics/ParameterLists
-                            tool_calls: :tool_calls, tool_call_class: nil,
-                            model: :model, model_class: nil)
+        def acts_as_message(chat: :chat, chat_class: nil, chat_foreign_key: nil, touch_chat: false, # rubocop:disable Metrics/ParameterLists
+                            tool_calls: :tool_calls, tool_call_class: nil, tool_calls_foreign_key: nil,
+                            model: :model, model_class: nil, model_foreign_key: nil)
           include RubyLLM::ActiveRecord::MessageMethods
 
           class_attribute :chat_association_name, :tool_calls_association_name, :model_association_name,
@@ -106,12 +104,12 @@ module RubyLLM
 
           belongs_to chat,
                      class_name: self.chat_class,
-                     foreign_key: ActiveSupport::Inflector.foreign_key(chat.to_s.singularize),
+                     foreign_key: chat_foreign_key,
                      touch: touch_chat
 
           has_many tool_calls,
                    class_name: self.tool_call_class,
-                   foreign_key: ActiveSupport::Inflector.foreign_key(table_name.singularize),
+                   foreign_key: tool_calls_foreign_key,
                    dependent: :destroy
 
           belongs_to :parent_tool_call,
@@ -126,7 +124,7 @@ module RubyLLM
 
           belongs_to model,
                      class_name: self.model_class,
-                     foreign_key: ActiveSupport::Inflector.foreign_key(model.to_s.singularize),
+                     foreign_key: model_foreign_key,
                      optional: true
 
           delegate :tool_call?, :tool_result?, to: :to_llm
@@ -144,8 +142,8 @@ module RubyLLM
           end
         end
 
-        def acts_as_tool_call(message: :message, message_class: nil,
-                              result: :result, result_class: nil)
+        def acts_as_tool_call(message: :message, message_class: nil, message_foreign_key: nil, # rubocop:disable Metrics/ParameterLists
+                              result: :result, result_class: nil, result_foreign_key: nil)
           class_attribute :message_association_name, :result_association_name, :message_class, :result_class
 
           self.message_association_name = message
@@ -155,11 +153,11 @@ module RubyLLM
 
           belongs_to message,
                      class_name: self.message_class,
-                     foreign_key: ActiveSupport::Inflector.foreign_key(message.to_s.singularize)
+                     foreign_key: message_foreign_key
 
           has_one result,
                   class_name: self.result_class,
-                  foreign_key: ActiveSupport::Inflector.foreign_key(table_name.singularize),
+                  foreign_key: result_foreign_key,
                   dependent: :nullify
 
           define_method :message_association do

data/lib/ruby_llm/aliases.json

@@ -11,11 +11,6 @@
   "claude-3-5-haiku-latest": {
     "anthropic": "claude-3-5-haiku-latest"
   },
-  "claude-3-5-sonnet": {
-    "anthropic": "claude-3-5-sonnet-20241022",
-    "openrouter": "anthropic/claude-3.5-sonnet",
-    "bedrock": "anthropic.claude-3-5-sonnet-20240620-v1:0:200k"
-  },
   "claude-3-7-sonnet": {
     "anthropic": "claude-3-7-sonnet-20250219",
     "openrouter": "anthropic/claude-3.7-sonnet",
@@ -245,6 +240,22 @@
     "openai": "gpt-5-nano",
     "openrouter": "openai/gpt-5-nano"
   },
+  "gpt-5-pro": {
+    "openai": "gpt-5-pro",
+    "openrouter": "openai/gpt-5-pro"
+  },
+  "gpt-oss-120b": {
+    "openai": "gpt-oss-120b",
+    "openrouter": "openai/gpt-oss-120b"
+  },
+  "gpt-oss-20b": {
+    "openai": "gpt-oss-20b",
+    "openrouter": "openai/gpt-oss-20b"
+  },
+  "imagen-4.0-generate-001": {
+    "gemini": "imagen-4.0-generate-001",
+    "vertexai": "imagen-4.0-generate-001"
+  },
   "grok-2": {
     "xai": "grok-2-1212"
   },
@@ -323,14 +334,6 @@
     "openai": "o1",
     "openrouter": "openai/o1"
   },
-  "o1-mini": {
-    "openai": "o1-mini",
-    "openrouter": "openai/o1-mini"
-  },
-  "o1-mini-2024-09-12": {
-    "openai": "o1-mini-2024-09-12",
-    "openrouter": "openai/o1-mini-2024-09-12"
-  },
   "o1-pro": {
     "openai": "o1-pro",
     "openrouter": "openai/o1-pro"

data/lib/ruby_llm/attachment.rb

@@ -7,17 +7,8 @@ module RubyLLM
 
     def initialize(source, filename: nil)
       @source = source
-      if url?
-        @source = URI source
-        @filename = filename || File.basename(@source.path).to_s
-      elsif path?
-        @source = Pathname.new source
-        @filename = filename || @source.basename.to_s
-      elsif active_storage?
-        @filename = filename || extract_filename_from_active_storage
-      else
-        @filename = filename
-      end
+      @source = source_type_cast
+      @filename = filename || source_filename
 
       determine_mime_type
     end
@@ -168,6 +159,38 @@ module RubyLLM
                  end
     end
 
+    def source_type_cast
+      if url?
+        URI(@source)
+      elsif path?
+        Pathname.new(@source)
+      else
+        @source
+      end
+    end
+
+    def source_filename
+      if url?
+        File.basename(@source.path).to_s
+      elsif path?
+        @source.basename.to_s
+      elsif io_like?
+        extract_filename_from_io
+      elsif active_storage?
+        extract_filename_from_active_storage
+      end
+    end
+
+    def extract_filename_from_io
+      if defined?(ActionDispatch::Http::UploadedFile) && @source.is_a?(ActionDispatch::Http::UploadedFile)
+        @source.original_filename.to_s
+      elsif @source.respond_to?(:path)
+        File.basename(@source.path).to_s
+      else
+        'attachment'
+      end
+    end
+
     def extract_filename_from_active_storage # rubocop:disable Metrics/PerceivedComplexity
       return 'attachment' unless defined?(ActiveStorage)