Skip to content

Commit d1133dc

Browse files
wpanasilayaperumalg
authored andcommitted
Fix #4346: Optimize EmbeddingUtils.toPrimitive(List<Float>)
Auto-cherry-pick to 1.0.x Fixes #4346 Signed-off-by: Waldemar Panas <waldemar.panas@allegro.com>
1 parent 62acfbc commit d1133dc

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

spring-ai-model/src/main/java/org/springframework/ai/model/EmbeddingUtils.java

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,17 @@ public static List<Float> doubleToFloat(final List<Double> doubles) {
3939
}
4040

4141
public static float[] toPrimitive(List<Float> floats) {
42-
return toPrimitive(floats.toArray(new Float[floats.size()]));
42+
if (floats == null) {
43+
return null;
44+
}
45+
if (floats.isEmpty()) {
46+
return EMPTY_FLOAT_ARRAY;
47+
}
48+
final float[] result = new float[floats.size()];
49+
for (int i = 0; i < result.length; i++) {
50+
result[i] = floats.get(i);
51+
}
52+
return result;
4353
}
4454

4555
public static float[] toPrimitive(final Float[] array) {

0 commit comments

Comments
 (0)