3333import static org .elasticsearch .xpack .inference .services .ServiceFields .URL ;
3434import static org .elasticsearch .xpack .inference .services .ServiceUtils .convertToUri ;
3535import static org .elasticsearch .xpack .inference .services .ServiceUtils .createUri ;
36+ import static org .elasticsearch .xpack .inference .services .ServiceUtils .extractOptionalPositiveInteger ;
3637import static org .elasticsearch .xpack .inference .services .ServiceUtils .extractRequiredString ;
3738import static org .elasticsearch .xpack .inference .services .ServiceUtils .extractSimilarity ;
38- import static org .elasticsearch .xpack .inference .services .ServiceUtils .removeAsType ;
3939
4040public class HuggingFaceServiceSettings extends FilteredXContentObject implements ServiceSettings , HuggingFaceRateLimitServiceSettings {
4141 public static final String NAME = "hugging_face_service_settings" ;
@@ -49,8 +49,13 @@ public static HuggingFaceServiceSettings fromMap(Map<String, Object> map, Config
4949 var uri = extractUri (map , URL , validationException );
5050
5151 SimilarityMeasure similarityMeasure = extractSimilarity (map , ModelConfigurations .SERVICE_SETTINGS , validationException );
52- Integer dims = removeAsType (map , DIMENSIONS , Integer .class );
53- Integer maxInputTokens = removeAsType (map , MAX_INPUT_TOKENS , Integer .class );
52+ Integer dims = extractOptionalPositiveInteger (map , DIMENSIONS , ModelConfigurations .SERVICE_SETTINGS , validationException );
53+ Integer maxInputTokens = extractOptionalPositiveInteger (
54+ map ,
55+ MAX_INPUT_TOKENS ,
56+ ModelConfigurations .SERVICE_SETTINGS ,
57+ validationException
58+ );
5459 RateLimitSettings rateLimitSettings = RateLimitSettings .of (
5560 map ,
5661 DEFAULT_RATE_LIMIT_SETTINGS ,
0 commit comments