File tree 1 file changed +5
-2
lines changed
models/spring-ai-openai/src/main/java/org/springframework/ai/openai
1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 23
23
import org .slf4j .LoggerFactory ;
24
24
25
25
import org .springframework .ai .chat .metadata .DefaultUsage ;
26
+ import org .springframework .ai .chat .metadata .EmptyUsage ;
27
+ import org .springframework .ai .chat .metadata .Usage ;
26
28
import org .springframework .ai .document .Document ;
27
29
import org .springframework .ai .document .MetadataMode ;
28
30
import org .springframework .ai .embedding .AbstractEmbeddingModel ;
@@ -167,8 +169,9 @@ public EmbeddingResponse call(EmbeddingRequest request) {
167
169
return new EmbeddingResponse (List .of ());
168
170
}
169
171
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 );
172
175
173
176
List <Embedding > embeddings = apiEmbeddingResponse .data ()
174
177
.stream ()
You can’t perform that action at this time.
0 commit comments