Add specs for embedding and image generation with per-request API key

Author: BrianBorge

spec/ruby_llm/embeddings_spec.rb

@@ -32,4 +32,16 @@
       end
     end
   end
+
+  describe 'embedding with per-request key' do
+    let(:override_openai_key) { ENV.fetch('OTHER_OPENAI_API_KEY', nil) }
+
+    it 'embeds text using an overridden key' do # rubocop:disable RSpec/MultipleExpectations
+      embedding = RubyLLM.embed("What's your favorite algorithm?",
+                                api_key: override_openai_key)
+
+      expect(embedding).to be_a(RubyLLM::Embedding)
+      expect(embedding.vectors).to be_an(Array)
+    end
+  end
 end

spec/ruby_llm/image_generation_spec.rb

@@ -65,4 +65,16 @@ def save_and_verify_image(image) # rubocop:disable Metrics/MethodLength
       end.to raise_error(RubyLLM::ModelNotFoundError)
     end
   end
+
+  describe 'painting with per-request key' do
+    let(:override_openai_key) { ENV.fetch('OTHER_OPENAI_API_KEY', nil) }
+
+    it 'generates an image using an overridden key' do # rubocop:disable RSpec/MultipleExpectations
+      image = RubyLLM.paint('A sunset in watercolor style', api_key: override_openai_key)
+      expect(image).to be_a(RubyLLM::Image)
+      expect(image.url).to match(%r{^https?://})
+
+      save_and_verify_image image
+    end
+  end
 end