File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
models/spring-ai-openai/src/test/java/org/springframework/ai/openai/chat/proxy
spring-ai-core/src/test/java/org/springframework/ai/chat/metadata Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -128,11 +128,11 @@ void streamingWithTokenUsage() {
128
128
var referenceTokenUsage = this .chatModel .call (prompt ).getMetadata ().getUsage ();
129
129
130
130
assertThat (streamingTokenUsage .getPromptTokens ()).isGreaterThan (0 );
131
- assertThat (streamingTokenUsage .getGenerationTokens ()).isGreaterThan (0 );
131
+ assertThat (streamingTokenUsage .getCompletionTokens ()).isGreaterThan (0 );
132
132
assertThat (streamingTokenUsage .getTotalTokens ()).isGreaterThan (0 );
133
133
134
134
assertThat (streamingTokenUsage .getPromptTokens ()).isEqualTo (referenceTokenUsage .getPromptTokens ());
135
- assertThat (streamingTokenUsage .getGenerationTokens ()).isEqualTo (referenceTokenUsage .getGenerationTokens ());
135
+ assertThat (streamingTokenUsage .getCompletionTokens ()).isEqualTo (referenceTokenUsage .getCompletionTokens ());
136
136
assertThat (streamingTokenUsage .getTotalTokens ()).isEqualTo (referenceTokenUsage .getTotalTokens ());
137
137
138
138
}
@@ -316,7 +316,7 @@ void validateCallResponseMetadata(String model) {
316
316
assertThat (response .getMetadata ().getId ()).isNotEmpty ();
317
317
assertThat (response .getMetadata ().getModel ()).containsIgnoringCase (model );
318
318
assertThat (response .getMetadata ().getUsage ().getPromptTokens ()).isPositive ();
319
- assertThat (response .getMetadata ().getUsage ().getGenerationTokens ()).isPositive ();
319
+ assertThat (response .getMetadata ().getUsage ().getCompletionTokens ()).isPositive ();
320
320
assertThat (response .getMetadata ().getUsage ().getTotalTokens ()).isPositive ();
321
321
}
322
322
Original file line number Diff line number Diff line change @@ -184,7 +184,7 @@ void testArbitraryNativeUsageMap() throws Exception {
184
184
assertThat (deserialized .getPromptTokens ()).isEqualTo (usage .getPromptTokens ());
185
185
assertThat (deserialized .getCompletionTokens ()).isEqualTo (usage .getCompletionTokens ());
186
186
assertThat (deserialized .getTotalTokens ()).isEqualTo (usage .getTotalTokens ());
187
- assertThat (deserialized .getGenerationTokens ()).isEqualTo (usage .getGenerationTokens ());
187
+ assertThat (deserialized .getCompletionTokens ()).isEqualTo (usage .getCompletionTokens ());
188
188
189
189
@ SuppressWarnings ("unchecked" )
190
190
Map <String , Object > deserializedMap = (Map <String , Object >) deserialized .getNativeUsage ();
You can’t perform that action at this time.
0 commit comments