File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
models/spring-ai-openai/src/main/java/org/springframework/ai/openai Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 2323import org .slf4j .LoggerFactory ;
2424
2525import org .springframework .ai .chat .metadata .DefaultUsage ;
26+ import org .springframework .ai .chat .metadata .EmptyUsage ;
27+ import org .springframework .ai .chat .metadata .Usage ;
2628import org .springframework .ai .document .Document ;
2729import org .springframework .ai .document .MetadataMode ;
2830import 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 ()
You can’t perform that action at this time.
0 commit comments