支持ollama超时时间配置,默认3分钟

master
daixiaoyi 1 month ago
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);
}
}

@ -21,6 +21,7 @@ spring:
top_p: 0.9
top_k: 40
temperature: 0.7
timeout: 180000
neo4j:
driver:
uri: bolt://192.168.10.137:17687

Loading…
Cancel
Save