diff --git a/models/spring-ai-mistral-ai/src/test/java/org/springframework/ai/mistralai/MistralAiChatModelIT.java b/models/spring-ai-mistral-ai/src/test/java/org/springframework/ai/mistralai/MistralAiChatModelIT.java index 03741aabd..e39cd2fb3 100644 --- a/models/spring-ai-mistral-ai/src/test/java/org/springframework/ai/mistralai/MistralAiChatModelIT.java +++ b/models/spring-ai-mistral-ai/src/test/java/org/springframework/ai/mistralai/MistralAiChatModelIT.java @@ -217,7 +217,7 @@ class MistralAiChatModelIT { assertThat(response.getResult().getOutput().getText()).containsAnyOf("30.0", "30"); assertThat(response.getMetadata()).isNotNull(); assertThat(response.getMetadata().getUsage()).isNotNull(); - assertThat(response.getMetadata().getUsage().getTotalTokens()).isLessThan(1050).isGreaterThan(800); + assertThat(response.getMetadata().getUsage().getTotalTokens()).isLessThan(1050).isGreaterThan(750); } @Test @@ -328,7 +328,7 @@ class MistralAiChatModelIT { logger.info("Response: " + chatResponse); assertThat(chatResponse.getMetadata()).isNotNull(); assertThat(chatResponse.getMetadata().getUsage()).isNotNull(); - assertThat(chatResponse.getMetadata().getUsage().getTotalTokens()).isLessThan(1050).isGreaterThan(800); + assertThat(chatResponse.getMetadata().getUsage().getTotalTokens()).isLessThan(1050).isGreaterThan(750); } record ActorsFilmsRecord(String actor, List movies) {