Skip to content

Commit a61437d

Browse files
committed
Fix usage metrics for the MistralAiChatModelIT tests
1 parent b902ca2 commit a61437d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models/spring-ai-mistral-ai/src/test/java/org/springframework/ai/mistralai/MistralAiChatModelIT.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ void functionCallTest() {
217217
assertThat(response.getResult().getOutput().getText()).containsAnyOf("30.0", "30");
218218
assertThat(response.getMetadata()).isNotNull();
219219
assertThat(response.getMetadata().getUsage()).isNotNull();
220-
assertThat(response.getMetadata().getUsage().getTotalTokens()).isLessThan(1050).isGreaterThan(800);
220+
assertThat(response.getMetadata().getUsage().getTotalTokens()).isLessThan(1050).isGreaterThan(750);
221221
}
222222

223223
@Test
@@ -328,7 +328,7 @@ void streamFunctionCallUsageTest() {
328328
logger.info("Response: " + chatResponse);
329329
assertThat(chatResponse.getMetadata()).isNotNull();
330330
assertThat(chatResponse.getMetadata().getUsage()).isNotNull();
331-
assertThat(chatResponse.getMetadata().getUsage().getTotalTokens()).isLessThan(1050).isGreaterThan(800);
331+
assertThat(chatResponse.getMetadata().getUsage().getTotalTokens()).isLessThan(1050).isGreaterThan(750);
332332
}
333333

334334
record ActorsFilmsRecord(String actor, List<String> movies) {

0 commit comments

Comments
 (0)