Class AzureOpenAiTokenCountEstimator

java.lang.Object

dev.langchain4j.model.azure.AzureOpenAiTokenCountEstimator

All Implemented Interfaces:

dev.langchain4j.model.TokenCountEstimator

public class AzureOpenAiTokenCountEstimator
extends Object
implements dev.langchain4j.model.TokenCountEstimator

This class can be used to estimate the cost (in tokens) before calling OpenAI or when using streaming. Magic numbers present in this class were found empirically while testing. There are integration tests in place that are making sure that the calculations here are very close to that of OpenAI.

Constructor Summary

Constructors

Constructor	Description
AzureOpenAiTokenCountEstimator(AzureOpenAiChatModelName modelName)	Creates an instance of the AzureOpenAiTokenCountEstimator for a given AzureOpenAiChatModelName.
AzureOpenAiTokenCountEstimator(AzureOpenAiEmbeddingModelName modelName)	Creates an instance of the AzureOpenAiTokenCountEstimator for a given AzureOpenAiEmbeddingModelName.
AzureOpenAiTokenCountEstimator(AzureOpenAiLanguageModelName modelName)	Creates an instance of the AzureOpenAiTokenCountEstimator for a given AzureOpenAiLanguageModelName.
AzureOpenAiTokenCountEstimator(String modelName)	Creates an instance of the AzureOpenAiTokenCountEstimator for a given model name.

Method Summary

Modifier and Type	Method	Description
int	estimateTokenCountInMessage(dev.langchain4j.data.message.ChatMessage message)
int	estimateTokenCountInMessages(Iterable<dev.langchain4j.data.message.ChatMessage> messages)
int	estimateTokenCountInText(String text)

Methods inherited from class Object

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

Constructor Details

AzureOpenAiTokenCountEstimator

public AzureOpenAiTokenCountEstimator(AzureOpenAiChatModelName modelName)

Creates an instance of the AzureOpenAiTokenCountEstimator for a given AzureOpenAiChatModelName.

AzureOpenAiTokenCountEstimator

public AzureOpenAiTokenCountEstimator(AzureOpenAiEmbeddingModelName modelName)

Creates an instance of the AzureOpenAiTokenCountEstimator for a given AzureOpenAiEmbeddingModelName.

AzureOpenAiTokenCountEstimator

public AzureOpenAiTokenCountEstimator(AzureOpenAiLanguageModelName modelName)

Creates an instance of the AzureOpenAiTokenCountEstimator for a given AzureOpenAiLanguageModelName.

AzureOpenAiTokenCountEstimator

public AzureOpenAiTokenCountEstimator(String modelName)

Creates an instance of the AzureOpenAiTokenCountEstimator for a given model name.

Method Details

estimateTokenCountInText

public int estimateTokenCountInText(String text)

Specified by:

estimateTokenCountInText in interface dev.langchain4j.model.TokenCountEstimator

estimateTokenCountInMessage

public int estimateTokenCountInMessage(dev.langchain4j.data.message.ChatMessage message)

Specified by:

estimateTokenCountInMessage in interface dev.langchain4j.model.TokenCountEstimator

estimateTokenCountInMessages

public int estimateTokenCountInMessages(Iterable<dev.langchain4j.data.message.ChatMessage> messages)

Specified by:

estimateTokenCountInMessages in interface dev.langchain4j.model.TokenCountEstimator