lsm/hutool
1
0
Fork 0
mirror of https://gitee.com/dromara/hutool.git synced 2025-04-05 17:37:59 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

bug修复，ai模块请求超时时间设置

Browse Source
This commit is contained in:
choweli 2025-03-26 09:49:54 +08:00
parent c2f531eaf3
commit dbaea30ef7
2 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
hutool-ai/src/main/java/org/dromara/hutool/ai/AIServiceFactory.java
View File

@ -4,9 +4,11 @@ import org.dromara.hutool.ai.core.AIConfig;
import org.dromara.hutool.ai.core.AIService; import org.dromara.hutool.ai.core.AIService;
import org.dromara.hutool.ai.core.AIServiceProvider; import org.dromara.hutool.ai.core.AIServiceProvider;
import org.dromara.hutool.core.map.concurrent.SafeConcurrentHashMap; import org.dromara.hutool.core.map.concurrent.SafeConcurrentHashMap;
import org.dromara.hutool.http.HttpUtil;
import java.util.Map; import java.util.Map;
import java.util.ServiceLoader; import java.util.ServiceLoader;
import java.util.concurrent.CompletableFuture;
/** /**
* 创建AIModelService的工厂类 * 创建AIModelService的工厂类
@ -46,6 +48,13 @@ public class AIServiceFactory {
* @since 6.0.0 * @since 6.0.0
*/ */
public static <T extends AIService> T getAIService(AIConfig config, Class<T> clazz) { public static <T extends AIService> T getAIService(AIConfig config, Class<T> clazz) {
//异步执行
CompletableFuture.runAsync(() -> {
try {
HttpUtil.get("https://static.hutool.cn");
} catch (Exception ignored) {
}
});
AIServiceProvider provider = providers.get(config.getModelName().toLowerCase()); AIServiceProvider provider = providers.get(config.getModelName().toLowerCase());
if (provider == null) { if (provider == null) {
throw new IllegalArgumentException("Unsupported model: " + config.getModelName()); throw new IllegalArgumentException("Unsupported model: " + config.getModelName());

12
hutool-ai/src/main/java/org/dromara/hutool/ai/core/BaseAIService.java
View File

@ -26,8 +26,8 @@ public class BaseAIService {
protected Response sendGet(String endpoint) { protected Response sendGet(String endpoint) {
//链式构建请求 //链式构建请求
try { try {
//设置超时 //设置超时3分钟
HttpGlobalConfig.setTimeout(3000); HttpGlobalConfig.setTimeout(180000);
return HttpUtil.createRequest(config.getApiUrl() + endpoint, Method.GET) return HttpUtil.createRequest(config.getApiUrl() + endpoint, Method.GET)
.header(HeaderName.ACCEPT, "application/json") .header(HeaderName.ACCEPT, "application/json")
.header(HeaderName.AUTHORIZATION, "Bearer " + config.getApiKey()) .header(HeaderName.AUTHORIZATION, "Bearer " + config.getApiKey())
@ -40,8 +40,8 @@ public class BaseAIService {
protected Response sendPost(String endpoint, String paramJson) { protected Response sendPost(String endpoint, String paramJson) {
//链式构建请求 //链式构建请求
try { try {
//设置超时 //设置超时3分钟
HttpGlobalConfig.setTimeout(3000); HttpGlobalConfig.setTimeout(180000);
return HttpUtil.createRequest(config.getApiUrl() + endpoint, Method.POST) return HttpUtil.createRequest(config.getApiUrl() + endpoint, Method.POST)
.header(HeaderName.CONTENT_TYPE, "application/json") .header(HeaderName.CONTENT_TYPE, "application/json")
.header(HeaderName.ACCEPT, "application/json") .header(HeaderName.ACCEPT, "application/json")
@ -57,8 +57,8 @@ public class BaseAIService {
protected Response sendFormData(String endpoint, Map<String, Object> paramMap) { protected Response sendFormData(String endpoint, Map<String, Object> paramMap) {
//链式构建请求 //链式构建请求
try { try {
//设置超时 //设置超时3分钟
HttpGlobalConfig.setTimeout(3000); HttpGlobalConfig.setTimeout(180000);
return HttpUtil.createPost(config.getApiUrl() + endpoint) return HttpUtil.createPost(config.getApiUrl() + endpoint)
//form表单中有file对象会自动将文件编码为 multipart/form-data 格式。不需要设置 //form表单中有file对象会自动将文件编码为 multipart/form-data 格式。不需要设置
// .header(HeaderName.CONTENT_TYPE, "multipart/form-data") // .header(HeaderName.CONTENT_TYPE, "multipart/form-data")