From 25a68d44aa6564d8aa96eb707de9fab882991d93 Mon Sep 17 00:00:00 2001
From: "Tahir Kardak (ETIYA)" <tahir.kardak@etiya.com>
Date: Thu, 3 Oct 2024 13:46:55 +0300
Subject: [PATCH] Update OnnxBertBiEncoder.java

IndexOutOfBoundsException execption is fixed. Add validation to ensure embeddings list is not empty.
changes:
-- added a check for an empty "embeddings"
-- Throws  RuntimeException if the embeddings list empty
---
 .../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..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<float[]> embeddings, List<Integer> weights) {
+        if (embeddings.isEmpty()) {
+            throw new RuntimeException("Embeddings list cannot be empty");
+        }
+
         if (embeddings.size() == 1) {
             return embeddings.get(0);
         }