Skip to content

Commit ecebadf

Browse files
committed
Changing from List<BigDecimal> to float[] for embeddings
1 parent 41f926b commit ecebadf

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

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

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,6 @@
3636
import com.sap.ai.sdk.foundationmodels.openai.generated.model.PromptFilterResult;
3737
import io.vavr.control.Try;
3838
import java.io.IOException;
39-
import java.math.BigDecimal;
4039
import java.util.List;
4140
import java.util.Map;
4241
import java.util.stream.Stream;
@@ -265,12 +264,7 @@ void embedding() {
265264
assertThat(embeddingData.getEmbedding())
266265
.isNotNull()
267266
.isNotEmpty()
268-
.containsExactly(
269-
new BigDecimal("0.0"),
270-
new BigDecimal("3.4028235E+38"),
271-
new BigDecimal("1.4E-45"),
272-
new BigDecimal("1.23"),
273-
new BigDecimal("-4.56"));
267+
.isEqualTo(new float[] {0.0f, 3.4028235E+38f, 1.4E-45f, 1.23f, -4.56f});
274268

275269
verify(
276270
postRequestedFor(urlPathEqualTo("/embeddings"))

0 commit comments

Comments
 (0)