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?