Fix embedding return format inconsistency for single-string arrays (#267)

## Summary
- Fixes inconsistent return format when passing an array with a single
string to the embed method
- Ensures `RubyLLM.embed([string])` returns `[[vector]]` instead of
`[vector]`
- Maintains backward compatibility for single string inputs

## Problem
As reported in #254, the embedding API had inconsistent behavior:
```ruby
RubyLLM.embed(string) -> [vector] ✓
RubyLLM.embed([string, string]) -> [[vector], [vector]] ✓
RubyLLM.embed([string]) -> [vector] ✗ (should be [[vector]])
```

## Solution
- Modified the base Provider class to pass the original `text` parameter
to `parse_embedding_response`
- Updated OpenAI and Gemini providers to check if input was an array
before unwrapping single embeddings
- Added comprehensive test coverage for this edge case

## Test plan
- [x] Added new test case specifically for single-string arrays
- [x] All existing tests pass
- [x] Verified fix works for both OpenAI and Gemini providers

Fixes #254

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Claude <[email protected]>
Co-authored-by: Carmine Paolino <[email protected]>

Author: 3 people

lib/ruby_llm/provider.rb

@@ -34,7 +34,7 @@ def list_models(connection:)
       def embed(text, model:, connection:, dimensions:)
         payload = render_embedding_payload(text, model:, dimensions:)
         response = connection.post(embedding_url(model:), payload)
-        parse_embedding_response(response, model:)
+        parse_embedding_response(response, model:, text:)
       end
 
       def paint(prompt, model:, size:, connection:)

lib/ruby_llm/providers/gemini/embeddings.rb

@@ -15,9 +15,11 @@ def render_embedding_payload(text, model:, dimensions:)
           { requests: [text].flatten.map { |t| single_embedding_payload(t, model:, dimensions:) } }
         end
 
-        def parse_embedding_response(response, model:)
+        def parse_embedding_response(response, model:, text:)
           vectors = response.body['embeddings']&.map { |e| e['values'] }
-          vectors in [vectors]
+          # If we only got one embedding AND the input was a single string (not an array),
+          # return it as a single vector
+          vectors = vectors.first if vectors&.length == 1 && !text.is_a?(Array)
 
           Embedding.new(vectors:, model:, input_tokens: 0)
         end

lib/ruby_llm/providers/openai/embeddings.rb

@@ -19,13 +19,14 @@ def render_embedding_payload(text, model:, dimensions:)
           }.compact
         end
 
-        def parse_embedding_response(response, model:)
+        def parse_embedding_response(response, model:, text:)
           data = response.body
           input_tokens = data.dig('usage', 'prompt_tokens') || 0
           vectors = data['data'].map { |d| d['embedding'] }
 
-          # If we only got one embedding, return it as a single vector
-          vectors in [vectors]
+          # If we only got one embedding AND the input was a single string (not an array),
+          # return it as a single vector
+          vectors = vectors.first if vectors.length == 1 && !text.is_a?(Array)
 
           Embedding.new(vectors:, model:, input_tokens:)
         end