Package org.elasticsearch.xpack.esql.inference.embedding

Class EmbeddingOperator

java.lang.Object

org.elasticsearch.compute.operator.AsyncOperator<InferenceOperator.OngoingInferenceResult>

org.elasticsearch.xpack.esql.inference.InferenceOperator

org.elasticsearch.xpack.esql.inference.embedding.EmbeddingOperator

All Implemented Interfaces:

Closeable, AutoCloseable, Operator, Releasable

public class EmbeddingOperator
extends InferenceOperator

EmbeddingOperator is an InferenceOperator that performs multimodal embedding inference. It evaluates a text expression for each input row, constructs typed embedding inference requests using EmbeddingAction.Request, and emits the dense vector embeddings as output.

Dispatch routes to InferenceService.executeEmbeddingInference(org.elasticsearch.xpack.core.inference.action.EmbeddingAction.Request, org.elasticsearch.action.ActionListener<org.elasticsearch.xpack.core.inference.action.InferenceAction.Response>) via an overridden dispatchInferenceRequest.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final record	EmbeddingOperator.Factory	Factory for creating EmbeddingOperator instances.

Nested classes/interfaces inherited from class org.elasticsearch.xpack.esql.inference.InferenceOperator

InferenceOperator.BulkInferenceOperation, InferenceOperator.BulkInferenceRequestItem, InferenceOperator.BulkInferenceRequestItemIterator, InferenceOperator.BulkInferenceResponseItem, InferenceOperator.OngoingInferenceResult, InferenceOperator.OutputBuilder

Nested classes/interfaces inherited from class org.elasticsearch.compute.operator.AsyncOperator

AsyncOperator.Status

Nested classes/interfaces inherited from interface org.elasticsearch.compute.operator.Operator

Operator.OperatorFactory

Field Summary

Fields inherited from class org.elasticsearch.xpack.esql.inference.InferenceOperator

DEFAULT_MAX_OUTSTANDING_PAGES, DEFAULT_MAX_OUTSTANDING_REQUESTS

Fields inherited from interface org.elasticsearch.compute.operator.Operator

MIN_TARGET_PAGE_SIZE, NOT_BLOCKED, TARGET_PAGE_SIZE

Method Summary

Modifier and Type	Method	Description
protected void	dispatchInferenceRequest(InferenceService inferenceService, BaseInferenceActionRequest request, ActionListener<InferenceAction.Response> listener)	Dispatches an inference request to the appropriate InferenceService method.
String	toString()

Methods inherited from class org.elasticsearch.xpack.esql.inference.InferenceOperator

doClose, getOutput, inferenceId, performAsync, releaseFetchedOnAnyThread

Methods inherited from class org.elasticsearch.compute.operator.AsyncOperator

addInput, canProduceMoreDataWithoutExtraInput, close, fetchFromBuffer, finish, isBlocked, isFinished, needsInput, releasePageOnAnyThread, status, status

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.elasticsearch.compute.operator.Operator

tryPromote

Method Details

dispatchInferenceRequest

protected void dispatchInferenceRequest(InferenceService inferenceService,
 BaseInferenceActionRequest request,
 ActionListener<InferenceAction.Response> listener)

Description copied from class: InferenceOperator

Dispatches an inference request to the appropriate InferenceService method. Subclasses may override this to route to a different service method (e.g. executeEmbeddingInference).

Overrides:

dispatchInferenceRequest in class InferenceOperator

toString

public String toString()

Overrides:

toString in class Object