Merge pull request #9 from icebaker/ib-gemini

Adding support for Google Gemini

1 files changed, 15 insertions, 4 deletions

diff --git a/spec/logic/cartridge/streaming_spec.rb b/spec/logic/cartridge/streaming_spec.rb
index 466dd0b..4b71dfd 100644
--- a/spec/logic/cartridge/streaming_spec.rb
+++ b/spec/logic/cartridge/streaming_spec.rb
@@ -7,11 +7,22 @@ require_relative '../../../logic/cartridge/streaming'
 RSpec.describe NanoBot::Logic::Cartridge::Streaming do
   context 'interfaces override' do
     context 'defaults' do
-      let(:cartridge) { {} }
+      context 'openai' do
+        let(:cartridge) { { provider: { id: 'openai' } } }
 
-      it 'uses default values when appropriate' do
-        expect(described_class.enabled?(cartridge, :repl)).to be(true)
-        expect(described_class.enabled?(cartridge, :eval)).to be(true)
+        it 'uses default values when appropriate' do
+          expect(described_class.enabled?(cartridge, :repl)).to be(true)
+          expect(described_class.enabled?(cartridge, :eval)).to be(true)
+        end
+      end
+
+      context 'google' do
+        let(:cartridge) { { provider: { id: 'google' } } }
+
+        it 'uses default values when appropriate' do
+          expect(described_class.enabled?(cartridge, :repl)).to be(true)
+          expect(described_class.enabled?(cartridge, :eval)).to be(true)
+        end
       end
     end