Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix broken test

Browse Source
This commit is contained in:
Christian Tzolov
2023-12-19 10:52:35 +01:00
parent 88869f10fa
commit 3b91c51295
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
spring-ai-core/src/test/java/org/springframework/ai/embedding/EmbeddingUtilTests.java
View File

@@ -83,7 +83,7 @@ public class EmbeddingUtilTests {
@Test
public void testUnknownModelDimension() {
when(embeddingClient.embed(eq("Test String"))).thenReturn(List.of(0.1, 0.1, 0.1));
when(embeddingClient.embed(eq("Hello world!"))).thenReturn(List.of(0.1, 0.1, 0.1));
assertThat(EmbeddingUtil.dimensions(embeddingClient, "unknown_model", "Hello world!")).isEqualTo(3);
}