diff --git a/src/GenerativeAI.Microsoft/GenerativeAIEmbeddingGenerator.cs b/src/GenerativeAI.Microsoft/GenerativeAIEmbeddingGenerator.cs
index 99fa0eea..57899f10 100644
--- a/src/GenerativeAI.Microsoft/GenerativeAIEmbeddingGenerator.cs
+++ b/src/GenerativeAI.Microsoft/GenerativeAIEmbeddingGenerator.cs
@@ -77,7 +77,8 @@ public async Task<GeneratedEmbeddings<Embedding<float>>> GenerateAsync(
                     Parts = [new Part { Text = text }]
                 },
                 Model = Model.Model,
-                TaskType = GetTaskType(options)
+                TaskType = GetTaskType(options),
+                OutputDimensionality = options?.Dimensions
             });
 
             // Call the batch embed API