From 56d3e75b330c3c3862dc6e1c51c1210e48f1068e Mon Sep 17 00:00:00 2001 From: Lucas Raschek Date: Fri, 14 Mar 2025 18:04:43 +0100 Subject: [PATCH] Map total tokens to prompt_tokens too --- litellm/llms/voyage/embedding/transformation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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