From ab9e5f4c12aab92eb99a4024bf14c58eaa81da5f Mon Sep 17 00:00:00 2001
From: Manish Bainsla <mbainsla@redhat.com>
Date: Sun, 8 Dec 2024 08:10:46 +0530
Subject: [PATCH] fix:: arrays index out of bound error fix in
 OnnxBertBiEncoder::weightedAverage

---
 .../langchain4j/model/embedding/onnx/OnnxBertBiEncoder.java   | 4 ++++
 1 file changed, 4 insertions(+)

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..e2fdddf 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<float[]> embeddings, List<Integer> weights) {
+        if(Objects.isNull(embeddings) || embeddings.isEmpty()) {
+            return new float[0];
+        }
+
         if (embeddings.size() == 1) {
             return embeddings.get(0);
         }