Package org.apache.flink.model.openai

Class OpenAIEmbeddingModelFunction

java.lang.Object

org.apache.flink.table.functions.UserDefinedFunction

org.apache.flink.table.functions.AsyncTableFunction<org.apache.flink.table.data.RowData>

org.apache.flink.table.functions.AsyncPredictFunction

org.apache.flink.model.openai.AbstractOpenAIModelFunction

org.apache.flink.model.openai.OpenAIEmbeddingModelFunction

All Implemented Interfaces:

Serializable, org.apache.flink.table.functions.FunctionDefinition

public class OpenAIEmbeddingModelFunction
extends AbstractOpenAIModelFunction

AsyncPredictFunction for OpenAI embedding task.

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.flink.model.openai.AbstractOpenAIModelFunction

AbstractOpenAIModelFunction.ErrorHandlingStrategy, AbstractOpenAIModelFunction.ErrorMessageMetadata, AbstractOpenAIModelFunction.RetryFallbackStrategy

Field Summary

Fields

Modifier and Type	Field	Description
static String	ENDPOINT_SUFFIX

Fields inherited from class org.apache.flink.model.openai.AbstractOpenAIModelFunction

client, outputColumnNames

Constructor Summary

Constructors

Constructor	Description
OpenAIEmbeddingModelFunction(org.apache.flink.table.factories.ModelProviderFactory.Context factoryContext, org.apache.flink.configuration.ReadableConfig config)

Method Summary

Modifier and Type	Method	Description
CompletableFuture<Collection<org.apache.flink.table.data.RowData>>	asyncPredictInternal(String input)
protected String	getEndpointSuffix()

Methods inherited from class org.apache.flink.model.openai.AbstractOpenAIModelFunction

asyncPredict, close, handleErrorsAndRespond, open, validateSingleColumnSchema

Methods inherited from class org.apache.flink.table.functions.AsyncPredictFunction

eval

Methods inherited from class org.apache.flink.table.functions.AsyncTableFunction

getKind, getTypeInference

Methods inherited from class org.apache.flink.table.functions.UserDefinedFunction

functionIdentifier, toString

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.flink.table.functions.FunctionDefinition

getRequirements, isDeterministic, supportsConstantFolding

Field Detail

ENDPOINT_SUFFIX

public static final String ENDPOINT_SUFFIX

See Also:

Constant Field Values

Constructor Detail

OpenAIEmbeddingModelFunction

public OpenAIEmbeddingModelFunction(org.apache.flink.table.factories.ModelProviderFactory.Context factoryContext,
                                    org.apache.flink.configuration.ReadableConfig config)

Method Detail

getEndpointSuffix

protected String getEndpointSuffix()

Specified by:

getEndpointSuffix in class AbstractOpenAIModelFunction

asyncPredictInternal

public CompletableFuture<Collection<org.apache.flink.table.data.RowData>> asyncPredictInternal(String input)

Specified by:

asyncPredictInternal in class AbstractOpenAIModelFunction