diff --git a/yudao-framework/yudao-common/src/main/java/cn/iocoder/yudao/framework/common/util/http/HttpUtils.java b/yudao-framework/yudao-common/src/main/java/cn/iocoder/yudao/framework/common/util/http/HttpUtils.java index cdc454a6c..79b2e662c 100644 --- a/yudao-framework/yudao-common/src/main/java/cn/iocoder/yudao/framework/common/util/http/HttpUtils.java +++ b/yudao-framework/yudao-common/src/main/java/cn/iocoder/yudao/framework/common/util/http/HttpUtils.java @@ -169,6 +169,15 @@ public class HttpUtils { } } + public static String getBody(String url, Map headers,String body) { + try (HttpResponse response = HttpRequest.get(url) + .addHeaders(headers) + .body(body) + .execute()) { + return response.body(); + } + } + public static String del(String url, Map headers) { try (HttpResponse response = HttpRequest.delete(url) .addHeaders(headers) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java index 9414a3a5d..d34bfc66d 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java @@ -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 headers){ + public String ragQueryMultiple(Map headers, RagQueryMultipleReqVo ragQueryReqVo){ String ragQueryMultiple = llmBackendProperties.getRagQueryMultiple(); - String res = HttpUtils.get(ragQueryMultiple, headers); + String res = HttpUtils.getBody(ragQueryMultiple, headers, JSON.toJSONString(ragQueryReqVo)); return res; } diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/RagQueryMultipleReqVo.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/RagQueryMultipleReqVo.java new file mode 100644 index 000000000..6bf7d30ce --- /dev/null +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/RagQueryMultipleReqVo.java @@ -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 file_ids; + + private Integer k; + +}