支持ollama超时时间配置,默认3分钟
parent
01566bba64
commit
3f8643b7a0
@ -0,0 +1,28 @@
|
||||
package com.supervision.pdfqaserver.config;
|
||||
|
||||
import org.springframework.beans.factory.annotation.Qualifier;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.http.client.JdkClientHttpRequestFactory;
|
||||
import org.springframework.web.client.RestClient;
|
||||
|
||||
import java.net.http.HttpClient;
|
||||
import java.time.Duration;
|
||||
|
||||
@Configuration
|
||||
public class OllamaConfig {
|
||||
|
||||
@Value("${spring.ai.ollama.chat.options.timeout:180000}")
|
||||
private int timeout;
|
||||
|
||||
@Bean
|
||||
@Qualifier("restClientBuilder")
|
||||
public RestClient.Builder restClientBuilder() {
|
||||
JdkClientHttpRequestFactory requestFactory = new JdkClientHttpRequestFactory(
|
||||
HttpClient.newHttpClient());
|
||||
requestFactory.setReadTimeout(Duration.ofMillis(timeout));
|
||||
return RestClient.builder().requestFactory(requestFactory);
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in New Issue