diff --git a/langchain4j-embeddings/src/main/java/dev/langchain4j/model/embedding/onnx/OnnxBertBiEncoder.java b/langchain4j-embeddings/src/main/java/dev/langchain4j/model/embedding/onnx/OnnxBertBiEncoder.java index 23d9adce..e2fdddf3 100644 --- a/langchain4j-embeddings/src/main/java/dev/langchain4j/model/embedding/onnx/OnnxBertBiEncoder.java +++ b/langchain4j-embeddings/src/main/java/dev/langchain4j/model/embedding/onnx/OnnxBertBiEncoder.java @@ -197,6 +197,10 @@ private static float[] meanPool(float[][] vectors) { } private float[] weightedAverage(List embeddings, List weights) { + if(Objects.isNull(embeddings) || embeddings.isEmpty()) { + return new float[0]; + } + if (embeddings.size() == 1) { return embeddings.get(0); }