scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

fix: correct module path in vertex_ai cost_per_character debug logs

Browse source
This commit is contained in:
rmaacario 2026-08-18 12:33:18 -03:00
parent 4fd7a73ef5
commit 2dfdfefc7a
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/vertex_ai/cost_calculator.py
View file

@ -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(

2
tests/local_testing/test_completion_cost.py
View file

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