Skip to content

Commit 2e26867

Browse files
committed
OpenAI: refactor code to clean
Signed-off-by: lambochen <[email protected]>
1 parent 2a2320e commit 2e26867

File tree

5 files changed

+32
-34
lines changed

5 files changed

+32
-34
lines changed

auto-configurations/models/spring-ai-autoconfigure-model-openai/src/main/java/org/springframework/ai/model/openai/autoconfigure/OpenAiAudioSpeechAutoConfiguration.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,8 @@ public OpenAiAudioSpeechModel openAiAudioSpeechModel(OpenAiConnectionProperties
7272
var openAiAudioApi = OpenAiAudioApi.builder()
7373
.baseUrl(resolved.baseUrl())
7474
.apiKey(new SimpleApiKey(resolved.apiKey()))
75-
.audioSpeechPath(speechProperties.getAudioSpeechPath())
7675
.headers(resolved.headers())
76+
.audioSpeechPath(speechProperties.getAudioSpeechPath())
7777
.restClientBuilder(restClientBuilderProvider.getIfAvailable(RestClient::builder))
7878
.webClientBuilder(webClientBuilderProvider.getIfAvailable(WebClient::builder))
7979
.responseErrorHandler(responseErrorHandler)

auto-configurations/models/spring-ai-autoconfigure-model-openai/src/main/java/org/springframework/ai/model/openai/autoconfigure/OpenAiAudioTranscriptionAutoConfiguration.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,8 +72,8 @@ public OpenAiAudioTranscriptionModel openAiAudioTranscriptionModel(OpenAiConnect
7272
var openAiAudioApi = OpenAiAudioApi.builder()
7373
.baseUrl(resolved.baseUrl())
7474
.apiKey(new SimpleApiKey(resolved.apiKey()))
75-
.audioTranscriptionPath(transcriptionProperties.getAudioTranscriptionPath())
7675
.headers(resolved.headers())
76+
.audioTranscriptionPath(transcriptionProperties.getAudioTranscriptionPath())
7777
.restClientBuilder(restClientBuilderProvider.getIfAvailable(RestClient::builder))
7878
.webClientBuilder(webClientBuilderProvider.getIfAvailable(WebClient::builder))
7979
.responseErrorHandler(responseErrorHandler)

auto-configurations/models/spring-ai-autoconfigure-model-openai/src/main/java/org/springframework/ai/model/openai/autoconfigure/OpenAiModerationAutoConfiguration.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,9 +69,9 @@ public OpenAiModerationModel openAiModerationModel(OpenAiConnectionProperties co
6969

7070
var openAiModerationApi = OpenAiModerationApi.builder()
7171
.baseUrl(resolved.baseUrl())
72-
.moderationPath(moderationProperties.getModerationPath())
7372
.apiKey(new SimpleApiKey(resolved.apiKey()))
7473
.headers(resolved.headers())
74+
.moderationPath(moderationProperties.getModerationPath())
7575
.restClientBuilder(restClientBuilderProvider.getIfAvailable(RestClient::builder))
7676
.responseErrorHandler(responseErrorHandler)
7777
.build();

models/spring-ai-openai/src/main/java/org/springframework/ai/openai/api/OpenAiAudioApi.java

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -112,25 +112,25 @@ public OpenAiAudioApi(String baseUrl, ApiKey apiKey, MultiValueMap<String, Strin
112112

113113
// @formatter:off
114114
this.restClient = restClientBuilder.clone()
115-
.baseUrl(baseUrl)
116-
.defaultHeaders(authHeaders)
117-
.defaultStatusHandler(responseErrorHandler)
118-
.defaultRequest(requestHeadersSpec -> {
119-
if (!(apiKey instanceof NoopApiKey)) {
120-
requestHeadersSpec.header(HttpHeaders.AUTHORIZATION, "Bearer " + apiKey.getValue());
121-
}
122-
})
123-
.build();
115+
.baseUrl(baseUrl)
116+
.defaultHeaders(authHeaders)
117+
.defaultStatusHandler(responseErrorHandler)
118+
.defaultRequest(requestHeadersSpec -> {
119+
if (!(apiKey instanceof NoopApiKey)) {
120+
requestHeadersSpec.header(HttpHeaders.AUTHORIZATION, "Bearer " + apiKey.getValue());
121+
}
122+
})
123+
.build();
124124

125125
this.webClient = webClientBuilder.clone()
126-
.baseUrl(baseUrl)
127-
.defaultHeaders(authHeaders)
128-
.defaultRequest(requestHeadersSpec -> {
129-
if (!(apiKey instanceof NoopApiKey)) {
130-
requestHeadersSpec.header(HttpHeaders.AUTHORIZATION, "Bearer " + apiKey.getValue());
131-
}
132-
})
133-
.build(); // @formatter:on
126+
.baseUrl(baseUrl)
127+
.defaultHeaders(authHeaders)
128+
.defaultRequest(requestHeadersSpec -> {
129+
if (!(apiKey instanceof NoopApiKey)) {
130+
requestHeadersSpec.header(HttpHeaders.AUTHORIZATION, "Bearer " + apiKey.getValue());
131+
}
132+
})
133+
.build(); // @formatter:on
134134
}
135135

136136
public static Builder builder() {

models/spring-ai-openai/src/main/java/org/springframework/ai/openai/api/OpenAiModerationApi.java

Lines changed: 12 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,6 @@
2525
import org.springframework.ai.model.ApiKey;
2626
import org.springframework.ai.model.NoopApiKey;
2727
import org.springframework.ai.model.SimpleApiKey;
28-
import org.springframework.ai.moderation.Categories;
29-
import org.springframework.ai.moderation.CategoryScores;
3028
import org.springframework.ai.openai.api.common.OpenAiApiConstants;
3129
import org.springframework.ai.retry.RetryUtils;
3230
import org.springframework.http.HttpHeaders;
@@ -88,18 +86,18 @@ public OpenAiModerationApi(String baseUrl, ApiKey apiKey, MultiValueMap<String,
8886

8987
// @formatter:off
9088
this.restClient = restClientBuilder.clone()
91-
.baseUrl(baseUrl)
92-
.defaultHeaders(h -> {
93-
h.setContentType(MediaType.APPLICATION_JSON);
94-
h.addAll(headers);
95-
})
96-
.defaultStatusHandler(responseErrorHandler)
97-
.defaultRequest(requestHeadersSpec -> {
98-
if (!(apiKey instanceof NoopApiKey)) {
99-
requestHeadersSpec.header(HttpHeaders.AUTHORIZATION, "Bearer " + apiKey.getValue());
100-
}
101-
})
102-
.build(); // @formatter:on
89+
.baseUrl(baseUrl)
90+
.defaultHeaders(h -> {
91+
h.setContentType(MediaType.APPLICATION_JSON);
92+
h.addAll(headers);
93+
})
94+
.defaultStatusHandler(responseErrorHandler)
95+
.defaultRequest(requestHeadersSpec -> {
96+
if (!(apiKey instanceof NoopApiKey)) {
97+
requestHeadersSpec.header(HttpHeaders.AUTHORIZATION, "Bearer " + apiKey.getValue());
98+
}
99+
})
100+
.build(); // @formatter:on
103101
}
104102

105103
public ResponseEntity<OpenAiModerationResponse> createModeration(OpenAiModerationRequest openAiModerationRequest) {

0 commit comments

Comments
 (0)