mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-07 02:59:05 +00:00
Update test_llm_cost_calc_utils.py
This commit is contained in:
parent
5247068418
commit
65a8730f3a
1 changed files with 1 additions and 2 deletions
|
|
@ -42,7 +42,7 @@ from litellm.litellm_core_utils.llm_cost_calc.utils import (
|
|||
|
||||
from litellm.types.utils import CacheCreationTokenDetails, Usage
|
||||
from litellm.cost_calculator import completion_cost, response_cost_calculator
|
||||
|
||||
|
||||
def test_reasoning_tokens_no_price_set():
|
||||
# Use o1 - o1-mini was deprecated/renamed; o1 has same reasoning-token semantics
|
||||
# (no separate output_cost_per_reasoning_token, so all completion tokens use output_cost_per_token)
|
||||
|
|
@ -130,7 +130,6 @@ def test_reasoning_tokens_gemini():
|
|||
10,
|
||||
)
|
||||
|
||||
|
||||
def test_reasoning_tokens_gemini_3_1_flash_lite():
|
||||
"""Test cost calculation for gemini-3.1-flash-lite-preview with reasoning tokens"""
|
||||
model = "gemini-3.1-flash-lite-preview"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue