DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

增加查询train modelTableQuery

Browse Source
This commit is contained in:
limin 2025-01-11 10:05:20 +08:00
parent a1df852642
commit 593328dff1
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
View File

@ -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);