Skip to content

Commit a11ceef

Browse files
allow null usage for open ai models
Signed-off-by: jonghoon park <[email protected]>
1 parent 690d242 commit a11ceef

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

models/spring-ai-openai/src/main/java/org/springframework/ai/openai/OpenAiChatModel.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -467,6 +467,10 @@ private OpenAiApi.ChatCompletion chunkToChatCompletion(OpenAiApi.ChatCompletionC
467467
}
468468

469469
private DefaultUsage getDefaultUsage(OpenAiApi.Usage usage) {
470+
if (usage == null) {
471+
logger.warn("No usage returned for request");
472+
return null;
473+
}
470474
return new DefaultUsage(usage.promptTokens(), usage.completionTokens(), usage.totalTokens(), usage);
471475
}
472476

models/spring-ai-openai/src/main/java/org/springframework/ai/openai/OpenAiEmbeddingModel.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,10 @@ public EmbeddingResponse call(EmbeddingRequest request) {
184184
}
185185

186186
private DefaultUsage getDefaultUsage(OpenAiApi.Usage usage) {
187+
if (usage == null) {
188+
logger.warn("No usage returned for request");
189+
return null;
190+
}
187191
return new DefaultUsage(usage.promptTokens(), usage.completionTokens(), usage.totalTokens(), usage);
188192
}
189193

0 commit comments

Comments
 (0)