diff --git a/output/openapi/elasticsearch-openapi.json b/output/openapi/elasticsearch-openapi.json index 2b4a0f12cd..4ceace402b 100644 --- a/output/openapi/elasticsearch-openapi.json +++ b/output/openapi/elasticsearch-openapi.json @@ -78096,32 +78096,25 @@ ] }, "inference._types:InferenceChunkingSettings": { - "allOf": [ - { - "$ref": "#/components/schemas/inference._types:InferenceEndpoint" + "type": "object", + "properties": { + "max_chunk_size": { + "description": "The maximum size of a chunk in words.\nThis value cannot be higher than `300` or lower than `20` (for `sentence` strategy) or `10` (for `word` strategy).", + "type": "number" }, - { - "type": "object", - "properties": { - "max_chunk_size": { - "description": "The maximum size of a chunk in words.\nThis value cannot be higher than `300` or lower than `20` (for `sentence` strategy) or `10` (for `word` strategy).", - "type": "number" - }, - "overlap": { - "description": "The number of overlapping words for chunks.\nIt is applicable only to a `word` chunking strategy.\nThis value cannot be higher than half the `max_chunk_size` value.", - "type": "number" - }, - "sentence_overlap": { - "description": "The number of overlapping sentences for chunks.\nIt is applicable only for a `sentence` chunking strategy.\nIt can be either `1` or `0`.", - "type": "number" - }, - "strategy": { - "description": "The chunking strategy: `sentence` or `word`.", - "type": "string" - } - } + "overlap": { + "description": "The number of overlapping words for chunks.\nIt is applicable only to a `word` chunking strategy.\nThis value cannot be higher than half the `max_chunk_size` value.", + "type": "number" + }, + "sentence_overlap": { + "description": "The number of overlapping sentences for chunks.\nIt is applicable only for a `sentence` chunking strategy.\nIt can be either `1` or `0`.", + "type": "number" + }, + "strategy": { + "description": "The chunking strategy: `sentence` or `word`.", + "type": "string" } - ] + } }, "inference._types:ServiceSettings": { "type": "object" @@ -85272,6 +85265,9 @@ "location": { "$ref": "#/components/schemas/ml._types:TrainedModelLocation" }, + "platform_architecture": { + "type": "string" + }, "prefix_strings": { "$ref": "#/components/schemas/ml._types:TrainedModelPrefixStrings" } @@ -85309,6 +85305,9 @@ "fill_mask": { "$ref": "#/components/schemas/ml._types:FillMaskInferenceOptions" }, + "learning_to_rank": { + "$ref": "#/components/schemas/ml._types:LearningToRankConfig" + }, "ner": { "$ref": "#/components/schemas/ml._types:NerInferenceOptions" }, @@ -85348,6 +85347,9 @@ "items": { "type": "string" } + }, + "vocabulary": { + "$ref": "#/components/schemas/ml._types:Vocabulary" } } }, @@ -85366,6 +85368,9 @@ }, "roberta": { "$ref": "#/components/schemas/ml._types:NlpRobertaTokenizationConfig" + }, + "xlm_roberta": { + "$ref": "#/components/schemas/ml._types:XlmRobertaTokenizationConfig" } }, "minProperties": 1, @@ -85429,6 +85434,27 @@ } ] }, + "ml._types:XlmRobertaTokenizationConfig": { + "allOf": [ + { + "$ref": "#/components/schemas/ml._types:CommonTokenizationConfig" + }, + { + "type": "object" + } + ] + }, + "ml._types:Vocabulary": { + "type": "object", + "properties": { + "index": { + "$ref": "#/components/schemas/_types:IndexName" + } + }, + "required": [ + "index" + ] + }, "ml._types:ZeroShotClassificationInferenceOptions": { "type": "object", "properties": { @@ -85492,15 +85518,55 @@ "vocabulary" ] }, - "ml._types:Vocabulary": { + "ml._types:LearningToRankConfig": { "type": "object", "properties": { - "index": { - "$ref": "#/components/schemas/_types:IndexName" + "default_params": { + "type": "object", + "additionalProperties": { + "type": "object" + } + }, + "feature_extractors": { + "type": "array", + "items": { + "type": "object", + "additionalProperties": { + "$ref": "#/components/schemas/ml._types:FeatureExtractor" + } + } + }, + "num_top_feature_importance_values": { + "type": "number" } }, "required": [ - "index" + "num_top_feature_importance_values" + ] + }, + "ml._types:FeatureExtractor": { + "allOf": [ + { + "$ref": "#/components/schemas/ml._types:QueryFeatureExtractor" + } + ] + }, + "ml._types:QueryFeatureExtractor": { + "type": "object", + "properties": { + "default_score": { + "type": "number" + }, + "feature_name": { + "type": "string" + }, + "query": { + "$ref": "#/components/schemas/_types.query_dsl:QueryContainer" + } + }, + "required": [ + "feature_name", + "query" ] }, "ml._types:NerInferenceOptions": { diff --git a/output/openapi/elasticsearch-serverless-openapi.json b/output/openapi/elasticsearch-serverless-openapi.json index 0b185ef774..9c22fbc9ca 100644 --- a/output/openapi/elasticsearch-serverless-openapi.json +++ b/output/openapi/elasticsearch-serverless-openapi.json @@ -49416,32 +49416,25 @@ ] }, "inference._types:InferenceChunkingSettings": { - "allOf": [ - { - "$ref": "#/components/schemas/inference._types:InferenceEndpoint" + "type": "object", + "properties": { + "max_chunk_size": { + "description": "The maximum size of a chunk in words.\nThis value cannot be higher than `300` or lower than `20` (for `sentence` strategy) or `10` (for `word` strategy).", + "type": "number" }, - { - "type": "object", - "properties": { - "max_chunk_size": { - "description": "The maximum size of a chunk in words.\nThis value cannot be higher than `300` or lower than `20` (for `sentence` strategy) or `10` (for `word` strategy).", - "type": "number" - }, - "overlap": { - "description": "The number of overlapping words for chunks.\nIt is applicable only to a `word` chunking strategy.\nThis value cannot be higher than half the `max_chunk_size` value.", - "type": "number" - }, - "sentence_overlap": { - "description": "The number of overlapping sentences for chunks.\nIt is applicable only for a `sentence` chunking strategy.\nIt can be either `1` or `0`.", - "type": "number" - }, - "strategy": { - "description": "The chunking strategy: `sentence` or `word`.", - "type": "string" - } - } + "overlap": { + "description": "The number of overlapping words for chunks.\nIt is applicable only to a `word` chunking strategy.\nThis value cannot be higher than half the `max_chunk_size` value.", + "type": "number" + }, + "sentence_overlap": { + "description": "The number of overlapping sentences for chunks.\nIt is applicable only for a `sentence` chunking strategy.\nIt can be either `1` or `0`.", + "type": "number" + }, + "strategy": { + "description": "The chunking strategy: `sentence` or `word`.", + "type": "string" } - ] + } }, "inference._types:ServiceSettings": { "type": "object" @@ -55277,6 +55270,9 @@ "location": { "$ref": "#/components/schemas/ml._types:TrainedModelLocation" }, + "platform_architecture": { + "type": "string" + }, "prefix_strings": { "$ref": "#/components/schemas/ml._types:TrainedModelPrefixStrings" } @@ -55314,6 +55310,9 @@ "fill_mask": { "$ref": "#/components/schemas/ml._types:FillMaskInferenceOptions" }, + "learning_to_rank": { + "$ref": "#/components/schemas/ml._types:LearningToRankConfig" + }, "ner": { "$ref": "#/components/schemas/ml._types:NerInferenceOptions" }, @@ -55353,6 +55352,9 @@ "items": { "type": "string" } + }, + "vocabulary": { + "$ref": "#/components/schemas/ml._types:Vocabulary" } } }, @@ -55371,6 +55373,9 @@ }, "roberta": { "$ref": "#/components/schemas/ml._types:NlpRobertaTokenizationConfig" + }, + "xlm_roberta": { + "$ref": "#/components/schemas/ml._types:XlmRobertaTokenizationConfig" } }, "minProperties": 1, @@ -55434,6 +55439,27 @@ } ] }, + "ml._types:XlmRobertaTokenizationConfig": { + "allOf": [ + { + "$ref": "#/components/schemas/ml._types:CommonTokenizationConfig" + }, + { + "type": "object" + } + ] + }, + "ml._types:Vocabulary": { + "type": "object", + "properties": { + "index": { + "$ref": "#/components/schemas/_types:IndexName" + } + }, + "required": [ + "index" + ] + }, "ml._types:ZeroShotClassificationInferenceOptions": { "type": "object", "properties": { @@ -55497,15 +55523,55 @@ "vocabulary" ] }, - "ml._types:Vocabulary": { + "ml._types:LearningToRankConfig": { "type": "object", "properties": { - "index": { - "$ref": "#/components/schemas/_types:IndexName" + "default_params": { + "type": "object", + "additionalProperties": { + "type": "object" + } + }, + "feature_extractors": { + "type": "array", + "items": { + "type": "object", + "additionalProperties": { + "$ref": "#/components/schemas/ml._types:FeatureExtractor" + } + } + }, + "num_top_feature_importance_values": { + "type": "number" } }, "required": [ - "index" + "num_top_feature_importance_values" + ] + }, + "ml._types:FeatureExtractor": { + "allOf": [ + { + "$ref": "#/components/schemas/ml._types:QueryFeatureExtractor" + } + ] + }, + "ml._types:QueryFeatureExtractor": { + "type": "object", + "properties": { + "default_score": { + "type": "number" + }, + "feature_name": { + "type": "string" + }, + "query": { + "$ref": "#/components/schemas/_types.query_dsl:QueryContainer" + } + }, + "required": [ + "feature_name", + "query" ] }, "ml._types:NerInferenceOptions": { diff --git a/output/schema/schema-serverless.json b/output/schema/schema-serverless.json index 2dd4f208f3..39cfa27cd2 100644 --- a/output/schema/schema-serverless.json +++ b/output/schema/schema-serverless.json @@ -69634,7 +69634,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L82-L91" + "specLocation": "ml/_types/inference.ts#L107-L116" }, { "kind": "interface", @@ -69707,7 +69707,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L93-L108" + "specLocation": "ml/_types/inference.ts#L118-L133" }, { "inherits": { @@ -103787,7 +103787,7 @@ "name": "DeploymentAllocationState", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L330-L343" + "specLocation": "ml/_types/TrainedModel.ts#L331-L344" }, { "kind": "enum", @@ -103813,7 +103813,7 @@ "name": "DeploymentAssignmentState", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L345-L362" + "specLocation": "ml/_types/TrainedModel.ts#L346-L363" }, { "kind": "type_alias", @@ -103986,6 +103986,65 @@ }, "specLocation": "ml/_types/Detector.ts#L145-L150" }, + { + "kind": "type_alias", + "name": { + "name": "FeatureExtractor", + "namespace": "ml._types" + }, + "specLocation": "ml/_types/inference.ts#L93-L96", + "type": { + "kind": "instance_of", + "type": { + "name": "QueryFeatureExtractor", + "namespace": "ml._types" + } + } + }, + { + "kind": "interface", + "name": { + "name": "QueryFeatureExtractor", + "namespace": "ml._types" + }, + "properties": [ + { + "name": "default_score", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "float", + "namespace": "_types" + } + } + }, + { + "name": "feature_name", + "required": true, + "type": { + "kind": "instance_of", + "type": { + "name": "string", + "namespace": "_builtins" + } + } + }, + { + "name": "query", + "required": true, + "type": { + "kind": "instance_of", + "type": { + "name": "QueryContainer", + "namespace": "_types.query_dsl" + } + } + } + ], + "specLocation": "ml/_types/inference.ts#L98-L105", + "variantName": "query_extractor" + }, { "kind": "enum", "members": [ @@ -104106,7 +104165,7 @@ "name": "PredictedValue", "namespace": "ml._types" }, - "specLocation": "ml/_types/inference.ts#L445-L445", + "specLocation": "ml/_types/inference.ts#L476-L476", "type": { "items": [ { @@ -104158,7 +104217,7 @@ "name": "RoutingState", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L407-L428" + "specLocation": "ml/_types/TrainedModel.ts#L408-L429" }, { "kind": "enum", @@ -104217,7 +104276,7 @@ "name": "TokenizationTruncate", "namespace": "ml._types" }, - "specLocation": "ml/_types/inference.ts#L338-L342" + "specLocation": "ml/_types/inference.ts#L369-L373" }, { "kind": "enum", @@ -104239,7 +104298,7 @@ "name": "TrainedModelType", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L314-L328" + "specLocation": "ml/_types/TrainedModel.ts#L315-L329" }, { "kind": "enum", @@ -104255,7 +104314,7 @@ "name": "TrainingPriority", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L364-L367" + "specLocation": "ml/_types/TrainedModel.ts#L365-L368" }, { "kind": "enum", @@ -123648,12 +123707,6 @@ }, { "description": "Chunking configuration object", - "inherits": { - "type": { - "name": "InferenceEndpoint", - "namespace": "inference._types" - } - }, "kind": "interface", "name": { "name": "InferenceChunkingSettings", @@ -137349,6 +137402,17 @@ } } }, + { + "name": "platform_architecture", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "string", + "namespace": "_builtins" + } + } + }, { "name": "prefix_strings", "required": false, @@ -137361,7 +137425,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L203-L239" + "specLocation": "ml/_types/TrainedModel.ts#L203-L240" }, { "description": "Inference configuration provided when storing the model config", @@ -137449,6 +137513,17 @@ } } }, + { + "name": "learning_to_rank", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "LearningToRankConfig", + "namespace": "ml._types" + } + } + }, { "availability": { "serverless": {}, @@ -137540,7 +137615,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L23-L80", + "specLocation": "ml/_types/inference.ts#L26-L85", "variants": { "kind": "container" } @@ -137603,9 +137678,20 @@ } } } + }, + { + "name": "vocabulary", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "Vocabulary", + "namespace": "ml._types" + } + } } ], - "specLocation": "ml/_types/inference.ts#L173-L183" + "specLocation": "ml/_types/inference.ts#L202-L214" }, { "description": "Tokenization options stored in inference configuration", @@ -137674,9 +137760,20 @@ "namespace": "ml._types" } } + }, + { + "name": "xlm_roberta", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "XlmRobertaTokenizationConfig", + "namespace": "ml._types" + } + } } ], - "specLocation": "ml/_types/inference.ts#L110-L131", + "specLocation": "ml/_types/inference.ts#L135-L158", "variants": { "kind": "container" } @@ -137695,7 +137792,7 @@ "namespace": "ml._types" }, "properties": [], - "specLocation": "ml/_types/inference.ts#L161-L162" + "specLocation": "ml/_types/inference.ts#L188-L189" }, { "kind": "interface", @@ -137770,7 +137867,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L133-L159" + "specLocation": "ml/_types/inference.ts#L160-L186" }, { "description": "RoBERTa tokenization configuration options", @@ -137800,7 +137897,43 @@ } } ], - "specLocation": "ml/_types/inference.ts#L164-L171" + "specLocation": "ml/_types/inference.ts#L191-L198" + }, + { + "inherits": { + "type": { + "name": "CommonTokenizationConfig", + "namespace": "ml._types" + } + }, + "kind": "interface", + "name": { + "name": "XlmRobertaTokenizationConfig", + "namespace": "ml._types" + }, + "properties": [], + "specLocation": "ml/_types/inference.ts#L200-L200" + }, + { + "kind": "interface", + "name": { + "name": "Vocabulary", + "namespace": "ml._types" + }, + "properties": [ + { + "name": "index", + "required": true, + "type": { + "kind": "instance_of", + "type": { + "name": "IndexName", + "namespace": "_types" + } + } + } + ], + "specLocation": "ml/_types/inference.ts#L248-L250" }, { "description": "Zero shot classification configuration options", @@ -137891,7 +138024,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L185-L206" + "specLocation": "ml/_types/inference.ts#L216-L237" }, { "description": "Fill mask inference options", @@ -137961,28 +138094,71 @@ } } ], - "specLocation": "ml/_types/inference.ts#L253-L268" + "specLocation": "ml/_types/inference.ts#L284-L299" }, { "kind": "interface", "name": { - "name": "Vocabulary", + "name": "LearningToRankConfig", "namespace": "ml._types" }, "properties": [ { - "name": "index", + "name": "default_params", + "required": false, + "type": { + "key": { + "kind": "instance_of", + "type": { + "name": "string", + "namespace": "_builtins" + } + }, + "kind": "dictionary_of", + "singleKey": false, + "value": { + "kind": "user_defined_value" + } + } + }, + { + "name": "feature_extractors", + "required": false, + "type": { + "kind": "array_of", + "value": { + "key": { + "kind": "instance_of", + "type": { + "name": "string", + "namespace": "_builtins" + } + }, + "kind": "dictionary_of", + "singleKey": false, + "value": { + "kind": "instance_of", + "type": { + "name": "FeatureExtractor", + "namespace": "ml._types" + } + } + } + } + }, + { + "name": "num_top_feature_importance_values", "required": true, "type": { "kind": "instance_of", "type": { - "name": "IndexName", + "name": "integer", "namespace": "_types" } } } ], - "specLocation": "ml/_types/inference.ts#L217-L219" + "specLocation": "ml/_types/inference.ts#L87-L91" }, { "description": "Named entity recognition options", @@ -138043,7 +138219,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L242-L251" + "specLocation": "ml/_types/inference.ts#L273-L282" }, { "description": "Pass through configuration options", @@ -138089,7 +138265,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L208-L215" + "specLocation": "ml/_types/inference.ts#L239-L246" }, { "description": "Text embedding inference options", @@ -138147,7 +138323,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L221-L231" + "specLocation": "ml/_types/inference.ts#L252-L262" }, { "description": "Text expansion inference options", @@ -138193,7 +138369,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L233-L240" + "specLocation": "ml/_types/inference.ts#L264-L271" }, { "description": "Question answering inference options", @@ -138252,7 +138428,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L270-L280" + "specLocation": "ml/_types/inference.ts#L301-L311" }, { "kind": "interface", @@ -138277,7 +138453,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L241-L244" + "specLocation": "ml/_types/TrainedModel.ts#L242-L245" }, { "kind": "interface", @@ -138354,7 +138530,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L246-L254" + "specLocation": "ml/_types/TrainedModel.ts#L247-L255" }, { "kind": "interface", @@ -138426,7 +138602,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L273-L287" + "specLocation": "ml/_types/TrainedModel.ts#L274-L288" }, { "kind": "interface", @@ -138478,7 +138654,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L289-L296" + "specLocation": "ml/_types/TrainedModel.ts#L290-L297" }, { "kind": "interface", @@ -138524,7 +138700,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L305-L312" + "specLocation": "ml/_types/TrainedModel.ts#L306-L313" }, { "kind": "interface", @@ -138561,7 +138737,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L298-L303" + "specLocation": "ml/_types/TrainedModel.ts#L299-L304" }, { "kind": "interface", @@ -138744,7 +138920,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L256-L271" + "specLocation": "ml/_types/TrainedModel.ts#L257-L272" }, { "kind": "interface", @@ -138778,7 +138954,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L486-L495" + "specLocation": "ml/_types/TrainedModel.ts#L487-L496" }, { "kind": "interface", @@ -138799,7 +138975,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L478-L480" + "specLocation": "ml/_types/TrainedModel.ts#L479-L481" }, { "kind": "interface", @@ -138820,7 +138996,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L482-L484" + "specLocation": "ml/_types/TrainedModel.ts#L483-L485" }, { "kind": "interface", @@ -139237,7 +139413,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L450-L457" + "specLocation": "ml/_types/TrainedModel.ts#L451-L458" }, { "kind": "interface", @@ -139549,7 +139725,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L430-L448" + "specLocation": "ml/_types/TrainedModel.ts#L431-L449" }, { "kind": "interface", @@ -139794,7 +139970,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L284-L306", + "specLocation": "ml/_types/inference.ts#L315-L337", "variants": { "kind": "container" } @@ -139858,7 +140034,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L351-L360" + "specLocation": "ml/_types/inference.ts#L382-L391" }, { "kind": "interface", @@ -139892,7 +140068,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L344-L349" + "specLocation": "ml/_types/inference.ts#L375-L380" }, { "kind": "interface", @@ -139953,7 +140129,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L362-L371" + "specLocation": "ml/_types/inference.ts#L393-L402" }, { "kind": "interface", @@ -139999,7 +140175,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L399-L406" + "specLocation": "ml/_types/inference.ts#L430-L437" }, { "kind": "interface", @@ -140033,7 +140209,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L392-L397" + "specLocation": "ml/_types/inference.ts#L423-L428" }, { "kind": "interface", @@ -140067,7 +140243,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L373-L378" + "specLocation": "ml/_types/inference.ts#L404-L409" }, { "kind": "interface", @@ -140100,7 +140276,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L380-L384" + "specLocation": "ml/_types/inference.ts#L411-L415" }, { "kind": "interface", @@ -140133,7 +140309,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L386-L390" + "specLocation": "ml/_types/inference.ts#L417-L421" }, { "kind": "interface", @@ -140203,7 +140379,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L408-L419" + "specLocation": "ml/_types/inference.ts#L439-L450" }, { "kind": "interface", @@ -140345,7 +140521,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L447-L495" + "specLocation": "ml/_types/inference.ts#L478-L526" }, { "kind": "interface", @@ -140410,7 +140586,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L421-L427" + "specLocation": "ml/_types/inference.ts#L452-L458" }, { "kind": "interface", @@ -140453,7 +140629,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L428-L432" + "specLocation": "ml/_types/inference.ts#L459-L463" }, { "kind": "interface", @@ -140499,7 +140675,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L439-L443" + "specLocation": "ml/_types/inference.ts#L470-L474" }, { "kind": "interface", @@ -140531,7 +140707,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L434-L437" + "specLocation": "ml/_types/inference.ts#L465-L468" }, { "kind": "interface", @@ -142116,7 +142292,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L459-L476" + "specLocation": "ml/_types/TrainedModel.ts#L460-L477" }, { "kind": "interface", @@ -142249,7 +142425,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L369-L405" + "specLocation": "ml/_types/TrainedModel.ts#L370-L406" }, { "kind": "interface", diff --git a/output/schema/schema.json b/output/schema/schema.json index 46ffa8a94e..66c141368f 100644 --- a/output/schema/schema.json +++ b/output/schema/schema.json @@ -149560,12 +149560,6 @@ { "kind": "interface", "description": "Chunking configuration object", - "inherits": { - "type": { - "name": "InferenceEndpoint", - "namespace": "inference._types" - } - }, "name": { "name": "InferenceChunkingSettings", "namespace": "inference._types" @@ -166671,7 +166665,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L93-L108" + "specLocation": "ml/_types/inference.ts#L118-L133" }, { "kind": "interface", @@ -166746,7 +166740,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L133-L159" + "specLocation": "ml/_types/inference.ts#L160-L186" }, { "kind": "enum", @@ -170090,7 +170084,7 @@ "name": "DeploymentAllocationState", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L330-L343" + "specLocation": "ml/_types/TrainedModel.ts#L331-L344" }, { "kind": "enum", @@ -170116,7 +170110,7 @@ "name": "DeploymentAssignmentState", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L345-L362" + "specLocation": "ml/_types/TrainedModel.ts#L346-L363" }, { "kind": "interface", @@ -170841,6 +170835,21 @@ ], "specLocation": "ml/_types/Datafeed.ts#L204-L208" }, + { + "kind": "type_alias", + "name": { + "name": "FeatureExtractor", + "namespace": "ml._types" + }, + "specLocation": "ml/_types/inference.ts#L93-L96", + "type": { + "kind": "instance_of", + "type": { + "name": "QueryFeatureExtractor", + "namespace": "ml._types" + } + } + }, { "kind": "interface", "description": "Fill mask inference options", @@ -170909,7 +170918,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L253-L268" + "specLocation": "ml/_types/inference.ts#L284-L299" }, { "kind": "interface", @@ -170955,7 +170964,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L399-L406" + "specLocation": "ml/_types/inference.ts#L430-L437" }, { "kind": "interface", @@ -171161,7 +171170,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L273-L287" + "specLocation": "ml/_types/TrainedModel.ts#L274-L288" }, { "kind": "interface", @@ -171457,6 +171466,17 @@ } } }, + { + "name": "learning_to_rank", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "LearningToRankConfig", + "namespace": "ml._types" + } + } + }, { "availability": { "serverless": {}, @@ -171548,7 +171568,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L23-L80", + "specLocation": "ml/_types/inference.ts#L26-L85", "variants": { "kind": "container" } @@ -171681,7 +171701,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L284-L306", + "specLocation": "ml/_types/inference.ts#L315-L337", "variants": { "kind": "container" } @@ -171826,7 +171846,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L447-L495" + "specLocation": "ml/_types/inference.ts#L478-L526" }, { "kind": "interface", @@ -173040,6 +173060,70 @@ ], "specLocation": "ml/_types/Job.ts#L332-L341" }, + { + "kind": "interface", + "name": { + "name": "LearningToRankConfig", + "namespace": "ml._types" + }, + "properties": [ + { + "name": "default_params", + "required": false, + "type": { + "kind": "dictionary_of", + "key": { + "kind": "instance_of", + "type": { + "name": "string", + "namespace": "_builtins" + } + }, + "singleKey": false, + "value": { + "kind": "user_defined_value" + } + } + }, + { + "name": "feature_extractors", + "required": false, + "type": { + "kind": "array_of", + "value": { + "kind": "dictionary_of", + "key": { + "kind": "instance_of", + "type": { + "name": "string", + "namespace": "_builtins" + } + }, + "singleKey": false, + "value": { + "kind": "instance_of", + "type": { + "name": "FeatureExtractor", + "namespace": "ml._types" + } + } + } + } + }, + { + "name": "num_top_feature_importance_values", + "required": true, + "type": { + "kind": "instance_of", + "type": { + "name": "integer", + "namespace": "_types" + } + } + } + ], + "specLocation": "ml/_types/inference.ts#L87-L91" + }, { "kind": "enum", "members": [ @@ -173240,7 +173324,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L256-L271" + "specLocation": "ml/_types/TrainedModel.ts#L257-L272" }, { "kind": "interface", @@ -173811,7 +173895,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L242-L251" + "specLocation": "ml/_types/inference.ts#L273-L282" }, { "kind": "interface", @@ -173845,7 +173929,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L392-L397" + "specLocation": "ml/_types/inference.ts#L423-L428" }, { "kind": "interface", @@ -173861,7 +173945,7 @@ "namespace": "ml._types" }, "properties": [], - "specLocation": "ml/_types/inference.ts#L161-L162" + "specLocation": "ml/_types/inference.ts#L188-L189" }, { "kind": "interface", @@ -173891,7 +173975,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L164-L171" + "specLocation": "ml/_types/inference.ts#L191-L198" }, { "kind": "interface", @@ -173925,7 +174009,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L344-L349" + "specLocation": "ml/_types/inference.ts#L375-L380" }, { "kind": "interface", @@ -174239,7 +174323,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L208-L215" + "specLocation": "ml/_types/inference.ts#L239-L246" }, { "kind": "interface", @@ -174273,7 +174357,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L373-L378" + "specLocation": "ml/_types/inference.ts#L404-L409" }, { "kind": "interface", @@ -174315,7 +174399,7 @@ "name": "PredictedValue", "namespace": "ml._types" }, - "specLocation": "ml/_types/inference.ts#L445-L445", + "specLocation": "ml/_types/inference.ts#L476-L476", "type": { "kind": "union_of", "items": [ @@ -174339,6 +174423,50 @@ ] } }, + { + "kind": "interface", + "name": { + "name": "QueryFeatureExtractor", + "namespace": "ml._types" + }, + "properties": [ + { + "name": "default_score", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "float", + "namespace": "_types" + } + } + }, + { + "name": "feature_name", + "required": true, + "type": { + "kind": "instance_of", + "type": { + "name": "string", + "namespace": "_builtins" + } + } + }, + { + "name": "query", + "required": true, + "type": { + "kind": "instance_of", + "type": { + "name": "QueryContainer", + "namespace": "_types.query_dsl" + } + } + } + ], + "specLocation": "ml/_types/inference.ts#L98-L105", + "variantName": "query_extractor" + }, { "kind": "interface", "description": "Question answering inference options", @@ -174396,7 +174524,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L270-L280" + "specLocation": "ml/_types/inference.ts#L301-L311" }, { "kind": "interface", @@ -174466,7 +174594,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L408-L419" + "specLocation": "ml/_types/inference.ts#L439-L450" }, { "kind": "interface", @@ -174503,7 +174631,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L82-L91" + "specLocation": "ml/_types/inference.ts#L107-L116" }, { "kind": "enum", @@ -174533,7 +174661,7 @@ "name": "RoutingState", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L407-L428" + "specLocation": "ml/_types/TrainedModel.ts#L408-L429" }, { "kind": "enum", @@ -174755,9 +174883,20 @@ } } } + }, + { + "name": "vocabulary", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "Vocabulary", + "namespace": "ml._types" + } + } } ], - "specLocation": "ml/_types/inference.ts#L173-L183" + "specLocation": "ml/_types/inference.ts#L202-L214" }, { "kind": "interface", @@ -174818,7 +174957,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L351-L360" + "specLocation": "ml/_types/inference.ts#L382-L391" }, { "kind": "interface", @@ -174876,7 +175015,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L221-L231" + "specLocation": "ml/_types/inference.ts#L252-L262" }, { "kind": "interface", @@ -174909,7 +175048,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L380-L384" + "specLocation": "ml/_types/inference.ts#L411-L415" }, { "kind": "interface", @@ -174955,7 +175094,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L233-L240" + "specLocation": "ml/_types/inference.ts#L264-L271" }, { "kind": "interface", @@ -174988,7 +175127,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L386-L390" + "specLocation": "ml/_types/inference.ts#L417-L421" }, { "kind": "interface", @@ -175109,9 +175248,20 @@ "namespace": "ml._types" } } + }, + { + "name": "xlm_roberta", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "XlmRobertaTokenizationConfig", + "namespace": "ml._types" + } + } } ], - "specLocation": "ml/_types/inference.ts#L110-L131", + "specLocation": "ml/_types/inference.ts#L135-L158", "variants": { "kind": "container" } @@ -175133,7 +175283,7 @@ "name": "TokenizationTruncate", "namespace": "ml._types" }, - "specLocation": "ml/_types/inference.ts#L338-L342" + "specLocation": "ml/_types/inference.ts#L369-L373" }, { "kind": "interface", @@ -175176,7 +175326,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L428-L432" + "specLocation": "ml/_types/inference.ts#L459-L463" }, { "kind": "interface", @@ -175228,7 +175378,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L289-L296" + "specLocation": "ml/_types/TrainedModel.ts#L290-L297" }, { "kind": "interface", @@ -175265,7 +175415,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L298-L303" + "specLocation": "ml/_types/TrainedModel.ts#L299-L304" }, { "kind": "interface", @@ -175311,7 +175461,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L305-L312" + "specLocation": "ml/_types/TrainedModel.ts#L306-L313" }, { "kind": "interface", @@ -175424,7 +175574,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L459-L476" + "specLocation": "ml/_types/TrainedModel.ts#L460-L477" }, { "kind": "interface", @@ -175482,7 +175632,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L430-L448" + "specLocation": "ml/_types/TrainedModel.ts#L431-L449" }, { "kind": "interface", @@ -175615,7 +175765,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L369-L405" + "specLocation": "ml/_types/TrainedModel.ts#L370-L406" }, { "kind": "interface", @@ -175862,6 +176012,17 @@ } } }, + { + "name": "platform_architecture", + "required": false, + "type": { + "kind": "instance_of", + "type": { + "name": "string", + "namespace": "_builtins" + } + } + }, { "name": "prefix_strings", "required": false, @@ -175874,7 +176035,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L203-L239" + "specLocation": "ml/_types/TrainedModel.ts#L203-L240" }, { "kind": "interface", @@ -175899,7 +176060,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L241-L244" + "specLocation": "ml/_types/TrainedModel.ts#L242-L245" }, { "kind": "interface", @@ -175976,7 +176137,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L246-L254" + "specLocation": "ml/_types/TrainedModel.ts#L247-L255" }, { "kind": "interface", @@ -176022,7 +176183,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L450-L457" + "specLocation": "ml/_types/TrainedModel.ts#L451-L458" }, { "kind": "interface", @@ -176590,7 +176751,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L421-L427" + "specLocation": "ml/_types/inference.ts#L452-L458" }, { "kind": "interface", @@ -176622,7 +176783,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L434-L437" + "specLocation": "ml/_types/inference.ts#L465-L468" }, { "kind": "interface", @@ -176668,7 +176829,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L439-L443" + "specLocation": "ml/_types/inference.ts#L470-L474" }, { "kind": "interface", @@ -176770,7 +176931,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L478-L480" + "specLocation": "ml/_types/TrainedModel.ts#L479-L481" }, { "kind": "interface", @@ -176791,7 +176952,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L482-L484" + "specLocation": "ml/_types/TrainedModel.ts#L483-L485" }, { "kind": "interface", @@ -176825,7 +176986,7 @@ } } ], - "specLocation": "ml/_types/TrainedModel.ts#L486-L495" + "specLocation": "ml/_types/TrainedModel.ts#L487-L496" }, { "kind": "interface", @@ -176972,7 +177133,7 @@ "name": "TrainedModelType", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L314-L328" + "specLocation": "ml/_types/TrainedModel.ts#L315-L329" }, { "kind": "enum", @@ -176988,7 +177149,7 @@ "name": "TrainingPriority", "namespace": "ml._types" }, - "specLocation": "ml/_types/TrainedModel.ts#L364-L367" + "specLocation": "ml/_types/TrainedModel.ts#L365-L368" }, { "kind": "interface", @@ -177095,7 +177256,22 @@ } } ], - "specLocation": "ml/_types/inference.ts#L217-L219" + "specLocation": "ml/_types/inference.ts#L248-L250" + }, + { + "kind": "interface", + "inherits": { + "type": { + "name": "CommonTokenizationConfig", + "namespace": "ml._types" + } + }, + "name": { + "name": "XlmRobertaTokenizationConfig", + "namespace": "ml._types" + }, + "properties": [], + "specLocation": "ml/_types/inference.ts#L200-L200" }, { "kind": "interface", @@ -177186,7 +177362,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L185-L206" + "specLocation": "ml/_types/inference.ts#L216-L237" }, { "kind": "interface", @@ -177247,7 +177423,7 @@ } } ], - "specLocation": "ml/_types/inference.ts#L362-L371" + "specLocation": "ml/_types/inference.ts#L393-L402" }, { "kind": "request", diff --git a/output/typescript/types.ts b/output/typescript/types.ts index bd1d8e0370..974928bfba 100644 --- a/output/typescript/types.ts +++ b/output/typescript/types.ts @@ -13080,7 +13080,7 @@ export type InferenceDenseByteVector = byte[] export type InferenceDenseVector = float[] -export interface InferenceInferenceChunkingSettings extends InferenceInferenceEndpoint { +export interface InferenceInferenceChunkingSettings { max_chunk_size?: integer overlap?: integer sentence_overlap?: integer @@ -15306,6 +15306,8 @@ export interface MlExponentialAverageCalculationContext { previous_exponential_average_ms?: DurationValue } +export type MlFeatureExtractor = MlQueryFeatureExtractor + export interface MlFillMaskInferenceOptions { mask_token?: string num_top_classes?: integer @@ -15371,6 +15373,7 @@ export interface MlInferenceConfigCreateContainer { text_classification?: MlTextClassificationInferenceOptions zero_shot_classification?: MlZeroShotClassificationInferenceOptions fill_mask?: MlFillMaskInferenceOptions + learning_to_rank?: MlLearningToRankConfig ner?: MlNerInferenceOptions pass_through?: MlPassThroughInferenceOptions text_embedding?: MlTextEmbeddingInferenceOptions @@ -15517,6 +15520,12 @@ export interface MlJobTimingStats { minimum_bucket_processing_time_ms?: DurationValue } +export interface MlLearningToRankConfig { + default_params?: Record + feature_extractors?: Record[] + num_top_feature_importance_values: integer +} + export type MlMemoryStatus = 'ok' | 'soft_limit' | 'hard_limit' export interface MlModelPackageConfig { @@ -15659,6 +15668,12 @@ export interface MlPerPartitionCategorization { export type MlPredictedValue = ScalarValue | ScalarValue[] +export interface MlQueryFeatureExtractor { + default_score?: float + feature_name: string + query: QueryDslQueryContainer +} + export interface MlQuestionAnsweringInferenceOptions { num_top_classes?: integer tokenization?: MlTokenizationConfigContainer @@ -15703,6 +15718,7 @@ export interface MlTextClassificationInferenceOptions { tokenization?: MlTokenizationConfigContainer results_field?: string classification_labels?: string[] + vocabulary?: MlVocabulary } export interface MlTextClassificationInferenceUpdateOptions { @@ -15745,6 +15761,7 @@ export interface MlTokenizationConfigContainer { bert_ja?: MlNlpBertTokenizationConfig mpnet?: MlNlpBertTokenizationConfig roberta?: MlNlpRobertaTokenizationConfig + xlm_roberta?: MlXlmRobertaTokenizationConfig } export type MlTokenizationTruncate = 'first' | 'second' | 'none' @@ -15822,6 +15839,7 @@ export interface MlTrainedModelConfig { model_size_bytes?: ByteSize model_package?: MlModelPackageConfig location?: MlTrainedModelLocation + platform_architecture?: string prefix_strings?: MlTrainedModelPrefixStrings } @@ -15957,6 +15975,9 @@ export interface MlVocabulary { index: IndexName } +export interface MlXlmRobertaTokenizationConfig extends MlCommonTokenizationConfig { +} + export interface MlZeroShotClassificationInferenceOptions { tokenization?: MlTokenizationConfigContainer hypothesis_template?: string diff --git a/specification/inference/_types/Services.ts b/specification/inference/_types/Services.ts index a5c72d352f..7c3ea7fc9d 100644 --- a/specification/inference/_types/Services.ts +++ b/specification/inference/_types/Services.ts @@ -60,7 +60,7 @@ export class InferenceEndpointInfo extends InferenceEndpoint { /** * Chunking configuration object */ -export class InferenceChunkingSettings extends InferenceEndpoint { +export class InferenceChunkingSettings { /** * The maximum size of a chunk in words. * This value cannot be higher than `300` or lower than `20` (for `sentence` strategy) or `10` (for `word` strategy). diff --git a/specification/ml/_types/TrainedModel.ts b/specification/ml/_types/TrainedModel.ts index 9374dd09b0..acc7fe5c4f 100644 --- a/specification/ml/_types/TrainedModel.ts +++ b/specification/ml/_types/TrainedModel.ts @@ -235,6 +235,7 @@ export class TrainedModelConfig { model_size_bytes?: ByteSize model_package?: ModelPackageConfig location?: TrainedModelLocation + platform_architecture?: string prefix_strings?: TrainedModelPrefixStrings } diff --git a/specification/ml/_types/inference.ts b/specification/ml/_types/inference.ts index 856120e27c..1cc539dc27 100644 --- a/specification/ml/_types/inference.ts +++ b/specification/ml/_types/inference.ts @@ -17,8 +17,11 @@ * under the License. */ +import { Dictionary } from '@spec_utils/Dictionary' +import { UserDefinedValue } from '@spec_utils/UserDefinedValue' import { Field, IndexName, ScalarValue } from '@_types/common' -import { double, integer } from '@_types/Numeric' +import { double, float, integer } from '@_types/Numeric' +import { QueryContainer } from '@_types/query_dsl/abstractions' /** * Inference configuration provided when storing the model config @@ -47,6 +50,8 @@ export class InferenceConfigCreateContainer { * @availability serverless */ fill_mask?: FillMaskInferenceOptions + + learning_to_rank?: LearningToRankConfig /** * Named entity recognition configuration for inference. * @availability stack since=8.0.0 @@ -79,6 +84,26 @@ export class InferenceConfigCreateContainer { question_answering?: QuestionAnsweringInferenceOptions } +export class LearningToRankConfig { + default_params?: Dictionary + feature_extractors?: Dictionary[] + num_top_feature_importance_values: integer +} + +/** + * @variants external + */ +export type FeatureExtractor = QueryFeatureExtractor + +/** + * @variant name=query_extractor + */ +export class QueryFeatureExtractor { + default_score?: float + feature_name: string + query: QueryContainer +} + export class RegressionInferenceOptions { /** The field that is added to incoming documents to contain the inference prediction. Defaults to predicted_value. */ results_field?: Field @@ -128,6 +153,8 @@ export class TokenizationConfigContainer { * @availability serverless * */ roberta?: NlpRobertaTokenizationConfig + + xlm_roberta?: XlmRobertaTokenizationConfig } export class CommonTokenizationConfig { @@ -170,6 +197,8 @@ export class NlpRobertaTokenizationConfig extends CommonTokenizationConfig { add_prefix_space?: boolean } +export class XlmRobertaTokenizationConfig extends CommonTokenizationConfig {} + /** Text classification configuration options */ export class TextClassificationInferenceOptions { /** Specifies the number of top class predictions to return. Defaults to 0. */ @@ -180,6 +209,8 @@ export class TextClassificationInferenceOptions { results_field?: string /** Classification labels to apply other than the stored labels. Must have the same deminsions as the default configured labels */ classification_labels?: string[] + + vocabulary?: Vocabulary } /** Zero shot classification configuration options */