From 697e945950efdafdda9547c64d46a428baee9628 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Fri, 16 Feb 2024 10:40:13 +0100 Subject: [PATCH] Fix EmbeddingIT after default openai embedding model switch --- .../springframework/ai/openai/embedding/EmbeddingIT.java | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/embedding/EmbeddingIT.java b/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/embedding/EmbeddingIT.java index caf98410c..cc154ab0b 100644 --- a/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/embedding/EmbeddingIT.java +++ b/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/embedding/EmbeddingIT.java @@ -42,7 +42,7 @@ class EmbeddingIT { assertThat(embeddingResponse.getResults()).hasSize(1); assertThat(embeddingResponse.getResults().get(0)).isNotNull(); assertThat(embeddingResponse.getResults().get(0).getOutput()).hasSize(1536); - assertThat(embeddingResponse.getMetadata()).containsEntry("model", "text-embedding-ada-002"); + assertThat(embeddingResponse.getMetadata()).containsEntry("model", "text-embedding-3-small"); assertThat(embeddingResponse.getMetadata()).containsEntry("total-tokens", 2); assertThat(embeddingResponse.getMetadata()).containsEntry("prompt-tokens", 2); @@ -65,15 +65,15 @@ class EmbeddingIT { } @Test - void embedding3Small() { + void textEmbeddingAda002() { EmbeddingResponse embeddingResponse = embeddingClient.call(new EmbeddingRequest(List.of("Hello World"), - OpenAiEmbeddingOptions.builder().withModel("text-embedding-3-small").build())); + OpenAiEmbeddingOptions.builder().withModel("text-embedding-ada-002").build())); assertThat(embeddingResponse.getResults()).hasSize(1); assertThat(embeddingResponse.getResults().get(0)).isNotNull(); assertThat(embeddingResponse.getResults().get(0).getOutput()).hasSize(1536); - assertThat(embeddingResponse.getMetadata()).containsEntry("model", "text-embedding-3-small"); + assertThat(embeddingResponse.getMetadata()).containsEntry("model", "text-embedding-ada-002"); assertThat(embeddingResponse.getMetadata()).containsEntry("total-tokens", 2); assertThat(embeddingResponse.getMetadata()).containsEntry("prompt-tokens", 2);