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 23d9adc..17499f3 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 (embeddings.isEmpty()) { + throw new RuntimeException("Embeddings list cannot be empty"); + } + if (embeddings.size() == 1) { return embeddings.get(0); }