diff --git a/src/main/java/com/olympus/hermione/config/AzureAIConfig.java b/src/main/java/com/olympus/hermione/config/AzureAIConfig.java new file mode 100644 index 0000000..9faf1cf --- /dev/null +++ b/src/main/java/com/olympus/hermione/config/AzureAIConfig.java @@ -0,0 +1,27 @@ +package com.olympus.hermione.config; + + import java.time.Duration; + +import org.springframework.ai.model.azure.openai.autoconfigure.AzureOpenAIClientBuilderCustomizer; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; + +import com.azure.core.http.HttpClient; +import com.azure.core.util.HttpClientOptions; + + +@Configuration +public class AzureAIConfig { + + @Bean + public AzureOpenAIClientBuilderCustomizer responseTimeoutCustomizer() { + return openAiClientBuilder -> { + HttpClientOptions clientOptions = new HttpClientOptions() + .setResponseTimeout(Duration.ofMinutes(5)); + openAiClientBuilder.httpClient(HttpClient.createDefault(clientOptions)); + }; + } + + + +} \ No newline at end of file