Skip to content

Commit 741284d

Browse files
committed
fix: 누락된 코드 추가
1 parent d41287b commit 741284d

File tree

3 files changed

+24
-0
lines changed

3 files changed

+24
-0
lines changed

src/main/java/com/techfork/global/llm/config/OpenAiConfig.java

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,10 @@
99
import org.springframework.beans.factory.annotation.Value;
1010
import org.springframework.context.annotation.Bean;
1111
import org.springframework.context.annotation.Configuration;
12+
import org.springframework.http.client.SimpleClientHttpRequestFactory;
13+
import org.springframework.web.client.RestClient;
14+
15+
import java.time.Duration;
1216

1317
@Configuration
1418
public class OpenAiConfig {
@@ -24,10 +28,16 @@ public class OpenAiConfig {
2428
@Value("${spring.ai.openai.chat.options.max-tokens}")
2529
private Integer maxTokens;
2630

31+
@Value("${spring.ai.openai.timeout:60}")
32+
private Integer timeoutSeconds;
33+
2734
@Bean
2835
public OpenAiChatModel openAiChatModel() {
36+
RestClient.Builder restClientBuilder = createRestClientBuilder();
37+
2938
OpenAiApi openAiApi = OpenAiApi.builder()
3039
.apiKey(apiKey)
40+
.restClientBuilder(restClientBuilder)
3141
.build();
3242

3343
OpenAiChatOptions options = OpenAiChatOptions.builder()
@@ -44,8 +54,11 @@ public OpenAiChatModel openAiChatModel() {
4454

4555
@Bean
4656
public OpenAiEmbeddingModel openAiEmbeddingModel() {
57+
RestClient.Builder restClientBuilder = createRestClientBuilder();
58+
4759
OpenAiApi openAiApi = OpenAiApi.builder()
4860
.apiKey(apiKey)
61+
.restClientBuilder(restClientBuilder)
4962
.build();
5063

5164
OpenAiEmbeddingOptions options = OpenAiEmbeddingOptions.builder()
@@ -55,4 +68,13 @@ public OpenAiEmbeddingModel openAiEmbeddingModel() {
5568

5669
return new OpenAiEmbeddingModel(openAiApi, MetadataMode.EMBED, options);
5770
}
71+
72+
private RestClient.Builder createRestClientBuilder() {
73+
SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();
74+
requestFactory.setConnectTimeout(Duration.ofSeconds(timeoutSeconds));
75+
requestFactory.setReadTimeout(Duration.ofSeconds(timeoutSeconds));
76+
77+
return RestClient.builder()
78+
.requestFactory(requestFactory);
79+
}
5880
}

src/main/resources/application-dev.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ spring:
5252
max-tokens: 8192
5353
openai:
5454
api-key: ${OPENAI_API_KEY}
55+
timeout: 60
5556
chat:
5657
options:
5758
model: gpt-4o-mini

src/main/resources/application-local.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ spring:
5252
max-tokens: 8192
5353
openai:
5454
api-key: ${OPENAI_API_KEY}
55+
timeout: 60
5556
chat:
5657
options:
5758
model: gpt-4o-mini

0 commit comments

Comments
 (0)