diff --git a/src/main/java/cn/kazusa/ai/assistant/config/OllamaConfig.java b/src/main/java/cn/kazusa/ai/assistant/config/OllamaConfig.java new file mode 100644 index 0000000..73d9aa0 --- /dev/null +++ b/src/main/java/cn/kazusa/ai/assistant/config/OllamaConfig.java @@ -0,0 +1,25 @@ +package cn.kazusa.ai.assistant.config; + +import java.time.Duration; +import org.springframework.boot.web.client.ClientHttpRequestFactories; +import org.springframework.boot.web.client.ClientHttpRequestFactorySettings; +import org.springframework.boot.web.client.RestClientCustomizer; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; + +/** + * @author zouzhiwen + */ +@Configuration +public class OllamaConfig { + + @Bean + public RestClientCustomizer restClientCustomizer() { + return restClientBuilder -> { + ClientHttpRequestFactorySettings settings = ClientHttpRequestFactorySettings.DEFAULTS + .withConnectTimeout(Duration.ofSeconds(30)) + .withReadTimeout(Duration.ofMinutes(5)); + restClientBuilder.requestFactory(ClientHttpRequestFactories.get(settings)); + }; + } +} diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml index 01c1340..fa83d6c 100644 --- a/src/main/resources/application.yml +++ b/src/main/resources/application.yml @@ -26,7 +26,7 @@ assistant: domains: - id: general name: 通用知识 - keywords: [] + keywords: [ ] - id: warehouse name: 仓库领域知识(订单、入库、移库、调拨、出库等) - keywords: [] + keywords: [ ]