scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Map total tokens to prompt_tokens too

Browse source
This commit is contained in:
Lucas Raschek 2025-03-14 18:04:43 +01:00
parent cc5c25a96c
commit 56d3e75b33
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/voyage/embedding/transformation.py
View file

@ -131,7 +131,7 @@ class VoyageEmbeddingConfig(BaseEmbeddingConfig):
model_response.object = raw_response_json.get("object")
usage = Usage(
prompt_tokens=raw_response_json.get("usage", {}).get("prompt_tokens", 0),
prompt_tokens=raw_response_json.get("usage", {}).get("total_tokens", 0),
total_tokens=raw_response_json.get("usage", {}).get("total_tokens", 0),
)
model_response.usage = usage