Skip to content

Commit d054074

Browse files
authored
Cast message to MessageType before creating StreamChunk in stream_broadcaster (#311)
1 parent eef2abc commit d054074

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

lib/active_agent/concerns/streaming.rb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,8 @@ def _stream_define_callback_wrapper_proc(callback_name, method_proc)
268268
# @return [Proc] callback proc that accepts (message, delta, type)
269269
def stream_broadcaster
270270
proc do |message, delta, type|
271-
self.stream_chunk = StreamChunk.new(message, delta)
271+
cast_message = message.is_a?(Hash) ? Providers::Common::Messages::Types::MessageType.new.cast(message) : message
272+
self.stream_chunk = StreamChunk.new(cast_message, delta)
272273

273274
run_callbacks(:stream_open) if type == :open
274275
run_callbacks(:stream)

0 commit comments

Comments
 (0)