ragas icon indicating copy to clipboard operation
ragas copied to clipboard

How can I calculate total tokens consumption for this evaluation process, in Azure Open AI.

Open wanjeakshay opened this issue 1 year ago • 1 comments

[ ] I checked the documentation and related resources and couldn't find an answer to my question.

Your Question How to calculate total tokens consumption for evaluating our RAG model? What parameter should I pass in token_usage_parser for Azure Open AI?

Code Examples result_2 = evaluate( dataset1, metrics=metrics, llm=azure_model, embeddings=azure_embeddings, token_usage_parser =model_name )

Additional context Anything else you want to share with us?

wanjeakshay avatar Sep 12 '24 09:09 wanjeakshay

To calculate the total tokens consumption for the evaluation process in Azure Open AI, you can use the total_tokens() method on the result object. The parameter you should pass in token_usage_parser is get_token_usage_for_openai.

Here is the relevant code:

from ragas import evaluate
from ragas.cost import get_token_usage_for_openai

result_2 = evaluate(
    dataset1,
    metrics=metrics,
    llm=azure_model,
    embeddings=azure_embeddings,
    token_usage_parser=get_token_usage_for_openai,
)

total_tokens = result_2.total_tokens()

This will give you the total tokens consumed during the evaluation process. The get_token_usage_for_openai function is used to parse the token usage from the LLM result, ensuring accurate token counting [1][2].

To continue talking to Dosu, mention @dosu.

dosubot[bot] avatar Sep 12 '24 09:09 dosubot[bot]

Token Usage Structure:

  • TokenUsage.input_tokens - Prompt tokens
  • TokenUsage.output_tokens - Completion tokens
  • TokenUsage.model - Model name used
  • Methods: cost(), __add__() for aggregation

Reference Links:

  • https://docs.ragas.io/en/latest/howtos/customizations/customize_models/#azure- openai
  • https://github.com/explodinggradients/ragas/blob/main/src/ragas/cost.py

Closing this as resolved.

anistark avatar Oct 30 '25 12:10 anistark