@@ -163,25 +163,26 @@ private float[] getSentenceEmbeddingAsFloatArrayFor(String text) {
163
163
private List <byte []> getEmbeddingsAsByteArrayFor (List <String > texts , EmbeddingModel model ) {
164
164
EmbeddingResponse embeddingResponse = model .embedForResponse (texts );
165
165
List <Embedding > embeddings = embeddingResponse .getResults ();
166
- return embeddings .stream ().map (e -> ObjectUtils .doubleListToByteArray (e .getOutput ())).toList ();
166
+
167
+ return embeddings .stream ().map (e -> ObjectUtils .floatArrayToByteArray (e .getOutput ())).toList ();
167
168
}
168
169
169
170
private List <float []> getEmbeddingAsFloatArrayFor (List <String > texts , EmbeddingModel model ) {
170
171
EmbeddingResponse embeddingResponse = model .embedForResponse (texts );
171
172
List <Embedding > embeddings = embeddingResponse .getResults ();
172
- return embeddings .stream ().map (e -> ObjectUtils . doubleListToFloatArray ( e . getOutput ()) ).toList ();
173
+ return embeddings .stream ().map (Embedding :: getOutput ).toList ();
173
174
}
174
175
175
176
private byte [] getEmbeddingsAsByteArrayFor (String text , EmbeddingModel model ) {
176
177
EmbeddingResponse embeddingResponse = model .embedForResponse (List .of (text ));
177
178
Embedding embedding = embeddingResponse .getResult ();
178
- return ObjectUtils .doubleListToByteArray (embedding .getOutput ());
179
+ return ObjectUtils .floatArrayToByteArray (embedding .getOutput ());
179
180
}
180
181
181
182
private float [] getEmbeddingAsFloatArrayFor (String text , EmbeddingModel model ) {
182
183
EmbeddingResponse embeddingResponse = model .embedForResponse (List .of (text ));
183
184
Embedding embedding = embeddingResponse .getResult ();
184
- return ObjectUtils . doubleListToFloatArray ( embedding .getOutput () );
185
+ return embedding .getOutput ();
185
186
}
186
187
187
188
@ Override
0 commit comments