Skip to content

Commit

Permalink
Remove extraction of unused fields from IbmWatsonxRerankServiceSettings
Browse files Browse the repository at this point in the history
  • Loading branch information
saikatsarkar056 committed Jan 30, 2025
1 parent 3924962 commit 1be2bf3
Showing 1 changed file with 0 additions and 9 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -27,15 +27,11 @@
import java.util.Map;
import java.util.Objects;

import static org.elasticsearch.xpack.inference.services.ServiceFields.DIMENSIONS;
import static org.elasticsearch.xpack.inference.services.ServiceFields.MAX_INPUT_TOKENS;
import static org.elasticsearch.xpack.inference.services.ServiceFields.MODEL_ID;
import static org.elasticsearch.xpack.inference.services.ServiceFields.URL;
import static org.elasticsearch.xpack.inference.services.ServiceUtils.convertToUri;
import static org.elasticsearch.xpack.inference.services.ServiceUtils.createUri;
import static org.elasticsearch.xpack.inference.services.ServiceUtils.extractRequiredString;
import static org.elasticsearch.xpack.inference.services.ServiceUtils.extractSimilarity;
import static org.elasticsearch.xpack.inference.services.ServiceUtils.removeAsType;
import static org.elasticsearch.xpack.inference.services.ibmwatsonx.IbmWatsonxServiceFields.API_VERSION;
import static org.elasticsearch.xpack.inference.services.ibmwatsonx.IbmWatsonxServiceFields.PROJECT_ID;

Expand All @@ -56,11 +52,6 @@ public static IbmWatsonxRerankServiceSettings fromMap(Map<String, Object> map, C
URI uri = convertToUri(url, URL, ModelConfigurations.SERVICE_SETTINGS, validationException);
String apiVersion = extractRequiredString(map, API_VERSION, ModelConfigurations.SERVICE_SETTINGS, validationException);

// We need to extract/remove those fields to avoid unknown service settings errors
extractSimilarity(map, ModelConfigurations.SERVICE_SETTINGS, validationException);
removeAsType(map, DIMENSIONS, Integer.class);
removeAsType(map, MAX_INPUT_TOKENS, Integer.class);

String modelId = extractRequiredString(map, MODEL_ID, ModelConfigurations.SERVICE_SETTINGS, validationException);
String projectId = extractRequiredString(map, PROJECT_ID, ModelConfigurations.SERVICE_SETTINGS, validationException);

Expand Down

0 comments on commit 1be2bf3

Please sign in to comment.