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")