Introduce Utils.TokenUsageCollector
for token usage tracking
#251
+108
−0
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR introduces
LangChain.Utils.TokenUsageCollector
, a utility module designed to simplify token usage tracking.Motivation:
Currently, while the
on_llm_token_usage
callback provides valuable token usage information, there isn't a readily available, best-practice approach for aggregating this data, especially in scenarios involving multiple LLM interactions within a single task. This PR aims to address this by providing a reusable and easy-to-use solution.Benefits:
on_llm_token_usage
callback.I believe this module can be a valuable addition to
LangChain
, providing a more user-friendly experience for managing token usage. I'm open to feedback and suggestions for further improvements.