From c5bca45007a282abacadaefb1606794642c65c43 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Wed, 24 Jan 2024 21:33:55 +0100 Subject: [PATCH] Fix failing IT test --- .../org/springframework/ai/openai/embedding/EmbeddingIT.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 4c5d6f9af..7d2ca92f6 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 @@ -38,7 +38,7 @@ class EmbeddingIT { EmbeddingResponse embeddingResponse = embeddingClient.embedForResponse(List.of("Hello World")); assertThat(embeddingResponse.getData()).hasSize(1); assertThat(embeddingResponse.getData().get(0).getEmbedding()).isNotEmpty(); - assertThat(embeddingResponse.getMetadata()).containsEntry("generative", "text-embedding-ada-002-v2"); + assertThat(embeddingResponse.getMetadata()).containsEntry("model", "text-embedding-ada-002"); assertThat(embeddingResponse.getMetadata()).containsEntry("total-tokens", 2); assertThat(embeddingResponse.getMetadata()).containsEntry("prompt-tokens", 2);