Skip to content

Commit e09e766

Browse files
committed
Fix OpenAiChatModelIT#streamingWithTokenUsage
- Modify the streaming vs call usage assertion to match the close to 25% value - This change is done because the OpenAI chat response for call and stream invocation results in slight change in usage.
1 parent 7fe3b38 commit e09e766

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/OpenAiChatModelIT.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import java.util.stream.Collectors;
2828
import java.util.stream.IntStream;
2929

30+
import org.assertj.core.data.Percentage;
3031
import org.junit.jupiter.api.Test;
3132
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
3233
import org.junit.jupiter.params.ParameterizedTest;
@@ -216,9 +217,9 @@ void streamingWithTokenUsage() {
216217
assertThat(streamingTokenUsage.getGenerationTokens()).isGreaterThan(0);
217218
assertThat(streamingTokenUsage.getTotalTokens()).isGreaterThan(0);
218219

219-
assertThat(streamingTokenUsage.getPromptTokens()).isEqualTo(referenceTokenUsage.getPromptTokens());
220-
assertThat(streamingTokenUsage.getGenerationTokens()).isEqualTo(referenceTokenUsage.getGenerationTokens());
221-
assertThat(streamingTokenUsage.getTotalTokens()).isEqualTo(referenceTokenUsage.getTotalTokens());
220+
assertThat(streamingTokenUsage.getPromptTokens()).isCloseTo(referenceTokenUsage.getPromptTokens(), Percentage.withPercentage(25));
221+
assertThat(streamingTokenUsage.getGenerationTokens()).isCloseTo(referenceTokenUsage.getGenerationTokens(), Percentage.withPercentage(25));
222+
assertThat(streamingTokenUsage.getTotalTokens()).isCloseTo(referenceTokenUsage.getTotalTokens(), Percentage.withPercentage(25));
222223

223224
}
224225

0 commit comments

Comments
 (0)