logic/cartridge/streaming.rb in nano-bots-1.1.2 vs logic/cartridge/streaming.rb in nano-bots-1.2.0

- old
+ new

@@ -5,10 +5,17 @@ module NanoBot module Logic module Cartridge module Streaming def self.enabled?(cartridge, interface) - return false if Helpers::Hash.fetch(cartridge, %i[provider settings stream]) == false + provider_stream = case Helpers::Hash.fetch(cartridge, %i[provider id]) + when 'openai' + Helpers::Hash.fetch(cartridge, %i[provider settings stream]) + when 'google' + Helpers::Hash.fetch(cartridge, %i[provider options stream]) + end + + return false if provider_stream == false specific_interface = Helpers::Hash.fetch(cartridge, [:interfaces, interface, :output, :stream]) return specific_interface unless specific_interface.nil?