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 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 headers, String jobId){ login(headers);