topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-12 12:37:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(fix)(headless)Avoid IndexOutOfBound exception in some cases.

Browse Source
This commit is contained in:
jerryjzhang
2024-08-20 15:48:24 +08:00
parent ca098b576c
commit 7e013ca36a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/parser/llm/PromptHelper.java
View File

@@ -49,7 +49,7 @@ public class PromptHelper {
for (int i = 0; i < selfConsistencyNumber; i++) { for (int i = 0; i < selfConsistencyNumber; i++) {
List<Text2SQLExemplar> shuffledList = new ArrayList<>(exemplars); List<Text2SQLExemplar> shuffledList = new ArrayList<>(exemplars);
Collections.shuffle(shuffledList); Collections.shuffle(shuffledList);
results.add(shuffledList.subList(0, fewShotNumber)); results.add(shuffledList.subList(0, Math.min(shuffledList.size(), fewShotNumber)));
} }
return results; return results;