diff --git a/litellm/llms/ollama_chat.py b/litellm/llms/ollama_chat.py index 0311931b139..c9d6654c7c3 100644 --- a/litellm/llms/ollama_chat.py +++ b/litellm/llms/ollama_chat.py @@ -230,7 +230,7 @@ def get_ollama_response( model_response["model"] = "ollama/" + model prompt_tokens = response_json.get("prompt_eval_count", litellm.token_counter(messages=messages)) # type: ignore completion_tokens = response_json.get( - "eval_count", litellm.token_counter(text=response_json["message"]) + "eval_count", litellm.token_counter(text=response_json["message"]["content"]) ) model_response["usage"] = litellm.Usage( prompt_tokens=prompt_tokens,