Skip to content

Commit 3135663

Browse files
authored
Fix inflector (#159)
Missed the foreign key on tool call
1 parent 1678ea5 commit 3135663

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

lib/ruby_llm/active_record/acts_as.rb

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,8 @@ def acts_as_message(chat_class: 'Chat', tool_call_class: 'ToolCall', **options)
5353

5454
def acts_as_tool_call(message_class: 'Message', **options) # rubocop:disable Metrics/MethodLength
5555
@message_class = message_class.to_s
56-
@message_foreign_key = options[:message_foreign_key] || "#{@message_class.underscore}_id"
56+
@message_foreign_key = options[:message_foreign_key] || @message_class.foreign_key
57+
@result_foreign_key = options[:result_foreign_key] || 'id'
5758

5859
belongs_to :message,
5960
class_name: @message_class,
@@ -62,7 +63,7 @@ def acts_as_tool_call(message_class: 'Message', **options) # rubocop:disable Met
6263

6364
has_one :result,
6465
class_name: @message_class,
65-
foreign_key: @message_foreign_key,
66+
foreign_key: @result_foreign_key,
6667
inverse_of: :parent_tool_call,
6768
dependent: :nullify
6869
end

0 commit comments

Comments
 (0)