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() {
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
Original file line number Diff line number Diff 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 ();
You can’t perform that action at this time.
0 commit comments