Skip to content

Commit 9899307

Browse files
committed
[FEL] remove apikey header while input apikey is null
1 parent 4be7382 commit 9899307

File tree

1 file changed

+12
-4
lines changed
  • framework/fel/java/fel-community/model-openai/src/main/java/modelengine/fel/community/model/openai

1 file changed

+12
-4
lines changed

framework/fel/java/fel-community/model-openai/src/main/java/modelengine/fel/community/model/openai/OpenAiModel.java

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,9 @@ public List<Embedding> generate(List<String> inputs, EmbedOption option) {
132132
notBlank(option.model(), "The embed model name cannot be null.");
133133
HttpClassicClientRequest request = this.httpClient.get()
134134
.createRequest(HttpRequestMethod.POST, UrlUtils.combine(this.baseUrl, OpenAiApi.EMBEDDING_ENDPOINT));
135-
HttpUtils.setBearerAuth(request, StringUtils.blankIf(option.apiKey(), this.defaultApiKey));
135+
if (!StringUtils.isBlank(option.apiKey())) {
136+
HttpUtils.setBearerAuth(request, option.apiKey());
137+
}
136138
request.jsonEntity(new OpenAiEmbeddingRequest(inputs, option.model()));
137139
Class<OpenAiEmbeddingResponse> clazz = OpenAiEmbeddingResponse.class;
138140
try (HttpClassicClientResponse<OpenAiEmbeddingResponse> response = request.exchange(clazz)) {
@@ -151,7 +153,9 @@ public Choir<ChatMessage> generate(Prompt prompt, ChatOption chatOption) {
151153
String modelSource = StringUtils.blankIf(chatOption.baseUrl(), this.baseUrl);
152154
HttpClassicClientRequest request = this.getHttpClient(chatOption.secureConfig())
153155
.createRequest(HttpRequestMethod.POST, UrlUtils.combine(modelSource, OpenAiApi.CHAT_ENDPOINT));
154-
HttpUtils.setBearerAuth(request, StringUtils.blankIf(chatOption.apiKey(), this.defaultApiKey));
156+
if (!StringUtils.isBlank(chatOption.apiKey())) {
157+
HttpUtils.setBearerAuth(request, chatOption.apiKey());
158+
}
155159
request.jsonEntity(new OpenAiChatCompletionRequest(prompt, chatOption));
156160
return chatOption.stream() ? this.createChatStream(request) : this.createChatCompletion(request);
157161
}
@@ -163,7 +167,9 @@ public List<Media> generate(String prompt, ImageOption option) {
163167
String modelSource = StringUtils.blankIf(option.baseUrl(), this.baseUrl);
164168
HttpClassicClientRequest request = this.httpClient.get()
165169
.createRequest(HttpRequestMethod.POST, UrlUtils.combine(modelSource, OpenAiApi.IMAGE_ENDPOINT));
166-
HttpUtils.setBearerAuth(request, StringUtils.blankIf(option.apiKey(), this.defaultApiKey));
170+
if (!StringUtils.isBlank(option.apiKey())) {
171+
HttpUtils.setBearerAuth(request, option.apiKey());
172+
}
167173
request.jsonEntity(new OpenAiImageRequest(option.model(), option.size(), prompt));
168174
Class<OpenAiImageResponse> clazz = OpenAiImageResponse.class;
169175
try (HttpClassicClientResponse<OpenAiImageResponse> response = request.exchange(clazz)) {
@@ -182,7 +188,9 @@ public List<MeasurableDocument> generate(List<MeasurableDocument> documents, Rer
182188
String modelSource = StringUtils.blankIf(rerankOption.baseUri(), this.baseUrl);
183189
HttpClassicClientRequest request = this.getHttpClient(rerankOption.secureConfig())
184190
.createRequest(HttpRequestMethod.POST, UrlUtils.combine(modelSource, OpenAiApi.RERANK_ENDPOINT));
185-
HttpUtils.setBearerAuth(request, StringUtils.blankIf(rerankOption.apiKey(), this.defaultApiKey));
191+
if (!StringUtils.isBlank(rerankOption.apiKey())) {
192+
HttpUtils.setBearerAuth(request, rerankOption.apiKey());
193+
}
186194
List<String> docs = documents.stream().map(MeasurableDocument::text).collect(Collectors.toList());
187195
OpenAiRerankRequest fields = new OpenAiRerankRequest(rerankOption, docs);
188196
request.entity(Entity.createObject(request, fields));

0 commit comments

Comments
 (0)