Refactor Embeddings spec to use Constant and custom model dimensions

Author: infinityrobot

spec/ruby_llm/embeddings_spec.rb

@@ -7,13 +7,12 @@
 
   let(:test_text) { "Ruby is a programmer's best friend" }
   let(:test_texts) { %w[Ruby Python JavaScript] }
-  let(:test_dimensions) { 768 }
-
-  embedding_models = %w[text-embedding-004 text-embedding-3-small].freeze
 
   describe 'basic functionality' do
-    embedding_models.each do |model|
-      provider = RubyLLM::Models.provider_for(model).slug
+    EMBEDDINGS_MODELS.each do |model_info|
+      model = model_info[:model]
+      provider = model_info[:provider]
+      dimensions = model_info[:default_dimensions]
       it "#{provider}/#{model} can handle a single text" do # rubocop:disable RSpec/MultipleExpectations
         embedding = RubyLLM.embed(test_text, model: model)
         expect(embedding.vectors).to be_an(Array)
@@ -23,9 +22,9 @@
       end
 
       it "#{provider}/#{model} can handle a single text with custom dimensions" do # rubocop:disable RSpec/MultipleExpectations
-        embedding = RubyLLM.embed(test_text, model: model, dimensions: test_dimensions)
+        embedding = RubyLLM.embed(test_text, model: model, dimensions: dimensions)
         expect(embedding.vectors).to be_an(Array)
-        expect(embedding.vectors.length).to eq(test_dimensions)
+        expect(embedding.vectors.length).to eq(dimensions)
       end
 
       it "#{provider}/#{model} can handle multiple texts" do # rubocop:disable RSpec/ExampleLength,RSpec/MultipleExpectations
@@ -38,10 +37,10 @@
       end
 
       it "#{provider}/#{model} can handle multiple texts with custom dimensions" do # rubocop:disable RSpec/MultipleExpectations
-        embeddings = RubyLLM.embed(test_texts, model: model, dimensions: test_dimensions)
+        embeddings = RubyLLM.embed(test_texts, model: model, dimensions: dimensions)
         expect(embeddings.vectors).to be_an(Array)
         embeddings.vectors.each do |vector|
-          expect(vector.length).to eq(test_dimensions)
+          expect(vector.length).to eq(dimensions)
         end
       end
     end

spec/spec_helper.rb

@@ -144,6 +144,12 @@
   { provider: :ollama, model: 'qwen3' }
 ].freeze
 
+EMBEDDINGS_MODELS = [
+  { provider: :openai, model: 'text-embedding-004', default_dimensions: 768 },
+  { provider: :gemini, model: 'text-embedding-3-small', default_dimensions: 1024 },
+  { provider: :cohere, model: 'embed-v4.0', default_dimensions: 1024 }
+].freeze
+
 PDF_MODELS = [
   { provider: :anthropic, model: 'claude-3-5-haiku-20241022' },
   { provider: :gemini, model: 'gemini-2.0-flash' },