Skip to content

Commit 51a0c4b

Browse files
committed
Get defaults first
1 parent 5e43ea5 commit 51a0c4b

File tree

1 file changed

+15
-12
lines changed
  • x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/registry

1 file changed

+15
-12
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/registry/ModelRegistry.java

Lines changed: 15 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -249,15 +249,17 @@ public MinimalServiceSettings getMinimalServiceSettings(String inferenceEntityId
249249
* @param listener Model listener
250250
*/
251251
public void getModelWithSecrets(String inferenceEntityId, ActionListener<UnparsedModel> listener) {
252+
var maybeDefault = defaultConfigIds.get(inferenceEntityId);
253+
if (maybeDefault != null) {
254+
getDefaultConfig(true, maybeDefault, listener);
255+
logger.debug("Returning default inference endpoint [{}] with secrets", inferenceEntityId);
256+
return;
257+
}
258+
252259
ActionListener<SearchResponse> searchListener = ActionListener.wrap((searchResponse) -> {
253260
// There should be a hit for the configurations
254261
if (searchResponse.getHits().getHits().length == 0) {
255-
var maybeDefault = defaultConfigIds.get(inferenceEntityId);
256-
if (maybeDefault != null) {
257-
getDefaultConfig(true, maybeDefault, listener);
258-
} else {
259-
listener.onFailure(inferenceNotFoundException(inferenceEntityId));
260-
}
262+
listener.onFailure(inferenceNotFoundException(inferenceEntityId));
261263
return;
262264
}
263265

@@ -289,15 +291,16 @@ public void getModelWithSecrets(String inferenceEntityId, ActionListener<Unparse
289291
* @param listener Model listener
290292
*/
291293
public void getModel(String inferenceEntityId, ActionListener<UnparsedModel> listener) {
294+
var maybeDefault = defaultConfigIds.get(inferenceEntityId);
295+
if (maybeDefault != null) {
296+
getDefaultConfig(true, maybeDefault, listener);
297+
return;
298+
}
299+
292300
ActionListener<SearchResponse> searchListener = ActionListener.wrap((searchResponse) -> {
293301
// There should be a hit for the configurations
294302
if (searchResponse.getHits().getHits().length == 0) {
295-
var maybeDefault = defaultConfigIds.get(inferenceEntityId);
296-
if (maybeDefault != null) {
297-
getDefaultConfig(true, maybeDefault, listener);
298-
} else {
299-
listener.onFailure(inferenceNotFoundException(inferenceEntityId));
300-
}
303+
listener.onFailure(inferenceNotFoundException(inferenceEntityId));
301304
return;
302305
}
303306

0 commit comments

Comments
 (0)