From 593328dff19930e573b0551fe37dc6ca4017d6c7 Mon Sep 17 00:00:00 2001
From: limin <limin@qq.com>
Date: Sat, 11 Jan 2025 10:05:20 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=9F=A5=E8=AF=A2train=20mod?=
 =?UTF-8?q?elTableQuery?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../module/llm/service/http/TrainHttpService.java   | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
index d2d63a9b8..806b04056 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
@@ -7,10 +7,13 @@ import cn.iocoder.yudao.module.llm.framework.backend.config.LLMBackendProperties
 import cn.iocoder.yudao.module.llm.service.http.vo.*;
 import com.alibaba.fastjson.JSON;
 import lombok.extern.slf4j.Slf4j;
+import org.hibernate.validator.constraints.NotBlank;
+import org.hibernate.validator.constraints.NotEmpty;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 
 /**
@@ -155,6 +158,16 @@ public class TrainHttpService {
         }
         return null;
     }
+    /**
+     * 根据表名称查询数据
+     */
+    public String modelTableQuery(Map<String, String> headers, String tableName,String query){
+        String url = llmBackendProperties.getTableDataQuery();
+        String res = HttpUtils.getBody(String.format(url,tableName), headers, query);
+        log.info(" model query info :{}", res);
+        return res;
+
+    }
 
     public AigcFineTuningDetailRespVO finetuningDetail(Map<String, String> headers, String jobId){
         login(headers);