Skip to content

Commit 77ee57c

Browse files
dev-jonghoonparkilayaperumalg
authored andcommitted
allow null usage for openai embedding model
Signed-off-by: jonghoon park <[email protected]>
1 parent cdac6ee commit 77ee57c

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

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

+5-2
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@
2323
import org.slf4j.LoggerFactory;
2424

2525
import org.springframework.ai.chat.metadata.DefaultUsage;
26+
import org.springframework.ai.chat.metadata.EmptyUsage;
27+
import org.springframework.ai.chat.metadata.Usage;
2628
import org.springframework.ai.document.Document;
2729
import org.springframework.ai.document.MetadataMode;
2830
import org.springframework.ai.embedding.AbstractEmbeddingModel;
@@ -167,8 +169,9 @@ public EmbeddingResponse call(EmbeddingRequest request) {
167169
return new EmbeddingResponse(List.of());
168170
}
169171

170-
var metadata = new EmbeddingResponseMetadata(apiEmbeddingResponse.model(),
171-
getDefaultUsage(apiEmbeddingResponse.usage()));
172+
OpenAiApi.Usage usage = apiEmbeddingResponse.usage();
173+
Usage embeddingResponseUsage = usage != null ? getDefaultUsage(usage) : new EmptyUsage();
174+
var metadata = new EmbeddingResponseMetadata(apiEmbeddingResponse.model(), embeddingResponseUsage);
172175

173176
List<Embedding> embeddings = apiEmbeddingResponse.data()
174177
.stream()

0 commit comments

Comments
 (0)