From 7e013ca36ab9eea62ea8bcc22a17726b473533d7 Mon Sep 17 00:00:00 2001
From: jerryjzhang <jerryjzhang@tencent.com>
Date: Tue, 20 Aug 2024 15:48:24 +0800
Subject: [PATCH] (fix)(headless)Avoid IndexOutOfBound exception in some cases.

---
 .../supersonic/headless/chat/parser/llm/PromptHelper.java       | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java
index ae52b386b..56d91671b 100644
--- a/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java
+++ b/headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java
@@ -49,7 +49,7 @@ public class PromptHelper {
         for (int i = 0; i < selfConsistencyNumber; i++) {
             List<Text2SQLExemplar> shuffledList = new ArrayList<>(exemplars);
             Collections.shuffle(shuffledList);
-            results.add(shuffledList.subList(0, fewShotNumber));
+            results.add(shuffledList.subList(0, Math.min(shuffledList.size(), fewShotNumber)));
         }
 
         return results;