Skip to content

Commit 14ec6b2

Browse files
committed
Use Usage#getCompletionTokens instead of deprecated Usage#getGenerationTokens
1 parent 78a58a1 commit 14ec6b2

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,11 +128,11 @@ void streamingWithTokenUsage() {
128128
var referenceTokenUsage = this.chatModel.call(prompt).getMetadata().getUsage();
129129

130130
assertThat(streamingTokenUsage.getPromptTokens()).isGreaterThan(0);
131-
assertThat(streamingTokenUsage.getGenerationTokens()).isGreaterThan(0);
131+
assertThat(streamingTokenUsage.getCompletionTokens()).isGreaterThan(0);
132132
assertThat(streamingTokenUsage.getTotalTokens()).isGreaterThan(0);
133133

134134
assertThat(streamingTokenUsage.getPromptTokens()).isEqualTo(referenceTokenUsage.getPromptTokens());
135-
assertThat(streamingTokenUsage.getGenerationTokens()).isEqualTo(referenceTokenUsage.getGenerationTokens());
135+
assertThat(streamingTokenUsage.getCompletionTokens()).isEqualTo(referenceTokenUsage.getCompletionTokens());
136136
assertThat(streamingTokenUsage.getTotalTokens()).isEqualTo(referenceTokenUsage.getTotalTokens());
137137

138138
}
@@ -316,7 +316,7 @@ void validateCallResponseMetadata(String model) {
316316
assertThat(response.getMetadata().getId()).isNotEmpty();
317317
assertThat(response.getMetadata().getModel()).containsIgnoringCase(model);
318318
assertThat(response.getMetadata().getUsage().getPromptTokens()).isPositive();
319-
assertThat(response.getMetadata().getUsage().getGenerationTokens()).isPositive();
319+
assertThat(response.getMetadata().getUsage().getCompletionTokens()).isPositive();
320320
assertThat(response.getMetadata().getUsage().getTotalTokens()).isPositive();
321321
}
322322

spring-ai-core/src/test/java/org/springframework/ai/chat/metadata/DefaultUsageTests.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ void testArbitraryNativeUsageMap() throws Exception {
184184
assertThat(deserialized.getPromptTokens()).isEqualTo(usage.getPromptTokens());
185185
assertThat(deserialized.getCompletionTokens()).isEqualTo(usage.getCompletionTokens());
186186
assertThat(deserialized.getTotalTokens()).isEqualTo(usage.getTotalTokens());
187-
assertThat(deserialized.getGenerationTokens()).isEqualTo(usage.getGenerationTokens());
187+
assertThat(deserialized.getCompletionTokens()).isEqualTo(usage.getCompletionTokens());
188188

189189
@SuppressWarnings("unchecked")
190190
Map<String, Object> deserializedMap = (Map<String, Object>) deserialized.getNativeUsage();

0 commit comments

Comments
 (0)