Initial

Author: a-d

foundation-models/openai/src/main/java/com/sap/ai/sdk/foundationmodels/openai/model/OpenAiEmbeddingData.java

@@ -2,7 +2,6 @@
 
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.google.common.annotations.Beta;
-import java.util.List;
 import javax.annotation.Nonnull;
 import lombok.EqualsAndHashCode;
 import lombok.Getter;
@@ -23,7 +22,7 @@ public class OpenAiEmbeddingData {
   /** Array of size `1536` (OpenAI's embedding size) containing embedding vector. */
   @JsonProperty("embedding")
   @Getter(onMethod_ = @Nonnull)
-  private List<Double> embedding;
+  private float[] embedding;
 
   /** Index of choice. */
   @JsonProperty("index")

foundation-models/openai/src/test/java/com/sap/ai/sdk/foundationmodels/openai/OpenAiClientTest.java

@@ -373,9 +373,7 @@ void embedding() {
     assertThat(embeddingData.getEmbedding())
         .isNotNull()
         .isNotEmpty()
-        .containsExactly(-0.0000000070958645d, 2.123e-300d, -0.0069813123d, -3.385849e-05d)
-        // ensure double precision
-        .hasToString("[-7.0958645E-9, 2.123E-300, -0.0069813123, -3.385849E-5]");
+        .isEqualTo(new float[] {0.0f, 3.4028235E38f, 1.4E-45f, 1.23f, -4.56f});
 
     verify(
         postRequestedFor(urlPathEqualTo("/embeddings"))

foundation-models/openai/src/test/resources/__files/embeddingResponse.json

@@ -1,12 +1,7 @@
 {
   "data": [
     {
-      "embedding": [
-        -0.0000000070958645,
-        2.123e-300,
-        -0.0069813123,
-        -3.385849e-05
-      ],
+      "embedding": [ 0.0, 3.4028235E38, 1.4E-45, 1.23, -4.56 ],
       "index": 0,
       "object": "embedding"
     }