feat(voyageai): add output_dimension support for embeddings (#904)

Author: markritterman

src/Providers/VoyageAI/Embeddings.php

@@ -51,6 +51,7 @@ protected function sendRequest(): void
             'model' => $this->request->model(),
             'input' => $this->request->inputs(),
             'input_type' => $providerOptions['inputType'] ?? null,
+            'output_dimension' => $providerOptions['outputDimension'] ?? null,
             'truncation' => $providerOptions['truncation'] ?? null,
         ]));
 

tests/Fixtures/voyageai/embeddings-with-output-dimension-1.json

@@ -0,0 +1 @@
+{"object":"list","data":[{"object":"embedding","embedding":[0.1,0.2,0.3,0.4],"index":0}],"model":"voyage-3-lite","usage":{"total_tokens":8}}

tests/Providers/VoyageAI/EmbeddingsTest.php

@@ -98,6 +98,41 @@
     expect($response->usage->tokens)->toBe(7);
 });
 
+it('returns embeddings with outputDimension set', function (): void {
+    FixtureResponse::fakeResponseSequence('*', 'voyageai/embeddings-with-output-dimension');
+
+    $response = Prism::embeddings()
+        ->using(Provider::VoyageAI, 'voyage-3-lite')
+        ->fromInput('The food was delicious and the waiter...')
+        ->withProviderOptions(['outputDimension' => 4])
+        ->asEmbeddings();
+
+    expect($response->embeddings)->toBeArray();
+    expect($response->embeddings[0]->embedding)->toHaveCount(4);
+    expect($response->usage->tokens)->toBe(8);
+
+    Http::assertSent(function ($request): bool {
+        $body = json_decode((string) $request->body(), true);
+
+        return $body['output_dimension'] === 4;
+    });
+});
+
+it('does not include output_dimension when not set', function (): void {
+    FixtureResponse::fakeResponseSequence('*', 'voyageai/embeddings-from-input');
+
+    Prism::embeddings()
+        ->using(Provider::VoyageAI, 'voyage-3-lite')
+        ->fromInput('The food was delicious and the waiter...')
+        ->asEmbeddings();
+
+    Http::assertSent(function ($request): bool {
+        $body = json_decode((string) $request->body(), true);
+
+        return ! array_key_exists('output_dimension', $body);
+    });
+});
+
 it('throws a PrismRateLimitedException for a 429 response code', function (): void {
     Http::fake([
         '*' => Http::response(