From fc56bbc6edaa0c43616e2647e9735e1b5ed26a11 Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Mon, 13 Jan 2025 18:00:21 +0800 Subject: [PATCH] =?UTF-8?q?Csv=20=E8=A7=A3=E6=9E=90=20=E6=A0=B9=E6=8D=AE?= =?UTF-8?q?=E6=A0=87=E9=A2=98=E8=A1=8C=E8=BF=9B=E8=A1=8C=E8=A7=A3=E6=9E=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../module/llm/utils/DataSetReadFileUtils.java | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/utils/DataSetReadFileUtils.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/utils/DataSetReadFileUtils.java index 1dc9d046a..e893139ad 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/utils/DataSetReadFileUtils.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/utils/DataSetReadFileUtils.java @@ -84,8 +84,20 @@ public class DataSetReadFileUtils { // 动态读取，当行长度与标题行长度相等时 if (line.length == headers.length) { + // 获取系统列索引 + int systemIndex = getIndex(headers, "system"); + // 获取问题列索引 + int questionIndex = getIndex(headers, "question"); + // 获取答案列索引 + int answerIndex = getIndex(headers, "answer"); + // 存储系统列的值 + String systemValue = systemIndex == -1? "" : line[systemIndex]; + // 存储问题列的值 + String questionValue = questionIndex == -1? "" : line[questionIndex]; + // 存储答案列的值 + String answerValue = answerIndex == -1? "" : line[answerIndex]; // 根据标题行找到相应列的索引创建 CsvDataSetVO 对象 - CsvDataSetVO dataSetVO = new CsvDataSetVO(line[getIndex(headers, "system")], line[getIndex(headers, "question")], line[getIndex(headers, "answer")]); + CsvDataSetVO dataSetVO = new CsvDataSetVO(systemValue, questionValue, answerValue); // 将对象添加到列表中 dataSetVos.add(dataSetVO); }