ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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