dev.langchain4j.model.openai

Class OpenAiEmbeddingModel

java.lang.Object

dev.langchain4j.model.openai.OpenAiEmbeddingModel

All Implemented Interfaces:

EmbeddingModel, TokenCountEstimator

public class OpenAiEmbeddingModel
extends Object
implements EmbeddingModel, TokenCountEstimator

Represents an OpenAI embedding model, such as text-embedding-ada-002.

Constructor Summary

Constructors

Constructor and Description
OpenAiEmbeddingModel(String baseUrl, String apiKey, String organizationId, String modelName, String user, Duration timeout, Integer maxRetries, Proxy proxy, Boolean logRequests, Boolean logResponses, Tokenizer tokenizer)

Method Summary

Modifier and Type	Method and Description
Response<List<Embedding>>	embedAll(List<TextSegment> textSegments)
int	estimateTokenCount(String text)
static OpenAiEmbeddingModel	withApiKey(String apiKey)

Methods inherited from class java.lang.Object

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

Methods inherited from interface dev.langchain4j.model.embedding.EmbeddingModel

embed, embed

Methods inherited from interface dev.langchain4j.model.embedding.TokenCountEstimator

estimateTokenCount, estimateTokenCount

Constructor Detail

OpenAiEmbeddingModel

public OpenAiEmbeddingModel(String baseUrl,
                            String apiKey,
                            String organizationId,
                            String modelName,
                            String user,
                            Duration timeout,
                            Integer maxRetries,
                            Proxy proxy,
                            Boolean logRequests,
                            Boolean logResponses,
                            Tokenizer tokenizer)

Method Detail

embedAll

public Response<List<Embedding>> embedAll(List<TextSegment> textSegments)

Specified by:

embedAll in interface EmbeddingModel

estimateTokenCount

public int estimateTokenCount(String text)

Specified by:

estimateTokenCount in interface TokenCountEstimator

withApiKey

public static OpenAiEmbeddingModel withApiKey(String apiKey)