From 2dfdfefc7af832564c60283e70fe6158a5f907a0 Mon Sep 17 00:00:00 2001 From: rmaacario Date: Tue, 18 Aug 2026 12:33:18 -0300 Subject: [PATCH] fix: correct module path in vertex_ai cost_per_character debug logs --- litellm/llms/vertex_ai/cost_calculator.py | 4 ++-- tests/local_testing/test_completion_cost.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/llms/vertex_ai/cost_calculator.py b/litellm/llms/vertex_ai/cost_calculator.py index 86a5bb207ec..401887bde37 100644 --- a/litellm/llms/vertex_ai/cost_calculator.py +++ b/litellm/llms/vertex_ai/cost_calculator.py @@ -114,7 +114,7 @@ def cost_per_character( prompt_cost = prompt_characters * model_info["input_cost_per_character"] except Exception as e: verbose_logger.debug( - "litellm.litellm_core_utils.llm_cost_calc.google.py::cost_per_character(): Exception occured - %s\nDefaulting to None", + "litellm.llms.vertex_ai.cost_calculator.py::cost_per_character(): Exception occured - %s\nDefaulting to None", e, ) prompt_cost, _ = cost_per_token( @@ -153,7 +153,7 @@ def cost_per_character( completion_cost = completion_characters * model_info["output_cost_per_character"] except Exception as e: verbose_logger.debug( - "litellm.litellm_core_utils.llm_cost_calc.google.py::cost_per_character(): Exception occured - %s\nDefaulting to None", + "litellm.llms.vertex_ai.cost_calculator.py::cost_per_character(): Exception occured - %s\nDefaulting to None", e, ) _, completion_cost = cost_per_token( diff --git a/tests/local_testing/test_completion_cost.py b/tests/local_testing/test_completion_cost.py index cf0c645615d..93b311059dc 100644 --- a/tests/local_testing/test_completion_cost.py +++ b/tests/local_testing/test_completion_cost.py @@ -736,7 +736,7 @@ def test_vertex_ai_embedding_completion_cost(caplog): for item in captured_logs: print("\nitem:{}\n".format(item)) if ( - "litellm.litellm_core_utils.llm_cost_calc.google.cost_per_character(): Exception occured " + "litellm.llms.vertex_ai.cost_calculator.py::cost_per_character(): Exception occured " in item ): raise Exception("Error log raised for calculating embedding cost")