Skip to content

Commit 55508d1

Browse files
committed
improved ability to use openAiApi to support DeepSeek
1 parent e830bb6 commit 55508d1

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

models/spring-ai-openai/src/test/java/org/springframework/ai/openai/metadata/OpenAiUsageTests.java

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ void whenTotalTokensIsNull() {
7474

7575
@Test
7676
void whenPromptAndCompletionTokensDetailsIsNull() {
77-
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null, null);
77+
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null, null, null, null);
7878
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
7979
assertThat(usage.getTotalTokens()).isEqualTo(300);
8080
assertThat(usage.getCompletionTokenDetails().reasoningTokens()).isEqualTo(0);
@@ -85,7 +85,7 @@ void whenPromptAndCompletionTokensDetailsIsNull() {
8585

8686
@Test
8787
void whenCompletionTokenDetailsIsNull() {
88-
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null, null);
88+
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null, null, null, null);
8989
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
9090
assertThat(usage.getTotalTokens()).isEqualTo(300);
9191
assertThat(usage.getReasoningTokens()).isEqualTo(0);
@@ -94,15 +94,15 @@ void whenCompletionTokenDetailsIsNull() {
9494
@Test
9595
void whenReasoningTokensIsNull() {
9696
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null,
97-
new OpenAiApi.Usage.CompletionTokenDetails(null, null, null, null));
97+
new OpenAiApi.Usage.CompletionTokenDetails(null, null, null, null), null, null);
9898
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
9999
assertThat(usage.getReasoningTokens()).isEqualTo(0);
100100
}
101101

102102
@Test
103103
void whenCompletionTokenDetailsIsPresent() {
104104
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null,
105-
new OpenAiApi.Usage.CompletionTokenDetails(50, null, null, null));
105+
new OpenAiApi.Usage.CompletionTokenDetails(50, null, null, null), null, null);
106106
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
107107
assertThat(usage.getCompletionTokenDetails().reasoningTokens()).isEqualTo(50);
108108
assertThat(usage.getCompletionTokenDetails().acceptedPredictionTokens()).isEqualTo(0);
@@ -113,7 +113,7 @@ void whenCompletionTokenDetailsIsPresent() {
113113
@Test
114114
void whenAcceptedPredictionTokensIsPresent() {
115115
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null,
116-
new OpenAiApi.Usage.CompletionTokenDetails(null, 75, null, null));
116+
new OpenAiApi.Usage.CompletionTokenDetails(null, 75, null, null), null, null);
117117
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
118118
assertThat(usage.getCompletionTokenDetails().reasoningTokens()).isEqualTo(0);
119119
assertThat(usage.getCompletionTokenDetails().acceptedPredictionTokens()).isEqualTo(75);
@@ -124,7 +124,7 @@ void whenAcceptedPredictionTokensIsPresent() {
124124
@Test
125125
void whenAudioTokensIsPresent() {
126126
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null,
127-
new OpenAiApi.Usage.CompletionTokenDetails(null, null, 125, null));
127+
new OpenAiApi.Usage.CompletionTokenDetails(null, null, 125, null), null, null);
128128
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
129129
assertThat(usage.getCompletionTokenDetails().reasoningTokens()).isEqualTo(0);
130130
assertThat(usage.getCompletionTokenDetails().acceptedPredictionTokens()).isEqualTo(0);
@@ -135,7 +135,7 @@ void whenAudioTokensIsPresent() {
135135
@Test
136136
void whenRejectedPredictionTokensIsNull() {
137137
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null,
138-
new OpenAiApi.Usage.CompletionTokenDetails(null, null, null, null));
138+
new OpenAiApi.Usage.CompletionTokenDetails(null, null, null, null), null, null);
139139
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
140140
assertThat(usage.getCompletionTokenDetails().reasoningTokens()).isEqualTo(0);
141141
assertThat(usage.getCompletionTokenDetails().acceptedPredictionTokens()).isEqualTo(0);
@@ -147,7 +147,7 @@ void whenRejectedPredictionTokensIsNull() {
147147
@Test
148148
void whenRejectedPredictionTokensIsPresent() {
149149
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300, null,
150-
new OpenAiApi.Usage.CompletionTokenDetails(null, null, null, 25));
150+
new OpenAiApi.Usage.CompletionTokenDetails(null, null, null, 25), null, null);
151151
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
152152
assertThat(usage.getCompletionTokenDetails().reasoningTokens()).isEqualTo(0);
153153
assertThat(usage.getCompletionTokenDetails().acceptedPredictionTokens()).isEqualTo(0);
@@ -158,7 +158,7 @@ void whenRejectedPredictionTokensIsPresent() {
158158
@Test
159159
void whenCacheTokensIsNull() {
160160
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300,
161-
new OpenAiApi.Usage.PromptTokensDetails(null, null), null);
161+
new OpenAiApi.Usage.PromptTokensDetails(null, null), null, null, null);
162162
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
163163
assertThat(usage.getPromptTokensDetails().audioTokens()).isEqualTo(0);
164164
assertThat(usage.getPromptTokensDetails().cachedTokens()).isEqualTo(0);
@@ -167,7 +167,7 @@ void whenCacheTokensIsNull() {
167167
@Test
168168
void whenCacheTokensIsPresent() {
169169
OpenAiApi.Usage openAiUsage = new OpenAiApi.Usage(100, 200, 300,
170-
new OpenAiApi.Usage.PromptTokensDetails(99, 15), null);
170+
new OpenAiApi.Usage.PromptTokensDetails(99, 15), null, null, null);
171171
OpenAiUsage usage = OpenAiUsage.from(openAiUsage);
172172
assertThat(usage.getPromptTokensDetails().audioTokens()).isEqualTo(99);
173173
assertThat(usage.getPromptTokensDetails().cachedTokens()).isEqualTo(15);

0 commit comments

Comments
 (0)