Skip to content

Commit 7cf2afa

Browse files
committed
fix(Message): Enhance file handling in create_user_message to reject nil and blank values
1 parent 775498f commit 7cf2afa

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

lib/ruby_llm/active_record/acts_as.rb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -143,14 +143,16 @@ def on_end_message(...)
143143
self
144144
end
145145

146-
def create_user_message(content, with: nil)
146+
def create_user_message(content, with: nil) # rubocop:disable Metrics/PerceivedComplexity
147147
message_record = messages.create!(
148148
role: :user,
149149
content: content
150150
)
151151

152152
if with.present?
153-
files = Array(with).reject(&:empty?)
153+
files = Array(with).reject do |f|
154+
f.nil? || (f.respond_to?(:empty?) && f.empty?) || (f.respond_to?(:blank?) && f.blank?)
155+
end
154156

155157
if files.any?
156158
if files.first.is_a?(ActionDispatch::Http::UploadedFile)

0 commit comments

Comments
 (0)