diff --git a/litellm/llms/voyage/embedding/transformation.py b/litellm/llms/voyage/embedding/transformation.py index 623dfe73af..532b3cb8df 100644 --- a/litellm/llms/voyage/embedding/transformation.py +++ b/litellm/llms/voyage/embedding/transformation.py @@ -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