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

Merge remote-tracking branch 'xnjz/master'

Browse Source
This commit is contained in:
zhangtao 2024-12-29 19:06:55 +08:00
commit fa2d624618
3 changed files with 30 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
yudao-framework/yudao-common/src/main/java/cn/iocoder/yudao/framework/common/util/http/HttpUtils.java
View File

@ -169,6 +169,15 @@ public class HttpUtils {
}
}
public static String getBody(String url, Map<String, String> headers,String body) {
try (HttpResponse response = HttpRequest.get(url)
.addHeaders(headers)
.body(body)
.execute()) {
return response.body();
}
}
public static String del(String url, Map<String, String> headers) {
try (HttpResponse response = HttpRequest.delete(url)
.addHeaders(headers)

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

@ -4,6 +4,7 @@ package cn.iocoder.yudao.module.llm.service.http;
import cn.iocoder.yudao.framework.common.util.http.HttpUtils;
import cn.iocoder.yudao.module.llm.framework.backend.config.LLMBackendProperties;
import cn.iocoder.yudao.module.llm.service.http.vo.RagEmbedReqVo;
import cn.iocoder.yudao.module.llm.service.http.vo.RagQueryMultipleReqVo;
import cn.iocoder.yudao.module.llm.service.http.vo.RagQueryReqVo;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.toolkit.BeanUtils;
@ -76,9 +77,9 @@ public class RagHttpService {
/**
* 支持多个文件id查询向量
*/
public String ragQueryMultiple(Map<String, String> headers){
public String ragQueryMultiple(Map<String, String> headers, RagQueryMultipleReqVo ragQueryReqVo){
String ragQueryMultiple = llmBackendProperties.getRagQueryMultiple();
String res = HttpUtils.get(ragQueryMultiple, headers);
String res = HttpUtils.getBody(ragQueryMultiple, headers, JSON.toJSONString(ragQueryReqVo));
return res;
}

18
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/RagQueryMultipleReqVo.java Normal file
View File

@ -0,0 +1,18 @@
package cn.iocoder.yudao.module.llm.service.http.vo;
import lombok.Data;
import lombok.ToString;
import java.util.List;
@Data
@ToString(callSuper = true)
public class RagQueryMultipleReqVo {
private String query;
private List<String> file_ids;
private Integer k;
}